mirror of https://github.com/stijndcl/didier
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from sqlalchemy import func, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.exceptions import NoResultFoundException
|
|
from database.schemas import Link
|
|
|
|
__all__ = ["add_link", "edit_link", "get_all_links", "get_link_by_name"]
|
|
|
|
|
|
async def get_all_links(session: AsyncSession) -> list[Link]:
|
|
"""Get a list of all links"""
|
|
statement = select(Link)
|
|
return list((await session.execute(statement)).scalars().all())
|
|
|
|
|
|
async def add_link(session: AsyncSession, name: str, url: str) -> Link:
|
|
"""Add a new link into the database"""
|
|
if name.islower():
|
|
name = name.capitalize()
|
|
|
|
instance = Link(name=name, url=url)
|
|
session.add(instance)
|
|
await session.commit()
|
|
|
|
return instance
|
|
|
|
|
|
async def get_link_by_name(session: AsyncSession, name: str) -> Optional[Link]:
|
|
"""Get a link by its name"""
|
|
statement = select(Link).where(func.lower(Link.name) == name.lower())
|
|
return (await session.execute(statement)).scalar_one_or_none()
|
|
|
|
|
|
async def edit_link(session: AsyncSession, name: str, new_url: str):
|
|
"""Edit an existing link"""
|
|
link: Optional[Link] = await get_link_by_name(session, name)
|
|
|
|
if link is None:
|
|
raise NoResultFoundException
|
|
|
|
link.url = new_url
|
|
session.add(link)
|
|
await session.commit()
|