Spaces:
Sleeping
Sleeping
File size: 1,518 Bytes
55d0d9e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | from sqlalchemy import select, cast, Integer
from sqlalchemy.ext.asyncio import AsyncSession
from app.models.models import Chapters
import logging
class ChapterService:
def __init__(self):
self.logger = logging.getLogger(__name__)
async def get_chapters(self, session: AsyncSession):
query = select(Chapters).where(Chapters.chapter.isnot("")).order_by(cast(Chapters.chapter, Integer).asc())
result = await session.execute(query)
return result.scalars().all()
async def get_appendix(self, session: AsyncSession):
query = select(Chapters).where(Chapters.appendix.isnot("")).order_by(Chapters.appendix.asc())
result = await session.execute(query)
return result.scalars().all()
async def get_chapter_appendix(self, session: AsyncSession, chapter: str):
query = select(Chapters).where(Chapters.chapter == chapter)
result = await session.execute(query)
return result.scalar_one_or_none()
async def get_chapter_appendix_by_search(self, session: AsyncSession, search: str):
if search.isdigit():
query = select(Chapters).where(Chapters.chapter == search)
result = await session.execute(query)
return result.scalar_one_or_none(), "chapter"
else:
query = select(Chapters).where(Chapters.appendix == search.upper())
result = await session.execute(query)
return result.scalar_one_or_none(), "appendix"
chapter_service = ChapterService() |