crud.py 720 B

1234567891011121314151617181920212223
  1. from app.api.v1.module_system.auth.schema import AuthSchema
  2. from app.core.base_crud import CRUDBase
  3. from .model import QuotaModel
  4. from .schema import QuotaCreateSchema, QuotaUpdateSchema
  5. class QuotaCRUD(CRUDBase[QuotaModel, QuotaCreateSchema, QuotaUpdateSchema]):
  6. """额度 CRUD 操作"""
  7. def __init__(self, auth: AuthSchema) -> None:
  8. self.auth = auth
  9. super().__init__(model=QuotaModel, auth=auth)
  10. async def get_by_out_biz_no(
  11. self, out_biz_no: str
  12. ) -> QuotaModel | None:
  13. return await self.get(out_biz_no=out_biz_no)
  14. async def get_by_employee_id(
  15. self, employee_id: str
  16. ) -> QuotaModel | None:
  17. return await self.get(employee_id=employee_id)