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