Skip to content

Commit 07c92a8

Browse files
committed
add checks for title on usecase update
1 parent 826dd14 commit 07c92a8

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

api/schema/usecase_schema.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import strawberry_django
1111
from django.db import models
1212
from strawberry import auto
13+
from strawberry.file_uploads import Upload
1314
from strawberry.types import Info
1415
from strawberry_django.mutations import mutations
1516
from strawberry_django.pagination import OffsetPaginationInput
@@ -59,13 +60,16 @@ class UpdateUseCaseMetadataInput:
5960
sectors: List[uuid.UUID]
6061

6162

63+
use_case_running_status = strawberry.enum(UseCaseStatus) # type: ignore
64+
65+
6266
@strawberry_django.partial(UseCase, fields="__all__", exclude=["datasets"])
6367
class UseCaseInputPartial:
6468
"""Input type for use case updates."""
6569

6670
id: str
67-
logo: auto
68-
running_status: auto
71+
logo: Optional[Upload] = strawberry.field(default=None)
72+
running_status: Optional[use_case_running_status] = UseCaseStatus.DRAFT
6973
title: Optional[str] = None
7074
summary: Optional[str] = None
7175
platform_url: Optional[str] = None
@@ -367,11 +371,14 @@ def update_use_case(self, info: Info, data: UseCaseInputPartial) -> TypeUseCase:
367371
usecase.platform_url = data.platform_url.strip()
368372
if data.started_on is not None:
369373
usecase.started_on = data.started_on
370-
if data.completed_on is not None:
374+
if data.completed_on is not None and data.completed_on is not strawberry.UNSET:
371375
usecase.completed_on = data.completed_on
372-
if data.running_status is not None:
373-
usecase.running_status = data.running_status.value
374-
if data.logo is not None:
376+
if (
377+
data.running_status is not None
378+
and data.running_status is not strawberry.UNSET
379+
):
380+
usecase.running_status = data.running_status
381+
if data.logo is not None and data.logo is not strawberry.UNSET:
375382
usecase.logo = data.logo
376383
usecase.save()
377384
return TypeUseCase.from_django(usecase)

0 commit comments

Comments
 (0)