from feedgen.feed import FeedGenerator

from models.postmodel import Post


def update_rss_feed():
    fg = create_rss_feed()
    posts = Post.query.all()
    for post in posts:
        fe = fg.add_entry()
        fe.id("http://lernfunk.de/media/654321/1")
        fe.title(f"Crunk website:{post.id}")
        fe.content(post.content)
        fe.link(href="http://crunk.website")
    write_rss_feed(fg)


def create_rss_feed():
    print("creating rss feed")
    fg = FeedGenerator()
    fg.id("http://crunk.website")
    fg.title("Crunk website")
    fg.link(href="http://crunk.website", rel="alternate")
    fg.subtitle("Some things crunk is doing!")
    fg.link(href="http://crunk.website/test.atom", rel="self")
    fg.language("en")
    return fg


def write_rss_feed(fg):
    atomfeed = fg.atom_str(pretty=True)
    rssfeed = fg.rss_str(pretty=True)
    fg.atom_file("static/atom.xml")
    fg.rss_file("static/rss.xml")