# -*- coding: utf-8 -*-
from django.contrib.auth.models import AnonymousUser
from django.http import Http404
from django.utils.translation import activate
from django.utils.timezone import now
from parler.tests.utils import override_parler_settings
from parler.utils.conf import add_default_language_settings
from parler.utils.context import switch_language
from djangocms_blog.feeds import LatestEntriesFeed, TagFeed
from djangocms_blog.sitemaps import BlogSitemap
from djangocms_blog.views import (PostListView, PostDetailView,
                                  PostArchiveView, CategoryEntriesView,
                                  AuthorEntriesView, TaggedListView)

from . import BaseTest


class ViewTest(BaseTest):

    def test_post_list_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        request = self.get_page_request(page1, AnonymousUser(), r'/en/blog/', edit=False)
        activate('en')
        view_obj = PostListView()
        view_obj.request = request

        self.assertEqual(list(view_obj.get_queryset()), [post1])

        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=False)
        activate('en')
        view_obj.request = request
        view_obj.kwargs = {}
        qs = view_obj.get_queryset()
        self.assertEqual(qs.count(), 1)
        self.assertEqual(set(qs), set([post1]))

        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=True)
        view_obj.request = request
        self.assertEqual(set(view_obj.get_queryset()), set([post1, post2]))

        view_obj.kwargs = {}
        view_obj.object_list = view_obj.get_queryset()
        view_obj.paginate_by = 1
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertTrue(context['is_paginated'])
        self.assertEqual(list(context['post_list']), [post2])
        self.assertEqual(context['paginator'].count, 2)
        self.assertEqual(context['post_list'][0].title, 'Second post')
        response = view_obj.render_to_response(context)
        self.assertContains(response, context['post_list'][0].get_absolute_url())

        request = self.get_page_request(page1, self.user, r'/it/blog/', lang_code='it', edit=True)
        activate('it')
        view_obj.request = request
        view_obj.object_list = view_obj.get_queryset()
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertEqual(context['post_list'][0].title, 'Secondo post')
        response = view_obj.render_to_response(context)
        self.assertContains(response, context['post_list'][0].get_absolute_url())

    def test_post_list_view_fallback(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        PARLER_FALLBACK = {
            1: (
                {'code': 'en'},
                {'code': 'it'},
                {'code': 'fr', 'hide_untranslated': True,},
            ),
            'default': {
                'fallback': 'en',
                'hide_untranslated': False,
            }
        }

        view_obj = PostListView()
        request = self.get_page_request(page1, self.user, r'/fr/blog/', lang_code='fr', edit=True)
        activate('fr')
        view_obj.request = request
        view_obj.kwargs = {}
        view_obj.object_list = view_obj.get_queryset()
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertEqual(view_obj.get_queryset().count(), 2)

        PARLER_FALLBACK = add_default_language_settings(PARLER_FALLBACK)
        with override_parler_settings(PARLER_LANGUAGES=PARLER_FALLBACK):

            view_obj = PostListView()
            request = self.get_page_request(page1, self.user, r'/fr/blog/', lang_code='fr', edit=True)
            activate('fr')
            view_obj.request = request
            view_obj.kwargs = {}
            view_obj.object_list = view_obj.get_queryset()
            context = view_obj.get_context_data(object_list=view_obj.object_list)
            self.assertEqual(view_obj.get_queryset().count(), 0)

    def test_post_detail_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        with switch_language(post1, 'en'):
            request = self.get_page_request(page1, AnonymousUser(), r'/en/blog/', edit=False)
            view_obj = PostDetailView()
            view_obj.request = request

            with self.assertRaises(Http404):
                view_obj.kwargs = {'slug': 'not-existing'}
                post_obj = view_obj.get_object()

            view_obj.kwargs = {'slug': post1.slug}
            post_obj = view_obj.get_object()
            self.assertEqual(post_obj, post1)
            self.assertEqual(post_obj.language_code, 'en')

        with switch_language(post1, 'it'):
            request = self.get_page_request(page1, AnonymousUser(), r'/it/blog/', lang_code='it', edit=False)
            view_obj.request = request
            view_obj.kwargs = {'slug': post1.slug}
            post_obj = view_obj.get_object()
            self.assertEqual(post_obj, post1)
            self.assertEqual(post_obj.language_code, 'it')

            view_obj.object = post_obj
            context = view_obj.get_context_data()
            self.assertEqual(context['post'], post1)
            self.assertEqual(context['post'].language_code, 'it')
            self.assertTrue(context['meta'])

    def test_post_archive_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        request = self.get_page_request(page1, AnonymousUser(), r'/en/blog/', edit=False)
        activate('en')
        view_obj = PostArchiveView()
        view_obj.request = request
        view_obj.kwargs = {'year': now().year, 'month': now().month}

        # One post only, anonymous request
        qs = view_obj.get_queryset()
        self.assertEqual(qs.count(), 1)
        self.assertEqual(list(qs), [post1])

        view_obj.object_list = qs
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertEqual(context['archive_date'].date(), now().replace(year=now().year, month=now().month, day=1).date())
        
    def test_category_entries_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=True)
        activate('en')
        view_obj = CategoryEntriesView()
        view_obj.request = request
        view_obj.kwargs = {'category': 'category-1'}
        qs = view_obj.get_queryset()
        self.assertEqual(qs.count(), 2)
        self.assertEqual(set(qs), set([post1, post2]))

        view_obj.paginate_by = 1
        view_obj.object_list = qs
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertTrue(context['category'])
        self.assertEqual(context['category'], self.category_1)
        self.assertTrue(context['is_paginated'])
        self.assertEqual(list(context['post_list']), [post2])
        self.assertEqual(context['paginator'].count, 2)
        self.assertEqual(context['post_list'][0].title, 'Second post')

    def test_author_entries_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()

        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=True)
        activate('en')
        view_obj = AuthorEntriesView()
        view_obj.request = request
        view_obj.kwargs = {'username': self.user.get_username()}
        qs = view_obj.get_queryset()
        self.assertEqual(qs.count(), 2)
        self.assertEqual(set(qs), set([post1, post2]))

        view_obj.paginate_by = 1
        view_obj.object_list = qs
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertTrue(context['author'])
        self.assertEqual(context['author'], self.user)
        self.assertTrue(context['is_paginated'])
        self.assertEqual(list(context['post_list']), [post2])
        self.assertEqual(context['paginator'].count, 2)
        self.assertEqual(context['post_list'][0].title, 'Second post')

    def test_taggedlist_view(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()
        post1.tags.add('tag 1', 'tag 2', 'tag 3', 'tag 4')
        post1.save()
        post2.tags.add('tag 6', 'tag 2', 'tag 5', 'tag 8')
        post2.save()

        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=True)
        activate('en')
        view_obj = TaggedListView()
        view_obj.request = request
        view_obj.kwargs = {'tag': 'tag-2'}
        qs = view_obj.get_queryset()
        self.assertEqual(qs.count(), 2)
        self.assertEqual(set(qs), set([post1, post2]))

        view_obj.paginate_by = 1
        view_obj.object_list = qs
        context = view_obj.get_context_data(object_list=view_obj.object_list)
        self.assertTrue(context['tagged_entries'], 'tag-2')
        self.assertTrue(context['is_paginated'])
        self.assertEqual(list(context['post_list']), [post2])
        self.assertEqual(context['paginator'].count, 2)
        self.assertEqual(context['post_list'][0].title, 'Second post')

    def test_feed(self):
        page1, page2 = self.get_pages()
        post1, post2 = self.get_posts()
        post1.tags.add('tag 1', 'tag 2', 'tag 3', 'tag 4')
        post1.save()
        post2.tags.add('tag 6', 'tag 2', 'tag 5', 'tag 8')
        post2.save()
        post1.set_current_language('en')

        feed = LatestEntriesFeed()
        self.assertEqual(list(feed.items()), [post1])
        request = self.get_page_request(page1, self.user, r'/en/blog/', lang_code='en', edit=False)
        xml = feed(request)
        self.assertContains(xml, post1.get_absolute_url())
        self.assertContains(xml, 'Blog articles on example.com')

        activate('it')
        post1.set_current_language('it')
        feed = LatestEntriesFeed()
        self.assertEqual(list(feed.items()), [post1])
        request = self.get_page_request(page1, self.user, r'/it/blog/', lang_code='it', edit=False)
        xml = feed(request)
        self.assertContains(xml, post1.get_absolute_url())
        self.assertContains(xml, 'Articoli del blog su example.com')

        feed = TagFeed()
        self.assertEqual(list(feed.items('tag-2')), [post1])
        
    def test_sitemap(self):
        post1, post2 = self.get_posts()
        post1.tags.add('tag 1', 'tag 2', 'tag 3', 'tag 4')
        post1.save()
        post2.tags.add('tag 6', 'tag 2', 'tag 5', 'tag 8')
        post2.publish = True
        post2.save()
        post1.set_current_language('en')
        
        sitemap = BlogSitemap()
        self.assertEqual(sitemap.items().count(), 2)
        for item in sitemap.items():
            self.assertTrue(sitemap.lastmod(item).date(), now().today())
