fleetmqsdk 0.0.1__tar.gz

fleetmqsdk-0.0.1/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.2
+Name: fleetmqsdk
+Version: 0.0.1
+Summary: A convenient SDK for FleetMQ
+Author: Nicole
+Author-email: nicole@fleetmq.io
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Dist: google-api-python-client
+Requires-Dist: pyzmq
+Requires-Dist: setuptools
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: requires-dist
+Dynamic: summary

fleetmqsdk-0.0.1/fleetmqsdk/Abc_config_pb2.py

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: Abc_config.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x10\x41\x62\x63_config.proto\x12\x0e\x61\x62\x63onnectproto\x1a\x1bgoogle/protobuf/empty.proto\"\xb6\x01\n\x0f\x41\x42\x43onnectConfig\x12\x10\n\x08group_id\x18\x03 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\x12=\n\x07\x64\x65vices\x18\x01 \x03(\x0b\x32,.abconnectproto.ABConnectConfig.DevicesEntry\x1a\x46\n\x0c\x44\x65vicesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.abconnectproto.Device:\x02\x38\x01\"\xba\x02\n\x06\x44\x65vice\x12;\n\x0bsend_topics\x18\x01 \x03(\x0b\x32&.abconnectproto.Device.SendTopicsEntry\x12\x41\n\x0ereceive_topics\x18\x02 \x03(\x0b\x32).abconnectproto.Device.ReceiveTopicsEntry\x12\x0e\n\x06joined\x18\x03 \x01(\x08\x1aL\n\x0fSendTopicsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.abconnectproto.Publisher:\x02\x38\x01\x1aR\n\x12ReceiveTopicsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.abconnectproto.Subscription:\x02\x38\x01\"\x9f\x03\n\x06Source\x12\x31\n\x06source\x18\x01 \x01(\x0e\x32!.abconnectproto.Source.SourceType\x12\x10\n\x08zmq_port\x18\x04 \x01(\x05\x12\x1a\n\x12source_device_name\x18\x05 \x01(\t\x12\x13\n\x0bsource_file\x18\x06 \x01(\t\x12\x36\n\x0bvideo_codec\x18\x07 \x01(\x0e\x32!.abconnectproto.Source.VideoCodec\"\x8f\x01\n\nSourceType\x12\r\n\tUNDEFINED\x10\x00\x12\t\n\x05STDIN\x10\x01\x12\t\n\x05VIDEO\x10\x02\x12\r\n\tZMQ_VIDEO\x10\x03\x12\x0c\n\x08ROS_DATA\x10\x04\x12\r\n\tZMQ_AUDIO\x10\x05\x12\t\n\x05\x41UDIO\x10\x06\x12\x0b\n\x07MAVLINK\x10\x07\x12\x0e\n\nTIME_PROBE\x10\x08\x12\x08\n\x04\x46ILE\x10\t\"1\n\nVideoCodec\x12\x19\n\x15VIDEO_CODEC_UNDEFINED\x10\x00\x12\x08\n\x04MMAL\x10\x01J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04R\x08video_idR\x0cros_msg_type\"\xe9\x01\n\tPublisher\x12\r\n\x05topic\x18\x01 \x01(\t\x12&\n\x06source\x18\x03 \x01(\x0b\x32\x16.abconnectproto.Source\x12M\n\x13subscriber_peer_ids\x18\x04 \x03(\x0b\x32\x30.abconnectproto.Publisher.SubscriberPeerIdsEntry\x1aP\n\x16SubscriberPeerIdsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.Empty:\x02\x38\x01J\x04\x08\x02\x10\x03\"\xf5\x01\n\x04Sink\x12+\n\x04sink\x18\x01 \x01(\x0e\x32\x1d.abconnectproto.Sink.SinkType\x12\x10\n\x08zmq_port\x18\x03 \x01(\x05\x12\x11\n\tsink_file\x18\x04 \x01(\t\"p\n\x08SinkType\x12\r\n\tUNDEFINED\x10\x00\x12\n\n\x06STDOUT\x10\x01\x12\x0c\n\x08ROS_DATA\x10\x02\x12\t\n\x05VIDEO\x10\x03\x12\t\n\x05\x41UDIO\x10\x04\x12\x0b\n\x07MAVLINK\x10\x05\x12\x0e\n\nTIME_PROBE\x10\x06\x12\x08\n\x04\x46ILE\x10\x07J\x04\x08\x02\x10\x03J\x04\x08\x05\x10\x06R\x0cros_msg_typeR\x0flocal_web_media\"\\\n\x0cSubscription\x12\r\n\x05topic\x18\x01 \x01(\t\x12\x19\n\x11publisher_peer_id\x18\x02 \x01(\t\x12\"\n\x04sink\x18\x03 \x01(\x0b\x32\x14.abconnectproto.SinkB\x0cZ\n/messagepbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'Abc_config_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\n/messagepb'
+  _globals['_ABCONNECTCONFIG_DEVICESENTRY']._loaded_options = None
+  _globals['_ABCONNECTCONFIG_DEVICESENTRY']._serialized_options = b'8\001'
+  _globals['_DEVICE_SENDTOPICSENTRY']._loaded_options = None
+  _globals['_DEVICE_SENDTOPICSENTRY']._serialized_options = b'8\001'
+  _globals['_DEVICE_RECEIVETOPICSENTRY']._loaded_options = None
+  _globals['_DEVICE_RECEIVETOPICSENTRY']._serialized_options = b'8\001'
+  _globals['_PUBLISHER_SUBSCRIBERPEERIDSENTRY']._loaded_options = None
+  _globals['_PUBLISHER_SUBSCRIBERPEERIDSENTRY']._serialized_options = b'8\001'
+  _globals['_ABCONNECTCONFIG']._serialized_start=66
+  _globals['_ABCONNECTCONFIG']._serialized_end=248
+  _globals['_ABCONNECTCONFIG_DEVICESENTRY']._serialized_start=178
+  _globals['_ABCONNECTCONFIG_DEVICESENTRY']._serialized_end=248
+  _globals['_DEVICE']._serialized_start=251
+  _globals['_DEVICE']._serialized_end=565
+  _globals['_DEVICE_SENDTOPICSENTRY']._serialized_start=405
+  _globals['_DEVICE_SENDTOPICSENTRY']._serialized_end=481
+  _globals['_DEVICE_RECEIVETOPICSENTRY']._serialized_start=483
+  _globals['_DEVICE_RECEIVETOPICSENTRY']._serialized_end=565
+  _globals['_SOURCE']._serialized_start=568
+  _globals['_SOURCE']._serialized_end=983
+  _globals['_SOURCE_SOURCETYPE']._serialized_start=753
+  _globals['_SOURCE_SOURCETYPE']._serialized_end=896
+  _globals['_SOURCE_VIDEOCODEC']._serialized_start=898
+  _globals['_SOURCE_VIDEOCODEC']._serialized_end=947
+  _globals['_PUBLISHER']._serialized_start=986
+  _globals['_PUBLISHER']._serialized_end=1219
+  _globals['_PUBLISHER_SUBSCRIBERPEERIDSENTRY']._serialized_start=1133
+  _globals['_PUBLISHER_SUBSCRIBERPEERIDSENTRY']._serialized_end=1213
+  _globals['_SINK']._serialized_start=1222
+  _globals['_SINK']._serialized_end=1467
+  _globals['_SINK_SINKTYPE']._serialized_start=1312
+  _globals['_SINK_SINKTYPE']._serialized_end=1424
+  _globals['_SUBSCRIPTION']._serialized_start=1469
+  _globals['_SUBSCRIPTION']._serialized_end=1561
+# @@protoc_insertion_point(module_scope)

