testdata_upload_t.py

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

# config

# the testdata will append at the end of the testdata list
input_suffix = '.in'
output_suffix = '.out'
input_prefix = ''
output_prefix = ''
filename_format = '%s%02d%s' # % (prefix, id, suffix)
time_limit = '2500'
memory_limit = '262144'

# end config

session = requests.Session()

def login():
    TIOJusername = input('Username: ')
    TIOJpassword = getpass('Password: ')
    print('logging in...')
    global session
    rel = session.get('https://tioj.ck.tp.edu.tw/users/sign_in')
    soup = BeautifulSoup(rel.text, "html.parser")
    inputs = soup.find('form').find_all('input')
    rel = session.post('https://tioj.ck.tp.edu.tw/users/sign_in', data = {
        inputs[0].attrs['name']: inputs[0].attrs['value'],
        inputs[1].attrs['name']: inputs[1].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: '))

sign_up_get_url = 'https://tioj.ck.tp.edu.tw/problems/%s/testdata/new' % problem_id
sign_up_post_url = 'https://tioj.ck.tp.edu.tw/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 % (input_prefix, i, input_suffix), 'rb'),
        'testdatum[test_output]': open(filename_format % (output_prefix, i, output_suffix), 'rb')
    })
    print('Create %d.' % i)