Spaces:
Sleeping
Sleeping
File size: 2,195 Bytes
338036b | 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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | from .BaseDataModel import BaseDataModel
from .db_schemes.minirag.schemes import Project
from sqlalchemy.future import select
from sqlalchemy import func
class ProjectModel(BaseDataModel):
def __init__(self, db_client: object):
super().__init__(db_client=db_client)
self.db_client = db_client
@classmethod
async def create_instance(cls, db_client: object):
instance = cls(db_client) # class init
return instance
async def create_project(self, project: Project):
async with self.db_client() as session:
async with session.begin():
session.add(project)
await session.commit()
await session.refresh(project)
return project
async def get_project_or_create_one(self, project_id: str):
async with self.db_client() as session:
async with session.begin():
query = select(Project).where(Project.project_id==project_id)
result = await session.execute(query)
project = result.scalar_one_or_none()
if project == None:
# create project
project_rec = Project(
project_id = project_id
)
project = await self.create_project(project=project_rec)
return project
else:
return project
async def get_all_projects(self, page: int=1, page_size: int=10):
async with self.db_client() as session:
async with session.begin():
total_documents = await session.execute(select(
func.count(Project.roject_id)
))
total_documents = total_documents.scalar_one()
total_pages = total_documents // page_size
if total_documents % total_pages > 0:
total_pages += 1
query = select(Project).offset((page-1)*page_size).limit(page_size)
projects = await session.execute(query).scalars().all()
return projects,total_pages
|