From 44698cfd5b83869f9ba14e1000dda3f1b861b8a9 Mon Sep 17 00:00:00 2001 From: Quido Zientek <quido@codero.cz> Date: Mon, 29 Aug 2022 15:42:13 +0200 Subject: [PATCH] main models.py reformatted --- .isort.cfg | 2 +- .pre-commit-config.yaml | 2 +- main/models.py | 66 +++++++++++++++++++++++------------------ 3 files changed, 39 insertions(+), 31 deletions(-) diff --git a/.isort.cfg b/.isort.cfg index 642718cf..340422d3 100644 --- a/.isort.cfg +++ b/.isort.cfg @@ -3,4 +3,4 @@ line_length = 88 multi_line_output = 3 include_trailing_comma = true -known_third_party = PyPDF2,arrow,bleach,bs4,captcha,celery,django,environ,faker,fastjsonschema,icalevnt,markdown,modelcluster,pirates,pytest,pytz,requests,sentry_sdk,taggit,tweepy,wagtail,wagtailmetadata,weasyprint,yaml,zoneinfo +known_third_party = PyPDF2,arrow,bleach,bs4,captcha,celery,django,environ,faker,fastjsonschema,icalevnt,markdown,modelcluster,pirates,pytest,pytz,requests,sentry_sdk,taggit,tweepy,wagtail,wagtailmetadata,weasyprint,yaml diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 547e2ea1..12840563 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -1,5 +1,5 @@ default_language_version: - python: python3.8 + python: python3.9 exclude: snapshots/ repos: diff --git a/main/models.py b/main/models.py index 6f9546a3..9ac004f0 100644 --- a/main/models.py +++ b/main/models.py @@ -180,39 +180,41 @@ class MainHomePage(MenuMixin, ExtendedMetadataHomePageMixin, MetadataPageMixin, def get_region_response(self, request): last_month = timezone.now().today().replace(day=1) - timedelta(days=1) first_day_of_last_month = last_month.replace(day=1) - if request.GET.get('region', None) == 'VSK': + if request.GET.get("region", None) == "VSK": sorted_article_qs = MainArticlePage.objects.filter( date__gt=first_day_of_last_month ).order_by("-date") else: sorted_article_qs = MainArticlePage.objects.filter( - date__gt=first_day_of_last_month, region=request.GET.get('region', None) + date__gt=first_day_of_last_month, region=request.GET.get("region", None) ).order_by("-date") - context = { - "article_data_list": sorted_article_qs[:3] - } + context = {"article_data_list": sorted_article_qs[:3]} data = { - 'html': render(request, 'main/includes/small_article_preview.html', context).content.decode("utf-8") + "html": render( + request, "main/includes/small_article_preview.html", context + ).content.decode("utf-8") } return JsonResponse(data=data, safe=False) def get_twitter_response(self, request): - tweet_paginator = Paginator(Tweet.objects.order_by("-twitter_id"), self.max_items) - tweet_page = tweet_paginator.get_page(request.GET.get('page', 1)) - context = { - "tweet_list": tweet_page.object_list - } - html_content = render(request, 'main/includes/twitter_widget.html', context).content + tweet_paginator = Paginator( + Tweet.objects.order_by("-twitter_id"), self.max_items + ) + tweet_page = tweet_paginator.get_page(request.GET.get("page", 1)) + context = {"tweet_list": tweet_page.object_list} + html_content = render( + request, "main/includes/twitter_widget.html", context + ).content data = { - 'html': html_content.decode("utf-8"), - 'last_page': tweet_page.paginator.num_pages, + "html": html_content.decode("utf-8"), + "last_page": tweet_page.paginator.num_pages, } return JsonResponse(data=data, safe=False) def serve(self, request, *args, **kwargs): - if not request.META.get('HTTP_X_REQUESTED_WITH') == 'XMLHttpRequest': + if not request.META.get("HTTP_X_REQUESTED_WITH") == "XMLHttpRequest": return super().serve(request, *args, **kwargs) - if 'region' in request.GET: + if "region" in request.GET: return self.get_region_response(request) else: return self.get_twitter_response(request) @@ -336,24 +338,30 @@ class MainArticlesPage( ctx = super().get_context(request, args, kwargs) article_list = self.get_article_data_list(1) ctx["article_data_list"] = article_list - ctx["show_next_articles"] = MainArticlePage.objects.filter(article_type=ARTICLE_TYPES.WORK_TIMELINE).count() > len(article_list) + ctx["show_next_articles"] = MainArticlePage.objects.filter( + article_type=ARTICLE_TYPES.WORK_TIMELINE + ).count() > len(article_list) + return ctx def get_articles_response(self, request): - article_list = self.get_article_data_list(int(request.GET.get('months', None))) - context = { - "article_data_list": article_list - } + article_list = self.get_article_data_list(int(request.GET.get("months", None))) + context = {"article_data_list": article_list} data = { - 'html': render(request, 'main/blocks/articles_timeline_block.html', context).content.decode("utf-8"), - 'last_article': article_list[-1] == MainArticlePage.objects.filter(article_type=ARTICLE_TYPES.WORK_TIMELINE).order_by('-date').last() + "html": render( + request, "main/blocks/articles_timeline_block.html", context + ).content.decode("utf-8"), + "last_article": article_list[-1] + == MainArticlePage.objects.filter(article_type=ARTICLE_TYPES.WORK_TIMELINE) + .order_by("-date") + .last(), } return JsonResponse(data=data, safe=False) def serve(self, request, *args, **kwargs): - if not request.META.get('HTTP_X_REQUESTED_WITH') == 'XMLHttpRequest': + if not request.META.get("HTTP_X_REQUESTED_WITH") == "XMLHttpRequest": return super().serve(request, *args, **kwargs) - if 'months' in request.GET: + if "months" in request.GET: return self.get_articles_response(request) @staticmethod @@ -507,7 +515,7 @@ class MainPersonPage(ExtendedMetadataPageMixin, SubpageMixin, MetadataPageMixin, blank=True, null=True, verbose_name="Hlavní obrázek", - related_name="+" + related_name="+", ) profile_image = models.ForeignKey( @@ -516,7 +524,7 @@ class MainPersonPage(ExtendedMetadataPageMixin, SubpageMixin, MetadataPageMixin, blank=True, null=True, verbose_name="Profilový obrázek", - related_name="+" + related_name="+", ) before_name = models.CharField( "Tituly před jménem", max_length=32, blank=True, null=True @@ -535,7 +543,7 @@ class MainPersonPage(ExtendedMetadataPageMixin, SubpageMixin, MetadataPageMixin, blank=True, null=True, max_length=32, - help_text="Uživatelské jméno zadejte bez @ na začátku" + help_text="Uživatelské jméno zadejte bez @ na začátku", ) social_links = StreamField( @@ -622,7 +630,7 @@ class MainContactPage(ExtendedMetadataPageMixin, SubpageMixin, MetadataPageMixin text = StreamField( [("two_columns_text", blocks.TwoTextColumnBlock())], verbose_name="Kontaktní informace", - blank=True + blank=True, ) ### PANELS -- GitLab