fleetmqsdk-0.0.1/fleetmqsdk/Message_pb2.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: Message.proto
+# Protobuf Python Version: 5.26.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+import Abc_config_pb2 as Abc__config__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\rMessage.proto\x12\x0e\x61\x62\x63onnectproto\x1a\x10\x41\x62\x63_config.proto\"\x9a\x04\n\x10\x41\x42\x43onnectMessage\x12&\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x15.abconnectproto.ErrorH\x00\x12\"\n\x03icm\x18\x01 \x01(\x0b\x32\x13.abconnectproto.ICMH\x00\x12.\n\tsignaling\x18\x02 \x01(\x0b\x32\x19.abconnectproto.SignalingH\x00\x12*\n\x02wf\x18\x03 \x01(\x0b\x32\x1c.abconnectproto.WebFrameworkH\x00\x12\x31\n\x06\x63onfig\x18\x07 \x01(\x0b\x32\x1f.abconnectproto.ABConnectConfigH\x00\x12\x37\n\x0e\x63onfig_request\x18\n \x01(\x0b\x32\x1d.abconnectproto.ConfigRequestH\x00\x12\x35\n\x10\x63reate_publisher\x18\x0b \x01(\x0b\x32\x19.abconnectproto.PublisherH\x00\x12;\n\x13\x63reate_subscription\x18\x0c \x01(\x0b\x32\x1c.abconnectproto.SubscriptionH\x00\x12/\n\rdevice_config\x18\r \x01(\x0b\x32\x16.abconnectproto.DeviceH\x00\x12\x0c\n\x04\x66rom\x18\x05 \x01(\t\x12\n\n\x02to\x18\x06 \x01(\t\x12\x11\n\tconfig_id\x18\x08 \x01(\t\x12\x10\n\x08group_id\x18\t \x01(\tB\x0e\n\x0cmessage_type\"\x92\x02\n\x03ICM\x12+\n\x05\x65vent\x18\x01 \x01(\x0e\x32\x1c.abconnectproto.ICM.ICMEvent\x12<\n\rmedia_sources\x18\x02 \x03(\x0b\x32%.abconnectproto.ICM.MediaSourcesEntry\x12\x0f\n\x07version\x18\x03 \x01(\t\x1aK\n\x11MediaSourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.abconnectproto.Source:\x02\x38\x01\"B\n\x08ICMEvent\x12\r\n\tUNDEFINED\x10\x00\x12\x08\n\x04JOIN\x10\x01\x12\t\n\x05LEAVE\x10\x02\x12\x08\n\x04PING\x10\x03\x12\x08\n\x04PONG\x10\x04\"\xe5\x03\n\tSignaling\x12\x36\n\x04type\x18\x01 \x01(\x0e\x32(.abconnectproto.Signaling.ConnectionType\x12=\n\rconnection_id\x18\x04 \x01(\x0e\x32&.abconnectproto.Signaling.ConnectionID\x12\x37\n\x05\x65vent\x18\x02 \x01(\x0e\x32(.abconnectproto.Signaling.SignalingEvent\x12\x0f\n\x07payload\x18\x03 \x01(\t\"[\n\x0e\x43onnectionType\x12\x1d\n\x19\x43ONNECTION_TYPE_UNDEFINED\x10\x00\x12\x11\n\rCONNECT_BY_ID\x10\x01\x12\x17\n\x13\x43ONNECT_BY_CATEGORY\x10\x02\"G\n\x0c\x43onnectionID\x12\x1b\n\x17\x43ONNECTION_ID_UNDEFINED\x10\x00\x12\x0b\n\x07PRIMARY\x10\x01\x12\r\n\tSECONDARY\x10\x02\"q\n\x0eSignalingEvent\x12\x1d\n\x19SIGNALING_EVENT_UNDEFINED\x10\x00\x12\x10\n\x0cOPEN_REQUEST\x10\x01\x12\x11\n\rOPEN_RESPONSE\x10\x02\x12\t\n\x05\x43LOSE\x10\x03\x12\x07\n\x03SDP\x10\x04\x12\x07\n\x03ICE\x10\x05\"\x94\x01\n\x0cWebFramework\x12=\n\x05\x65vent\x18\x01 \x01(\x0e\x32..abconnectproto.WebFramework.WebFrameworkEvent\x12\x0f\n\x07payload\x18\x02 \x01(\t\"4\n\x11WebFrameworkEvent\x12\r\n\tUNDEFINED\x10\x00\x12\x10\n\x0cGRAPH_UPDATE\x10\x01\"\xf4\x02\n\x05\x45rror\x12.\n\x05\x65rror\x18\x01 \x01(\x0e\x32\x1f.abconnectproto.Error.ErrorType\x12\x15\n\rerror_message\x18\x02 \x01(\t\"\xa3\x02\n\tErrorType\x12\r\n\tUNDEFINED\x10\x00\x12\x14\n\x10UNIQUENESS_ERROR\x10\x01\x12\x1a\n\x16\x44\x45VICE_NOT_FOUND_ERROR\x10\x02\x12\x1c\n\x18\x43\x41TEGORY_NOT_FOUND_ERROR\x10\x03\x12\x15\n\x11\x41LREADY_CONNECTED\x10\x04\x12\x12\n\x0eUNKNOWN_CONFIG\x10\x05\x12\x14\n\x10INCORRECT_CONFIG\x10\x08\x12\x11\n\rUNKNOWN_GROUP\x10\x07\x12\x13\n\x0fINVALID_MESSAGE\x10\x06\x12\x1a\n\x16\x43REATE_PUBLISHER_ERROR\x10\t\x12\x1d\n\x19\x43REATE_SUBSCRIPTION_ERROR\x10\n\x12\x13\n\x0fINVALID_VERSION\x10\x0b\"\x0f\n\rConfigRequestB\x0cZ\n/messagepbb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'Message_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\n/messagepb'
+  _globals['_ICM_MEDIASOURCESENTRY']._loaded_options = None
+  _globals['_ICM_MEDIASOURCESENTRY']._serialized_options = b'8\001'
+  _globals['_ABCONNECTMESSAGE']._serialized_start=52
+  _globals['_ABCONNECTMESSAGE']._serialized_end=590
+  _globals['_ICM']._serialized_start=593
+  _globals['_ICM']._serialized_end=867
+  _globals['_ICM_MEDIASOURCESENTRY']._serialized_start=724
+  _globals['_ICM_MEDIASOURCESENTRY']._serialized_end=799
+  _globals['_ICM_ICMEVENT']._serialized_start=801
+  _globals['_ICM_ICMEVENT']._serialized_end=867
+  _globals['_SIGNALING']._serialized_start=870
+  _globals['_SIGNALING']._serialized_end=1355
+  _globals['_SIGNALING_CONNECTIONTYPE']._serialized_start=1076
+  _globals['_SIGNALING_CONNECTIONTYPE']._serialized_end=1167
+  _globals['_SIGNALING_CONNECTIONID']._serialized_start=1169
+  _globals['_SIGNALING_CONNECTIONID']._serialized_end=1240
+  _globals['_SIGNALING_SIGNALINGEVENT']._serialized_start=1242
+  _globals['_SIGNALING_SIGNALINGEVENT']._serialized_end=1355
+  _globals['_WEBFRAMEWORK']._serialized_start=1358
+  _globals['_WEBFRAMEWORK']._serialized_end=1506
+  _globals['_WEBFRAMEWORK_WEBFRAMEWORKEVENT']._serialized_start=1454
+  _globals['_WEBFRAMEWORK_WEBFRAMEWORKEVENT']._serialized_end=1506
+  _globals['_ERROR']._serialized_start=1509
+  _globals['_ERROR']._serialized_end=1881
+  _globals['_ERROR_ERRORTYPE']._serialized_start=1590
+  _globals['_ERROR_ERRORTYPE']._serialized_end=1881
+  _globals['_CONFIGREQUEST']._serialized_start=1883
+  _globals['_CONFIGREQUEST']._serialized_end=1898
+# @@protoc_insertion_point(module_scope)

