Esempio n. 1
0
 def __init__(self,*args,**kwargs):
     self.post_list = []
     self.post_name_list = []
     self.rss_generator = RssGenerator()
     loader = FileSystemLoader(os.path.abspath(settings.TEMPLATE_DIR))
     self.env = Environment(loader=loader)
     self.env.globals["settings"] = settings
Esempio n. 2
0
class Composer(object):
    def __init__(self,*args,**kwargs):
        self.post_list = []
        self.post_name_list = []
        self.rss_generator = RssGenerator()
        loader = FileSystemLoader(os.path.abspath(settings.TEMPLATE_DIR))
        self.env = Environment(loader=loader)
        self.env.globals["settings"] = settings

    def get_output_file_name(self,post, filename):
        if filename in self.post_name_list:
            self.get_output_file_name(post, filename+"_1")
        else:
            post["output_file_name"] = filename

    def build_index_page(self, post):
        self.build_post(post,"index")

    def build_post(self,post, post_name=None):
        template = self.env.get_template('post_page.html')
        if post["format"]=="markdown":
            post["content"] = markdown.markdown(post["content"])
        rendered_template =  template.render({"post":post})
        if not post_name:
            post_name = post["output_file_name"]
        f = codecs.open(os.path.join(os.path.abspath(settings.OUTPUT_DIR), post_name+".html"),"w", encoding="utf-8")
        f.write(rendered_template)
        f.close()

    def is_new_post(self, post):
        return self.post_list.index(post) == len(self.post_list) - 1

    def build_rss_feed(self,posts=[]):
        rss_data = self.rss_generator.generate(posts)
        f = open(os.path.abspath(settings.RSS_PATH),"w")
        f.write(rss_data)
        f.close()

    def build(self, post):
        if self.is_new_post(post):
            self.build_index_page(post)
        self.build_post(post, post["output_file_name"])


    def build_all(self):
        storage = get_storage()
        storage.clear_storage()
        self.post_list = storage.get_all_posts_by_type(status="publish")
        for post in self.post_list:
            base_slug_title = slugify(post["title"],"_") if post["title"] else slugify(post["content"][:24],"_")
            self.get_output_file_name(post,base_slug_title)
            self.post_name_list.append(post["output_file_name"])

        self.env.globals["post_manager"] = PostOrganizer(self.post_list)
        for p in self.post_list:
            print p["pub_date"], p["output_file_name"]
            self.build(p)

        self.build_rss_feed(self.post_list[-10:][::-1])

        print "-" * 50
        print "len of posts", len(self.post_list)
        print "len of post names", len(self.post_name_list)