"Module to define some objects in for use in the archive pages"

class Archive(object):

    "An archive for a queryset of objects"

    def __init__(self, queryset):

        "Build an Archive object out of the passed queryset"

        self.queryset = queryset

    def __iter__(self):

        return iter(self.years)

    @property

    def years(self):

        if not hasattr(self, "_years"):

            self.build_years()

        return sorted(self._years.values(), key=lambda y: -y.year)

    def build_years(self):

        self._years = {}

        for entry in self.queryset:

            year = entry.pub_date.year

            month = entry.pub_date.month

            if year not in self._years:

                self._years[year] = Year(year)

            yearobj = self._years[year]

            if month not in yearobj.months:

                yearobj.months[month] = Month(year, month)

            monthobj = yearobj.months[month]

            monthobj.entries.append(entry)

class Year(object):

    def __init__(self, year):

        self.year = year

        self.months = {}

    def __unicode__(self):

        return unicode(self.year)

    def __iter__(self):

        return iter(sorted(self.months.values(), key=lambda m: m.month))

    @property

    def count(self):

        return sum([month.count for month in self])

class Month(object):

    def __init__(self, year, month):

        self.year = year

        self.month = month

        self.entries = []

    @property

    def count(self):

        return len(self.entries)

    def __iter__(self):

        return iter(sorted(self.entries, key=lambda x: x.pub_date))

    def __unicode__(self):

        from datetime import date

        return "%s %s" % (date(2000, self.month, 1).strftime("%B"), unicode(self.year))

    @property

    def abbrev(self):

        from datetime import date

        return date(2000, self.month, 1).strftime("%b").lower()

    @property

    def url(self):

        from django.core.urlresolvers import reverse

        return reverse("blog-archive-month", args=[self.year, self.abbrev])