fleetmqsdk-0.0.1/fleetmqsdk/fleetmqsdk.py

@@ -0,0 +1,249 @@
+import Message_pb2
+import zmq
+
+tcpPrefix = "tcp://0.0.0.0:"
+ipcPrefix = "ipc:///tmp/feeds/"
+
+def _getAddress(port, ipc):
+    if ipc:
+        return ipcPrefix + port
+    else:
+        return tcpPrefix + port
+
+class FleetMQ:
+    def __init__(self):
+        print("FleetMQ SDK")
+        self._context = zmq.Context()
+        self._rpcSocket = self._initRpcReq()
+        self._publishers = {}
+        self._subscribers = {}
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        self.close()
+
+    def close(self):
+        self._rpcSocket.close()
+        for publisher in self._publishers.values():
+            publisher.close()
+        for subscriber in self._subscribers.values():
+            subscriber.close()
+
+    def _initRpcReq(self, rpcReqPort="5558"):
+        address = 'tcp://localhost:' + rpcReqPort
+        try:
+            rpcReqInterface = self._context.socket(zmq.REQ)
+            rpcReqInterface.connect(address)
+            print("Connected to RPC request address: " + address)
+        except zmq.ZMQError as e:
+            print(f"Failed to open RPC request socket on port {rpcReqPort}: {e}")
+            exit(1)
+        return rpcReqInterface    
+    
+    def _createPublisherFromConfig(self, port, topic, ipc=False):        
+        if topic in self._publishers:
+            # Publisher already exists
+            print("Publisher for topic " + topic + " already exists")
+            return
+        self._publishers[topic] = _Publisher(self._context, port, topic, ipc)    
+    
+    def _createSubscriberFromConfig(self, port, topic, ipc=False):
+        if topic in self._subscribers:
+            # Subscriber already exists
+            print("Subscriber for topic " + topic + " already exists")
+            return
+        self._subscribers[topic] = _Subscriber(self._context, port, topic, ipc)
+    
+    def _createConfigState(self, config, ports):
+        for topic in config.send_topics:
+            if topic not in ports:
+                print("Error: send topic " + topic + " not found in ports, won't create publisher")
+                continue
+            self._createPublisherFromConfig(ports[topic], topic, config.send_topics[topic].source.source)
+        for topic in config.receive_topics:
+            if topic not in ports:
+                print("Error: receive topic " + topic + " not found in ports, won't create subscriber")
+                continue
+            self._createSubscriberFromConfig(ports[topic], topic, config.receive_topics[topic].sink.sink)
+
+    def _receiveConfig(self):
+        try:
+            configBytes = self._rpcSocket.recv_multipart()
+            if len(configBytes) != 2:
+                print("Error getting config from datastreamer, expected 2 frames, got " + str(len(configBytes)))
+                return None, None
+            
+            # Parse received config data into ABConnectMessage.
+            msg = Message_pb2.ABConnectMessage()
+            msg.ParseFromString(configBytes[0])
+
+            # Evaluate the message type.
+            if msg.HasField("device_config"):
+                print("Received config response from datastreamer")
+            elif msg.HasField("error"):            
+                print("Received Error from datastreamer: " + msg.error.error_message)
+                return None, None
+            else:
+                print("Unexpected message type received from datastreamer: " + msg)
+                exit(1)
+
+        except zmq.ZMQError as e:
+            print(f"Failed to receive config: {e}")
+            exit(1)
+
+        
+        portsString = configBytes[1].decode()
+        print("Ports: " + portsString)    
+        
+        print("Device config:")
+        print(msg.device_config)        
+
+        return msg.device_config, eval(portsString)        
+
+    def getConfig(self, createConfigState=False):
+        req = Message_pb2.ABConnectMessage()
+        req.config_request.SetInParent()
+        print("Sending config request to datastreamer...")    
+        try:
+            self._rpcSocket.send(req.SerializeToString())
+        except zmq.ZMQError as e:
+            print(f"Failed to receive config: {e}")
+            exit(1)
+
+        # Wait for response
+        config, addresses = self._receiveConfig()
+        if config is None or addresses is None:
+            print("Failed to get config from datastreamer")
+            exit(1)
+
+        if createConfigState:
+            self._createConfigState(config, addresses)
+
+        return config, addresses
+    
+    # CreatePublisher creates a publisher for the given topic, executing an rpc to the datastreamer
+    # to provision a zmq port.    
+    def createPublisher(self, topic, source_type):    
+        if topic in self._publishers:
+            # Publisher already exists
+            print("Publisher for topic " + topic + " already exists")
+            return        
+             
+        req = Message_pb2.ABConnectMessage()
+        req.create_publisher.topic = topic
+        req.create_publisher.source.source = source_type
+        print("Sending create publisher request to datastreamer...")    
+        try:
+            self._rpcSocket.send(req.SerializeToString())
+        except zmq.ZMQError as e:
+            print(f"Failed to send CreatePublisher request to datastreamer: {e}")
+            exit(1)
+
+        config, ports = self._receiveConfig()
+        if config is None or ports is None:
+            print("Failed to create publisher")
+            exit(1)
+        self._createConfigState(config, ports)
+
+        if topic not in self._publishers:
+            print("Internal error, publisher for topic " + topic + " not created")
+            exit(1)
+    
+    # CreateSubscriber creates a subscriber for the given topic, executing an rpc to the
+    # datastreamer to provision a zmq port.    
+    def createSubscriber(self, topic, sink_type):
+        if topic in self._subscribers:
+            # Subscriber already exists
+            print("Subscriber for topic " + topic + " already exists")
+            return
+        
+        req = Message_pb2.ABConnectMessage()
+        req.create_subscription.topic = topic
+        req.create_subscription.sink.sink = sink_type
+        print("Sending create subscriber request to datastreamer...")
+        try:
+            self._rpcSocket.send(req.SerializeToString())
+        except zmq.ZMQError as e:
+            print(f"Failed to send CreateSubscriber request to datastreamer: {e}")
+            exit(1)
+        
+        config, ports = self._receiveConfig()
+        if config is None or ports is None:
+            print("Failed to create subscriber")
+            exit(1)
+        self._createConfigState(config, ports)
+
+        if topic not in self._subscribers:
+            print("Internal error, subscriber for topic " + topic + " not created")
+            exit(1)    
+
+    def publish(self, topic, data):
+        if topic not in self._publishers:
+            print("Publisher for topic " + topic + " does not exist")
+            return
+        
+        self._publishers[topic].publish(data)
+
+    def receive(self, topic):
+        if topic not in self._subscribers:
+            print("Subscriber for topic " + topic + " does not exist")
+            return
+
+        self._subscribers[topic].receive()
+
+
+class _Publisher:
+    def __init__(self, context, port, topic, ipc=False):        
+        self._topic = topic
+        self._topicBytes = topic.encode()
+        self._publisherSocket = context.socket(zmq.PUB)
+        address = _getAddress(port, ipc)
+        try:
+            self._publisherSocket.bind(address)
+        except zmq.ZMQError as e:
+            print(f"Failed to open publish socket on address {address} error: {e}")
+            exit(1)    
+        print(f"Bound to publish socket on address {address} for topic {topic}")
+
+    def __enter__(self):
+        return self
+    
+    def __exit__(self, type, value, traceback):
+        self.close()
+
+    def close(self):
+        self._publisherSocket.close()
+    
+    def publish(self, msg):
+        if type(msg) == type(""):
+            msgData = msg.encode()
+        else:
+            msgData = msg
+        self._publisherSocket.send_multipart([self._topicBytes, msgData])
+
+class _Subscriber:
+    def __init__(self, context, port, topic, ipc=False):            
+        self._topic = topic
+        self._subscriberSocket = context.socket(zmq.SUB)
+        address = _getAddress(port, ipc)
+        try:
+            self._subscriberSocket.connect(address)
+        except zmq.ZMQError as e:
+            print(f"Failed to open subscribe socket on address {address} error: {e}")
+            exit(1)    
+        self._subscriberSocket.setsockopt(zmq.SUBSCRIBE, topic.encode())        
+        print(f"Bound to subscribe socket on address {address} for topic {topic}")        
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        self.close()
+
+    def close(self):
+        self._subscriberSocket.close()
+
+    def receive(self):
+        return self._subscriberSocket.recv_multipart()[1].decode()

