def create_user():
"""
update user data to the database
"""
user = User.from_dict(g.args)
db.session.add(user)
db.session.commit()
def wrapper(*args, **kwargs):
if not getattr(func, 'auth', True):
return func(*args, **kwargs)
token = request.headers.get('Authorization') or request.args.get('token')
# to process the authorization correctly
if not token:
return 'Please login first!', 401
token = token.split(' ')[-1]
# logout for user
if token in redis_db.get('token_black_list'):
return 'Invalid token', 401
user, refresh = User.verify_auth_token(token)
if user:
g.user = user
try:
res = func(*args, **kwargs)
except Exception as e:
current_app.logger.exception(e)
res = '', 500
if refresh:
res.setdefault('token', user.generate_auth_token())
redis_db.get('token_black_list').append(token)
return res
else:
return 'Invalid token or token has expired', 401
def create_user(db_session: Session, *, user_in: UserCreate, otp: int) -> User:
user = User(email=user_in.email,
password_hash=get_password_hash(user_in.password),
otp=otp)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def create(db: Session, *, user_in: UserCreate) -> User:
user = User(email=user_in.email,
hashed_password=get_password_hash(user_in.password),
full_name=user_in.full_name)
db.add(user)
db.commit()
db.refresh(user)
return user
def create(db_session: Session, *, user_in: UserCreate) -> User:
user = User(email=user_in.email,
hashed_password=user_in.hashed_password,
role=user_in.role)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def create(self, db: Session) -> None:
user = User(
email=self.email,
password_hash=get_password_hash(self.password),
otp=self.otp,
)
db.add(user)
db.commit()
db.refresh(user)
self.id = user.id
def create(db_session: Session, *, user_in: UserCreate) -> User:
user = User(
email=user_in.email,
hashed_password=get_password_hash(user_in.password),
full_name=user_in.full_name,
is_superuser=user_in.is_superuser,
)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def update_user_info(db_session: Session,
*,
user: User,
data: UserUpdate,
otp: int = None) -> User:
"""If `otp` sent is not None:
`user.otp = otp`
`user.email_verified = False`
`user.otp_created_at = datetime.utcnow()`
"""
data = jsonable_encoder(data, exclude_none=True)
for field in data:
setattr(user, field, data[field])
if otp:
user.otp = otp
user.email_verified = False
user.otp_created_at = datetime.utcnow()
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def update(db_session: Session, *, user: User, user_in: UserUpdate) -> User:
user_data = jsonable_encoder(user)
update_data = user_in.dict(skip_defaults=True)
for field in user_data:
if field in update_data:
setattr(user, field, update_data[field])
if user_in.password:
passwordhash = get_password_hash(user_in.password)
user.hashed_password = passwordhash
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def update(db_session, *, user: User, user_in: UserInUpdate) -> User:
user_data = jsonable_encoder(user)
for field in user_data:
if field in user_in.fields:
value_in = getattr(user_in, field)
if value_in is not None:
setattr(user, field, value_in)
if user_in.password:
passwordhash = get_password_hash(user_in.password)
user.hashed_password = passwordhash
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def create(db_session, *, user_in: UserInCreate) -> User:
create_user_in_cognito(db_session,
user_to_create=user_in.email,
user_pool_id=None)
user = User(email=user_in.email,
hashed_password=get_password_hash(user_in.password),
full_name=user_in.full_name,
is_superuser=user_in.is_superuser,
business_id=user_in.business_id,
created_by=user_in.created_by,
modified_by=user_in.modified_by,
is_active=user_in.is_active)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def create(db_session: Session, *, user_in: UserCreate) -> User:
user_in_city = user_in.city.lower().strip()
city = db_session.query(City).filter_by(name=user_in_city).first()
if not city:
city = City(name=user_in_city)
db_session.add(city)
db_session.commit()
db_session.refresh(city)
user = User(email=user_in.email,
hashed_password=get_password_hash(user_in.password),
full_name=user_in.full_name,
city_id=city.id)
db_session.add(user)
db_session.commit()
db_session.refresh(user)
return user
def setup():
db.drop_all()
db.create_all()
User.generate_fake()
def user_email_verified(db_session: Session, *, user: User) -> User:
user.email_verified = True
db_session.commit()
return user
