def put(self):
args = allParser.putUserParser.parse_args()
m_users = dbclient.list_column(table, ['username'])
if args['username'] in m_users:
return make_result(code=Code.ERROR, msg="已经存在此用户")
args["password"] = encode_password(args["password"])
m_result = dbclient.insert(table, args)
if m_result:
response = make_result(code=Code.SUCCESS)
else:
response = make_result(code=Code.ERROR, msg="新增失败")
return response
def post(self):
args = allParser.postLoginParser.parse_args()
m_user = dbclient.list_one(table, {"username": args["username"]})
if not m_user:
return make_result(code=Code.ERROR, msg="没有该用户")
m_user = m_user[0]
if m_user['password'] == encode_password(args['password']):
token = make_token()
dbclient.update(table, token, {"username": args["username"]})
# token["username"]
back = {}
back["name"] = m_user["name"]
back["token"] = token["token"]
back["role"] = m_user["role"]
response = make_result(data=back, code=Code.SUCCESS)
else:
response = make_result(code=Code.ERROR, msg="账户密码不一致")
return response
def post(self):
args = allParser.postUserParser.parse_args()
verify_result = verify_token(args["token"])
if not verify_result:
return make_result(code=Code.ERROR, msg="token无效")
args.pop('token')
values_keys = ['username', 'password', 'name', 'role']
condition_keys = ['id']
values = {
key: value
for key, value in args.items() if key in values_keys and args[key]
}
condition = {
key: value
for key, value in args.items() if key in condition_keys
}
values["password"] = encode_password(values["password"])
m_result = dbclient.update(table, values, condition)
if m_result:
response = make_result(code=Code.SUCCESS)
else:
response = make_result(code=Code.ERROR, msg="更新失败")
return response