Added hub

vishr · vishr · commit 7da6acabdf94 · 2018-05-26T18:20:50.000-07:00
Signed-off-by: Vishal Rana &lt;vr@labstack.com&gt;
diff --git a/labstack/__init__.py b/labstack/__init__.py
@@ -1 +1,2 @@
-from .client import Client, APIError
+from .client import Client, APIError
+from .hub import Hub
diff --git a/labstack/client.py b/labstack/client.py
@@ -1,7 +1,7 @@
 import requests
 from .currency import Currency
 from .geocode import Geocode 
-from .post import Post 
+from .email import Email 
 from .watermark import Watermark 
 from .webpage import Webpage 
 
@@ -37,8 +37,8 @@ def currency(self):
   def geocode(self):
     return Geocode(self)
 
-  def post(self):
-    return Post(self)
+  def email(self):
+    return Email(self)
   
   def watermark(self):
     return Watermark(self)
diff --git a/labstack/email.py b/labstack/email.py
@@ -1,8 +1,8 @@
 import requests
 
-class Post():
+class Email():
   def __init__(self, client):
     self.client = client 
   
   def verify(self, email):
-    return self.client._request('GET', '/post/verify', params={'email': email})
+    return self.client._request('GET', '/email/verify', params={'email': email})
diff --git a/labstack/hub.py b/labstack/hub.py
@@ -0,0 +1,50 @@
+import signal
+import time
+import paho.mqtt.client as mqtt
+
+class Hub():
+  def __init__(self, account_id, api_key, client_id=None):
+    self.account_id = account_id
+    self.client = mqtt.Client(client_id=client_id, clean_session=True)
+    self.client.username_pw_set(account_id, api_key)
+    # self.client.tls_set(ca_certs='labstack.com/cert.pem')
+    self.handlers = {}
+    def handler(client, userdata, msg):
+      self.handlers[msg.topic](msg.payload)
+    self.client.on_message = handler
+    self._run = True
+    signal.signal(signal.SIGINT, self._stop)
+    signal.signal(signal.SIGTERM, self._stop)
+  
+  def connect(self, handler=None):
+    self.client.connect("hub.labstack.com", 1883)
+    self.client.loop_start()
+    def on_connect(client, userdata, flags, rc):
+      handler()
+    self.client.on_connect = on_connect 
+
+  def publish(self, topic, message):
+    self.client.publish('{}/{}'.format(self.account_id, topic), message)
+
+  # def subscribe(self, topic, handler, shared=False):
+  def subscribe(self, topic, handler):
+    topic = '{}/{}'.format(self.account_id, topic)
+    # if shared:
+    #   topic = '$queue/' + topic
+    self.client.subscribe(topic)
+    self.handlers[topic] = handler
+
+  def unsubscribe(self, topic):
+    self.client.unsubscribe(topic)
+
+  def disconnect(self):
+    self.client.loop_stop()
+    self.client.disconnect()
+
+  def run(self):
+    while self._run:
+      time.sleep(1)
+
+  def _stop(self, signum, frame):
+    self._run = False
+    self.disconnect()
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
   name='labstack',
-  version='0.31.4',
+  version='0.31.5',
   description='Official Python client library for the LabStack platform',
   long_description='`<https://github.com/labstack/labstack-python>`_',
   keywords='image compress, image resize, text summary, barcode generate, barcode scan',

Original file line number	Diff line number	Diff line change
`@@ -1 +1,2 @@`
`1`		`-from .client import Client, APIError`
	`1`	`+from .client import Client, APIError`
	`2`	`+from .hub import Hub`