Add promo management and unified user lookup

db/dal/promo_code_dal.py

@@ -49,6 +49,27 @@ async def get_all_active_promo_codes(session: AsyncSession,
     return result.scalars().all()
 
 
+async def update_promo_code(session: AsyncSession, promo_id: int,
+                            update_data: Dict[str, Any]) -> Optional[PromoCode]:
+    promo = await get_promo_code_by_id(session, promo_id)
+    if not promo:
+        return None
+    for key, value in update_data.items():
+        setattr(promo, key, value)
+    await session.flush()
+    await session.refresh(promo)
+    return promo
+
+
+async def delete_promo_code(session: AsyncSession, promo_id: int) -> Optional[PromoCode]:
+    promo = await get_promo_code_by_id(session, promo_id)
+    if not promo:
+        return None
+    await session.delete(promo)
+    await session.flush()
+    return promo
+
+
 async def increment_promo_code_usage(
         session: AsyncSession, promo_code_id: int) -> Optional[PromoCode]:
     promo = await get_promo_code_by_id(session, promo_code_id)

db/dal/user_dal.py

@@ -30,6 +30,22 @@ async def get_user_by_panel_uuid(
     return result.scalar_one_or_none()
 
 
+async def get_user(
+    session: AsyncSession,
+    *,
+    user_id: Optional[int] = None,
+    username: Optional[str] = None,
+    panel_uuid: Optional[str] = None,
+) -> Optional[User]:
+    if user_id is not None:
+        return await get_user_by_id(session, user_id)
+    if username is not None:
+        return await get_user_by_username(session, username)
+    if panel_uuid is not None:
+        return await get_user_by_panel_uuid(session, panel_uuid)
+    return None
+
+
 async def create_user(session: AsyncSession, user_data: Dict[str, Any]) -> User:
 
     if "registration_date" not in user_data: