class Standings(Ncaa):
'''Standings class that will pull and show standings for the brackets in a defined pool'''
def __init__(self, **kwargs):
self.__db = MysqlPython()
self.__pool = Pool()
def get_standings(self, **kwargs):
'''Get standings data for the pool'''
bracket_type = kwargs['bracket_type'] + 'Bracket'
pool_name = self.__pool.get_pool_name()
pool_status = self.__pool.check_pool_status(bracket_type)
# fetch the user standings
data = self.__db.query(
proc='Standings',
params=[pool_status['is_open'], pool_name, bracket_type])
# get number of games played to see if there are games left
# this will help determine if we show the best possible score column in the standings
result = self.__db.query(proc='AreThereGamesLeft', params=[])
games_left = result[0]['status']
# we are using the seed bonus scoring just ignore the best possible score
pool_info = self.__pool.get_pool_info()
if pool_info['seedBonusScoring'] == 1:
games_left = 0
return (data, games_left)
class User(Ncaa):
'''User class to get/set information for a user that is tied to their pool. A user (email) can exist in multiple pools'''
def __init__(self):
self.__db = MysqlPython()
self.__pool = Pool()
# set user specific attributes
self.__user_edit_token = ''
self.__username = ''
self.__first_name = ''
self.__email_address = ''
# get admin edit token
config = configparser.ConfigParser()
config.read("site.cfg")
self.__admin_edit_token = config.get('DEFAULT', 'ADMIN_EDIT_TOKEN')
def get_admin_edit_token(self):
'''Return the edit token for the admin user'''
return self.__admin_edit_token
def setup_user_edit_token(self, **kwargs):
'''
Creates a token string to be encoded from the time, pool name, user email, username, bracket type and display type.
This combo ensures a unique token for every user
'''
token = [
str(time.time()), session['pool_name'], kwargs['email_address'],
kwargs['username'], kwargs['bracket_type_name'], 'edit'
]
edit_token = self.set_token('.'.join(token))
self.set_edit_token(edit_token)
return edit_token
def setup_user_display_token(self, **kwargs):
'''
Creates a token string to be encoded from the time, pool name, user email, username, bracket type and display type.
This combo ensures a unique token for every user
'''
token = [
str(time.time()), session['pool_name'], kwargs['email_address'],
kwargs['username'], kwargs['bracket_type_name'], 'display'
]
return self.set_token('.'.join(token))
def set_token(self, string):
'''Creates an md5 hash from string parameter'''
h = hashlib.new('ripemd160')
h.update(string.encode('utf-8'))
return h.hexdigest()
def set_username(self, username):
'''Method to set username'''
self.__username = username
def get_edit_token(self):
'''Method to get user edit token'''
return self.__user_edit_token
def set_edit_token(self, token):
'''Method to set user edit token'''
self.__user_edit_token = token
def set_user_bracket_name(self):
'''
Fixes the usernmame for display.
If the username doesnt end in an 's' append one for the bracket name
ex: Jim -> Jim's
ex: Balls stays Balls'
'''
# get rid of spaces at the end of the user name and append '
username = re.sub(r'\s+$', '', self.__username)
username += "'"
# append 's' at the end of the string if the string doesnt already end with "'s"
if not re.search(r"s'$", username):
username += 's'
return username
def update_user_data(self):
'''
Add/update the user information
Ex: email, username, etc
'''
edit_type = request.values['edit_type']
# add/edit user data
if edit_type == 'add':
user_id = self.insert_new_user()
else:
user_id = self.update_user()
#self.debug(f"Working with user {user_id}")
return user_id
def insert_new_user(self, **kwargs):
'''Insert new user to DB'''
bracket_type_name = request.values['bracket_type_name']
email_address = request.values['email_address']
username = request.values['username']
tie_breaker_points = request.values['tie_breaker_points']
first_name = request.values['first_name']
transaction_order_id = request.values['transaction_order_id']
# handle error "Failed calling stored routine; bytearray index out of range error"
# TODO: find a proper fix
if first_name == '':
first_name = ' '
if transaction_order_id == '':
transaction_order_id = ' '
# setup user tokens
display_token = self.setup_user_display_token(
email_address=email_address,
username=username,
bracket_type_name=bracket_type_name)
edit_token = self.setup_user_edit_token(
email_address=email_address,
username=username,
bracket_type_name=bracket_type_name)
user_id = self.__db.insert(proc='InsertUser',
params=[
self.__pool.get_pool_name(), username,
email_address, tie_breaker_points,
first_name, edit_token, display_token,
bracket_type_name, transaction_order_id
])
return user_id
def update_user(self, **kwargs):
'''Update the user information based on their edit token value'''
edit_token = self.get_edit_token()
email_address = request.values['email_address']
username = request.values['username']
first_name = request.values['first_name']
tie_breaker_points = request.values['tie_breaker_points']
user_id = self.__db.update(proc='UpdateUser',
params=[
edit_token, username, email_address,
tie_breaker_points, first_name
])
return user_id