process_ratings.py

import MySQLdb
from server_info import server_info
import sys

if len(sys.argv) == 2:
  elapsed_time = sys.argv[1]
else:
  elapsed_time = -1
connection = MySQLdb.connect(host = server_info["db_host"],
                             user = server_info["db_username"],
                             passwd = server_info["db_password"],
                             db = server_info["db_name"])
cursor = connection.cursor(MySQLdb.cursors.DictCursor)
cursor.execute("""
  INSERT INTO leaderboards
  (timestamp,algorithm_name,calculation_time)
  VALUES
  (CURRENT_TIMESTAMP,'Bayeselo',""" + str(elapsed_time) + """)
""")
leaderboard_id = connection.insert_id()
f = open("ratings.txt", "r")
f.readline()
for line in f:
  csv = ','.join(line.strip().split())
  tokens = csv.split(",")
  if len(tokens) != 11:
    continue
  (rank, user_id, username, submission_id, elo, plus_bound, minus_bound, \
    num_games, score, oppo, draws) = tokens
  elo = int(elo) + 2000
  values = str(leaderboard_id) + "," + \
    str(submission_id) + "," + \
    str(rank) + "," + \
    "0,0,0," + str(elo)
  cursor.execute("""
    INSERT INTO rankings
    (leaderboard_id,submission_id,rank,wins,losses,draws,score)
    VALUES
    (""" + values + """)
  """)
cursor.close()
connection.close()