from typing import List, Optional

from sqlalchemy.orm import Session

from app.crud.base import CRUDBase
from app.models.task import Task
from app.schemas.task import TaskCreate, TaskUpdate


class CRUDTask(CRUDBase[Task, TaskCreate, TaskUpdate]):
    def get_multi_by_owner(self, db: Session, *, owner_id: int, skip: int = 0, limit: int = 100) -> List[Task]:
        return db.query(self.model).filter(Task.owner_id == owner_id).offset(skip).limit(limit).all()

    def create_with_owner(self, db: Session, *, obj_in: TaskCreate, owner_id: int) -> Task:
        obj_in_data = obj_in.model_dump()
        db_obj = self.model(**obj_in_data, owner_id=owner_id)
        db.add(db_obj)
        db.commit()
        db.refresh(db_obj)
        return db_obj

    def get_by_id_and_owner(self, db: Session, *, task_id: int, owner_id: int) -> Optional[Task]:
        return db.query(self.model).filter(Task.id == task_id, Task.owner_id == owner_id).first()


task = CRUDTask(Task)