2018-03-27 05:53:30 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
|
|
|
|
|
|
def fix_rejudge_bugs(apps, schema_editor):
|
|
|
|
Submission = apps.get_model("submission", "Submission")
|
|
|
|
User = apps.get_model("account", "User")
|
|
|
|
|
|
|
|
for user in User.objects.all():
|
2018-03-29 13:54:38 +00:00
|
|
|
submissions = Submission.objects.filter(user_id=user.id, contest__isnull=True)
|
2018-03-27 05:53:30 +00:00
|
|
|
profile = user.userprofile
|
|
|
|
profile.submission_number = submissions.count()
|
|
|
|
profile.accepted_number = submissions.filter(result=0).count()
|
|
|
|
profile.save(update_fields=["submission_number", "accepted_number"])
|
|
|
|
|
|
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
|
|
('submission', '0009_delete_user_output'),
|
2018-03-30 12:07:37 +00:00
|
|
|
('problem', '0010_problem_spj_compile_ok'),
|
2018-03-27 05:53:30 +00:00
|
|
|
]
|
|
|
|
|
|
|
|
operations = [
|
|
|
|
migrations.RunPython(fix_rejudge_bugs, reverse_code=migrations.RunPython.noop)
|
|
|
|
]
|