from sqlalchemy import Select
from sqlalchemy.ext.asyncio import AsyncSession

from models import AttributeSelect
from repositories.mixins import RepCrudMixin
from schemas.attr_select import UpdateAttrSelectSchema, CreateAttrSelectSchema


class AttrSelectRepository(
    RepCrudMixin[AttributeSelect, CreateAttrSelectSchema, UpdateAttrSelectSchema],
):
    session: AsyncSession
    entity_class = AttributeSelect

    def _process_get_all_stmt(self, stmt: Select) -> Select:
        return stmt.where(AttributeSelect.is_deleted.is_(False))