dnaStreaming 2.2.0__tar.gz → 2.2.1__tar.gz

{dnastreaming-2.2.0/dnaStreaming.egg-info → dnastreaming-2.2.1}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: dnaStreaming
-Version: 2.2.0
+Version: 2.2.1
 Summary: Dow Jones DNA Streaming Project
 Home-page: https://github.com/dowjones/dj-dna-streams-python
-Download-URL: https://github.com/dowjones/dj-dna-streams-python/archive/release-2.2.0.tar.gz
+Download-URL: https://github.com/dowjones/dj-dna-streams-python/archive/release-2.2.1.tar.gz
 Author: Zachary Kagan
 Author-email: zachary.kagan@dowjones.com
 License: MIT

{dnastreaming-2.2.0 → dnastreaming-2.2.1}/dnaStreaming/demo/show_ha_stream_async.py

@@ -4,7 +4,7 @@ from dnaStreaming.listener import Listener
 
 listener = Listener()
 quiet_demo = os.getenv('QUIET_DEMO', "false") == "true"
-max_secs = 5
+max_secs = 1000
 print("\n[ACTIVITY] Receiving messages (ASYNC) for {} seconds...\n[0]".format(max_secs), end='')
 
 
@@ -12,9 +12,9 @@ def callback(message, subscription_id):
     callback.counter += 1
     if not quiet_demo:
         if message['action'] != 'del':
-            print('[INFO] [MSG] [{}]: AN: {}, TITLE: {}'.format(callback.counter, message['an'], message['title']))
+            print('[INFO] [SUBSCRIPTION]: {} [MSG] [{}]: AN: {}, TITLE: {}'.format(subscription_id, callback.counter, message['an'], message['title']))
         else:
-            print('[INFO] [MSG] [{}]: AN: {}, *** DELETE ***'.format(callback.counter, message['an']))
+            print('[INFO] [SUBSCRIPTION]: {} [MSG] [{}]: AN: {}, *** DELETE ***'.format(subscription_id, callback.counter, message['an']))
     else:
         if callback.counter % 10 == 0:
             print('[{}]'.format(callback.counter), end='')

{dnastreaming-2.2.0 → dnastreaming-2.2.1}/dnaStreaming/listener.py

@@ -125,19 +125,20 @@ class Listener(object):
             'Listeners for subscriptions have been configured, set and await message arrival.')
         return subscription
 
-    def listen_async_ha(self, on_message_callback, subscription_id=""):
-        def ack_message_and_callback(message):
+    def listen_async_ha(self, on_message_callback):
+        def ack_message_and_callback(message, subscription_path):
             pubsub_msg = json.loads(message.data)
             logger.info("Received news message with ID: {}".format(
                 pubsub_msg['data'][0]['id']))
             news_msg = pubsub_msg['data'][0]['attributes']
-            on_message_callback(news_msg, subscription_id)
+            short_subscription_id = subscription_path.split("/")[-1]
+            on_message_callback(news_msg, short_subscription_id)
             message.ack()
 
         main_pubsub_client = pubsub_service.get_client(self.config, MAIN_REGION)
         backup_pubsub_client = pubsub_service.get_client(self.config, BACKUP_REGION)
 
-        subscription_id = subscription_id or self.config.subscription()
+        subscription_id = self.config.subscription()
 
         if not subscription_id:
             raise Exception(

{dnastreaming-2.2.0 → dnastreaming-2.2.1}/dnaStreaming/services/availability_service.py

@@ -38,8 +38,14 @@ def ha_listen(api_host, user_key, subscription_id, stop_event, main_subscription
 
     current_region = MAIN_REGION
 
+    def wrapped_callback(subscription_path):
+        def inner_callback(message):
+            callback(message, subscription_path)
+
+        return inner_callback
+
     streaming_pull_future = main_pubsub_client.subscribe(
-            main_subscription_path, callback=callback)
+        main_subscription_path, callback=wrapped_callback(main_subscription_path))
     
     while not stop_event.is_set():
 
@@ -65,10 +71,10 @@ def ha_listen(api_host, user_key, subscription_id, stop_event, main_subscription
 
             if active_region == MAIN_REGION:
                 streaming_pull_future = main_pubsub_client.subscribe(
-                    main_subscription_path, callback=callback)
+                    main_subscription_path, callback=wrapped_callback(main_subscription_path))
             else: # active_region == BACKUP_REGION
                 streaming_pull_future = backup_pubsub_client.subscribe(
-                    backup_subscription_path, callback=callback)
+                    backup_subscription_path, callback=wrapped_callback(backup_subscription_path))
                 
             logger.warning(f"Started listener in region {active_region}")
 

{dnastreaming-2.2.0 → dnastreaming-2.2.1/dnaStreaming.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: dnaStreaming
-Version: 2.2.0
+Version: 2.2.1
 Summary: Dow Jones DNA Streaming Project
 Home-page: https://github.com/dowjones/dj-dna-streams-python
-Download-URL: https://github.com/dowjones/dj-dna-streams-python/archive/release-2.2.0.tar.gz
+Download-URL: https://github.com/dowjones/dj-dna-streams-python/archive/release-2.2.1.tar.gz
 Author: Zachary Kagan
 Author-email: zachary.kagan@dowjones.com
 License: MIT

{dnastreaming-2.2.0 → dnastreaming-2.2.1}/setup.py

@@ -6,7 +6,7 @@ this_directory = path.abspath(path.dirname(__file__))
 with open(path.join(this_directory, 'README.rst'), encoding='utf-8') as f:
     long_description = f.read()
 
-VERSION = "2.2.0"
+VERSION = "2.2.1"
 RELEASE_TAG = f"release-{VERSION}"
 
 setup(