fleetmqsdk-0.0.1/fleetmqsdk.egg-info/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.2
+Name: fleetmqsdk
+Version: 0.0.1
+Summary: A convenient SDK for FleetMQ
+Author: Nicole
+Author-email: nicole@fleetmq.io
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Dist: google-api-python-client
+Requires-Dist: pyzmq
+Requires-Dist: setuptools
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: requires-dist
+Dynamic: summary

fleetmqsdk-0.0.1/fleetmqsdk.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+setup.py
+fleetmqsdk/Abc_config_pb2.py
+fleetmqsdk/Message_pb2.py
+fleetmqsdk/__init__.py
+fleetmqsdk/fleetmqsdk.py
+fleetmqsdk.egg-info/PKG-INFO
+fleetmqsdk.egg-info/SOURCES.txt
+fleetmqsdk.egg-info/dependency_links.txt
+fleetmqsdk.egg-info/requires.txt
+fleetmqsdk.egg-info/top_level.txt

fleetmqsdk-0.0.1/fleetmqsdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

fleetmqsdk-0.0.1/fleetmqsdk.egg-info/requires.txt

@@ -0,0 +1,3 @@
+google-api-python-client
+pyzmq
+setuptools

fleetmqsdk-0.0.1/fleetmqsdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+fleetmqsdk

fleetmqsdk-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

fleetmqsdk-0.0.1/setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='fleetmqsdk',
+    version='0.0.1',
+    description='A convenient SDK for FleetMQ',
+    author='Nicole',
+    author_email='nicole@fleetmq.io',
+    packages=find_packages(),
+    install_requires=[
+        'google-api-python-client',
+        'pyzmq',
+        'setuptools'
+    ],
+    classifiers=[
+        'Programming Language :: Python :: 3',
+        'License :: OSI Approved :: MIT License',
+    ],
+)