from django.urls import reverse
from django.test.client import Client
from django.apps import apps
from builder.base import BuildNew
from itertools import chain

from django.conf import settings
from .models import PostType, PostTopic


class BuildSrc(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.SRC).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("src:list"),
            paginate_by=50
        )
        self.build_detail_view()
        self.build_feed("src:feed")


class BuildGuide(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type__in=[PostType.GUIDE, PostType.REVIEW]).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("guides:guide-base"),
            paginate_by=50
        )
        self.build_detail_view()
        

class BuildFieldNotes(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.FIELD_NOTE).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_detail_view()
        self.build_list_view(
            base_path=reverse("fieldnotes:list"),
            paginate_by=24
        )
        self.build_year_view("fieldnotes:list_year")
        self.build_month_view("fieldnotes:list_month")


class BuildJrnl(BuildNew):
    '''
    Write jrnl to disk
    '''
    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.JRNL).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("jrnl:list"),
            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_latest()

    def build_arc(self):
        self.build_list_view(
            base_path=reverse("jrnl:list"),
            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,
                    post_type=PostType.JRNL,
                    location__state__country=c
                )
            except:
                qs = self.model.objects.filter(
                    status__exact=1,
                    post_type=PostType.JRNL,
                    location__state__country__lux_region=c.id
                )
            print(c)
            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)
                print(response.content)
                if page == 0:
                    self.write_file(base_path, response.content)
                else:
                    self.write_file(base_path, response.content)

    def build_latest(self):
        response = self.client.get('/jrnl/latest/')
        self.write_file(reverse("jrnl:latest"), response.content)


class BuildEssays(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.ESSAY).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("range:list"),
            paginate_by=50
        )
        self.build_list_view(
            base_path=reverse("essay-list:list"),
            paginate_by=50
        )
        for t in PostTopic.labels:
            self.build_list_view(
                base_path=reverse("essay-list:category-detail", kwargs={'topic':t}),
                paginate_by=50
            )
        self.build_detail_view()


class BuildRange(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.RANGE).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("range:range-list"),
            paginate_by=30
        )
        self.build_detail_view()
        self.build_feed("range:feed")


class BuildFriends(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.FRIENDS).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("friends:friends-list"),
            paginate_by=30
        )
        self.build_detail_view()
        self.build_feed("friends:feed")


class BuildFilms(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.filter(post_type=PostType.FILM).filter(status__exact=1).order_by('-pub_date')

    def build(self):
        self.build_list_view(
            base_path=reverse("films:list"),
            paginate_by=30
        )
        self.build_detail_view()


class BuildTrips(BuildNew):

    def get_model_queryset(self):
        return self.model.objects.all()

    def build(self):
        self.build_list_view(
            base_path=reverse("trips:list"),
            paginate_by=30
        )
        self.build_detail_view()


class BuildSitemap(BuildNew):
    def build(self):
        c = Client()
        response = c.get('/sitemap.xml', HTTP_HOST='127.0.0.1')
        self.write_file('/', response.content, 'xml', 'sitemap')