Done
This commit is contained in:
@ -2,17 +2,13 @@ from enum import Enum
|
||||
from typing import Annotated, Optional
|
||||
|
||||
import sqlalchemy
|
||||
from litestar.contrib.repository import FilterTypes
|
||||
from litestar.contrib.sqlalchemy.base import BigIntBase
|
||||
from litestar.contrib.sqlalchemy.dto import SQLAlchemyDTO
|
||||
from litestar.contrib.sqlalchemy.repository import SQLAlchemyAsyncRepository
|
||||
from litestar.contrib.sqlalchemy.repository.types import SelectT
|
||||
from litestar.dto import DTOConfig
|
||||
from sqlalchemy import ColumnElement
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
|
||||
from app.lib import service
|
||||
from app.lib.filters import ExactFilter
|
||||
from app.lib.filter_repository import FilterRepository
|
||||
|
||||
|
||||
class FiscalModuleEnum(str, Enum):
|
||||
@ -52,24 +48,9 @@ class FiscalPaymentMapping(BigIntBase):
|
||||
payment_device_code: Mapped[Optional[int]]
|
||||
|
||||
|
||||
class Repository(SQLAlchemyAsyncRepository[FiscalPaymentMapping]):
|
||||
class Repository(FilterRepository[FiscalPaymentMapping]):
|
||||
model_type = FiscalPaymentMapping
|
||||
|
||||
def _apply_filters(
|
||||
self, *filters: FilterTypes, apply_pagination: bool = True, statement: SelectT
|
||||
) -> SelectT:
|
||||
standard_filters = []
|
||||
for filter_ in filters:
|
||||
if isinstance(filter_, ExactFilter):
|
||||
field: ColumnElement = getattr(self.model_type, filter_.field_name)
|
||||
statement = statement.where(field == filter_.value)
|
||||
else:
|
||||
standard_filters.append(filter_)
|
||||
|
||||
return super()._apply_filters(
|
||||
*standard_filters, apply_pagination=apply_pagination, statement=statement
|
||||
)
|
||||
|
||||
|
||||
class Service(service.Service[FiscalPaymentMapping]):
|
||||
repository_type = Repository
|
||||
|
||||
Reference in New Issue
Block a user