def parse_file(filename, mode="text", player_name=None):
out = []
try:
with file(filename) as f:
parser = GamesParser(f, filename)
games = parser.parse_file()
for error in parser.errors:
print error
rater = Rater(games)
ratings = rater.rate_games(parser.score_type)
if mode == "gnuplot":
commands = 'set key left\n'
plot_command = 'plot "%s" with lines lw 2 title "%s"\n'
for name in rater.players:
player = rater.players[name]
if player.get_game_count() > 0:
player_file_name = "plot_rating_%s.data" % player.name
commands += plot_command % (player_file_name, player.name)
if plot_command.startswith('plot'):
plot_command = 're' + plot_command
player_file = open(player_file_name, "w")
for game_num in range(player.get_first_game() - 1, len(games) + 1):
rating = player.get_rating(game_num)
print >> player_file, "%d %d" % (game_num, rating)
commands += 'pause mouse\n'
command_file_name = "plot_commands.gnuplot"
command_file = open(command_file_name, "w")
print >> command_file, commands
out.append("Run 'gnuplot %s' to plot all ratings." % command_file_name)
elif mode == "json":
rating_dict = {}
for rating in ratings:
r = {"name":rating[0], "matches":rating[1], "rating":rating[2], "delta":rating[3]}
rating_dict[rating[0]] = r
out.append(json.dumps(rating_dict))
elif player_name is not None:
player = rater.players[player_name]
if player:
for game_num in sorted(player.rating_history.keys()):
rating = player.rating_history[game_num]
out.append("%d %d" % (game_num, rating))
else:
out.append("Game: %s" % parser.game_name)
out.append("Total number of games: %s" % len(games))
hformat = "%-15s%-10s%-10s%-10s"
rformat = "%-15s%7d%9d%9d"
out.append(hformat % ("Name", "Games", "Rating", "Delta"))
for rating in ratings:
out.append(rformat % rating)
except IOError as ioe:
out.append(str(ioe))
return "\n".join(out)
