=h#ddlZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZmZmZddlmZmZedZedZGd d eeeefZy) N)ABCabstractmethod)GenericTypeVarListOptionalTypeUnionAny)Session)selectdeletefunc) DatabaseErrorNotFoundExceptionTDBModelc ~eZdZdZdedeefdZddee dee de e fd Z de fd Z d ee efdee fd Zd ee efdeefd Zddededee eezfdZde de fdZd ee efde de fdZd ee efdefdZdedefdZde de fdZde dede fdZedeedee fdZede defdZy)BaseRepositoryz Abstract base repository for handling database operations using SQLAlchemy. Provides generic methods for CRUD operations while ensuring proper error handling. db_session model_classc ||_||_y)z Initializes the repository with a database session and model class. :param db_session: SQLAlchemy database session. :param model_class: The SQLAlchemy model class associated with the repository. N)dbr)selfrrs J/var/www/html/phonemate/phone_mate_backend/repositories/base_repository.py__init__zBaseRepository.__init__s&NstartlimitreturncK t|j}||j|}||dkDr|j|}|jj |}|j j}|Dcgc]}|j|c}Scc}w#t$r2tjtjtwxYww)aK Retrieves all records of the given model with optional pagination. :param start: Offset for pagination. :param limit: Maximum number of records to retrieve. :return: List of Pydantic entities representing the database records. :raises DatabaseError: If a database operation fails. r)r roffsetrrexecutescalarsall _to_entity Exceptionloggingerror traceback format_excr)rrrqueryresult db_modelsmodels rget_allzBaseRepository.get_alls "4++,E  U+ UQY E*WW__U+F(,,.I8AB uDOOE* B BB " MM)..0 1/ ! "s/CA=B!BB!CB!!;CCcK ttjj|j}|j j |}|jS#t$r}t|d}~wwxYww)z Retrieves the total count of records in the table. :return: Total number of records. :raises DatabaseError: If a database operation fails. N) r rcount select_fromrrr# scalar_oner'r)rr,r-es r get_all_countzBaseRepository.get_all_count6sf )4::<(44T5E5EFEWW__U+F$$& & )/q ( )s)BA A%$B% A>. A99A>>Bidc0K t|jj|jj|k(}|jj |}|j }|r|j|SdS#t$r twxYww)z Retrieves a record by its ID. :param id: The unique identifier of the record. :return: The corresponding Pydantic entity, or None if not found. :raises DatabaseError: If a database operation fails. N) r rwherer7rr#scalar_one_or_noner&r'r)rr7r,r-db_models r get_by_idzBaseRepository.get_by_idDs "4++,2243C3C3F3F"3LMEWW__U+F002H084??8, Bd B "/ ! "s)BA8A?<B=A?>B?BBcK t|jj|jj|k(}|jj |}|j S#t$r}td}~wwxYww)a  Retrieves a raw SQLAlchemy model by its ID without conversion. :param id: The unique identifier of the record. :return: The SQLAlchemy model instance, or None if not found. :raises DatabaseError: If a database operation fails. N) r rr9r7rr#r:r'r)rr7r,r-r5s r get_by_id_rawzBaseRepository.get_by_id_rawTso "4++,2243C3C3F3F"3LMEWW__U+F,,. . "/ ! "s)BA%A*)B* B3 A==BB field_namevaluecK t|j|}t|jj||k(}|jj |}|j }|r|r|j|SdS|r|SdS#t$r2tjtjtwxYww)az Retrieves a record by a specific field. :param field_name: The field to filter by. :param value: The value to match. :param convert_to_entity: True/False weather to convert db model to entity or not. :return: The corresponding Pydantic entity, or None if not found. :raises DatabaseError: If a database operation fails. N)getattrrr r9rr#r:r&r'r(r)r*r+r)rr?r@convert_to_entitycolumnr,r-r;s r get_by_fieldzBaseRepository.get_by_fieldcs "T--z:F4++,226U?CEWW__U+F002H 4C E; A,E66E;;E>cK|j|}|t t|jj |jj |k(}|j j||j jy#t$r)}|j jtd}~wwxYww)a Deletes a record by its ID. :param id: The unique identifier of the record to delete. :raises DatabaseError: If a database operation fails. :raises NotFoundException: If a model class record not found against that id. NT) r<rrrr9r7rr#rKr'rNr)rr7rr,r5s rrzBaseRepository.deletesnnR(  # # "4++,2243C3C3F3F"3LME GGOOE " GGNN  " GG   / ! "s)CA0B C C$B;;CCr;c|jjDcic]\}}|jds||c}}Scc}}w)z Removes private and SQLAlchemy-specific fields from a model before an update. :param db_model: The SQLAlchemy model instance. :return: A dictionary of cleaned attributes. _)__dict__rS startswith)rr;rVr@s rrRz$BaseRepository._get_clean_attributessL*2):):)@)@)B )B:3>>#& J)B   sAc|S)z Hook method for preprocessing an entity before saving. Override in subclasses to implement custom logic. :param entity: The Pydantic entity before saving. :return: The processed entity. rrFs rrHzBaseRepository.before_saves  rc$|j|S)a! Hook method for post-processing an entity after saving. Override in subclasses to implement custom logic. :param entity: The Pydantic entity before saving. :param db_model: The saved SQLAlchemy model. :return: The final processed entity. )r&)rrFr;s rrMzBaseRepository.after_savesx((rcy)z Converts a database model instance into a Pydantic entity. :param db_model: The SQLAlchemy model instance. :return: The corresponding Pydantic entity. Nr^)rr;s rr&zBaseRepository._to_entity rcy)z Converts a Pydantic entity into a database model. :param entity: The Pydantic entity. :return: The corresponding SQLAlchemy model instance. Nr^r_s rrIzBaseRepository._to_db_modelrbr)NN)T)__name__ __module__ __qualname____doc__r r rrrintrrr0r6r rOr<r>r rErPrWboolrdictrRrHrMrr&rIr^rrrrs '7'g'"8C=" "Y]^_Y`". )S )"%S/"hqk" "eCHo "(7:K ""S""YabcYdgopwgxYx","1""*"uS#X""a"<"uS#X"4"*  g  $   !   ) )g )! ) 8G#4 !   1   rr)r(r*abcrrtypingrrrrr r r sqlalchemy.ormr sqlalchemyr rrutils.exceptionsrrrrrr^rrrpsO#EEE"++= CL ) h S'!W*-h r