def test_encode_auth_token(self):
user = User(userId=Counter.getNextSequence('userId'),
userName='******',
email='*****@*****.**',
dateRegistered=datetime.datetime.utcnow())
user.userPassword = '******'
user.save()
auth_token = user.encode_auth_token(user.userId)
self.assertTrue(isinstance(auth_token, bytes))
def save_new_user(data):
user = User.objects(email=data["email"])
if not user:
hashed_password = flask_bcrypt.generate_password_hash(
data['password']).decode('utf-8')
new_user = User(username=data["username"],
email=data["email"],
first_name=data["first_name"],
last_name=data["last_name"],
password=hashed_password)
new_user.save()
logger.debug("NEW USER CREATED: ", new_user)
return generate_token(new_user)
else:
return ResponseObject.user_reg(success=False)
def save_new_user(data):
new_user = User(userId=Counter.getNextSequence('userId'),
userName=data['userName'],
email=data['email'],
dateRegistered=datetime.datetime.utcnow())
new_user.userPassword = data['userPassword']
try:
new_user.save()
except DuplicateKeyError:
response_object = {
'status': 'fail',
'message': 'User already exists. Please Log in.'
}
return response_object, 409 # 409 status : Conflict
else:
return generate_token(new_user)
def save_new_user(data):
user = User.query.filter_by(email=data['email']).first()
token = ''
if not user:
email = data['email']
password_hash = flask_bcrypt.generate_password_hash(
data['password']).decode('utf-8')
is_active = data['is_active']
is_tfa = data['is_tfa']
is_admin = data['is_admin']
user = User(email, password_hash, is_active, is_tfa, is_admin)
user.save()
if user.id:
token = generate_confirmation_token(data['email'])
body = "Welcome! Thanks for signing up. Please click this link to activate your account: " + request.base_url + "confirm/{}".format(
token)
subject = "Please confirm your email"
email_send = 0
send_email(user.email, subject, body)
response = jsonify({
'status':
'success',
'message':
'User Successfully Registered ... ! check your email [' +
email + ']',
'account_name':
user.email,
'created_at':
user.created_at
})
response.status_code = 201
return response
else:
response = jsonify({
'status': 'fail',
'message': 'User already exists. Please Log in.',
})
response.status_code = 409
return response
def save_new_user(data):
user = User.objects.filter(email=data['email']).first()
if not user:
password = flask_bcrypt.generate_password_hash(
data['password']).decode('utf-8')
new_user = User(public_id=str(uuid.uuid4()),
email=data['email'],
username=data['username'],
password=password,
registered_on=datetime.datetime.utcnow())
new_user.save()
response_object = {
'status': 'success',
'message': 'Successfully registered.'
}
return response_object, 201
else:
response_object = {
'status': 'fail',
'message': 'User already exists. Please Log in.',
}
return response_object, 409
