def post(self):
args = parse_base.parse_args()
password = args.get('password')
action = args.get('action').lower()
if action == USER_ACTION_REGISTER:
args_register = parse_register.parse_args()
username = args_register.get('username')
phone = args_register.get('phone')
movie_user = MovieUser()
movie_user.username = username
movie_user.password = password
movie_user.phone = phone
if not movie_user.save():
abort(400, msg='创建失败')
data = {
'status': HTTP_CREATE_OK,
'msg': '用户创建成功',
'data': movie_user,
}
return marshal(data, single_movie_user_fields)
elif action == USER_ACTION_LOGIN:
args_login = parse_login.parse_args()
username = args_login.get('username')
phone = args_login.get('phone')
user = get_movie_user(username) or get_movie_user(phone)
if not user:
abort(400, msg='用户不存在')
if not user.check_password(password):
abort(401, msg='密码错误')
if user.is_delete:
abort(401, msg='用户不存在')
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {
'msg': '登录成功',
'status': HTTP_OK,
'token': token,
}
return data
else:
abort(400, msg='请提供正确的参数')
def post(self):
args = parse_base.parse_args()
password = args.get('password')
print(password)
action = args.get('action').lower()
print(action)
if action == USER_ACTION_REGISTER:
args_register = parse_register.parse_args()
phone = args_register.get('phone')
username = args_register.get('username')
movie_user = MovieUser()
movie_user.username = username
movie_user.password = password
movie_user.phone = phone
if not movie_user.save():
abort(400, msg="create fail")
data = {
"status": HTTP_CREATE_OK,
"msg": "用户创建成功",
"data": movie_user
}
return marshal(data, single_movie_user_fields)
elif action == USER_ACTION_LOGIN:
args_login = parse_login.parse_args()
username = args_login.get("username")
phone = args_login.get("phone")
user = get_movie_user(username) or get_movie_user(phone)
if not user:
abort(400, msg="用户不存在")
if not user.check_password(password):
abort(401, msg="密码错误")
if user.is_delete:
abort(401, msg="用户不存在")
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {"msg": "login success", "status": HTTP_OK, "token": token}
return data
else:
abort(400, msg="其提供正确的参数")
def post(self):
args = parser_base.parse_args()
password = args.get("password")
action = args.get("action").lower()
if action == USER_ACTION_REGISTER:
args_register = parser_register.parse_args()
username = args_register.get("username")
phone = args_register.get("phone")
movie_user = CinemaUser()
movie_user.username = username
movie_user.password = password
movie_user.phone = phone
if not movie_user.save():
abort(400, msg="创建用户失败")
data = {
"status": HTTP_CREATE_OK,
"msg": "用户创建成功",
"data": movie_user,
}
return marshal(data, single_movie_user_fields)
elif action == USER_ACTION_LOGIN:
args_login = parser_login.parse_args()
username = args_login.get("username")
phone = args_login.get("phone")
user = get_movie_user(phone)
if not user:
user = get_movie_user(username)
if not user:
abort(400, msg="用户不存在")
if not user.check_password(password):
abort(401, msg="用户名或密码错误")
if user.is_delete:
abort(401, msg="用户不存在")
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {
"msg": "success",
"status": HTTP_OK,
"token": token,
}
return data
else:
abort(400, msg="参数错误")
def post(self):
args = parse_base.parse_args()
password = args.get('password')
action = args.get('action').lower()
if action == USER_REGISTER:
args_registe = parse_regist.parse_args()
name = args_registe.get('name')
phone = args_registe.get('phone')
user = MovieUser()
user.name = name
user.phone = phone
user.password = password
if not user.save():
abort(400, msg="create fail")
data = {
'msg': "创建成功",
'status': HTTP_CREAT_OK,
'data': user,
}
return marshal(data, single_movie_user_fileds)
elif action == USER_LOGIN:
args_login = parse_login.parse_args()
name = args_login.get('name')
phone = args_login.get('phone')
user = get_user(phone) or get_user(name)
if not user:
abort(400, msg="请输入用户名")
if not user.check_password(password):
abort(400, msg="密码错误")
if user.is_delete:
abort(400, msg="用户不存在")
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {
'status': HTTP_OK,
'msg': 'http_ok',
'token': token,
}
return data
else:
abort(400, msg="请提供正确的参数")
def post(self):
args = parse_base.parse_args()
password = args.get("password")
action = args.get("action")
if action == "register":
register_args = parse_register.parse_args()
username = register_args.get("username")
phone = register_args.get("phone")
movie_user = MovieUser()
movie_user.username = username
movie_user.password = password
movie_user.phone = phone
if not movie_user.save():
abort(400, msg="create fail")
data = {"status": '201', "msg": "用户创建成功", "data": movie_user}
return marshal(data, single_movie_user_fields)
elif action == "login":
login_args = parse_login.parse_args()
username = login_args.get("username")
phone = login_args.get("phone")
user = get_movie_user(username) or get_movie_user(phone)
if not user:
abort(400, msg="用户不存在")
if not user.check_password(password):
abort(401, msg="密码错误")
if user.is_delete:
abort(400, msg="用户不存在")
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {"msg": "login success", "status": 200, "token": token}
return data
else:
abort(400, msg="请提供正确的参数")
def post(self):
args = parse_base.parse_args()
password = args.get("password")
action = args.get("action").lower()
# 注册
if action == USER_ACTION_REGISTER:
args_register = parse_register.parse_args()
phone = args_register.get("phone")
username = args_register.get("username")
movie_user = MovieUser()
movie_user.username = username
movie_user.password = password
movie_user.phone = phone
if not movie_user.save():
abort(400, msg="create fail")
data = {
"status": HTTP_CREATE_OK,
"msg": "用户创建成功",
"data": marshal(movie_user, movie_user_fields),
}
return data
# 登录
elif action == USER_ACTION_LOGIN:
args_login = parse_login.parse_args()
phone = args_login.get("phone")
username = args_login.get("username")
user = get_user(username) or get_user(phone)
if not user:
abort(400, msg="用户不存在")
if not user.check_password(password):
abort(401, msg="密码错误")
if user.is_delete:
abort(401, msg="用户不存在")
# 给用户发送token,并存到缓存中
token = generate_movie_user_token()
cache.set(token, user.id, timeout=60 * 60 * 24 * 7)
data = {
"msg": "login success",
"status": HTTP_OK,
"token": token,
}
return data
else:
abort(400, msg="请提供正确的参数")
