new pushi support

hivesolutions · May 22, 2018 · 1077d05 · 1077d05
1 parent 2b00441
commit 1077d05
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/digitalriver/models/provision.py b/src/digitalriver/models/provision.py
@@ -224,15 +224,18 @@ def logger(message):
 
     def create_pushi(self):
 
-        def on_connect(connection):
-            connection.subscribe_pushi(self.pid)
+        def on_connect(protocol):
+            protocol.subscribe_pushi(self.pid)
 
         client_key = appier.conf("PUSHI_KEY")
-        client = pushi.PushiClient(client_key = client_key)
-        connection = client.connect_pushi(callback = on_connect)
+
+        _loop, protocol = pushi.net.PushiClient.connect_pushi_s(
+            client_key = client_key,
+            callback = on_connect
+        )
 
         def logger(message):
-            connection.send_channel("stdout", message, self.pid, persist = False)
+            protocol.send_channel("stdout", message, self.pid, persist = False)
 
         return logger