summaryrefslogtreecommitdiff
path: root/kpov_judge/test_task.py
blob: d3dcec75612cd79d4073c37087d5d49ff7ecafb6 (plain)
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
#!/usr/bin/env python3

import argparse
import collections
import getpass
import inspect
import io
import json
import locale
import os
import random
import readline
import sys
import urllib.request

import yaml
import kpov_util

locale.setlocale(locale.LC_ALL, ['C', 'utf8'])
readline.set_completer_delims(readline.get_completer_delims().replace('/', ''))
readline.parse_and_bind('tab: complete')

TASK_URL = "file://" + os.getcwd() + '/tasks'
PARAMS_FILE = os.path.expanduser("~/.kpov_params.yaml")
DEFAULT_LANGUAGE = 'si'

# get the parameters for a task either from the user or from a file
def get_params(params, meta, language=None):
    # prefill input() prompt with given text
    def rlinput(prompt, prefill=''):
        readline.set_startup_hook(lambda: readline.insert_text(prefill))
        try:
            return input(prompt)
        finally:
            readline.set_startup_hook()

    param_name_list = list(meta.keys())
    if language is None:
        language = params.get('language', DEFAULT_LANGUAGE)

    for name in param_name_list:
        m = meta.get(name, {})
        description = m.get('descriptions', {}).get(language, name)
        try:
            if m.get('masked', False):
                s = getpass.getpass('{}: '.format(description))
            else:
                s = rlinput('{}: '.format(description), params.get(name, ''))
            if s and m.get('w', True):
                params[name] = s
            elif name not in params:
                params[name] = None
        except EOFError:
            print()
    return params

def add_meta_to_argparser(argparser, meta, defaults={}):
    language = defaults.get('language', DEFAULT_LANGUAGE)
    for k, v in meta.items():
        try:
            desc = v['descriptions'][language].encode("utf-8")
        except:
            desc = k
        argparser.add_argument('--'+k, nargs='?', type=str, help=desc,
            default=defaults.get(k, None))

def http_auth(url, username, password):
    password_mgr = urllib.request.HTTPPasswordMgrWithDefaultRealm()
    password_mgr.add_password(None, url, username, password)
    handler = urllib.request.HTTPBasicAuthHandler(password_mgr)
    opener = urllib.request.build_opener(handler)
    urllib.request.install_opener(opener) # now all calls to urlopen use our opener

def load_task(stream):
    # the stream should definitions for the functions task(…),
    # task_check and gen_params, and a dictionary params_meta
    source = stream.read()
    d = {}
    exec(compile(source, 'task.py', 'exec'), globals(), d)
    return d['task'], d['task_check'], d['params_meta'], d['gen_params']

def locate_task(params, argparser, quiet=False):
    # first the URL where all tasks are stored
    url_meta = {
        'task_url': {'descriptions': {'si': 'URL z nalogami', 'en': 'Root URL for all tasks'}}
    }
    if 'task_url' not in params:
        params['task_url'] = TASK_URL
    add_meta_to_argparser(argparser, meta=url_meta, defaults=params)
    args, unknown_args = argparser.parse_known_args()
    params['task_url'] = args.task_url
    if not quiet:
        params = get_params(params, url_meta)
    # then the student's ID (and password if neccessarry)
    fetch_params_meta = collections.OrderedDict({
            'username': {'descriptions': {'si': 'Uporabniško ime', 'en': 'Username'}},
    })
    if params.get('task_url', '').startswith('http'):
        fetch_params_meta['password'] = {'descriptions': {'si': 'Geslo', 'en': 'Password'}, 'masked': True}
    # and finally, the name of the task
    fetch_params_meta['task_name'] = {'descriptions': {'si': 'Ime naloge', 'en': 'Task name'}, }
    add_meta_to_argparser(argparser, meta=fetch_params_meta, defaults=params)
    args, unknown_args = argparser.parse_known_args()
    # update params with the now known args
    for k, v in fetch_params_meta.items():
        params[k] = vars(args).get(k, params.get(k, None))
    if not quiet:
        params = get_params(params, fetch_params_meta)
    return params

