BIh |ddlZddlmZddlmZddlmZddlZddlm Z eje Z GddZ y)N)Optional)Session) HTTPException)UserceZdZdZddefdZdededdfdZd e d ed ede e fd Z dd e d e de dedede e f dZ d e de de e fdZd e dede fdZy)BaseRepoz@Base repo class that provides common functionality for all reposNdbc ||_||_yN)r current_user)selfr r s ii :; $$ridc |j|d|S#t$r*}|j|d|jdYd}~yd}~wwxYw)z Generic method to get an entity by ID Args: model_class: The SQLAlchemy model class id: The ID to look up Returns: The found entity or None r:rz by IDN)r&r#rr$)r rr:r%s r get_by_idzBaseRepo.get_by_idDsR M$$[$; ; M  h{/C/C.DF$K L L Ms A AAdatacV |di|}|jj||jj|jj||S#t$rC}|jj |j |d|jYd}~yd}~wwxYw)z Generic method to create an entity Args: model_class: The SQLAlchemy model class data: Dict containing the entity data Returns: The created entity z creating N)r addcommitrefreshr#rollbackrr$)r rr=entityr%s rcreatezBaseRepo.createTs H (4(F GGKK  GGNN  GGOOF #M H GG     i 0D0D/E$F G G HsAA B(%9B##B(r ) created_atr-N)r$ __module__ __qualname____doc__rrr#rranyrr&intr9r<dictrEr?rrrr sJ)7) cY c3 c4 cRRCRRQT R%S%%%WZ%|%gopsgt%<MSMcMhsmM H#HTHcHrr)r3typingrsqlalchemy.ormrfastapirloggingapp.common.modelsr getLoggerr$rrr?rrrSs7 "!"   8 $\H\Hr