Dynastie/generators/archive.py

import os
from datetime import datetime
from xml.dom.minidom import parse, parseString
from dynastie.generators.generator import DynastieGenerator
from dynastie.generators.index import Index
from django.db import models

class Archive(Index):

    cur_page = 0
    nb_pages = 0
    cur_post = 0
    posts_per_page = 0
    filename = 'index'
    dirname = ''
    cur_year = 0

    def createArchive(self, posts, dom, root, node):
        if node.hasAttribute('year'):
            self.replaceByText(dom, root, node, str(self.cur_year))

        return None

    def generate(self, blog, src, output):
        from dynastie.models import Post, Blog

        self.hooks['archive'] = self.createArchive

        dom = self.parseTemplate(blog, src, output, 'archive', 'archive')
        if dom is None: return self.report

        posts = Post.objects.filter(published=True, front_page=True).order_by('creation_date')

        if posts.count() != 0:
            self.cur_year = int(posts[0].creation_date.year)

        my_post = []
        now = datetime.now()
        nb_post = len(posts)
        for i in range(0, nb_post):
            if self.cur_year == now.year:
                break

            if i < nb_post-1:
                if posts[i].creation_date.year != posts[i+1].creation_date.year:
                    dom = parse(src + '/_archive.html')
                    my_post.reverse()
                    self.nb_pages = 0
                    self.cur_page = 0
                    self.cur_post = 0

                    self.dirname = '/archive/' + str(self.cur_year)
                
                    self.generatePages(dom, my_post, src, output, 'archive', 'archive')
                    self.cur_year = int(posts[i+1].creation_date.year)
                    #print 'New year ' + str(self.cur_year)
                    my_post = []
                    if self.cur_year == int(now.year):
                        break
                else:
                    my_post.append(posts[i])
            else:
                # Last post
                my_post.append(posts[i])
                if nb_post != 1 and posts[i].creation_date.year != posts[i-1].creation_date.year:
                    self.cur_year = int(posts[i].creation_date.year)

        if not self.somethingWrote:
            self.addReport('Nothing changed')

        return self.report