def resp(token, did): user = UserDao.get_user_by_token(token) if not user: return RESPONSE.WRONG_TOKEN devices = DeviceDao.get_devices_by_user(user) for device in devices: if device['did'] == did: #TODO 需要进行一些数据清理工作 return RESPONSE.SUCCESS return RESPONSE.WRONG_DID
def update(token,did,loc_info): user = UserDao.get_user_by_token(token) if not user: return {'status':0, 'desc':'wrong token'} device_loc = LocCollection.find_one({'did':did}) if device_loc: device_loc['loc_info'].append(loc_info) else: device_loc = { 'did':did, 'loc_info': loc_info } LocCollection.update({'did':did}, device_loc, upsert=True) return {'status':1}
def pull(token, did): user = UserDao.get_user_by_token(token) if not user: return RESPONSE.WRONG_TOKEN devices = DeviceDao.get_devices_by_user(user) for device in devices: if device['did'] == did: #TODO 需要将具体的信息返回回去 RESPONSE.PULL_SUCCESS['data'] = device['pull_info'] device['pull_info']=[] UserCollection.update({'token':user['token']},user) return RESPONSE.PULL_SUCCESS return RESPONSE.WRONG_DID
def initial(token,phone,did,cid,imei): user = UserDao.get_user_by_token(token) if not user: return RESPONSE.WRONG_TOKEN devices = DeviceDao.get_devices_by_user(user) for device in devices: if device['phone'] == phone: device['did'] = did device['cid'] = cid device['imei'] = imei UserCollection.update({'tid':user['tid']},user) #TODO 需要增加client对初始化配置信息 RESPONSE.INITIAL_SUCCESS['initial'] = {} return RESPONSE.INITIAL_SUCCESS return RESPONSE.WRONG_PHONE
def test_get_user_by_token(): user = UserDao.get_user_by_email(email) user2 = UserDao.get_user_by_token(user['token']) assert user2['email'] == email