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())