def create_team():
post_body = json.loads(request.data)
playerOneId = post_body.get('playerOneId', 0)
playerTwoId = post_body.get('playerTwoId', 0)
team = Team(playerOneId=playerOneId, playerTwoId=playerTwoId)
db.session.add(team)
db.session.commit()
return json.dumps({'success': True, 'data': team.serialize()}), 201
def init_db(session):
Base.metadata.create_all(engine)
# skip if there are games already
if session.query(Game).first():
return
game = Game(name="Test Game", uuid="A97E57A2-E440-4975-A624-6F99999D64DA")
questions = [Question(question="Who won the quiz?", game=game)]
session.add(game)
game = Game(name="new", uuid="C0CFAF27-0770-400E-A9B9-82461ED9FB6F")
questions = [
Question(question="Who is good at this?", game=game),
Question(question="Who read the next question?", game=game),
]
session.add(game)
session.commit()
t = Team(name="ee22dc", team_code="111", game_id=game.id)
session.commit()
p = Player(color="#ee22dc", name="p1")
sp = PlayerInGame(team=t, player=p, game=game)
session.add(p)
session.commit()
session.add(
GivenAnswer(answer="This is my answer",
question_uuid=questions[0].uuid,
player_id=p.id))
session.commit()
def parse_teams(data, session):
"""
Adds new teams to the Teams table and stats to the Teamstats table
"""
game_id = data["gameId"]
for team in data["gamepackageJSON"]["boxscore"]["players"]:
team_id = team["team"]["abbreviation"]
team_name = team["team"]["shortDisplayName"]
# Only add team if they are not yet in the db
t = session.query(Team).filter_by(team_id=team_id).first()
if not t:
t = Team(team_id=team_id, name=team_name)
session.add(t)
s = parse_stats(team["statistics"][0]["totals"])
team_stats = TeamIn(team=team_id,
game=game_id,
fgm=s["fgm"],
fga=s["fga"],
fgm3=s["fgm3"],
fga3=s["fga3"],
ftm=s["ftm"],
fta=s["fta"],
tp=s["tp"],
blk=s["blk"],
stl=s["stl"],
ast=s["ast"],
oreb=s["oreb"],
dreb=s["dreb"],
treb=s["treb"],
pf=s["pf"],
to=s["to"])
session.add(team_stats)
def create_team_handler(user):
team = Team.create(team_leader_id=user.user_id)
u = str(uuid.uuid4())
code = str(team.team_id) + u[0: 4]
team.code = code
team.save()
user.team_id = team.team_id
user.save()
data = {'code_card':team,'team_member_cards':[]}
return bot_postbacks.view_team_postback(data, user)
def team_code_handler(receiver, user):
text = receiver.get_text()
team_query = Team.select().where(Team.code == text)
if len(team_query) > 0:
team = team_query[0]
members = User.select().where(User.team_id == team.team_id)
if len(members) == 6:
# Team full
return bot_messages.team_full_message(user)
user.team_id = team.team_id
user.save()
return bot_messages.team_code_ask_success_message(team.team_id, user)
return bot_messages.team_code_ask_failure_message(user)
def addteam_(self, **kwargs):
success, serial = self._addteamform.Validate(kwargs)
if not success:
raise cpy.HTTPRedirect("/admin/addteam?%s" % Form.Encode(serial))
session = Session()
team = Team(kwargs['name'],
kwargs['thumb'],
kwargs['fullsize'],
encrypt(kwargs['password']),
1,
kwargs['description'])
session.save(team)
session.commit()
session.close()
raise cpy.HTTPRedirect("/admin/lsteam/")
def refresh_stats():
tid = request.form['tid']
stats = sbteamdata.get_team_data(tid)
team = Team.query.filter_by(teamid=tid).order_by(Team.season.desc()).all()
players = Player.query.filter_by(teamid=tid, season=stats.season).all()
pitchers = Pitcher.query.filter_by(teamid=tid, season=stats.season).all()
for entry in team:
if entry.season == int(stats.season):
db.session.delete(entry)
db.session.commit()
new_entry = Team(datetime.now(), stats.season, tid, stats.name,
stats.stars, stats.stadium, stats.pl, stats.pl_position,
stats.trained, stats.games, stats.ab, stats.hits,
stats.avg, stats.singles, stats.doubles, stats.triples,
stats.hr, stats.rbi, stats.so, stats.era, stats.slg,
stats.wins, stats.losses, stats.pct, stats.rs, stats.ra)
db.session.add(new_entry)
if players:
for player in players:
db.session.delete(player)
db.session.commit()
for player in stats.players:
new_entry = Player(stats.season, tid, player.id, player.name,
player.position, player.games, player.ab,
player.hits, player.avg, player.singles,
player.doubles, player.triples, player.hr,
player.rbi, player.so, player.era, player.slg)
db.session.add(new_entry)
if pitchers:
for player in pitchers:
db.session.delete(player)
db.session.commit()
for player in stats.pitchers:
new_entry = Pitcher(stats.season, tid, player.id, player.name,
player.games, player.so, player.era)
db.session.add(new_entry)
db.session.commit()
return redirect('/stats/{0}'.format(tid))
async def join_team(player: "PlayerConnection", session, *, team_code):
game = player.current_game(session)
if not game:
return
team = (session.query(Team).filter(Team.team_code == team_code).filter(
Team.game_id == game.id).first())
if team is None:
team = Team(team_code=team_code, name="", game_id=game.id)
session.add(team)
sub_player = player.player_in_game(session)
team.players.append(sub_player)
TEAM_IDS[player.websocket] = team.id
await send_team_info(player, session, team=team)
await send_questions(player, session, team=team)
await send_answers(player, session, team=team)
await send_selected_answers(player, session, team=team)
def xml_to_database(xml_file):
game_info = parse_game_file(xml_file)
# Extract information for the Game table
venue = game_info['venue']
home = venue["home_id"]
vis = venue["vis_id"]
# Check if information for this game has already been added - if it has, then exit the function
if session.query(Game).filter_by(date=venue['date'],
home=venue['home_id']).first():
return "ERR: Game data already documented. Aborting upload"
g = Game(date=venue['date'],
home=venue['home_id'],
visitor=venue['vis_id'],
isLeague=venue['is_league'],
isPlayoff=venue['is_playoff'])
session.add(g)
# Extract information for Team table, adding the playing teams to the database if they don't already exist
t1 = session.query(Team).filter_by(team_id=venue['home_id']).first(
) # Should only be one team with each id, so we can use first()
t2 = session.query(Team).filter_by(team_id=venue['vis_id']).first()
if not t1:
t1 = Team(team_id=venue['home_id'], name=venue['home_name'])
session.add(t1)
if not t2:
t2 = Team(team_id=venue['vis_id'], name=venue['vis_name'])
session.add(t2)
# Extract information for the TeamIn table
""" TODO: Wrap everything in its own adder function, maybe put this in a file like py2db.py, which converts from
the python dictionary to the database"""
team1 = game_info['t1']
spec = team1['special']
stats = team1['stats']
p1_vh = True if spec['vh'] == 'H' else False
plays_in_team_one = TeamIn(team=team1["id"],
game=g.id,
fgm=stats['fgm'],
fga=stats['fga'],
fgm3=stats['fgm3'],
fga3=stats['fga3'],
fta=stats['fta'],
ftm=stats['ftm'],
tp=stats['tp'],
blk=stats['blk'],
stl=stats['stl'],
ast=stats['ast'],
oreb=stats['oreb'],
dreb=stats['dreb'],
treb=stats['treb'],
pf=stats['pf'],
tf=stats['tf'],
to=stats['to'],
is_home=p1_vh,
pts_to=spec['pts_to'],
pts_paint=spec['pts_paint'],
pts_ch2=spec['pts_ch2'],
pts_fastb=spec['pts_fastb'],
pts_bench=spec['pts_bench'],
ties=spec['ties'],
leads=spec['leads'],
poss_count=spec['poss_count'],
poss_time=spec['poss_time'],
score_count=spec['score_count'],
score_time=spec['score_time'])
session.add(plays_in_team_one)
team2 = game_info['t2']
spec = team2['special']
stats = team2['stats']
p2_vh = True if spec['vh'] == 'H' else False
plays_in_team_two = TeamIn(team=team2["id"],
game=g.id,
fgm=stats['fgm'],
fga=stats['fga'],
fgm3=stats['fgm3'],
fga3=stats['fga3'],
fta=stats['fta'],
ftm=stats['ftm'],
tp=stats['tp'],
blk=stats['blk'],
stl=stats['stl'],
ast=stats['ast'],
oreb=stats['oreb'],
dreb=stats['dreb'],
treb=stats['treb'],
pf=stats['pf'],
tf=stats['tf'],
to=stats['to'],
is_home=p2_vh,
pts_to=spec['pts_to'],
pts_paint=spec['pts_paint'],
pts_ch2=spec['pts_ch2'],
pts_fastb=spec['pts_fastb'],
pts_bench=spec['pts_bench'],
ties=spec['ties'],
leads=spec['leads'],
poss_count=spec['poss_count'],
poss_time=spec['poss_time'],
score_count=spec['score_count'],
score_time=spec['score_time'])
session.add(plays_in_team_two)
session.commit()
# Put in information on total game scores
if team1['special']['vh'] == 'H':
# team1 is the home team
g.home_score = team1['stats']['score']
g.visitor_score = team2['stats']['score']
else:
g.home_score = team2['stats']['score']
g.visitor_score = team1['stats']['score']
if team1['stats']['score'] > team2['stats']['score']:
g.winner = team1['id']
g.loser = team2['id']
else:
g.winner = team2['id']
g.loser = team1['id']
session.add(g)
session.commit()
# Loop through Players and add them to the database if they don't already exist, repeat for team2
starters_team_1 = []
for player in team1['players']:
name_formatted = player["checkname"].title()
if player["checkname"] != "TEAM":
comma = name_formatted.index(",")
name_formatted = name_formatted[:comma +
1] + " " + name_formatted[comma +
1:]
p = session.query(Player).filter_by(name=name_formatted.title(),
team=team1["id"]).first()
if not p:
# If the player's not already in the database add him
p = Player(name=name_formatted, team=team1["id"])
session.add(p)
session.commit()
# Some players don't have stats for the game - we ignore those by checking arbitrarily for the fgm stat to exist
# Example: Keion Green from CENTPENN
if "fgm" in player:
game_stats = PlayerIn(player=p.id,
game=g.id,
fgm=player["fgm"],
fga=player["fga"],
fgm3=player["fgm3"],
fga3=player["fga3"],
ftm=player["ftm"],
fta=player["fta"],
tp=player["tp"],
blk=player["blk"],
stl=player["stl"],
ast=player["ast"],
oreb=player["oreb"],
dreb=player["dreb"],
treb=player["treb"],
pf=player["pf"],
tf=player["tf"],
to=player["to"],
dq=player["dq"],
number=player["uni"],
mins=player["min"])
session.add(game_stats)
if "gs" in player:
starters_team_1.append(p.id)
session.commit()
# Add stats for the player for the game
# Now do the same thing for team2
starters_team_2 = []
for player in team2['players']:
name_formatted = player["checkname"].title()
if player["checkname"] != "TEAM":
comma = name_formatted.index(",")
name_formatted = name_formatted[:comma +
1] + " " + name_formatted[comma +
1:]
p = session.query(Player).filter_by(name=name_formatted,
team=team2["id"]).first()
if not p:
# If the player's not already in the database add him
p = Player(name=name_formatted, team=team2["id"])
session.add(p)
session.commit()
# Some players don't have stats for the game - we ignore those by checking arbitrarily for the fgm stat to exist
# Example: Keion Green from CENTPENN
if "fgm" in player:
game_stats = PlayerIn(player=p.id,
game=g.id,
fgm=player["fgm"],
fga=player["fga"],
fgm3=player["fgm3"],
fga3=player["fga3"],
ftm=player["ftm"],
fta=player["fta"],
tp=player["tp"],
blk=player["blk"],
stl=player["stl"],
ast=player["ast"],
oreb=player["oreb"],
dreb=player["dreb"],
treb=player["treb"],
pf=player["pf"],
tf=player["tf"],
to=player["to"],
dq=player["dq"],
number=player["uni"],
mins=player["min"])
if "gs" in player:
starters_team_2.append(p.id)
session.add(game_stats)
session.commit()
# print("TEAM ONE STARTERS", starters_team_1)
# print("TEAM TWO STARTERS", starters_team_2)
if team1["id"] == home:
home_on_court = starters_team_1
away_on_court = starters_team_2
else:
home_on_court = starters_team_2
away_on_court = starters_team_1
# Now create a dummy play that initializes the starters
starters_play = Play(game_id=g.id,
period=1,
time="20:00",
scoring_play=False,
shooting_play=False,
home_score=0,
away_score=0,
text="Starters",
action="Starters",
type="",
h1=home_on_court[0],
h2=home_on_court[1],
h3=home_on_court[2],
h4=home_on_court[3],
h5=home_on_court[4],
v1=away_on_court[0],
v2=away_on_court[1],
v3=away_on_court[2],
v4=away_on_court[3],
v5=away_on_court[4])
session.add(starters_play)
session.commit()
plays = game_info["plays"]
last_v_score = 0
last_h_score = 0
# TODO: add a dummy play to the start of the second period
for period in plays:
if team1["id"] == home:
home_on_court = starters_team_1
away_on_court = starters_team_2
else:
home_on_court = starters_team_2
away_on_court = starters_team_1
for play in plays[period]:
# print(play)
name_formatted = play["checkname"].title()
if play["checkname"] != "TEAM":
comma = name_formatted.index(",")
name_formatted = name_formatted[:comma +
1] + " " + name_formatted[
comma + 1:]
player_id = session.query(Player).filter_by(
name=name_formatted, team=play["team"]).first().id
# Update home_on_court and away_on_court as necessary
if play["action"] == "SUB":
if play["type"] == "OUT":
if player_id in home_on_court:
home_on_court.remove(player_id)
elif player_id in away_on_court:
away_on_court.remove(player_id)
if play["type"] == "IN":
team = session.query(Player).filter_by(
id=player_id).first().team
# print(team)
if team == home:
home_on_court.append(player_id)
else:
away_on_court.append(player_id)
# TODO: make sure this loops in order of increasing period, dicts are unpredictable
if play["action"] == "GOOD":
# Update the last known score after someone scores
last_v_score = play["vscore"]
last_h_score = play["hscore"]
this_play = Play(
game_id=g.id,
period=period,
time=play["time"],
scoring_play=play["action"] == "GOOD",
shooting_play=(
play["type"] == "LAYUP" or play["type"] == "3PTR"
or play["type"] == "JUMPER") if "type" in play else False,
home_score=last_h_score,
away_score=last_v_score,
text="",
action=play["action"],
type=play["type"] if "type" in play else "",
player_id=player_id,
h1=home_on_court[0] if len(home_on_court) > 0 else -1,
h2=home_on_court[1] if len(home_on_court) > 1 else -1,
h3=home_on_court[2] if len(home_on_court) > 2 else -1,
h4=home_on_court[3] if len(home_on_court) > 3 else -1,
h5=home_on_court[4] if len(home_on_court) > 4 else -1,
v2=away_on_court[0] if len(away_on_court) > 0 else -1,
v1=away_on_court[1] if len(away_on_court) > 1 else -1,
v3=away_on_court[2] if len(away_on_court) > 2 else -1,
v4=away_on_court[3] if len(away_on_court) > 3 else -1,
v5=away_on_court[4] if len(away_on_court) > 4 else -1)
this_play.convert_time(int(this_play.period), this_play.time)
session.add(this_play)
session.commit()
def stats(tid):
current_season = sbteamdata.get_current_season()
seasons = Team.query.filter_by(teamid=tid).order_by(
Team.season.desc()).all()
team = Team.query.filter_by(teamid=tid, season=current_season).first()
if team and (team.time - datetime.now()).total_seconds() > -3600:
players = Player.query.filter_by(teamid=tid,
season=current_season).all()
pitchers = Pitcher.query.filter_by(teamid=tid,
season=current_season).all()
updated = team.time.strftime('Updated on %B %d at %I:%M %p')
else:
stats = sbteamdata.get_team_data(tid)
team = Team.query.filter_by(teamid=tid).order_by(
Team.season.desc()).all()
players = Player.query.filter_by(teamid=tid, season=stats.season).all()
pitchers = Pitcher.query.filter_by(teamid=tid,
season=stats.season).all()
for entry in team:
if entry.season == int(stats.season):
db.session.delete(entry)
db.session.commit()
new_entry = Team(datetime.now(), stats.season, tid, stats.name,
stats.stars, stats.stadium, stats.pl,
stats.pl_position, stats.trained, stats.games,
stats.ab, stats.hits, stats.avg, stats.singles,
stats.doubles, stats.triples, stats.hr, stats.rbi,
stats.so, stats.era, stats.slg, stats.wins,
stats.losses, stats.pct, stats.rs, stats.ra)
db.session.add(new_entry)
if players:
for player in players:
db.session.delete(player)
db.session.commit()
for player in stats.players:
new_entry = Player(stats.season, tid, player.id, player.name,
player.position, player.games, player.ab,
player.hits, player.avg, player.singles,
player.doubles, player.triples, player.hr,
player.rbi, player.so, player.era, player.slg)
db.session.add(new_entry)
if pitchers:
for player in pitchers:
db.session.delete(player)
db.session.commit()
for player in stats.pitchers:
new_entry = Pitcher(stats.season, tid, player.id, player.name,
player.games, player.so, player.era)
db.session.add(new_entry)
db.session.commit()
team = Team.query.filter_by(teamid=tid, season=current_season).first()
players = Player.query.filter_by(teamid=tid,
season=current_season).all()
pitchers = Pitcher.query.filter_by(teamid=tid,
season=current_season).all()
updated = team.time.strftime('Updated on %B %d at %I:%M %p')
return render_template('stats.html',
updated=updated,
seasons=seasons,
team=team,
players=players,
pitchers=pitchers)
def view_team_handler(user):
team = Team.select().where(Team.team_id == user.team_id).get()
team_members = User.select().where((User.team_id == user.team_id) & (User.user_id != user.user_id))
data = {'code_card':team,'team_member_cards':team_members}
return bot_postbacks.view_team_postback(data, user)