From a749ade30be7851ca9667787106b62bc4e2c9e7f Mon Sep 17 00:00:00 2001 From: Untone Date: Sat, 18 May 2024 11:00:46 +0300 Subject: [PATCH] fmt --- orm/author.py | 1 - resolvers/__init__.py | 14 ++++++++++++-- resolvers/notifier.py | 7 ++++++- resolvers/stat.py | 4 +--- services/search.py | 17 +++++++++++++---- services/viewed.py | 7 ++++++- 6 files changed, 38 insertions(+), 12 deletions(-) diff --git a/orm/author.py b/orm/author.py index 4bf79657..c96fbcf5 100644 --- a/orm/author.py +++ b/orm/author.py @@ -7,7 +7,6 @@ from services.db import Base # from sqlalchemy_utils import TSVectorType - class AuthorRating(Base): __tablename__ = "author_rating" diff --git a/resolvers/__init__.py b/resolvers/__init__.py index bff24eef..dbc3f35e 100644 --- a/resolvers/__init__.py +++ b/resolvers/__init__.py @@ -11,7 +11,12 @@ from resolvers.author import ( # search_authors, ) from resolvers.community import get_communities_all, get_community from resolvers.editor import create_shout, delete_shout, update_shout -from resolvers.follower import follow, get_shout_followers, get_topic_followers, unfollow +from resolvers.follower import ( + follow, + get_shout_followers, + get_topic_followers, + unfollow, +) from resolvers.notifier import ( load_notifications, notification_mark_seen, @@ -35,7 +40,12 @@ from resolvers.reader import ( load_shouts_search, load_shouts_unrated, ) -from resolvers.topic import get_topic, get_topics_all, get_topics_by_author, get_topics_by_community +from resolvers.topic import ( + get_topic, + get_topics_all, + get_topics_by_author, + get_topics_by_community, +) from services.triggers import events_register events_register() diff --git a/resolvers/notifier.py b/resolvers/notifier.py index 8a3b9a91..8ce824e3 100644 --- a/resolvers/notifier.py +++ b/resolvers/notifier.py @@ -8,7 +8,12 @@ from sqlalchemy.orm import aliased from sqlalchemy.sql import not_ from orm.author import Author -from orm.notification import Notification, NotificationAction, NotificationEntity, NotificationSeen +from orm.notification import ( + Notification, + NotificationAction, + NotificationEntity, + NotificationSeen, +) from orm.shout import Shout from services.auth import login_required from services.db import local_session diff --git a/resolvers/stat.py b/resolvers/stat.py index f88cdc6a..37e65234 100644 --- a/resolvers/stat.py +++ b/resolvers/stat.py @@ -31,9 +31,7 @@ def add_author_stat_columns(q): func.count(distinct(aliased_shout.shout)).label("shouts_stat") ) aliased_follower = aliased(AuthorFollower) - q = q.outerjoin( - aliased_follower, aliased_follower.author == Author.id - ).add_columns( + q = q.outerjoin(aliased_follower, aliased_follower.author == Author.id).add_columns( func.count(distinct(aliased_follower.follower)).label("followers_stat") ) diff --git a/services/search.py b/services/search.py index 8ae7dccf..80678ba4 100644 --- a/services/search.py +++ b/services/search.py @@ -12,7 +12,10 @@ ELASTIC_HOST = os.environ.get("ELASTIC_HOST", "").replace("https://", "") ELASTIC_USER = os.environ.get("ELASTIC_USER", "") ELASTIC_PASSWORD = os.environ.get("ELASTIC_PASSWORD", "") ELASTIC_PORT = os.environ.get("ELASTIC_PORT", 9200) -ELASTIC_URL = os.environ.get("ELASTIC_URL", f"https://{ELASTIC_USER}:{ELASTIC_PASSWORD}@{ELASTIC_HOST}:{ELASTIC_PORT}") +ELASTIC_URL = os.environ.get( + "ELASTIC_URL", + f"https://{ELASTIC_USER}:{ELASTIC_PASSWORD}@{ELASTIC_HOST}:{ELASTIC_PORT}", +) REDIS_TTL = 86400 # 1 день в секундах index_settings = { @@ -113,7 +116,9 @@ class SearchService: mapping = result.get(self.index_name, {}).get("mappings") if mapping and mapping != expected_mapping: logger.debug(f"Найдена структура индексации: {mapping}") - logger.warn("Требуется другая структура индексации, переиндексация") + logger.warn( + "Требуется другая структура индексации, переиндексация" + ) await self.recreate_index() else: logger.error("клиент не инициализован, невозможно проверить индекс") @@ -121,7 +126,9 @@ class SearchService: async def recreate_index(self): if self.client: async with self.lock: - self.client.indices.delete(index=self.index_name, ignore_unavailable=True) + self.client.indices.delete( + index=self.index_name, ignore_unavailable=True + ) await self.check_index() else: logger.error("клиент не инициализован, невозможно пересоздать индекс") @@ -136,7 +143,9 @@ class SearchService: async def perform_index(self, shout): if self.client: - self.client.index(index=self.index_name, id=str(shout.id), body=shout.dict()) + self.client.index( + index=self.index_name, id=str(shout.id), body=shout.dict() + ) async def search(self, text, limit, offset): logger.debug(f"Ищем: {text}") diff --git a/services/viewed.py b/services/viewed.py index f04de2db..f9540fe2 100644 --- a/services/viewed.py +++ b/services/viewed.py @@ -7,7 +7,12 @@ from typing import Dict # ga from google.analytics.data_v1beta import BetaAnalyticsDataClient -from google.analytics.data_v1beta.types import DateRange, Dimension, Metric, RunReportRequest +from google.analytics.data_v1beta.types import ( + DateRange, + Dimension, + Metric, + RunReportRequest, +) from orm.author import Author from orm.shout import Shout, ShoutAuthor, ShoutTopic