def test_register_on_error(self, client):
password = getmd5("123456")
rv = client.post("/api/user/register",
json={
"email": self._email,
"password": password
})
password = getmd5("123456")
rv = client.post("/api/user/register",
json={
"email": self._email,
"password": password
})
assert rv.status_code == 400
def register():
params = request.values or request.get_json() or {}
email = params.get("email")
password = params.get("password")
if len(password) != 32:
return CommonError.get_error(40000)
exsist_user = db.session.query(User).filter_by(email=email).first()
if exsist_user:
return UserError.get_error(40200)
salt = current_app.config['SECURITY_PASSWORD_SALT'] or 'token'
token = getmd5("{}{}{}".format(salt, email, get_random_num(5)))
user = User(email=email, password=password, status=1)
user.token = token
db.session.add(user)
db.session.commit()
payload = {}
payload['user_id'] = user.id
return response_succ(body=payload)
def login():
params = request.values or request.get_json() or {}
email = params.get("email")
password = params.get("password")
exsist_user: User = db.session.query(User).filter_by(
email=email, password=password).first()
if exsist_user:
# update token
salt = current_app.config['SECURITY_PASSWORD_SALT'] or 'token'
token = getmd5("{}{}{}".format(salt, email, get_random_num(5)))
exsist_user.token = token
# update log time
record = LoginRecord()
record.user_id = exsist_user.id
record.login_time = get_unix_time_tuple()
record.log_ip = request.args.get("user_ip") or request.remote_addr
db.session.add(record)
db.session.commit()
return response_succ(body={'token': token})
else:
return UserError.get_error(40203)
def setup_method(self):
self._email = "*****@*****.**"
self._password = getmd5("123456")
def setup_method(self):
self._email = "*****@*****.**"
self._password = getmd5("123456")
self._app = create_app("testing")
self._client = self._app.test_client()