from django.core.urlresolvers import reverse
from django.apps import apps
from builder.base import BuildNew
from itertools import chain


class BuildJrnl(BuildNew):

    def build(self):
        self.build_list_view(
            base_path=reverse("jrnl:live_redirect"),
            paginate_by=24
        )
        self.build_year_view("jrnl:list_year")
        self.build_month_view("jrnl:list_month")
        self.build_detail_view()
        self.build_location_view()
        self.build_feed("jrnl:feed")

    def build_arc(self):
        self.build_list_view(
            base_path=reverse("jrnl:live_redirect"),
            paginate_by=24
        )
        self.build_year_view("jrnl:list_year")
        self.build_month_view("jrnl:list_month")
        self.build_location_view()

    def build_location_view(self):
        c = apps.get_model('locations', 'Country')
        r = apps.get_model('locations', 'Region')
        countries = c.objects.filter(visited=True)
        regions = r.objects.all()
        locations = list(chain(countries, regions))
        for c in locations:
            try:
                qs = self.model.objects.filter(
                    status__exact=1,
                    location__state__country=c
                )
            except:
                qs = self.model.objects.filter(
                    status__exact=1,
                    location__state__country__lux_region=c.id
                )
            pages = self.get_pages(qs, 24)
            for page in range(pages):
                base_path = reverse("jrnl:list_country", kwargs={'slug': c.slug, 'page': page + 1})
                response = self.client.get(base_path, HTTP_HOST='127.0.0.1')
                if page == 0:
                    self.write_file(base_path[:-2], response.content)
                else:
                    self.write_file(base_path, response.content)

    def build_homepage(self):
        response = self.client.get('/')
        self.write_file('', response.content)


def archive_builder():
    j = BuildJrnl("jrnl", "entry")
    j.build_arc()


def detail_builder():
    j = BuildJrnl("jrnl", "entry")
    j.build_detail_view()


def home_builder():
    j = BuildJrnl("jrnl", "entry")
    j.build_detail_view()