add support for attachments

Author: Matt Sokoloff

labelbox/orm/model.py

@@ -104,8 +104,8 @@ def DateTime(*args):
         return Field(Field.Type.DateTime, *args)
 
     @staticmethod
-    def Enum(enum_cls: type, *args, **kwargs):
-        return Field(Field.EnumType(enum_cls), *args, **kwargs)
+    def Enum(enum_cls: type, *args):
+        return Field(Field.EnumType(enum_cls), *args)
 
     @staticmethod
     def Json(*args):

labelbox/schema/data_row.py

@@ -49,7 +49,6 @@ class DataRow(DbObject, Updateable, BulkDeletable):
                           name="metadata",
                           graphql_name="customMetadata",
                           result_subquery="customMetadata { schemaId value }")
-    media_type = Field.Enum(MediaType, "media_type", result_subquery="")
 
     # Relationships
     dataset = Relationship.ToOne("Dataset")
@@ -67,7 +66,6 @@ def __init__(self, *args, **kwargs):
         self.attachments.supports_sorting = False
 
     def _set_field_values(self, field_values):
-        field_values.update({'mediaType': MediaType.Unknown})
         super()._set_field_values(field_values)
 
     @staticmethod

labelbox/schema/dataset.py

@@ -15,6 +15,7 @@
 from labelbox.exceptions import InvalidQueryError, LabelboxError, ResourceNotFoundError, InvalidAttributeError
 from labelbox.orm.db_object import DbObject, Updateable, Deletable
 from labelbox.orm.model import Entity, Field, Relationship
+from labelbox.orm import query
 from labelbox.exceptions import MalformedQueryException
 
 if TYPE_CHECKING:
@@ -102,15 +103,35 @@ def convert_field_keys(items):
         elif os.path.exists(row_data):
             # If row data is a local file path, upload it to server.
             args[DataRow.row_data.name] = self.client.upload_file(row_data)
-        args[DataRow.dataset.name] = self
 
         # Parse metadata fields, if they are provided
         if DataRow.metadata_fields.name in args:
             mdo = self.client.get_data_row_metadata_ontology()
             args[DataRow.metadata_fields.name] = mdo.parse_upsert_metadata(
                 args[DataRow.metadata_fields.name])
 
-        return self.client._create(DataRow, args)
+        query_str = """mutation CreateDataRowPyApi(
+            $row_data: String!,
+            $metadata_fields: [DataRowCustomMetadataUpsertInput!]!,
+            $attachments: [DataRowAttachmentInput!],
+            $media_type : MediaType,
+            $dataset: ID!
+            ){
+                createDataRow(
+                    data:
+                      {
+                        rowData: $row_data
+                        mediaType: $media_type
+                        metadataFields: $metadata_fields
+                        attachments: $attachments
+                        dataset: {connect: {id: $dataset}}
+                    }
+                   )
+                {%s}
+            }
+        """ % query.results_query_part(Entity.DataRow)
+        res = self.client.execute(query_str, {**args, 'dataset': self.uid})
+        return DataRow(self.client, res['createDataRow'])
 
     def create_data_rows_sync(self, items) -> None:
         """ Synchronously bulk upload data rows.