7>hhddlmZddlmZddlmZddlZejeZ GddZ y))Optional)Session) HTTPExceptionNc |eZdZdZdefdZdededdfdZd e d ed ede e fd Z d e d e de e fdZ d e dede fdZy) BaseServicezFBase service class that provides common functionality for all servicesdbc||_y)N)r)selfrs K/var/www/html/DP/alpha_backend/app/advisor_service/services/base_service.py__init__zBaseService.__init__ s error operationreturnNcjtjd|dt|tdd|)z Handle database errors in a consistent way Args: error: The exception that was raised operation: Description of the operation that failed zDatabase error during z: izDatabase error occurred during ) status_codedetail)loggerrstrr)r rrs r handle_db_errorzBaseService.handle_db_errors9  -i[3u:,GH6UV_U`4abbr model_classfieldvaluec |jj|jt|||k(j S#t $r,}|j |d|jd|Yd}~yd}~wwxYw)z< Generic method to get an entity by a field getting z by N)rqueryfiltergetattrfirst Exceptionr__name__)r rrres r get_by_fieldzBaseService.get_by_fieldsu R77==-44W[%5PTY5YZ``b b R  h{/C/C.DD$P Q Q RsAA A<"A77A<idc |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_idzBaseService.get_by_id"sR 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)raddcommitrefreshr rollbackrr!)r rr'entityr"s r createzBaseService.create2s H (4(F GGKK  GGNN  GGOOF #M H GG     i 0D0D/E$F G G HsAA B(%9B##B()r! __module__ __qualname____doc__rr r rranyrr#intr&dictr/r)r r rrsP7 cY c3 c4 cRRCRRQT RMSMcMhsmM H#HTHcHr r) typingrsqlalchemy.ormrfastapirlogging getLoggerr!rrr)r r r;s/"!   8 $=H=Hr