add retry_db to all pg_crud functions

externals/databases/pg_crud.py

@@ -149,6 +149,7 @@ async def mark_file_extracted(
     await db.commit()
 
 
+@retry_db(retries=2, delay=2)
 async def create_cv_file(
     db: AsyncSession,
     *,
@@ -182,6 +183,8 @@ async def delete_file_by_filename(
     await db.commit()
     return result.rowcount > 0
 
+
+@retry_db(retries=2, delay=2)
 async def get_file_by_user_id(
     db: AsyncSession,
     user_id: str,
@@ -191,6 +194,8 @@ async def get_file_by_user_id(
     )
     return result.scalars().all()
 
+
+@retry_db(retries=2, delay=2)
 async def get_file_by_filename(
     db: AsyncSession,
     filename: str,
@@ -207,6 +212,7 @@ async def get_file_by_filename(
     return result.scalar_one_or_none()
 
 
+@retry_db(retries=2, delay=2)
 async def count_files_by_user(
     db: AsyncSession,
     user_id: str,
@@ -218,6 +224,7 @@ async def count_files_by_user(
     return int(result.scalar_one())
 
 
+@retry_db(retries=2, delay=2)
 async def count_profiles_by_user(
     db: AsyncSession,
     user_id: str,
@@ -231,6 +238,7 @@ async def count_profiles_by_user(
     return int(result.scalar_one())
 
 
+@retry_db(retries=2, delay=2)
 async def mark_file_extracted(db: AsyncSession, file_id: UUID):
     await db.execute(
         update(CVFile)
@@ -253,6 +261,7 @@ async def mark_file_extracted(db: AsyncSession, file_id: UUID):
 
 from services.models.data_model import AIProfile
 
+@retry_db(retries=2, delay=2)
 async def create_profile(
     db: AsyncSession,
     filename: str,
@@ -290,6 +299,7 @@ async def create_profile(
     return profile
 
 
+@retry_db(retries=2, delay=2)
 async def get_profile_by_filename(
     db: AsyncSession,
     filename: str,
@@ -303,6 +313,8 @@ async def get_profile_by_filename(
     result = await db.execute(stmt)
     return result.scalar_one_or_none()
 
+
+@retry_db(retries=2, delay=2)
 async def get_profiles(
     db: AsyncSession,
 ) -> List[CVProfile]:
@@ -310,13 +322,14 @@ async def get_profiles(
     result = await db.execute(stmt)
     return result.scalars().all()
 
-from sqlalchemy import func
 
 from sqlalchemy import select, and_, func, cast, String
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.dialects.postgresql import ARRAY
 from typing import List
 
+
+@retry_db(retries=2, delay=2)
 async def get_profiles_by_criteria_id(
     db: AsyncSession,
     criteria_id: str,
@@ -448,7 +461,7 @@ async def get_profiles_by_criteria_id(
 #     result = await db.execute(stmt)
 #     return result.scalar_one_or_none()
 
-from typing import Tuple
+# from typing import Tuple
 
 # async def get_profile_by_id(
 #     db: AsyncSession,
@@ -469,7 +482,7 @@ from typing import Tuple
     
 #     return result 
 
-
+@retry_db(retries=2, delay=2)
 async def get_profile_by_id(
     db: AsyncSession,
     profile_id: str,
@@ -501,6 +514,7 @@ async def get_profile_by_id(
     return profile_dict
 
 
+@retry_db(retries=2, delay=2)
 async def list_profiles(
     db: AsyncSession,
     limit: int = 50,
@@ -517,6 +531,7 @@ async def list_profiles(
 from sqlalchemy import select, and_
 from sqlalchemy.ext.asyncio import AsyncSession
 
+@retry_db(retries=2, delay=2)
 async def get_filter(
     db: AsyncSession,
     filter: CVFilter,
@@ -577,6 +592,7 @@ async def get_filter(
     return result.scalar_one_or_none()
 
 
+@retry_db(retries=2, delay=2)
 async def get_filter_by_id(
     db: AsyncSession,
     criteria_id: str,
@@ -586,7 +602,7 @@ async def get_filter_by_id(
     return result.scalar_one_or_none()
 
 
-
+@retry_db(retries=2, delay=2)
 async def create_filter(
     db: AsyncSession,
     filter: CVFilter,
@@ -596,6 +612,7 @@ async def create_filter(
     return filter
     
 
+@retry_db(retries=2, delay=2)
 async def get_filter_and_weight(
     db: AsyncSession,
     criteria_id: str,
@@ -609,6 +626,7 @@ async def get_filter_and_weight(
     return result.scalar_one_or_none()
 
 
+@retry_db(retries=2, delay=2)
 async def get_weight_by_id(
     db: AsyncSession,
     weight_id: str,
@@ -620,6 +638,8 @@ async def get_weight_by_id(
     result = await db.execute(stmt)
     return result.scalar_one_or_none()
 
+
+@retry_db(retries=2, delay=2)
 async def create_filter_and_weight(
     db: AsyncSession,
     cv_filter: CVFilter,
@@ -628,6 +648,8 @@ async def create_filter_and_weight(
     db.add_all([cv_filter, cv_weight])
     await db.commit()
 
+
+@retry_db(retries=2, delay=2)
 async def create_weight(
     db: AsyncSession,
     cv_weight: CVWeight,
@@ -640,6 +662,7 @@ async def create_weight(
 # MATCHING
 # =========================
 
+@retry_db(retries=2, delay=2)
 async def create_matching(
     db: AsyncSession,
     matching: CVMatching,
@@ -650,8 +673,7 @@ async def create_matching(
     return matching
 
 
-from services.models.data_model import AIMatchProfile
-
+@retry_db(retries=2, delay=2)
 async def create_matchings(
     db: AsyncSession,
     matchings: list,
@@ -671,6 +693,7 @@ async def create_matchings(
         raise
 
 
+@retry_db(retries=2, delay=2)
 async def get_matching_by_profile_and_criteria(
     db: AsyncSession,
     profile_id: str,
@@ -692,7 +715,7 @@ async def get_matching_by_profile_and_criteria(
 # =========================
 # SCORE
 # =========================
-
+@retry_db(retries=2, delay=2)
 async def create_score(
     db: AsyncSession,
     score: CVScore,
@@ -703,6 +726,7 @@ async def create_score(
     return score
 
 
+@retry_db(retries=2, delay=2)
 async def create_scores(
     db: AsyncSession,
     scores: list[CVScore],
@@ -718,6 +742,7 @@ async def create_scores(
         raise
 
 
+@retry_db(retries=2, delay=2)
 async def get_scores_by_criteria(
     db: AsyncSession,
     criteria_id: str,