top.py (733B)
1 #!/usr/bin/env python3 2 3 import os 4 import sys 5 import pickle 6 from trueskill import Rating 7 8 os.environ['DJANGO_SETTINGS_MODULE'] = 'prologin.concours.settings' 9 sys.path.insert(0, '/root/sadm/python-lib') 10 from django.contrib.auth.models import User 11 from prologin.concours.stechec.models import * 12 13 filepath=sys.argv[1] 14 with open(filepath, 'rb') as f: 15 ratings = pickle.load(f) 16 filepath=sys.argv[2] 17 18 limit=int(sys.argv[3]) 19 leaderboard = list(ratings.keys()) 20 #leaderboard.sort(key=conservative_estimate, reverse=True) 21 leaderboard.sort(key=lambda x: ratings[x].mu, reverse=True) 22 23 selected = leaderboard[0:limit] 24 new_ratings = dict( (key,ratings[key]) for key in selected ) 25 26 with open(filepath, 'wb') as f: 27 pickle.dump(new_ratings, f)