def setUp(self):
now = datetime.datetime.now()
UserProfile.objects(username='******').update(
upsert=True,
set__username='******',
set__password=
'******',
last_login=now,
)
self.login('123', '15d77445c11f15134e765d6acaea7a68')
def put(self):
req = pwd_parser.parse_args(strict=True)
user_obj = UserProfile.objects(username=req["userName"]).first()
if user_obj and user_obj.check_password(req["oldPwd"]):
user_obj.set_password(req["newPwd"])
return make_response()
else:
return make_response(status=const.AUTH_ERROR)
def post(self):
req = user_parser.parse_args(strict=True)
user_obj = UserProfile.objects(username=req["username"]).first()
if user_obj and user_obj.check_password(req["password"]):
login_user(user_obj, remember=True)
return make_response()
else:
return make_response(status=const.AUTH_ERROR)
def insert(username, password):
obj = UserProfile.objects(username=username).first()
print("账号：{}".format(username))
pass_hashed = get_password(password)
# pass_save = flask_bcrypt.generate_password_hash(pass_hashed)
if obj:
obj.set_password(password=pass_hashed)
else:
obj = UserProfile(username=username).save()
obj.set_password(password=pass_hashed)
print("密码:{}".format(password))
return 0
def tearDown(self):
self.logout()
UserProfile.objects(username='******').delete()
def load_user(user_id):
return UserProfile.objects(id=user_id).first()
