QingdaoU/OnlineJudge
1
0
Fork 0
mirror of https://github.com/QingdaoU/OnlineJudge.git synced 2025-01-07 13:02:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

增加初始化用户排名的命令脚本

Browse Source
This commit is contained in:
virusdefender 2016-04-24 14:32:27 +08:00
parent 203ed3b35e
commit a5a313d2fe
No known key found for this signature in database
GPG Key ID: 1686FB5677979E61
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
utils/management/commands/inituserrank.py Normal file
View File

@ -0,0 +1,19 @@
# coding=utf-8
from django.core.management.base import BaseCommand
from account.models import UserProfile
from submission.models import Submission
class Command(BaseCommand):
def handle(self, *args, **options):
self.stdout.write(self.style.SUCCESS("Please wait a minute"))
for profile in UserProfile.objects.all():
submissions = Submission.objects.filter(user_id=profile.user.id)
profile.submission_number = submissions.count()
accepted_problem_number = len(set(Submission.objects.filter(user_id=profile.user.id, contest_id__isnull=True)\
.values_list("problem_id", flat=True)))
accepted_contest_problem_number = len(set(Submission.objects.filter(user_id=profile.user.id, contest_id__isnull=False)\
.values_list("problem_id", flat=True)))
profile.accepted_problem_number = accepted_problem_number + accepted_contest_problem_number
profile.save()
self.stdout.write(self.style.SUCCESS("Succeeded"))