Merge pull request #1 from hotgluexyz/feature/inTransit

added incoming

.gitignore

@@ -1,3 +1,4 @@
+.secrets
 todo.org
 # Byte-compiled / optimized / DLL files
 __pycache__/

setup.py

@@ -3,7 +3,7 @@
 
 setup(
     name="tap-shopify",
-    version="1.4.1",
+    version="1.4.2",
     description="Singer.io tap for extracting Shopify data",
     author="Stitch",
     url="http://github.com/singer-io/tap-shopify",

tap_shopify/schemas/inventory_levels.json

@@ -13,6 +13,9 @@
     "location_id": {
       "type": ["null", "integer"]
     },
+    "incoming": {
+      "type": ["null", "integer"]
+    },
     "admin_graphql_api_id": {
       "type": ["null", "string"]
     }

tap_shopify/streams/inventory_levels.py

@@ -1,3 +1,4 @@
+import json
 import shopify
 from singer.utils import strftime, strptime_to_utc
 from tap_shopify.streams.base import (Stream,
@@ -10,6 +11,7 @@ class InventoryLevels(Stream):
     replication_key = 'updated_at'
     key_properties = ['location_id', 'inventory_item_id']
     replication_object = shopify.InventoryLevel
+    gql_query = "query inventoryLevel($id: ID!){inventoryLevel(id: $id){incoming}}"
 
     @shopify_error_handling
     def api_call_for_inventory_levels(self, parent_object_id, bookmark):
@@ -41,10 +43,15 @@ def get_objects(self):
             yield from self.get_inventory_levels(parent_object.id, bookmark)
 
     def sync(self):
+        gql_client = shopify.GraphQL()
         bookmark = self.get_bookmark()
         max_bookmark = bookmark
         for inventory_level in self.get_objects():
             inventory_level_dict = inventory_level.to_dict()
+            variables = dict(id = inventory_level_dict.get("admin_graphql_api_id"))
+            response = gql_client.execute(self.gql_query, variables)
+            response_dict = json.loads(response)
+            inventory_level_dict["incoming"] = response_dict["data"]["inventoryLevel"].get("incoming")
             replication_value = strptime_to_utc(inventory_level_dict[self.replication_key])
             if replication_value >= bookmark:
                 yield inventory_level_dict