def load_params(filename):
    try:
        return yaml.load(open(filename))
    except:
        return {}
 
if __name__ == '__main__':
    # get the parameters needed to get to the task, such as the URLs, the name of the task and optionally an ID from the student
    # start with the the parameters needed for the dialog gui to work
    argparser = argparse.ArgumentParser(conflict_handler='resolve', add_help=False,
        description='Get task parameters')
    argparser.add_argument('-h', '--help', action='store_true')
    argparser.add_argument('-q', '--quiet', action='store_true',
        help='disable prompts')
    argparser.add_argument('-g', '--generate_params', action='store_true',
        help='generate initial values for the task parameters')
    argparser.add_argument('-pf','--params_file', nargs='?', default=PARAMS_FILE,
        help='a local file containing saved param values')
    basic_args, unknown_args = argparser.parse_known_args()

    # get default parameters including language
    params = load_params(basic_args.params_file)
    argparser.add_argument('-l', '--language', nargs='?',
        default=params.get('language', DEFAULT_LANGUAGE),
        help='the language used')
    basic_args, unknown_args = argparser.parse_known_args()
    params['language'] = basic_args.language

    # continue with the parameters needed to get the task
    params = locate_task(params, argparser, quiet=basic_args.quiet)
    # TODO: if the task name is missing or invalid, try to get a list of tasks 
    # get task source and generate params if neccessarry
    try:
        task_url = params['task_url']
        task_name = params['task_name']
        if task_url.startswith('http'):
            http_auth(task_url, params['username'], params['password'])
        print("Fetching {task_url}/{task_name}/task.py…".format(**params))
        source = urllib.request.urlopen("{task_url}/{task_name}/task.py".format(**params))
        task, task_check, task_params_meta, gen_params = load_task(source)
    except Exception as e:
        print(str(e))
        print()
        for k, v in params.items():
            if k not in ('password', 'task_params'):
                print('{}: {}'.format(k, v))
        with open(basic_args.params_file, 'w') as f:
            yaml.dump(params, f)
        exit(1)

    # get task parameters
    params['task_params'] = params.get('task_params', {})
    params['task_params'][params['task_name']] = params['task_params'].get(params['task_name'], {})
    task_params = params['task_params'][params['task_name']]
    if basic_args.generate_params:
	#prejema lahko samo stringe in ne številk (potrebno je str(int)
        # print ("params before: {} {}".format(params, task_params))
        task_params.update(gen_params(params['username'], task_params_meta))
        # print ("params after: {} {}".format(params, task_params))
    if task_url.startswith('http'):
        response = urllib.request.urlopen('{task_url}/{task_name}/params.json'.format(**params))
        web_task_params = json.load(io.TextIOWrapper(response))
        task_params.update(web_task_params)

    task_argparser = argparse.ArgumentParser(parents=[argparser], conflict_handler='resolve', add_help=True)
    add_meta_to_argparser(task_argparser, task_params_meta, defaults=task_params)
    args = vars(task_argparser.parse_args())
    for k in task_params_meta:
        if args.get(k):
            task_params[k] = args[k]
    if not basic_args.quiet:
        task_params = get_params(task_params, task_params_meta, language=params['language'])
    if basic_args.help:
        print(task_argparser.format_help())
        exit(0)

    public_params = {}
    for k in inspect.getargs(task.__code__)[0]:
        public_params[k] = task_params[k]
    params['task_params'][params['task_name']] = task_params
    # save parameters for the next run
    with open(basic_args.params_file, 'w') as f:
        yaml.dump(params, f)

    try:
        print()
        print('Running task… ')
        task_result = task(**public_params)
        print('Checking task… ')
        score, hints = task_check(task_result, task_params)
        print('Done!')
        print()
        print('Score: {}'.format(score))
        print('Hints:')
        for hint in hints:
            print(hint)
    except Exception as e:
        import traceback
        traceback.print_exc()