tioj-scripts/testdata_upload_sequence.py at master · CCUPCxDS/tioj-scripts · GitHub

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
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import requests
import random
import string
from bs4 import BeautifulSoup
from getpass import *

# config
judge_site = 'https://tioj.ck.tp.edu.tw/'
# the testdata will append at the end of the testdata list
input_suffix = '.in'
output_suffix = '.out'
filename_format = '%d%s' # % (prefix, id, suffix)
# end config

session = requests.Session()
def login():
    TIOJusername = input('Username: ')
    TIOJpassword = getpass('Password: ')
    print('logging in...')
    global session
    rel = session.get(judge_site + '/users/sign_in')
    soup = BeautifulSoup(rel.text, "html.parser")
    inputs = soup.find('form').find_all('input')
    rel = session.post(judge_site + '/users/sign_in', data = {
        'authenticity_token': inputs[0].attrs['value'],
        'user[username]': TIOJusername,
        'user[password]': TIOJpassword,
        'user[remember_me]': '1',
        'commit': 'Sign in'
    })

login()
print('Successful log in')

problem_id = input('Problem ID: ')
num_start = int(input('Testdata start number: '))
num_end = int(input('Testdata end number: '))
time_limit = input('Time limit: ')
memory_limit = input('Memory limit: ')

sign_up_get_url = judge_site + '/problems/%s/testdata/new' % problem_id
sign_up_post_url = judge_site + '/problems/%s/testdata' % problem_id
for i in range(num_start, num_end + 1):
    print('processing %d...' % i)
    rel = session.get(sign_up_get_url)
    soup = BeautifulSoup(rel.text, "html.parser")
    inputs = soup.find('form').find_all('input')
    rel = session.post(sign_up_post_url, data = {
        inputs[0].attrs['name']: inputs[0].attrs['value'],
        inputs[1].attrs['name']: inputs[1].attrs['value'],
        'testdatum[limit_attributes][time]': time_limit,
        'testdatum[limit_attributes][memory]': memory_limit,
        'testdatum[limit_attributes][output]': '65536',
        'testdatum[problem_id]': problem_id,
        'commit': 'Create Testdatum'
    }, files = {
        'testdatum[test_input]': open(filename_format % (i, input_suffix), 'rb'),
        'testdatum[test_output]': open(filename_format % (i, output_suffix), 'rb')
    })
    print('Create %d.' % i)