o i2@sdZddlmZmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZddlmZddlmZddlmZdd lmZdd lmZed ed Zed e d Zede d ZGdddeeeefZdS)aB Base CRUD operations with tenant awareness and permission filtering. Provides generic CRUD operations that can be inherited by domain-specific CRUD classes. Supports: - Tenant-aware filtering (automatic isolation) - Permission-based filtering (platform/customer admin/user) - Pagination - Soft delete - Batch operations )AnyDictGenericListOptionalTypeTypeVarUnion)jsonable_encoder) BaseModel)and_or_)Session)IntegrityError)Base)User)ConstraintViolationError ModelType)boundCreateSchemaTypeUpdateSchemaTypec@seZdZdZdeefddZ d4dedede e d e efd d Z d d ddddddede de de e de e eefde eded eefddZddddede e de e eefd e fddZdddddedede e de e ded ef dd Zdd!ded"edeee eeffded ef d#d$Zd%dd&dede d'eded e ef d(d)Zd%dd&ded*ee de d'eded e f d+d,Zde defd-d.Zd d dd/ded0ed1eede de de e d eefd2d3ZdS)5CRUDBasez Base CRUD class with generic database operations. Type Parameters: ModelType: SQLAlchemy model class CreateSchemaType: Pydantic schema for creation UpdateSchemaType: Pydantic schema for updates modelcCs ||_dS)zz Initialize CRUD object with a SQLAlchemy model. Args: model: SQLAlchemy model class N)r)selfrr7/lsinfo/ai/hellotax_ai/base_platform/./app/crud/base.py__init__&s zCRUDBase.__init__Ndbid current_userreturncCs4||j|jj|k}|r||||}|S)z Get a single record by ID. Args: db: Database session id: Record ID current_user: Current user for permission filtering Returns: Model instance or None if not found )queryrfilterr_apply_permission_filterfirst)rrrrr!rrrget/sz CRUDBase.getrdT)skiplimitrfiltersorder_by order_descr'r(r)r*r+c Cs||j}|r||||}|rY|D]C\} } t|j| rXt| tr1|t|j|  | }qt| t rM| drM| drM|t|j|  | }q|t|j| | k}q|rut|j|rut|j|} ||ro| n| }nt|jdr||jj}|||S)a Get multiple records with pagination and filtering. Args: db: Database session skip: Number of records to skip (offset) limit: Maximum number of records to return current_user: Current user for permission filtering filters: Dictionary of field:value filters order_by: Field name to order by (default: created_at if exists) order_desc: Whether to order descending (default: True) Returns: List of model instances % created_at)r!rr#itemshasattr isinstancelistr"getattrin_str startswithendswithliker*descascr-offsetr(all) rrr'r(rr)r*r+r!fieldvalue order_fieldrrr get_multiHs$     zCRUDBase.get_multi)rr)cCs||j}|r||||}|rY|D]C\}}t|j|rXt|tr1|t|j| |}qt|t rM| drM| drM|t|j| |}q|t|j||k}q|S)a Get count of records matching filters. Args: db: Database session current_user: Current user for permission filtering filters: Dictionary of field:value filters Returns: Count of matching records r,)r!rr#r.r/r0r1r"r2r3r4r5r6r7count)rrrr)r!r<r=rrr get_counts   zCRUDBase.get_count) created_by tenant_idcommitobj_inrBrCrDc Cst|}|rt|jdr||d<|durt|jdr||d<|jdi|}|||rjz||||||||W|Styi}z| t|dr]t |j nt |} t | d}~ww|S)a Create a new record. Args: db: Database session obj_in: Pydantic schema with creation data created_by: User ID who created the record tenant_id: Tenant ID for multi-tenant isolation commit: Whether to commit the transaction (default: True) Returns: Created model instance Raises: ConstraintViolationError: If database constraint is violated rBNrCorigr) r r/raddflushrDexpungerefreshrrollbackr4rFr) rrrErBrCrD obj_in_datadb_objeconstraint_namerrrcreates,    zCRUDBase.create)rDrMcCsjt|}t|tr |}n|jdd}|D]}||vr"t||||q|||r3||||S)aJ Update an existing record. Args: db: Database session db_obj: Existing model instance to update obj_in: Pydantic schema or dict with update data commit: Whether to commit the transaction (default: True) Returns: Updated model instance T) exclude_unset)r r0dict model_dumpsetattrrGrDrJ)rrrMrErDobj_data update_datar<rrrupdates    zCRUDBase.updateF)softrDrXcCsb||j|jj|k}|sdS|r$t|jdr$d|_||n|||r/| |S)ac Delete a record (hard or soft delete). Args: db: Database session id: Record ID to delete soft: Whether to perform soft delete (set is_deleted=True) commit: Whether to commit the transaction (default: True) Returns: Deleted model instance or None if not found N is_deletedT) r!rr"rr$r/rYrGdeleterD)rrrrXrDobjrrrrZs  zCRUDBase.deleteidsc Cs||j|jj|}t|jdr||jj|k}t|jdr2|jdkr2||jj|jk}|rDt|jdrD|j ddidd}n|j dd}|rP| |S)a Delete multiple records (batch operation). Args: db: Database session ids: List of record IDs to delete tenant_id: Tenant ID for multi-tenant isolation (required) current_user: Current user for ownership verification soft: Whether to perform soft delete commit: Whether to commit the transaction (default: True) Returns: Number of records deleted rCrBplatform_adminrYTF)synchronize_session) r!rr"rr3r/rCrolerBrWrZrD) rrr\rCrrXrDr!r@rrr delete_multi(s  zCRUDBase.delete_multicCsPt|jdr||jjdk}|jdkr|St|jdr&||jj|jk}|S)a Apply permission-based filtering to query. Filtering rules: - platform_admin: See all records - Others: See only records from their tenant - Automatically filter soft-deleted records Args: query: SQLAlchemy query object current_user: Current user db: Database session Returns: Filtered query object rYFr]rC)r/rr"rYr_rC)rr!rrrrrr#Vs   z!CRUDBase._apply_permission_filter)r'r(r search_term search_fieldsc Cs~||j}|r||||}g}|D]} t|j| r+|t|j| d|dq|r5|t|}| | | S)a Search records across multiple fields. Args: db: Database session search_term: Search term search_fields: List of field names to search in skip: Number of records to skip limit: Maximum number of records to return current_user: Current user for permission filtering Returns: List of matching model instances r,) r!rr#r/appendr2r7r"r r:r(r;) rrrarbr'r(rr!search_conditionsr<rrrsearchzs  zCRUDBase.search)N)__name__ __module__ __qualname____doc__rrrrrrrr%intrr4boolrr?rArrPr rrWrZr`r#rerrrrrs     < * < , - . * rN)ritypingrrrrrrrr fastapi.encodersr pydanticr sqlalchemyr r sqlalchemy.ormrsqlalchemy.excrapp.models.baserapp.models.userrapp.core.exceptionsrrrrrrrrrs(