def fill_ep(csv_filename, dbsession=session):
"""
fill the database with FPLs ep_next prediction, and also
write output to a csv.
"""
if not os.path.exists(csv_filename):
outfile = open(csv_filename, "w")
outfile.write("player_id,gameweek,EP\n")
else:
outfile = open(csv_filename, "a")
summary_data = fetcher.get_player_summary_data()
gameweek = NEXT_GAMEWEEK
for k, v in summary_data.items():
player = get_player_from_api_id(k)
player_id = player.player_id
outfile.write("{},{},{}\n".format(player_id, gameweek, v["ep_next"]))
pp = PlayerPrediction()
pp.player_id = player_id
pp.gameweek = gameweek
pp.predicted_points = v["ep_next"]
pp.method = "EP"
dbsession.add(pp)
dbsession.commit()
outfile.close()
def make_prediction(player, fixture, points, tag):
"""
fill one row in the player_prediction table
"""
pp = PlayerPrediction()
pp.predicted_points = points
pp.tag = tag
pp.player = player
pp.fixture = fixture
return pp