def get(self):
random.seed()
hash = random.getrandbits(128)
channel_name = "%016x" % hash
iterations = 0
found = False
while not found:
pin_code = random.randint(1000, 9999)
q = Pin.all()
q.filter('code = ', pin_code)
pin = q.get()
if pin:
time = datetime.utcnow() - timedelta(minutes=5)
if pin.date <= time:
pin.code = pin_code
pin.channel = channel_name
pin.date = datetime.utcnow()
pin.put()
found = True
else:
pin = None
else:
pin = Pin()
pin.code = pin_code
pin.channel = channel_name
pin.put()
found = True
iterations += 1
if iterations >= 10:
found = True
pin = None
if not pin:
self.response.set_status(409) # Conflict
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.out.write(simplejson.dumps("Try again."))
else:
token = channel.create_channel(pin.channel)
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.out.write(simplejson.dumps({
'token': token,
'channel': pin.channel,
'pin': pin.code,
}))
def post(self):
pin_code = self.request.get('pin')
if not pin_code:
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.set_status(400)
self.response.out.write(simplejson.dumps('Missing parameters.'))
return
try:
pin_code = int(pin_code)
except:
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.set_status(400)
self.response.out.write(simplejson.dumps('Bad parameters.'))
return
q = Pin.all()
q.filter('code = ', pin_code)
pin = q.get()
if not pin:
#channel_message = ChannelMessage('register', 'Failed to connect')
#channel.send_message(channel_name, simplejson.dumps(channel_message))
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.set_status(401)
self.response.out.write(simplejson.dumps('Unauthorize, must register again.'))
else:
channel_name = pin.channel
channel_message = ChannelMessage('register', 'Successfully connected.')
channel.send_message(channel_name, simplejson.dumps(channel_message))
pin.delete()
self.response.headers.add_header("Content-Type", 'application/json; charset=utf-8')
self.response.out.write(simplejson.dumps({
'channel': channel_name,
}))