def uid():
logger.info('rec a uid request')
res = {'code': 1, 'msg': 'success', 'uid': 'unkown'}
b = BaseDao()
try:
user_dao = User(base=b)
macid = request.args.get("macid") or ''
androidid = request.args.get("androidid") or ''
if validSqlInject(macid) or validSqlInject(androidid):
return jsonify({'code': 1, 'msg': 'invalid param'})
if macid or androidid:
user = user_dao.get_user_info(macid, androidid)
if not user:
uid = get_user_id(macid, androidid)
user_dao.add_user_id(uid, macid, androidid)
else:
uid = user.user_id
res['uid'] = uid
else:
res['uid'] = get_user_id()
logger.info('success|uid request serve success')
except:
logger.error('fail|exception|uid request serve error|%s' %
traceback.format_exc())
finally:
b.close()
return jsonify(res)
def generate_fund_userlist(data):
'''
将基金表数据整合为：{
基金ID1 : [{用户对象}],
基金ID2 : [{用户对象}]
}
'''
fund_user = {}
for r in data:
fid = r.get('id')
r['id'] = r.get('user_id')
if fund_user.get(fid) != None:
fund_user[fid].append(User(r))
else:
fund_user[fid] = [User(r)]
return fund_user
def post(self: tornado.web.RequestHandler):
client = Client(name=str(self.get_argument('FirstName')),
surname=str(self.get_argument('LastName')))
User.create_user(
User(login=str(self.get_argument('Login')),
password=str(self.get_argument('Password')),
client=Client.create_client(client)))
self.redirect('/login')
def create_user(self, json):
valid_params = self.verify_params(json, User.USER_REQUIRED_PARAMETERS)
if valid_params:
try:
new_user = User(**valid_params)
created_user = new_user.create()
result = {
"message": "Success!",
"user": created_user.to_dict(),
}
return jsonify(result), 201
except:
return jsonify(message="Server error!"), 500
else:
return jsonify(message="Bad Request!"), 400
def create_user(json):
valid_params = UserHandler.verify_params(json, User.USER_REQUIRED_PARAMETERS)
if valid_params:
try:
print(valid_params)
username_exists = User.verify_username(valid_params.get('username'))
if username_exists:
return jsonify(message="Username already taken."), 400
new_user = User(**valid_params)
created_user = new_user.create()
result = {
"message": "Success!",
"user": created_user.to_dict(),
}
return jsonify(result), 201
except Exception as err:
return jsonify(message="Server error!", error=err.__str__()), 500
else:
return jsonify(message="Bad Request!"), 400
def create_sample_data():
if get_engine().dialect.has_table(get_engine(), 'user'):
return
conn = get_engine()
create_tables(conn)
#add some client
session.begin()
for client in [
Client(name=('example_name_%s' % x),
surname=('example_surname_%s' % x)) for x in range(1, 4)
]:
session.add(client)
session.commit()
#add some user
session.begin()
for user in [
User(login=('user_%s' % x),
password=('user_%s' % x),
client=x,
permission_level=x) for x in range(1, 4)
]:
session.add(user)
session.commit()
#add some rooms
session.begin()
for room in [Room(name=('room_%s' % x)) for x in range(1, 4)]:
session.add(room)
session.commit()
#add some reservations
session.begin()
for roomstate in [
RoomState(reserved_from=datetime.date(2016, 11, 15),
reserved_to=datetime.date(2016, 11, 17 + x),
room=x,
client=3) for x in range(1, 4)
]:
session.add(roomstate)
session.commit()
class UserHandler(BaseHandler):
dao = User()
def get_all_users(self):
try:
users = self.dao.get_all_users()
result_list = []
for user in users:
result_list.append(user.to_dict())
result = {
"message": "Success!",
"users": result_list,
}
return jsonify(result), 200
except:
return jsonify(message="Server error!"), 500
def get_user_by_id(self, uid):
if uid:
try:
user = self.dao.get_user_by_id(uid)
if not user:
return jsonify(Error="User Not Found"), 404
else:
result = {
"message": "Success!",
"user": user.to_dict(),
}
return jsonify(result), 200
except:
return jsonify(message="Server Error!"), 500
else:
return jsonify(message="Bad request!"), 400
def create_user(self, json):
valid_params = self.verify_params(json, User.USER_REQUIRED_PARAMETERS)
if valid_params:
try:
new_user = User(**valid_params)
created_user = new_user.create()
result = {
"message": "Success!",
"user": created_user.to_dict(),
}
return jsonify(result), 201
except:
return jsonify(message="Server error!"), 500
else:
return jsonify(message="Bad Request!"), 400
def update_user(self, uid, json):
if uid:
try:
user_to_update = self.dao.get_user_by_id(uid)
if user_to_update:
valid_params = self.verify_params(
json, User.USER_REQUIRED_PARAMETERS)
for key, value in valid_params.items():
if key == "password":
user_to_update.update_password(value)
else:
setattr(user_to_update, key, value)
user_to_update.update()
result = {
"message": "Success!",
"user": user_to_update.to_dict(),
}
return jsonify(result), 200
else:
return jsonify(message="Not Found!"), 404
except:
return jsonify(message="Server Error!"), 500
else:
return jsonify(message="Bad Request!"), 400
def delete_user(self, uid):
if uid:
try:
user_to_delete = self.dao.get_user_by_id(uid)
if user_to_delete:
user_to_delete.delete()
return jsonify(message="Success!"), 200
else:
return jsonify(message="Not Found!"), 404
except:
return jsonify(message="Server Error!"), 500
else:
return jsonify(message="Bad Request!"), 400
