forked from wangzitian0/BOJ-V4
-
Notifications
You must be signed in to change notification settings - Fork 2
/
add_test_data.py
66 lines (58 loc) · 1.86 KB
/
add_test_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import os, sys
import django
from django.core.files.base import ContentFile
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
os.environ['DJANGO_SETTINGS_MODULE'] = 'bojv4.settings'
django.setup()
from filer.models.filemodels import File
from submission.models import Submission as OldSubmission
from submission.abstract_models import NormalSubmission
from contest.models import Submission as ContestSubmission
from contest.models import ContestSubmission as OldContestSubmission
if __name__ == '__main__':
cs_ids = set()
cs = OldContestSubmission.objects.all()
for o in cs:
cs_ids.add(o.submission_id)
s = ContestSubmission()
oc = o.submission
s.user = oc.user
s.problem = o.problem
s.code_file = oc.code_file
s.status = oc.status
s.create_time = oc.create_time
s.score = oc.score
s.language = oc.language
s.info = oc.info
for c in oc.cases.all():
case = {}
case['position'] = c.position
case['time'] = c.running_time
case['memory'] = c.running_memory
case['status'] = c.status
s.add_case(case)
s.save()
'''
ss = OldSubmission.objects.all()
for o in ss:
if o.pk in cs_ids:
continue
s = NormalSubmission()
s.user = o.user
s.problem = o.problem
s.code_file = o.code_file
s.status = o.status
s.create_time = o.create_time
s.score = o.score
s.language = o.language
s.info = o.info
for c in o.cases.all():
case = {}
case['position'] = c.position
case['time'] = c.running_time
case['memory'] = c.running_memory
case['status'] = c.status
s.add_case(case)
s.save()
'''