Merge branch 'kgs/update-source'

Author: jsbroks

datatorch/api/entity/sources/source.py

@@ -23,6 +23,21 @@
   }
 """
 
+_UPDATE_SOURCE = """
+  mutation UpdateSource(
+    $id: ID!
+    $type: String!
+    $data: JSON!
+  ) {
+    source: updateSource(
+      id: $id
+      type: $type
+      data: $data
+    ) {
+      id
+    }
+  }
+"""
 
 class Source(BaseEntity):
 
@@ -56,3 +71,16 @@ def create(self, client=None):
         )
         r_source = results.get("source")
         self.id = r_source.get("id")
+
+    def save(self, client=None):
+      super().save(client=client)
+
+      assert self.type is not None, "Source must have a type"
+      self.client.execute(
+          _UPDATE_SOURCE,
+          params={
+              "id": self.id,
+              "type": self.type,
+              "data": self.data(),
+          },
+      )

setup.py

@@ -24,7 +24,7 @@
 
 setup(
     name="datatorch",
-    version="0.4.0",
+    version="0.4.1",
     description="A CLI and library for interacting with DataTorch",
     author="DataTorch",
     author_email="support@datatorch.io",