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
