init.py (582B)
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 def has_bot(user): 14 return len(Champion.objects.filter(author=user)) > 0 15 16 filepath=sys.argv[1] 17 with open(filepath, 'wb') as f: 18 users = User.objects.filter(is_staff=False) 19 ratings = dict([(user.username, Rating()) for user in users if has_bot(user)]) 20 pickle.dump(ratings, f)