def is_user_can_join_referral_program(user):
r = Referral.find_referral_by_uid(user.id)
# we remove is_valid_email(user.email) temporary
if r is None:
return True
return False
def reward_user_redeem_code(user_id, code):
u = User.find_user_with_id(user_id)
if u is not None and is_valid_email(u.email):
r = Referral.find_referral_by_uid(u.id)
if r is not None:
send_reward_redeem.delay(
u.email, code,
'{}prediction?refer={}'.format(g.BASE_URL, r.code))
def issue_referral_code_for_user(user):
r = Referral.find_referral_by_uid(user.id)
if r is None:
code = generate_referral_code(user.id)
r = Referral(code=code, user_id=user.id)
db.session.add(r)
db.session.flush()
return r.code
def check_referral():
"""
" check user has joined referral program or not
"""
try:
uid = int(request.headers['Uid'])
r = Referral.find_referral_by_uid(uid)
response = {
"code": None,
"referral_link": None,
"referred_users": referral_bl.all_referred_users_by_user(uid)
}
if r is None:
return response_error(response)
response['code'] = r.code
response['referral_link'] = '{}prediction?refer={}'.format(
g.BASE_URL, r.code)
return response_ok(response)
except Exception, ex:
return response_error(ex.message)
def clear_data_before_test(self):
r = Referral.find_referral_by_uid(88)
if r is not None:
db.session.delete(r)
db.session.commit()
