83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
from operator import and_
|
|
from graphql import GraphQLError
|
|
from sqlalchemy import delete, insert, select
|
|
|
|
from orm.author import AuthorBookmark
|
|
from orm.shout import Shout
|
|
from resolvers.feed import apply_options
|
|
from resolvers.reader import get_shouts_with_links, has_field, query_with_stat
|
|
from services.auth import login_required
|
|
from services.common_result import CommonResult
|
|
from services.db import local_session
|
|
from services.schema import mutation, query
|
|
|
|
|
|
@query.field("load_shouts_bookmarked")
|
|
@login_required
|
|
def load_shouts_bookmarked(_, info, options):
|
|
"""
|
|
Load bookmarked shouts for the authenticated user.
|
|
|
|
Args:
|
|
limit (int): Maximum number of shouts to return.
|
|
offset (int): Number of shouts to skip.
|
|
|
|
Returns:
|
|
list: List of bookmarked shouts.
|
|
"""
|
|
author_dict = info.context.get("author", {})
|
|
author_id = author_dict.get("id")
|
|
if not author_id:
|
|
raise GraphQLError("User not authenticated")
|
|
|
|
q = query_with_stat() if has_field(info, "stat") else select(Shout).filter(and_(Shout.published_at.is_not(None), Shout.deleted_at.is_(None)))
|
|
q = q.join(AuthorBookmark)
|
|
q = q.filter(
|
|
and_(
|
|
Shout.id == AuthorBookmark.shout,
|
|
AuthorBookmark.author == author_id,
|
|
)
|
|
)
|
|
q, limit, offset = apply_options(q, options, author_id)
|
|
return get_shouts_with_links(info, q, limit, offset)
|
|
|
|
|
|
@mutation.field("toggle_bookmark_shout")
|
|
def toggle_bookmark_shout(_, info, slug: str) -> CommonResult:
|
|
"""
|
|
Toggle bookmark status for a specific shout.
|
|
|
|
Args:
|
|
slug (str): Unique identifier of the shout.
|
|
|
|
Returns:
|
|
CommonResult: Result of the operation with bookmark status.
|
|
"""
|
|
author_dict = info.context.get("author", {})
|
|
author_id = author_dict.get("id")
|
|
if not author_id:
|
|
raise GraphQLError("User not authenticated")
|
|
|
|
with local_session() as db:
|
|
shout = db.query(Shout).filter(Shout.slug == slug).first()
|
|
if not shout:
|
|
raise GraphQLError("Shout not found")
|
|
|
|
existing_bookmark = (
|
|
db.query(AuthorBookmark)
|
|
.filter(AuthorBookmark.author == author_id, AuthorBookmark.shout == shout.id)
|
|
.first()
|
|
)
|
|
|
|
if existing_bookmark:
|
|
db.execute(
|
|
delete(AuthorBookmark).where(AuthorBookmark.author == author_id, AuthorBookmark.shout == shout.id)
|
|
)
|
|
result = False
|
|
else:
|
|
db.execute(insert(AuthorBookmark).values(author=author_id, shout=shout.id))
|
|
result = True
|
|
|
|
db.commit()
|
|
return result
|