1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
## www.pubnub.com - PubNub Real-time push service in the cloud.
# coding=utf8
## PubNub Real-time Push APIs and Notifications Framework
## Copyright (c) 2010 Stephen Blum
## http://www.pubnub.com/
## -----------------------------------
## PubNub 3.3.4 Real-time Push Cloud API
## -----------------------------------
import sys
import json
import time
import hashlib
import urllib2
import uuid
try:
from hashlib import sha256
digestmod = sha256
except ImportError:
import Crypto.Hash.SHA256 as digestmod
sha256 = digestmod.new
import hmac
from PubnubCrypto import PubnubCrypto
from PubnubBase import PubnubBase
class PubnubCoreAsync(PubnubBase):
def start(self): pass
def stop(self): pass
def timeout( self, delay, callback ):
pass
def __init__(
self,
publish_key,
subscribe_key,
secret_key = False,
cipher_key = False,
ssl_on = False,
origin = 'pubsub.pubnub.com',
uuid = None
) :
"""
#**
#* Pubnub
#*
#* Init the Pubnub Client API
#*
#* @param string publish_key required key to send messages.
#* @param string subscribe_key required key to receive messages.
#* @param string secret_key required key to sign messages.
#* @param boolean ssl required for 2048 bit encrypted messages.
#* @param string origin PUBNUB Server Origin.
#**
## Initiat Class
pubnub = Pubnub( 'PUBLISH-KEY', 'SUBSCRIBE-KEY', 'SECRET-KEY', False )
"""
super(PubnubCoreAsync, self).__init__(
publish_key=publish_key,
subscribe_key=subscribe_key,
secret_key=secret_key,
cipher_key=cipher_key,
ssl_on=ssl_on,
origin=origin,
UUID=uuid
)
self.subscriptions = {}
self.timetoken = 0
self.version = '3.4'
self.accept_encoding = 'gzip'
def subscribe( self, args ) :
"""
#**
#* Subscribe
#*
#* This is NON-BLOCKING.
#* Listen for a message on a channel.
#*
#* @param array args with channel and message.
#* @return false on fail, array on success.
#**
## Subscribe Example
def receive(message) :
print(message)
return True
## On Connect Callback
def connected() :
pubnub.publish({
'channel' : 'hello_world',
'message' : { 'some_var' : 'text' }
})
## Subscribe
pubnub.subscribe({
'channel' : 'hello_world',
'connect' : connected,
'callback' : receive
})
"""
## Fail if missing channel
if not 'channel' in args :
return 'Missing Channel.'
## Fail if missing callback
if not 'callback' in args :
return 'Missing Callback.'
## Capture User Input
channel = str(args['channel'])
callback = args['callback']
connectcb = args['connect']
if 'errorback' in args:
errorback = args['errorback']
else:
errorback = lambda x: x
## New Channel?
if not (channel in self.subscriptions) :
self.subscriptions[channel] = {
'first' : False,
'connected' : False,
}
## Ensure Single Connection
if self.subscriptions[channel]['connected'] :
return "Already Connected"
self.subscriptions[channel]['connected'] = 1
## SUBSCRIPTION RECURSION
def _subscribe():
## STOP CONNECTION?
if not self.subscriptions[channel]['connected']:
return
def sub_callback(response):
if not self.subscriptions[channel]['first'] :
self.subscriptions[channel]['first'] = True
connectcb()
## STOP CONNECTION?
if not self.subscriptions[channel]['connected']:
return
## PROBLEM?
if not response:
def time_callback(_time):
if not _time:
self.timeout( 1, _subscribe )
return errorback("Lost Network Connection")
else:
self.timeout( 1, _subscribe)
## ENSURE CONNECTED (Call Time Function)
return self.time({ 'callback' : time_callback })
self.timetoken = response[1]
_subscribe()
pc = PubnubCrypto()
out = []
for message in response[0]:
callback(self.decrypt(message))
## CONNECT TO PUBNUB SUBSCRIBE SERVERS
try:
self._request( { "urlcomponents" : [
'subscribe',
self.subscribe_key,
channel,
'0',
str(self.timetoken)
], "urlparams" : {"uuid":self.uuid} }, sub_callback )
except :
self.timeout( 1, _subscribe)
return
## BEGIN SUBSCRIPTION (LISTEN FOR MESSAGES)
_subscribe()
def unsubscribe( self, args ):
channel = str(args['channel'])
if not (channel in self.subscriptions):
return False
## DISCONNECT
self.subscriptions[channel]['connected'] = 0
self.subscriptions[channel]['timetoken'] = 0
self.subscriptions[channel]['first'] = False
|
