mirror of https://github.com/stijndcl/didier
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
from typing import Optional
|
|
|
|
import sqlalchemy.exc
|
|
from sqlalchemy import delete, func, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.crud.users import get_or_add_user
|
|
from database.exceptions import (
|
|
DuplicateInsertException,
|
|
Forbidden,
|
|
ForbiddenNameException,
|
|
NoResultFoundException,
|
|
)
|
|
from database.schemas import Bookmark
|
|
|
|
__all__ = ["create_bookmark", "get_bookmarks", "get_bookmark_by_name"]
|
|
|
|
|
|
async def create_bookmark(session: AsyncSession, user_id: int, label: str, jump_url: str) -> Bookmark:
|
|
"""Create a new bookmark to a message"""
|
|
# Don't allow bookmarks with names of subcommands
|
|
if label.lower() in ["create", "delete", "ls", "list", "Rm", "search"]:
|
|
raise ForbiddenNameException
|
|
|
|
await get_or_add_user(session, user_id)
|
|
|
|
try:
|
|
bookmark = Bookmark(label=label, jump_url=jump_url, user_id=user_id)
|
|
session.add(bookmark)
|
|
await session.commit()
|
|
await session.refresh(bookmark)
|
|
except sqlalchemy.exc.IntegrityError as e:
|
|
raise DuplicateInsertException from e
|
|
|
|
return bookmark
|
|
|
|
|
|
async def delete_bookmark_by_id(session: AsyncSession, user_id: int, bookmark_id: int):
|
|
"""Find a bookmark by its id & delete it
|
|
|
|
This fails if you don't own this bookmark
|
|
"""
|
|
select_statement = select(Bookmark).where(Bookmark.bookmark_id == bookmark_id)
|
|
bookmark = (await session.execute(select_statement)).scalar_one_or_none()
|
|
|
|
# No bookmark with this id
|
|
if bookmark is None:
|
|
raise NoResultFoundException
|
|
|
|
# You don't own this bookmark
|
|
if bookmark.user_id != user_id:
|
|
raise Forbidden
|
|
|
|
# Delete it
|
|
delete_statement = delete(Bookmark).where(Bookmark.bookmark_id == bookmark_id)
|
|
await session.execute(delete_statement)
|
|
await session.commit()
|
|
|
|
|
|
async def get_bookmarks(session: AsyncSession, user_id: int, *, query: Optional[str] = None) -> list[Bookmark]:
|
|
"""Get all a user's bookmarks"""
|
|
statement = select(Bookmark).where(Bookmark.user_id == user_id)
|
|
|
|
if query is not None:
|
|
statement = statement.where(Bookmark.label.ilike(f"%{query.lower()}%"))
|
|
|
|
return list((await session.execute(statement)).scalars().all())
|
|
|
|
|
|
async def get_bookmark_by_name(session: AsyncSession, user_id: int, query: str) -> Optional[Bookmark]:
|
|
"""Try to find a bookmark by its name"""
|
|
statement = select(Bookmark).where(Bookmark.user_id == user_id).where(func.lower(Bookmark.label) == query.lower())
|
|
return (await session.execute(statement)).scalar_one_or_none()
|