544 lines
23 KiB
Python
544 lines
23 KiB
Python
import json
|
||
import time
|
||
|
||
from sqlalchemy import and_, desc, select
|
||
from sqlalchemy.orm import joinedload
|
||
from sqlalchemy.sql.functions import coalesce
|
||
|
||
from cache.cache import cache_author, cache_topic, invalidate_shout_related_cache, invalidate_shouts_cache
|
||
from orm.author import Author
|
||
from orm.shout import Shout, ShoutAuthor, ShoutTopic
|
||
from orm.topic import Topic
|
||
from resolvers.follower import follow, unfollow
|
||
from resolvers.stat import get_with_stat
|
||
from services.auth import login_required
|
||
from services.db import local_session
|
||
from services.notify import notify_shout
|
||
from services.schema import mutation, query
|
||
from services.search import search_service
|
||
from utils.logger import root_logger as logger
|
||
|
||
|
||
async def cache_by_id(entity, entity_id: int, cache_method):
|
||
caching_query = select(entity).filter(entity.id == entity_id)
|
||
result = get_with_stat(caching_query)
|
||
if not result or not result[0]:
|
||
logger.warning(f"{entity.__name__} with id {entity_id} not found")
|
||
return
|
||
x = result[0]
|
||
d = x.dict() # convert object to dictionary
|
||
cache_method(d)
|
||
return d
|
||
|
||
|
||
@query.field("get_my_shout")
|
||
@login_required
|
||
async def get_my_shout(_, info, shout_id: int):
|
||
# logger.debug(info)
|
||
user_id = info.context.get("user_id", "")
|
||
author_dict = info.context.get("author", {})
|
||
author_id = author_dict.get("id")
|
||
roles = info.context.get("roles", [])
|
||
shout = None
|
||
if not user_id or not author_id:
|
||
return {"error": "unauthorized", "shout": None}
|
||
with local_session() as session:
|
||
shout = (
|
||
session.query(Shout)
|
||
.filter(Shout.id == shout_id)
|
||
.options(joinedload(Shout.authors), joinedload(Shout.topics))
|
||
.filter(Shout.deleted_at.is_(None))
|
||
.first()
|
||
)
|
||
if not shout:
|
||
return {"error": "no shout found", "shout": None}
|
||
|
||
# Преобразуем media JSON в список объектов MediaItem
|
||
if hasattr(shout, "media") and shout.media:
|
||
if isinstance(shout.media, str):
|
||
try:
|
||
shout.media = json.loads(shout.media)
|
||
except Exception as e:
|
||
logger.error(f"Error parsing shout media: {e}")
|
||
shout.media = []
|
||
if not isinstance(shout.media, list):
|
||
shout.media = [shout.media] if shout.media else []
|
||
else:
|
||
shout.media = []
|
||
|
||
logger.debug(f"got {len(shout.authors)} shout authors, created by {shout.created_by}")
|
||
is_editor = "editor" in roles
|
||
logger.debug(f'viewer is{'' if is_editor else ' not'} editor')
|
||
is_creator = author_id == shout.created_by
|
||
logger.debug(f'viewer is{'' if is_creator else ' not'} creator')
|
||
is_author = bool(list(filter(lambda x: x.id == int(author_id), [x for x in shout.authors])))
|
||
logger.debug(f'viewer is{'' if is_creator else ' not'} author')
|
||
can_edit = is_editor or is_author or is_creator
|
||
|
||
if not can_edit:
|
||
return {"error": "forbidden", "shout": None}
|
||
|
||
logger.debug("got shout editor with data")
|
||
return {"error": None, "shout": shout}
|
||
|
||
|
||
@query.field("get_shouts_drafts")
|
||
@login_required
|
||
async def get_shouts_drafts(_, info):
|
||
# user_id = info.context.get("user_id")
|
||
author_dict = info.context.get("author")
|
||
if not author_dict:
|
||
return {"error": "author profile was not found"}
|
||
author_id = author_dict.get("id")
|
||
shouts = []
|
||
with local_session() as session:
|
||
if author_id:
|
||
q = (
|
||
select(Shout)
|
||
.options(joinedload(Shout.authors), joinedload(Shout.topics))
|
||
.filter(and_(Shout.deleted_at.is_(None), Shout.created_by == int(author_id)))
|
||
.filter(Shout.published_at.is_(None))
|
||
.order_by(desc(coalesce(Shout.updated_at, Shout.created_at)))
|
||
.group_by(Shout.id)
|
||
)
|
||
shouts = [shout for [shout] in session.execute(q).unique()]
|
||
return {"shouts": shouts}
|
||
|
||
|
||
@mutation.field("create_shout")
|
||
@login_required
|
||
async def create_shout(_, info, inp):
|
||
logger.info(f"Starting create_shout with input: {inp}")
|
||
user_id = info.context.get("user_id")
|
||
author_dict = info.context.get("author")
|
||
logger.debug(f"Context user_id: {user_id}, author: {author_dict}")
|
||
|
||
if not author_dict:
|
||
logger.error("Author profile not found in context")
|
||
return {"error": "author profile was not found"}
|
||
|
||
author_id = author_dict.get("id")
|
||
if user_id and author_id:
|
||
try:
|
||
with local_session() as session:
|
||
author_id = int(author_id)
|
||
current_time = int(time.time())
|
||
slug = inp.get("slug") or f"draft-{current_time}"
|
||
|
||
logger.info(f"Creating shout with input: {inp}")
|
||
# Создаем публикацию без topics
|
||
new_shout = Shout(
|
||
slug=slug,
|
||
body=inp.get("body", ""),
|
||
layout=inp.get("layout", "article"),
|
||
title=inp.get("title", ""),
|
||
created_by=author_id,
|
||
created_at=current_time,
|
||
community=1,
|
||
)
|
||
|
||
# Проверяем уникальность slug
|
||
logger.debug(f"Checking for existing slug: {slug}")
|
||
same_slug_shout = session.query(Shout).filter(Shout.slug == new_shout.slug).first()
|
||
c = 1
|
||
while same_slug_shout is not None:
|
||
logger.debug(f"Found duplicate slug, trying iteration {c}")
|
||
new_shout.slug = f"{slug}-{c}"
|
||
same_slug_shout = session.query(Shout).filter(Shout.slug == new_shout.slug).first()
|
||
c += 1
|
||
|
||
try:
|
||
logger.info("Creating new shout object")
|
||
session.add(new_shout)
|
||
session.commit()
|
||
logger.info(f"Created shout with ID: {new_shout.id}")
|
||
except Exception as e:
|
||
logger.error(f"Error creating shout object: {e}", exc_info=True)
|
||
return {"error": f"Database error: {str(e)}"}
|
||
|
||
# Связываем с автором
|
||
try:
|
||
logger.debug(f"Linking author {author_id} to shout {new_shout.id}")
|
||
sa = ShoutAuthor(shout=new_shout.id, author=author_id)
|
||
session.add(sa)
|
||
except Exception as e:
|
||
logger.error(f"Error linking author: {e}", exc_info=True)
|
||
return {"error": f"Error linking author: {str(e)}"}
|
||
|
||
# Связываем с темами
|
||
|
||
input_topics = inp.get("topics", [])
|
||
if input_topics:
|
||
try:
|
||
logger.debug(f"Linking topics: {[t.slug for t in input_topics]}")
|
||
main_topic = inp.get("main_topic")
|
||
for topic in input_topics:
|
||
st = ShoutTopic(
|
||
topic=topic.id,
|
||
shout=new_shout.id,
|
||
main=(topic.slug == main_topic) if main_topic else False,
|
||
)
|
||
session.add(st)
|
||
logger.debug(f"Added topic {topic.slug} {'(main)' if st.main else ''}")
|
||
except Exception as e:
|
||
logger.error(f"Error linking topics: {e}", exc_info=True)
|
||
return {"error": f"Error linking topics: {str(e)}"}
|
||
|
||
try:
|
||
session.commit()
|
||
logger.info("Final commit successful")
|
||
except Exception as e:
|
||
logger.error(f"Error in final commit: {e}", exc_info=True)
|
||
return {"error": f"Error in final commit: {str(e)}"}
|
||
|
||
# Получаем созданную публикацию
|
||
shout = session.query(Shout).filter(Shout.id == new_shout.id).first()
|
||
|
||
# Подписываем автора
|
||
try:
|
||
logger.debug("Following created shout")
|
||
await follow(None, info, "shout", shout.slug)
|
||
except Exception as e:
|
||
logger.warning(f"Error following shout: {e}", exc_info=True)
|
||
|
||
logger.info(f"Successfully created shout {shout.id}")
|
||
return {"shout": shout}
|
||
|
||
except Exception as e:
|
||
logger.error(f"Unexpected error in create_shout: {e}", exc_info=True)
|
||
return {"error": f"Unexpected error: {str(e)}"}
|
||
|
||
error_msg = "cant create shout" if user_id else "unauthorized"
|
||
logger.error(f"Create shout failed: {error_msg}")
|
||
return {"error": error_msg}
|
||
|
||
|
||
def patch_main_topic(session, main_topic_slug, shout):
|
||
logger.info(f"Starting patch_main_topic for shout#{shout.id} with slug '{main_topic_slug}'")
|
||
|
||
with session.begin():
|
||
# Получаем текущий главный топик
|
||
old_main = (
|
||
session.query(ShoutTopic).filter(and_(ShoutTopic.shout == shout.id, ShoutTopic.main.is_(True))).first()
|
||
)
|
||
if old_main:
|
||
logger.info(f"Found current main topic: {old_main.topic}")
|
||
|
||
# Находим новый главный топик
|
||
main_topic = session.query(Topic).filter(Topic.slug == main_topic_slug).first()
|
||
if not main_topic:
|
||
logger.error(f"Main topic with slug '{main_topic_slug}' not found")
|
||
return
|
||
|
||
logger.info(f"Found new main topic: {main_topic.id}")
|
||
|
||
# Находим связь с новым главным топиком
|
||
new_main = (
|
||
session.query(ShoutTopic)
|
||
.filter(and_(ShoutTopic.shout == shout.id, ShoutTopic.topic == main_topic.id))
|
||
.first()
|
||
)
|
||
|
||
if old_main and new_main and old_main is not new_main:
|
||
logger.info("Updating main topic flags")
|
||
old_main.main = False
|
||
session.add(old_main)
|
||
|
||
new_main.main = True
|
||
session.add(new_main)
|
||
|
||
session.flush()
|
||
logger.info(f"Main topic updated for shout#{shout.id}")
|
||
|
||
|
||
def patch_topics(session, shout, topics_input):
|
||
logger.info(f"Starting patch_topics for shout#{shout.id}")
|
||
logger.info(f"Received topics_input: {topics_input}")
|
||
|
||
# Создаем новые топики если есть
|
||
new_topics_to_link = [Topic(**new_topic) for new_topic in topics_input if new_topic["id"] < 0]
|
||
if new_topics_to_link:
|
||
logger.info(f"Creating new topics: {[t.dict() for t in new_topics_to_link]}")
|
||
session.add_all(new_topics_to_link)
|
||
session.flush()
|
||
|
||
# Получаем текущие связи
|
||
current_links = session.query(ShoutTopic).filter(ShoutTopic.shout == shout.id).all()
|
||
logger.info(f"Current topic links: {[{t.topic: t.main} for t in current_links]}")
|
||
|
||
# Удаляем старые связи
|
||
if current_links:
|
||
logger.info(f"Removing old topic links for shout#{shout.id}")
|
||
for link in current_links:
|
||
session.delete(link)
|
||
session.flush()
|
||
|
||
# Создаем новые связи
|
||
for topic_input in topics_input:
|
||
topic_id = topic_input["id"]
|
||
if topic_id < 0:
|
||
topic = next(t for t in new_topics_to_link if t.slug == topic_input["slug"])
|
||
topic_id = topic.id
|
||
|
||
logger.info(f"Creating new topic link: shout#{shout.id} -> topic#{topic_id}")
|
||
new_link = ShoutTopic(shout=shout.id, topic=topic_id, main=False)
|
||
session.add(new_link)
|
||
|
||
session.flush()
|
||
# Обновляем связи в объекте шаута
|
||
session.refresh(shout)
|
||
|
||
logger.info(f"Successfully updated topics for shout#{shout.id}")
|
||
logger.info(f"Final shout topics: {[t.dict() for t in shout.topics]}")
|
||
|
||
|
||
@mutation.field("update_shout")
|
||
@login_required
|
||
async def update_shout(_, info, shout_id: int, shout_input=None, publish=False):
|
||
logger.info(f"Starting update_shout with id={shout_id}, publish={publish}")
|
||
logger.debug(f"Full shout_input: {shout_input}")
|
||
|
||
user_id = info.context.get("user_id")
|
||
roles = info.context.get("roles", [])
|
||
author_dict = info.context.get("author")
|
||
if not author_dict:
|
||
logger.error("Author profile not found")
|
||
return {"error": "author profile was not found"}
|
||
|
||
author_id = author_dict.get("id")
|
||
shout_input = shout_input or {}
|
||
current_time = int(time.time())
|
||
shout_id = shout_id or shout_input.get("id", shout_id)
|
||
slug = shout_input.get("slug")
|
||
|
||
if not user_id:
|
||
logger.error("Unauthorized update attempt")
|
||
return {"error": "unauthorized"}
|
||
|
||
try:
|
||
with local_session() as session:
|
||
if author_id:
|
||
logger.info(f"Processing update for shout#{shout_id} by author #{author_id}")
|
||
shout_by_id = (
|
||
session.query(Shout)
|
||
.options(joinedload(Shout.authors), joinedload(Shout.topics))
|
||
.filter(Shout.id == shout_id)
|
||
.first()
|
||
)
|
||
|
||
if not shout_by_id:
|
||
logger.error(f"shout#{shout_id} not found")
|
||
return {"error": "shout not found"}
|
||
|
||
logger.info(f"Found shout#{shout_id}")
|
||
|
||
# Логируем текущие топики
|
||
current_topics = (
|
||
[{"id": t.id, "slug": t.slug, "title": t.title} for t in shout_by_id.topics]
|
||
if shout_by_id.topics
|
||
else []
|
||
)
|
||
logger.info(f"Current topics for shout#{shout_id}: {current_topics}")
|
||
|
||
if slug != shout_by_id.slug:
|
||
same_slug_shout = session.query(Shout).filter(Shout.slug == slug).first()
|
||
c = 1
|
||
while same_slug_shout is not None:
|
||
c += 1
|
||
slug = f"{slug}-{c}"
|
||
same_slug_shout = session.query(Shout).filter(Shout.slug == slug).first()
|
||
shout_input["slug"] = slug
|
||
logger.info(f"shout#{shout_id} slug patched")
|
||
|
||
if filter(lambda x: x.id == author_id, [x for x in shout_by_id.authors]) or "editor" in roles:
|
||
logger.info(f"Author #{author_id} has permission to edit shout#{shout_id}")
|
||
|
||
# topics patch
|
||
topics_input = shout_input.get("topics")
|
||
if topics_input:
|
||
logger.info(f"Received topics_input for shout#{shout_id}: {topics_input}")
|
||
try:
|
||
patch_topics(session, shout_by_id, topics_input)
|
||
logger.info(f"Successfully patched topics for shout#{shout_id}")
|
||
|
||
# Обновляем связи в сессии после patch_topics
|
||
session.refresh(shout_by_id)
|
||
|
||
except Exception as e:
|
||
logger.error(f"Error patching topics: {e}", exc_info=True)
|
||
return {"error": f"Failed to update topics: {str(e)}"}
|
||
|
||
del shout_input["topics"]
|
||
for tpc in topics_input:
|
||
await cache_by_id(Topic, tpc["id"], cache_topic)
|
||
else:
|
||
logger.warning(f"No topics_input received for shout#{shout_id}")
|
||
|
||
# main topic
|
||
main_topic = shout_input.get("main_topic")
|
||
if main_topic:
|
||
logger.info(f"Updating main topic for shout#{shout_id} to {main_topic}")
|
||
patch_main_topic(session, main_topic, shout_by_id)
|
||
|
||
shout_input["updated_at"] = current_time
|
||
if publish:
|
||
logger.info(f"Publishing shout#{shout_id}")
|
||
shout_input["published_at"] = current_time
|
||
# Проверяем наличие связи с автором
|
||
logger.info(f"Checking author link for shout#{shout_id} and author#{author_id}")
|
||
author_link = (
|
||
session.query(ShoutAuthor)
|
||
.filter(and_(ShoutAuthor.shout == shout_id, ShoutAuthor.author == author_id))
|
||
.first()
|
||
)
|
||
|
||
if not author_link:
|
||
logger.info(f"Adding missing author link for shout#{shout_id}")
|
||
sa = ShoutAuthor(shout=shout_id, author=author_id)
|
||
session.add(sa)
|
||
session.flush()
|
||
logger.info("Author link added successfully")
|
||
else:
|
||
logger.info("Author link already exists")
|
||
|
||
# Логируем финальное состояние перед сохранением
|
||
logger.info(f"Final shout_input for update: {shout_input}")
|
||
Shout.update(shout_by_id, shout_input)
|
||
session.add(shout_by_id)
|
||
|
||
try:
|
||
session.commit()
|
||
# Обновляем объект после коммита чтобы получить все связи
|
||
session.refresh(shout_by_id)
|
||
logger.info(f"Successfully committed updates for shout#{shout_id}")
|
||
except Exception as e:
|
||
logger.error(f"Commit failed: {e}", exc_info=True)
|
||
return {"error": f"Failed to save changes: {str(e)}"}
|
||
|
||
# После обновления проверяем топики
|
||
updated_topics = (
|
||
[{"id": t.id, "slug": t.slug, "title": t.title} for t in shout_by_id.topics]
|
||
if shout_by_id.topics
|
||
else []
|
||
)
|
||
logger.info(f"Updated topics for shout#{shout_id}: {updated_topics}")
|
||
|
||
# Инвалидация кэша после обновления
|
||
try:
|
||
logger.info("Invalidating cache after shout update")
|
||
|
||
cache_keys = [
|
||
"feed", # лента
|
||
f"author_{author_id}", # публикации автора
|
||
"random_top", # случайные топовые
|
||
"unrated", # неоцененные
|
||
]
|
||
|
||
# Добавляем ключи для тем публикации
|
||
for topic in shout_by_id.topics:
|
||
cache_keys.append(f"topic_{topic.id}")
|
||
cache_keys.append(f"topic_shouts_{topic.id}")
|
||
|
||
await invalidate_shouts_cache(cache_keys)
|
||
await invalidate_shout_related_cache(shout_by_id, author_id)
|
||
|
||
# Обновляем кэш тем и авторов
|
||
for topic in shout_by_id.topics:
|
||
await cache_by_id(Topic, topic.id, cache_topic)
|
||
for author in shout_by_id.authors:
|
||
await cache_author(author.dict())
|
||
|
||
logger.info("Cache invalidated successfully")
|
||
except Exception as cache_error:
|
||
logger.warning(f"Cache invalidation error: {cache_error}", exc_info=True)
|
||
|
||
if not publish:
|
||
await notify_shout(shout_by_id.dict(), "update")
|
||
else:
|
||
await notify_shout(shout_by_id.dict(), "published")
|
||
# search service indexing
|
||
search_service.index(shout_by_id)
|
||
for a in shout_by_id.authors:
|
||
await cache_by_id(Author, a.id, cache_author)
|
||
logger.info(f"shout#{shout_id} updated")
|
||
# Получаем полные данные шаута со связями
|
||
shout_with_relations = (
|
||
session.query(Shout)
|
||
.options(joinedload(Shout.topics), joinedload(Shout.authors))
|
||
.filter(Shout.id == shout_id)
|
||
.first()
|
||
)
|
||
|
||
# Создаем словарь с базовыми полями
|
||
shout_dict = shout_with_relations.dict()
|
||
|
||
# Явно добавляем связанные данные
|
||
shout_dict["topics"] = (
|
||
[
|
||
{"id": topic.id, "slug": topic.slug, "title": topic.title}
|
||
for topic in shout_with_relations.topics
|
||
]
|
||
if shout_with_relations.topics
|
||
else []
|
||
)
|
||
|
||
shout_dict["authors"] = (
|
||
[
|
||
{"id": author.id, "name": author.name, "slug": author.slug}
|
||
for author in shout_with_relations.authors
|
||
]
|
||
if shout_with_relations.authors
|
||
else []
|
||
)
|
||
|
||
logger.info(f"Final shout data with relations: {shout_dict}")
|
||
return {"shout": shout_dict, "error": None}
|
||
else:
|
||
logger.warning(f"Access denied: author #{author_id} cannot edit shout#{shout_id}")
|
||
return {"error": "access denied", "shout": None}
|
||
|
||
except Exception as exc:
|
||
logger.error(f"Unexpected error in update_shout: {exc}", exc_info=True)
|
||
logger.error(f"Failed input data: {shout_input}")
|
||
return {"error": "cant update shout"}
|
||
|
||
return {"error": "cant update shout"}
|
||
|
||
|
||
@mutation.field("delete_shout")
|
||
@login_required
|
||
async def delete_shout(_, info, shout_id: int):
|
||
user_id = info.context.get("user_id")
|
||
roles = info.context.get("roles", [])
|
||
author_dict = info.context.get("author")
|
||
if not author_dict:
|
||
return {"error": "author profile was not found"}
|
||
author_id = author_dict.get("id")
|
||
if user_id and author_id:
|
||
author_id = int(author_id)
|
||
with local_session() as session:
|
||
shout = session.query(Shout).filter(Shout.id == shout_id).first()
|
||
if not isinstance(shout, Shout):
|
||
return {"error": "invalid shout id"}
|
||
shout_dict = shout.dict()
|
||
# NOTE: only owner and editor can mark the shout as deleted
|
||
if shout_dict["created_by"] == author_id or "editor" in roles:
|
||
shout_dict["deleted_at"] = int(time.time())
|
||
Shout.update(shout, shout_dict)
|
||
session.add(shout)
|
||
session.commit()
|
||
|
||
for author in shout.authors:
|
||
await cache_by_id(Author, author.id, cache_author)
|
||
info.context["author"] = author.dict()
|
||
info.context["user_id"] = author.user
|
||
unfollow(None, info, "shout", shout.slug)
|
||
|
||
for topic in shout.topics:
|
||
await cache_by_id(Topic, topic.id, cache_topic)
|
||
|
||
await notify_shout(shout_dict, "delete")
|
||
return {"error": None}
|
||
else:
|
||
return {"error": "access denied"}
|