Merge pull request #763 from Labelbox/lgluszek/CCV-2223

[CCV-2223] Dont allow to call setup_editor multiple times

Author: msokoloff1

labelbox/schema/project.py

@@ -12,7 +12,7 @@
 
 from labelbox import utils
 from labelbox.exceptions import (InvalidQueryError, LabelboxError,
-                                 ProcessingWaitTimeout)
+                                 ProcessingWaitTimeout, ResourceConflict)
 from labelbox.orm import query
 from labelbox.orm.db_object import DbObject, Deletable, Updateable
 from labelbox.orm.model import Entity, Field, Relationship
@@ -511,6 +511,9 @@ def setup_editor(self, ontology) -> None:
         Args:
             ontology (Ontology): The ontology to attach to the project
         """
+        if self.labeling_frontend() is not None:
+            raise ResourceConflict("Editor is already set up.")
+
         labeling_frontend = next(
             self.client.get_labeling_frontends(
                 where=Entity.LabelingFrontend.name == "Editor"))
@@ -550,6 +553,9 @@ def setup(self, labeling_frontend, labeling_frontend_options) -> None:
                 to `str` using `json.dumps`.
         """
 
+        if self.labeling_frontend() is not None:
+            raise ResourceConflict("Editor is already set up.")
+
         if not isinstance(labeling_frontend_options, str):
             labeling_frontend_options = json.dumps(labeling_frontend_options)
 

tests/integration/test_project_setup.py

@@ -6,7 +6,7 @@
 import pytest
 
 from labelbox import LabelingFrontend
-from labelbox.exceptions import InvalidQueryError
+from labelbox.exceptions import InvalidQueryError, ResourceConflict
 
 
 def simple_ontology():
@@ -67,3 +67,12 @@ def test_project_editor_setup(client, project, rand_gen):
     time.sleep(3)  # Search takes a second
     assert [ontology.name for ontology in client.get_ontologies(ontology_name)
            ] == [ontology_name]
+
+
+def test_project_editor_setup_cant_call_multiple_times(client, project,
+                                                       rand_gen):
+    ontology_name = f"test_project_editor_setup_ontology_name-{rand_gen(str)}"
+    ontology = client.create_ontology(ontology_name, simple_ontology())
+    project.setup_editor(ontology)
+    with pytest.raises(ResourceConflict):
+        project.setup_editor(ontology)