def on_message(self, getdata): logininfo = json.loads(getdata) sha256 = hashlib.sha256() sha256.update(logininfo[0]['pwd'].encode('utf-8')) passwordsha = sha256.hexdigest() word = 'select id from `members-login` where loginuser = "******"' print(word) mydb = myDb() getdata = mydb.getData(word) print(getdata) word = 'select pass,userlimit,userlevel,mininame from `members-infor` where id = "' + str( getdata[0][0]) + '"' print(word) getdata = mydb.getData(word) print(getdata) if (getdata[0][0] == passwordsha): loginsend = {} loginsend['onlineuser'] = logininfo[0]['user'] loginsend['mininame'] = getdata[0][3] loginsend['userlimit'] = getdata[0][1] loginsend['userlevel'] = getdata[0][2] loginsend['time'] = time.time() cache = myCache() random = cache.randomkey() online = random + '_' + cache.dataconfig[ 'db'] + '_' + 'onlineuser' + '_user' + '_' online = online + logininfo[0]['user'] + '_0' #print(online) onlinedata = {} onlinedata[online] = time.time() temp = cache.createCache(onlinedata) self.write_message(json.dumps(loginsend).encode())
def insert_cache(self, key, tmpdata): cache = myCache() value = cache.getValue(key) if (value == ''): insertdata = {} insertdata[key] = tmpdata cache.createCache(insertdata) elif (value != ''): cache.updateValue(key, tmpdata)
def get_cache(self, key): cache = myCache() return cache.getValue(key)
def __init__(self): self.mydb = myDb() self.mycache = myCache()
def on_message(self, getdata): #print(type(getdata)) if (type(getdata) == str): data = json.loads(getdata) filebyte = {} if ('user' in data): cache = myCache() key = 'uploadexcel_' + data['user'] + '_' + str(data['size']) receivecache = {} receivecache[key] = '' cache.createCache(receivecache) filebyte['size'] = 0 self.write_message(json.dumps(filebyte).encode()) #print('数据对接成功') #保存修改的最终数据 elif (('date' in data) & ('xmb' in data)): #print('开始保存工资信息') if (os.path.exists('tmp/stamp.wenq') == False): fp = open('tmp/stamp.wenq', 'w') fp.write(str(100000000)) fp.close() if (os.path.exists('tmp/stamp.wenq')): fp = open('tmp/stamp.wenq', 'r') stamp = fp.read() fp.close() stamp = int(stamp) + 1 fp = open('tmp/stamp.wenq', 'w') fp.write(str(stamp)) fp.close() stamp = str( time.strftime('%Y%m%d', time.localtime( time.time()))) + str(stamp) if ((data['date'] != None) & (data['xmb'] != None) & (len(data['datainfo']) != 0)): cache = myCache() getdate = data['date'].split('-') xmb = data['xmb'] uploader = data['uploader'] redishead = cache.dataconfig[ 'db'] + '_' + 'members-wages' + '_' for item in data['datainfo']: numrandom = cache.randomkey() wages = str(item) redisset = {} redisdata = numrandom + '_' + redishead + 'batch' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = stamp redisdata = numrandom + '_' + redishead + 'name' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = item['姓名'] redisdata = numrandom + '_' + redishead + 'year' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = getdate[0] redisdata = numrandom + '_' + redishead + 'month' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = getdate[1] redisdata = numrandom + '_' + redishead + 'uploader' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = uploader redisdata = numrandom + '_' + redishead + 'wages' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = wages redisdata = numrandom + '_' + redishead + 'userlimit' + '_' + item[ '姓名'] + '_1' redisset[redisdata] = data['xmb'] cache.createCache(redisset) self.write_message(json.dumps({'log': '上传成功!'}).encode()) elif (type(getdata) == bytes): filename = str( time.strftime('%Y%m%d%H%M%S', time.localtime( time.time()))) + str(random.randint(10000, 99999)) + '.xls' url = 'tmp/' + filename #print(url) fp = open(url, 'wb') fp.write(getdata) fp.close() getdisc = self.fenxiexcel(url) self.write_message(json.dumps(getdisc).encode())