def save_new_user(data):
user = User.query.filter_by(email=data['email']).first()
if not user:
new_user = User(
public_id=data['public_id'],
email=data['email'],
username=data['username'],
password=data['password'],
level_cd=data['level_cd'],
user_role=data['user_role'],
reg_datetime=datetime.datetime.utcnow(),
memo="",
)
if data['level_cd'] == '99': # super admin
new_user.is_admin = True
else:
new_user.is_admin = False
user_detail = UserDetail(
public_id=data['public_id'],
address=data['address'],
phone_number=data['phone_number'],
reg_datetime=datetime.datetime.utcnow(),
)
save_changes(new_user, user_detail) # insert to DB
return generate_token(new_user)
else:
response_object = {
'status': 'fail',
'message': 'User email already exists. Please Log in.',
}
return response_object, 409
def load_db():
with app.app_context():
User.query.delete()
db.session.commit()
for data in samples:
user = User(**dict(zip(attrs, data)))
if user.username=='Admin':
user.is_admin = True
db.session.add(user)
db.session.commit()
print(User.query.all())
