158 lines
7.2 KiB
Python
158 lines
7.2 KiB
Python
import os, json, re, jsonpickle
|
|
import sqlite3 as sql
|
|
|
|
data_dir = "data"
|
|
league_dir = "leagues"
|
|
|
|
def create_connection(league_name):
|
|
#create connection, create db if doesn't exist
|
|
conn = None
|
|
try:
|
|
if not os.path.exists(os.path.join(data_dir, league_dir, league_name)):
|
|
|
|
os.makedirs(os.path.join(data_dir, league_dir, league_name))
|
|
conn = sql.connect(os.path.join(data_dir, league_dir, league_name, f"{league_name}.db"))
|
|
|
|
# enable write-ahead log for performance and resilience
|
|
conn.execute('pragma journal_mode=wal')
|
|
|
|
return conn
|
|
except:
|
|
print("oops, db connection no work")
|
|
return conn
|
|
|
|
def state(league_name):
|
|
if not os.path.exists(os.path.dirname(os.path.join(data_dir, league_dir, league_name, f"{league_name}.state"))):
|
|
os.makedirs(os.path.dirname(os.path.join(data_dir, league_dir, league_name, f"{league_name}.state")))
|
|
with open(os.path.join(data_dir, league_dir, league_name, f"{league_name}.state")) as state_file:
|
|
return json.load(state_file)
|
|
|
|
def init_league_db(league):
|
|
conn = create_connection(league.name)
|
|
|
|
player_stats_table_check_string = """ CREATE TABLE IF NOT EXISTS stats (
|
|
counter integer PRIMARY KEY,
|
|
id text,
|
|
name text,
|
|
team_name text,
|
|
outs_pitched integer DEFAULT 0,
|
|
walks_allowed integer DEFAULT 0,
|
|
hits_allowed integer DEFAULT 0,
|
|
strikeouts_given integer DEFAULT 0,
|
|
runs_allowed integer DEFAULT 0,
|
|
plate_appearances integer DEFAULT 0,
|
|
walks_taken integer DEFAULT 0,
|
|
sacrifices integer DEFAULT 0,
|
|
hits integer DEFAULT 0,
|
|
home_runs integer DEFAULT 0,
|
|
total_bases integer DEFAULT 0,
|
|
rbis integer DEFAULT 0,
|
|
strikeouts_taken integer DEFAULT 0
|
|
);"""
|
|
|
|
teams_table_check_string = """ CREATE TABLE IF NOT EXISTS teams (
|
|
counter integer PRIMARY KEY,
|
|
name text NOT NULL,
|
|
wins integer DEFAULT 0,
|
|
losses integer DEFAULT 0,
|
|
run_diff integer DEFAULT 0
|
|
); """
|
|
|
|
if conn is not None:
|
|
c = conn.cursor()
|
|
c.execute(player_stats_table_check_string)
|
|
c.execute(teams_table_check_string)
|
|
|
|
for team in league.teams_in_league():
|
|
c.execute("INSERT INTO teams (name) VALUES (?)", (team.name,))
|
|
|
|
player_string = "INSERT INTO stats (name, team_name) VALUES (?,?)"
|
|
for batter in team.lineup:
|
|
c.execute(player_string, (batter.name, team.name))
|
|
for pitcher in team.rotation:
|
|
c.execute(player_string, (pitcher.name, team.name))
|
|
|
|
state_dic = {
|
|
"day" : league.day,
|
|
"schedule" : league.schedule,
|
|
"game_length" : league.game_length,
|
|
"series_length" : league.series_length,
|
|
"games_per_hour" : league.games_per_hour,
|
|
"owner" : None,
|
|
"historic" : False
|
|
}
|
|
if not os.path.exists(os.path.dirname(os.path.join(data_dir, league_dir, league.name, f"{league.name}.state"))):
|
|
os.makedirs(os.path.dirname(os.path.join(data_dir, league_dir, league.name, f"{league.name}.state")))
|
|
with open(os.path.join(data_dir, league_dir, league.name, f"{league.name}.state"), "w") as state_file:
|
|
json.dump(state_dic, state_file, indent=4)
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def save_league(league):
|
|
if league_exists(league.name):
|
|
state_dic = {
|
|
"day" : league.day,
|
|
"schedule" : league.schedule,
|
|
"game_length" : league.game_length,
|
|
"series_length" : league.series_length,
|
|
"games_per_hour" : league.games_per_hour,
|
|
"owner" : league.owner,
|
|
"historic" : league.historic
|
|
}
|
|
with open(os.path.join(data_dir, league_dir, league.name, f"{league.name}.state"), "w") as state_file:
|
|
json.dump(state_dic, state_file, indent=4)
|
|
|
|
def add_stats(league_name, player_game_stats_list):
|
|
conn = create_connection(league_name)
|
|
if conn is not None:
|
|
c=conn.cursor()
|
|
for team_name in player_game_stats_list.keys():
|
|
for (name, player_stats_dic) in player_game_stats_list[team_name]:
|
|
c.execute("SELECT * FROM stats WHERE name=? AND team_name=?",(name, team_name))
|
|
this_player = c.fetchone()
|
|
if this_player is not None:
|
|
for stat in player_stats_dic.keys():
|
|
c.execute(f"SELECT {stat} FROM stats WHERE name=? AND team_name=?",(name, team_name))
|
|
old_value = int(c.fetchone()[0])
|
|
c.execute(f"UPDATE stats SET {stat} = ? WHERE name=? AND team_name=?",(player_stats_dic[stat]+old_value, name, team_name))
|
|
else:
|
|
c.execute("INSERT INTO stats(name) VALUES (?)",(name,))
|
|
for stat in player_stats_dic.keys():
|
|
c.execute(f"UPDATE stats SET {stat} = ? WHERE name=? AND team_name=?",(player_stats_dic[stat], name, team_name))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def update_standings(league_name, update_dic):
|
|
if league_exists(league_name):
|
|
conn = create_connection(league_name)
|
|
if conn is not None:
|
|
c = conn.cursor()
|
|
|
|
for team_name in update_dic.keys():
|
|
for stat_type in update_dic[team_name].keys(): #wins, losses, run_diff
|
|
c.execute(f"SELECT {stat_type} FROM teams WHERE name = ?", (team_name,))
|
|
old_value = int(c.fetchone()[0])
|
|
c.execute(f"UPDATE teams SET {stat_type} = ? WHERE name = ?", (update_dic[team_name][stat_type]+old_value, team_name))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def get_standings(league_name):
|
|
if league_exists(league_name):
|
|
conn = create_connection(league_name)
|
|
if conn is not None:
|
|
c = conn.cursor()
|
|
|
|
c.execute("SELECT name, wins, losses, run_diff FROM teams",)
|
|
standings_array = c.fetchall()
|
|
conn.close()
|
|
return standings_array
|
|
|
|
|
|
|
|
def league_exists(league_name):
|
|
with os.scandir(os.path.join(data_dir, league_dir)) as folder:
|
|
for subfolder in folder:
|
|
if league_name in subfolder.name:
|
|
return not state(league_name)["historic"]
|
|
return False |