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