def wine_expert():
expert = create_user(None, assign_wine_expert=False)
expert.first_name = 'wine_expert'
expert.email = '*****@*****.**'
db.session.commit()
yield expert
def user_no_stripe(wine_expert, sample_jwt_payload3):
_user = create_user(sample_jwt_payload3['sub'])
_user.first_name = 'tuser_no_stripe'
_user.cognito_sub = sample_jwt_payload3['sub']
_user.wine_expert_id = wine_expert.id
_user.registration_finished = True
db.session.commit()
yield _user
def user2(wine_expert, sample_jwt_payload2):
_user = create_user(sample_jwt_payload2['sub'])
_user.first_name = 'tuser2'
_user.cognito_sub = sample_jwt_payload2['sub']
_user.wine_expert_id = wine_expert.id
_user.stripe_customer_id = 'stripe_customer_id'
db.session.commit()
yield _user
def user_hardcoded_sub(wine_expert, sample_jwt_payload_hardcoded_sub):
_user = create_user(sample_jwt_payload_hardcoded_sub['sub'])
_user.first_name = 'tuser'
_user.wine_expert_id = wine_expert.id
_user.stripe_customer_id = 'stripe_customer_id'
_user.email = '*****@*****.**'
db.session.commit()
yield _user
def user(wine_expert, sample_jwt_payload):
_user = create_user(sample_jwt_payload['sub'])
_user.first_name = 'tuser'
_user.wine_expert_id = wine_expert.id
_user.stripe_customer_id = 'stripe_customer_id'
_user.email = '*****@*****.**'
_user.registration_finished = True
db.session.commit()
yield _user
def user4(wine_expert, sample_jwt_payload4):
_user = create_user(sample_jwt_payload4['sub'])
_user.first_name = 'tuser4'
_user.last_name = 'tlastname4'
_user.avatar = 'https://some.pic/url/4/'
_user.bio = 'test bio 4'
_user.cognito_sub = sample_jwt_payload4['sub']
_user.wine_expert_id = wine_expert.id
_user.stripe_customer_id = 'stripe_customer_id'
db.session.commit()
yield _user
def _process_absent_cognito_users(self):
for absent_user_subject, absent_user in self._cognito_user_dict.items(
):
try:
logging.warning(
"Cognito user with subject=%s is not present in the database - inserting...",
absent_user_subject,
)
user = create_user(absent_user_subject)
populate_most_of_cognito_fields(user, absent_user_subject,
absent_user)
populate_cognito_display_status(user)
self.successfully_added_count += 1
except:
self.exceptions_count += 1
logging.exception(
'Exception while adding user with cognito_sub=%s to the database',
absent_user_subject)