project request bir

Author: jtsodapop

labelbox/schema/project.py

@@ -126,8 +126,9 @@ def labels(self, datasets=None, order_by=None):
             id_param, id_param, where, order_by_str,
             query.results_query_part(Label))
 
-        return PaginatedCollection(self.client, query_str, {id_param: self.uid},
-                                   ["project", "labels"], Label)
+        return PaginatedCollection(self.client, query_str,
+                                   {id_param: self.uid}, ["project", "labels"],
+                                   Label)
 
     def export_queued_data_rows(self, timeout_seconds=120):
         """ Returns all data rows that are currently enqueued for this project.
@@ -368,11 +369,13 @@ def create_labeler_performance(client, result):
             # python isoformat doesn't accept Z as utc timezone
             result["lastActivityTime"] = datetime.fromisoformat(
                 result["lastActivityTime"].replace('Z', '+00:00'))
-            return LabelerPerformance(
-                **
-                {utils.snake_case(key): value for key, value in result.items()})
+            return LabelerPerformance(**{
+                utils.snake_case(key): value
+                for key, value in result.items()
+            })
 
-        return PaginatedCollection(self.client, query_str, {id_param: self.uid},
+        return PaginatedCollection(self.client, query_str,
+                                   {id_param: self.uid},
                                    ["project", "labelerPerformance"],
                                    create_labeler_performance)
 
@@ -384,7 +387,7 @@ def review_metrics(self, net_score):
         Returns:
             int, aggregation count of reviews for given `net_score`.
         """
-        if net_score not in (None,) + tuple(Entity.Review.NetScore):
+        if net_score not in (None, ) + tuple(Entity.Review.NetScore):
             raise InvalidQueryError(
                 "Review metrics net score must be either None "
                 "or one of Review.NetScore values")
@@ -509,8 +512,8 @@ def unset_labeling_parameter_overrides(self, data_rows):
         query_str = """mutation UnsetLabelingParameterOverridesPyApi($%s: ID!){
             project(where: { id: $%s}) {
             unsetLabelingParameterOverrides(data: [%s]) { success }}}""" % (
-            id_param, id_param, ",\n".join(
-                "{dataRowId: \"%s\"}" % row.uid for row in data_rows))
+            id_param, id_param, ",\n".join("{dataRowId: \"%s\"}" % row.uid
+                                           for row in data_rows))
         res = self.client.execute(query_str, {id_param: self.uid})
         return res["project"]["unsetLabelingParameterOverrides"]["success"]
 
@@ -582,6 +585,25 @@ def enable_model_assisted_labeling(self, toggle: bool = True) -> bool:
         return res["project"]["showPredictionsToLabelers"][
             "showingPredictionsToLabelers"]
 
+    def bulk_import_requests(self):
+        """ passpasspass
+        """
+
+        id_param = "project_id"
+        query_str = """query ListAllImportRequestsPyApi($%s: ID!) {
+            bulkImportRequests (
+                where: { projectId: $%s }
+                skip: %%d
+                first: %%d
+            ) {
+                %s
+            }
+        }""" % (id_param, id_param,
+                query.results_query_part(Entity.BulkImportRequest))
+        return PaginatedCollection(self.client, query_str,
+                                   {id_param: str(self.uid)},
+                                   ["bulkImportRequests"], BulkImportRequest)
+
     def upload_annotations(
             self,
             name: str,

tests/integration/bulk_import/test_bulk_import_request.py

@@ -17,9 +17,8 @@ def test_create_from_url(configured_project):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
 
-    bulk_import_request = configured_project.upload_annotations(name=name,
-                                                                annotations=url,
-                                                                validate=False)
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=url, validate=False)
 
     assert bulk_import_request.project() == configured_project
     assert bulk_import_request.name == name
@@ -126,9 +125,8 @@ def test_wait_till_done(rectangle_inference, configured_project):
     url = configured_project.client.upload_data(content=ndjson.dumps(
         [rectangle_inference]),
                                                 sign=True)
-    bulk_import_request = configured_project.upload_annotations(name=name,
-                                                                annotations=url,
-                                                                validate=False)
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=url, validate=False)
 
     assert len(bulk_import_request.inputs) == 1
     bulk_import_request.wait_until_done()
@@ -149,6 +147,29 @@ def assert_file_content(url: str, predictions):
     assert response.text == ndjson.dumps(predictions)
 
 
+def test_bulk_import_requests(client, configured_project, predictions):
+    result = configured_project.bulk_import_requests()
+    assert len(list(result)) == 0
+
+    name = str(uuid.uuid4())
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=predictions)
+    bulk_import_request.wait_until_done()
+
+    name = str(uuid.uuid4())
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=predictions)
+    bulk_import_request.wait_until_done()
+
+    name = str(uuid.uuid4())
+    bulk_import_request = configured_project.upload_annotations(
+        name=name, annotations=predictions)
+    bulk_import_request.wait_until_done()
+
+    result = configured_project.bulk_import_requests()
+    assert len(list(result)) == 3
+
+
 def test_delete(client, configured_project, predictions):
 
     id_param = "project_id"

tests/integration/test_project.py

@@ -107,3 +107,8 @@ def test_attach_instructions(client, project):
 def test_queued_data_row_export(configured_project):
     result = configured_project.export_queued_data_rows()
     assert len(result) == 1
+
+
+def test_bulk_import_requests(configured_project):
+    result = configured_project.bulk_import_requests()
+    assert len(list(result)) == 0