11

app/services/admin_oem_branding.py

@@ -18,7 +18,12 @@ def _map_image_error(exc: BrandingImageError) -> AdminOemBrandingError:
     return AdminOemBrandingError(exc.message)
 
 
-async def get_or_create_default_oem(db: AsyncSession, *, oem_id: int,user_id:int) -> Oem:
+async def get_or_create_default_oem(
+    db: AsyncSession,
+    *,
+    oem_id: int = DEFAULT_OEM_ID,
+    user_id: int,
+) -> Oem:
     oem = await db.get(Oem, oem_id)
     if oem is not None:
         return oem
@@ -38,18 +43,23 @@ async def get_or_create_default_oem(db: AsyncSession, *, oem_id: int,user_id:int
     return oem
 
 
-async def get_default_oem_branding(db: AsyncSession, *, oem_id: int,user_id:int) -> Oem:
-    return await get_or_create_default_oem(db, oem_id=oem_id,user_id=user_id)
+async def get_default_oem_branding(
+    db: AsyncSession,
+    *,
+    oem_id: int = DEFAULT_OEM_ID,
+    user_id: int,
+) -> Oem:
+    return await get_or_create_default_oem(db, oem_id=oem_id, user_id=user_id)
 
 
 async def update_default_oem_branding(
     db: AsyncSession,
     body: OemBrandingUpdate,
     *,
-    oem_id: int,
+    oem_id: int = DEFAULT_OEM_ID,
     user_id: int,
 ) -> Oem:
-    oem = await get_or_create_default_oem(db, oem_id=oem_id,user_id=user_id)
+    oem = await get_or_create_default_oem(db, oem_id=oem_id, user_id=user_id)
 
     try:
         logo_path = resolve_image_field(