From d8b5775428f603f28c116645b8721c50d4671dda Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Josu=C3=A9=20M=2E=20Sehnem?= Date: Wed, 6 Apr 2022 14:27:51 -0400 Subject: [PATCH] hide gql error print --- setup.py | 2 +- tap_shopify/streams/incoming_items.py | 15 ++++++++++++++- 2 files changed, 15 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 78ee8c04..7668668a 100755 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ setup( name="tap-shopify", - version="1.4.7", + version="1.4.8", description="Singer.io tap for extracting Shopify data", author="Stitch", url="http://github.com/singer-io/tap-shopify", diff --git a/tap_shopify/streams/incoming_items.py b/tap_shopify/streams/incoming_items.py index 3fb06264..c0325184 100644 --- a/tap_shopify/streams/incoming_items.py +++ b/tap_shopify/streams/incoming_items.py @@ -1,3 +1,5 @@ +import os +import sys import shopify import singer import json @@ -9,6 +11,16 @@ LOGGER = singer.get_logger() +class HiddenPrints: + def __enter__(self): + self._original_stdout = sys.stdout + sys.stdout = open(os.devnull, 'w') + + def __exit__(self, exc_type, exc_val, exc_tb): + sys.stdout.close() + sys.stdout = self._original_stdout + + class IncomingItems(Stream): name = 'incoming_items' replication_key = 'createdAt' @@ -17,7 +29,8 @@ class IncomingItems(Stream): @shopify_error_handling def call_api_for_incoming_items(self, parent_object): gql_client = shopify.GraphQL() - response = gql_client.execute(self.gql_query, dict(id=parent_object.admin_graphql_api_id)) + with HiddenPrints(): + response = gql_client.execute(self.gql_query, dict(id=parent_object.admin_graphql_api_id)) return json.loads(response) def get_objects(self):