import os import json from datetime import datetime from pymongo import MongoClient from channels.handler import AsgiHandler from channels.auth import channel_session_user from channels import Group host = os.environ.get('HOST_MONGO', 'localhost') CLIENT = MongoClient('{}:27017'.format(host)) DB = CLIENT['dbplots'] # Connected to websocket.connect def ws_connect(message, code, plot): message.reply_channel.send({'accept': True}) pk = message.content['query_string'].split('=')[1] Group('{}_{}'.format(pk, plot)).add(message.reply_channel) print('New connection from: {}, creating Group: {}_{}'.format(message.content['client'][0], pk, plot)) def ws_message(message, code, plot): # Accept the incoming connection print 'incoming message' dt = datetime.strptime(str(json.loads(message.content['text'])['date']), '%d/%m/%Y') exp = DB.exp_meta.find_one({'code': int(code), 'date': dt}) print exp if exp: if plot == 'spc': datas = DB.exp_data.find({'expmeta': exp['_id']}, ['time', 'data']).sort('time', -1).limit(1)[0] exp['time'] = [datas['time']] exp['spc'] = datas['data']['spc'] exp['rti'] = datas['data']['rti'] exp['noise'] = datas['data']['noise'] else: datas = DB.exp_data.find({'expmeta': exp['_id']}, ['time', 'data']).sort('time', 1) print 'Counting...' print datas.count() dum = [(d['time'], d['data'][plot]) for d in datas] exp['time'] = [d[0] for d in dum] dum = [d[1] for d in dum] exp[plot] = map(list, zip(*dum)) exp.pop('date', None) exp.pop('_id', None) message.reply_channel.send({'text': json.dumps(exp)}) else: message.reply_channel.send({'text': json.dumps({'interval': 0})}) # Connected to websocket.disconnect def ws_disconnect(message, code, plot): Group('{}_{}'.format(code, plot)).discard(message.reply_channel)