Skip to content

Commit b6832fe

Browse files
authored
Merge pull request #166 from superannotateai/re-design-sdk
Added project handeling
2 parents bc4248d + 416d8e1 commit b6832fe

File tree

2 files changed

+12
-7
lines changed

2 files changed

+12
-7
lines changed

src/superannotate/lib/core/usecases.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,12 @@ def execute(self):
125125
"team_id", self._team_id, EQ
126126
)
127127
projects = self._projects.get_all(condition)
128-
for project in projects:
129-
if project.name == self._name:
130-
self._response.data = project
128+
if not projects:
129+
self._response.errors = AppException("Project not found.")
130+
else:
131+
for project in projects:
132+
if project.name == self._name:
133+
self._response.data = project
131134
break
132135
return self._response
133136

src/superannotate/lib/infrastructure/controller.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,10 @@ def _get_project(self, name: str):
115115
team_id=self.team_id,
116116
projects=ProjectRepository(service=self._backend_client),
117117
)
118-
return use_case.execute().data
118+
response = use_case.execute()
119+
if response.errors:
120+
raise AppException(response.errors)
121+
return response.data
119122

120123
def _get_folder(self, project: ProjectEntity, name: str = None):
121124
name = self.get_folder_name(name)
@@ -624,9 +627,8 @@ def get_project_metadata(
624627
include_contributors: bool = False,
625628
include_complete_image_count: bool = False,
626629
):
627-
project = self.projects.get_one(
628-
uuid=self._get_project(project_name).uuid, team_id=self.team_id
629-
)
630+
project = self._get_project(project_name)
631+
630632
use_case = usecases.GetProjectMetadataUseCase(
631633
project=project,
632634
service=self._backend_client,

0 commit comments

Comments
 (0)