Spaces:
Sleeping
Sleeping
Increase db retries to 2
Browse files
externals/databases/pg_crud.py
CHANGED
|
@@ -25,7 +25,7 @@ from externals.databases.schemas.user import UserCreate, UserResponse
|
|
| 25 |
from utils.security import hash_password
|
| 26 |
|
| 27 |
|
| 28 |
-
@retry_db(retries=
|
| 29 |
async def get_user_by_username(
|
| 30 |
db: AsyncSession,
|
| 31 |
username: str,
|
|
@@ -36,7 +36,7 @@ async def get_user_by_username(
|
|
| 36 |
return result.scalar_one_or_none()
|
| 37 |
|
| 38 |
|
| 39 |
-
@retry_db(retries=
|
| 40 |
async def create_user(
|
| 41 |
db: AsyncSession,
|
| 42 |
user_in: UserCreate,
|
|
@@ -60,7 +60,7 @@ async def create_user(
|
|
| 60 |
return user
|
| 61 |
|
| 62 |
|
| 63 |
-
@retry_db(retries=
|
| 64 |
async def deactivate_user(
|
| 65 |
db: AsyncSession,
|
| 66 |
username: str,
|
|
@@ -80,7 +80,7 @@ async def deactivate_user(
|
|
| 80 |
return user
|
| 81 |
|
| 82 |
|
| 83 |
-
@retry_db(retries=
|
| 84 |
async def get_user_by_email(db, email: str):
|
| 85 |
result = await db.execute(
|
| 86 |
select(CVUser).where(CVUser.email == email)
|
|
@@ -88,7 +88,7 @@ async def get_user_by_email(db, email: str):
|
|
| 88 |
return result.scalar_one_or_none()
|
| 89 |
|
| 90 |
|
| 91 |
-
@retry_db(retries=
|
| 92 |
async def get_user_by_id(db, user_id: str):
|
| 93 |
result = await db.execute(
|
| 94 |
select(CVUser).where(CVUser.user_id == user_id)
|
|
@@ -102,7 +102,7 @@ async def get_user_by_id(db, user_id: str):
|
|
| 102 |
from externals.databases.schemas.tenant import TenantCreate
|
| 103 |
|
| 104 |
|
| 105 |
-
@retry_db(retries=
|
| 106 |
async def get_tenant_by_name(
|
| 107 |
db: AsyncSession,
|
| 108 |
tenant_name: str,
|
|
@@ -113,7 +113,7 @@ async def get_tenant_by_name(
|
|
| 113 |
return result.scalar_one_or_none()
|
| 114 |
|
| 115 |
|
| 116 |
-
@retry_db(retries=
|
| 117 |
async def create_tenant(
|
| 118 |
db: AsyncSession,
|
| 119 |
tenant_in: TenantCreate,
|
|
@@ -135,7 +135,7 @@ async def create_tenant(
|
|
| 135 |
# FILE
|
| 136 |
# =========================
|
| 137 |
|
| 138 |
-
@retry_db(retries=
|
| 139 |
async def mark_file_extracted(
|
| 140 |
db: AsyncSession,
|
| 141 |
file_id: UUID,
|
|
@@ -172,7 +172,7 @@ async def create_cv_file(
|
|
| 172 |
return cv_file
|
| 173 |
|
| 174 |
|
| 175 |
-
@retry_db(retries=
|
| 176 |
async def delete_file_by_filename(
|
| 177 |
db: AsyncSession,
|
| 178 |
filename: str,
|
|
|
|
| 25 |
from utils.security import hash_password
|
| 26 |
|
| 27 |
|
| 28 |
+
@retry_db(retries=2, delay=2)
|
| 29 |
async def get_user_by_username(
|
| 30 |
db: AsyncSession,
|
| 31 |
username: str,
|
|
|
|
| 36 |
return result.scalar_one_or_none()
|
| 37 |
|
| 38 |
|
| 39 |
+
@retry_db(retries=2, delay=2)
|
| 40 |
async def create_user(
|
| 41 |
db: AsyncSession,
|
| 42 |
user_in: UserCreate,
|
|
|
|
| 60 |
return user
|
| 61 |
|
| 62 |
|
| 63 |
+
@retry_db(retries=2, delay=2)
|
| 64 |
async def deactivate_user(
|
| 65 |
db: AsyncSession,
|
| 66 |
username: str,
|
|
|
|
| 80 |
return user
|
| 81 |
|
| 82 |
|
| 83 |
+
@retry_db(retries=2, delay=2)
|
| 84 |
async def get_user_by_email(db, email: str):
|
| 85 |
result = await db.execute(
|
| 86 |
select(CVUser).where(CVUser.email == email)
|
|
|
|
| 88 |
return result.scalar_one_or_none()
|
| 89 |
|
| 90 |
|
| 91 |
+
@retry_db(retries=2, delay=2)
|
| 92 |
async def get_user_by_id(db, user_id: str):
|
| 93 |
result = await db.execute(
|
| 94 |
select(CVUser).where(CVUser.user_id == user_id)
|
|
|
|
| 102 |
from externals.databases.schemas.tenant import TenantCreate
|
| 103 |
|
| 104 |
|
| 105 |
+
@retry_db(retries=2, delay=2)
|
| 106 |
async def get_tenant_by_name(
|
| 107 |
db: AsyncSession,
|
| 108 |
tenant_name: str,
|
|
|
|
| 113 |
return result.scalar_one_or_none()
|
| 114 |
|
| 115 |
|
| 116 |
+
@retry_db(retries=2, delay=2)
|
| 117 |
async def create_tenant(
|
| 118 |
db: AsyncSession,
|
| 119 |
tenant_in: TenantCreate,
|
|
|
|
| 135 |
# FILE
|
| 136 |
# =========================
|
| 137 |
|
| 138 |
+
@retry_db(retries=2, delay=2)
|
| 139 |
async def mark_file_extracted(
|
| 140 |
db: AsyncSession,
|
| 141 |
file_id: UUID,
|
|
|
|
| 172 |
return cv_file
|
| 173 |
|
| 174 |
|
| 175 |
+
@retry_db(retries=2, delay=2)
|
| 176 |
async def delete_file_by_filename(
|
| 177 |
db: AsyncSession,
|
| 178 |
filename: str,
|