Dynastie/generators/archive.py

# -*- coding: utf-8 -*-
"""
  Copyright 2012-2013 Grégory Soutadé

  This file is part of Dynastie.

  Dynastie is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  Dynastie is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with Dynastie.  If not, see <http://www.gnu.org/licenses/>.
"""
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

        first_post = Post.objects.filter(published=True, front_page=True).order_by('creation_date')[0]
        last_post = Post.objects.filter(published=True, front_page=True).order_by('-creation_date')[0]

        start_year = first_post.creation_date.year
        end_year = last_post.creation_date.year

        now = datetime.now()
        for i in range(start_year, end_year):
            if i == now.year: continue

            self.cur_year = i

            posts = Post.objects.filter(published=True, front_page=True, creation_date__gt=datetime(i, 1, 1), creation_date__lt=datetime(i+1, 1, 1)).order_by('-creation_date')

            self.nb_pages = 0
            self.cur_page = 0
            self.cur_post = 0

            self.dirname = '/archive/' + str(i)
                
            self.generatePages(dom, posts, src, output, 'archive')

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

        return self.report