plexflow 0.0.91__py3-none-any.whl → 0.0.93__py3-none-any.whl

plexflow/utils/google/pubsub/consume.py

@@ -0,0 +1,37 @@
+from google.cloud import pubsub_v1
+import json
+
+def consume_message(topic_id: str, project_id: str, as_json: bool = True, timeout: int = 5):
+    subscriber = pubsub_v1.SubscriberClient()
+    # The `subscription_path` method creates a fully qualified identifier
+    # in the form `projects/{project_id}/subscriptions/{subscription_id}`
+    subscription_path = subscriber.subscription_path(project_id, topic_id)
+
+    data = None
+
+    def callback(message: pubsub_v1.subscriber.message.Message) -> None:
+        nonlocal data
+
+        print(f"Received {message}.")
+        print(message.data.decode())
+        if as_json:
+            data = json.loads(message.data.decode())
+        else:
+            data = message.data.decode()
+
+        message.ack()
+
+    streaming_pull_future = subscriber.subscribe(subscription_path, callback=callback)
+    print(f"Listening for messages on {subscription_path}..\n")
+
+    # Wrap subscriber in a 'with' block to automatically call close() when done.
+    with subscriber:
+        try:
+            # When `timeout` is not set, result() will block indefinitely,
+            # unless an exception is encountered first.
+            streaming_pull_future.result(timeout=timeout)
+        except TimeoutError:
+            streaming_pull_future.cancel()  # Trigger the shutdown.
+            streaming_pull_future.result()  # Block until the shutdown is complete.
+    
+    return data

plexflow/utils/google/pubsub/produce.py

@@ -0,0 +1,21 @@
+from google.cloud import pubsub_v1
+import json
+
+def produce_message(value, topic_id: str, project_id: str, as_json: bool = True):
+    publisher = pubsub_v1.PublisherClient()
+    # The `topic_path` method creates a fully qualified identifier
+    # in the form `projects/{project_id}/topics/{topic_id}`
+    topic_path = publisher.topic_path(project_id, topic_id)
+
+    if as_json:
+        data_str = json.dumps(value)
+    else:
+        data_str = str(value)
+
+    # Data must be a bytestring
+    data = data_str.encode("utf-8")
+    # When you publish a message, the client returns a future.
+    future = publisher.publish(topic_path, data)
+    print(future.result())
+
+    print(f"Published messages to {topic_path}.")

{plexflow-0.0.91.dist-info → plexflow-0.0.93.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: plexflow
-Version: 0.0.91
+Version: 0.0.93
 Summary: A short description of the package.
 License: MIT
 Keywords: keyword1,keyword2,keyword3

{plexflow-0.0.91.dist-info → plexflow-0.0.93.dist-info}/RECORD

@@ -588,6 +588,8 @@ plexflow/utils/filesystem/search.py,sha256=ed2ixbx5r2z_xtk19--Dfi5KK9g3HcsaAk0sq
 plexflow/utils/gmail/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/utils/gmail/__pycache__/mails.cpython-311.pyc,sha256=FAYfZzjLVqpPMjrEYcrOIsohtWrZYYXOZSUHod3r8Qg,7239
 plexflow/utils/gmail/mails.py,sha256=hQIXiYR5LYeoiSH_lhbVM9criDBnTeCNmpZ5tvuUnsA,4562
+plexflow/utils/google/pubsub/consume.py,sha256=MG4bW0D2s_38xO88b_3-K8PtknNCxPnCVtPtnQzoHgA,1408
+plexflow/utils/google/pubsub/produce.py,sha256=AFlf6QZl6owA4QNzgrmed-d8lFbhsy-0bnMVVbAGNt4,713
 plexflow/utils/hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/utils/hooks/__pycache__/__init__.cpython-311.pyc,sha256=vlT8vFKsxMRss2XuHLM6ouyTl7Wk1To76MVYXowIABs,170
 plexflow/utils/hooks/__pycache__/__init__.cpython-312.pyc,sha256=c2gIcwOonxPVWWXQE6C3BgiQ7CBIM4EOrvTxD0TIbyw,158
@@ -675,7 +677,7 @@ plexflow/utils/video/__pycache__/audio.cpython-312.pyc,sha256=vXBnJwWgTDFdixMBs-
 plexflow/utils/video/__pycache__/subtitle.cpython-312.pyc,sha256=PCjpCLydGXaRsQy6cikhgsEs8WlComfOoYPiLFqfVMA,2515
 plexflow/utils/video/audio.py,sha256=tJ_lNwcjVuBQYD5cYOlXpr__eh8-hnReIgNRgIYOpqo,3380
 plexflow/utils/video/subtitle.py,sha256=LOGONGxs_RzmqtGP-DBKreOzS1eUFEKo75Q6AfnavW0,1290
-plexflow-0.0.91.dist-info/METADATA,sha256=Cth1VWhZ4Ehzv6ne1EbkYQwbAPmXNrk6e43rVJ-ewAc,2953
-plexflow-0.0.91.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.91.dist-info/entry_points.txt,sha256=aEqDHlozu_zjWrl2sibtrqtQHMgU8kSJZrE782CP47g,1362
-plexflow-0.0.91.dist-info/RECORD,,
+plexflow-0.0.93.dist-info/METADATA,sha256=OQlrJ20U88D_hDuRNKi1m4aOWdaXce0IRjObp9M8qUM,2953
+plexflow-0.0.93.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.93.dist-info/entry_points.txt,sha256=aEqDHlozu_zjWrl2sibtrqtQHMgU8kSJZrE782CP47g,1362
+plexflow-0.0.93.dist-info/RECORD,,