def main_run_cio(args):
if getattr(args, 'check', False):
args.check = False # to avoid recursion
return main(args)
mission = args.mission[0]
filename = get_filename(args)
if filename is None:
init_single_mission(mission)
domain_data = conf.default_domain_data
if 'executable' in domain_data:
return subprocess.call((domain_data['executable'], filename))
with open(filename, encoding="utf-8") as fh:
data = run_solution(code_for_check(fh.read()))
ret = False
while data:
block = data.pop(0)
com = block[0]
if com == 'err':
print(block[1], end='')
ret = True
elif com == 'out':
print(str(block[1]), end='')
elif com == 'wait':
data = restore(block[1])
else:
print(block)
print()
return ret
def main_run(args):
filename = get_filename(args)
mission = args.mission[0]
domain_data = conf.default_domain_data
print()
print('Start running...')
print()
if 'executable' in domain_data:
return os.system(' '.join((domain_data['executable'], filename)))
with open(filename, encoding="utf-8") as fh:
data = run_solution(code_for_check(fh.read()))
ret = False
while data:
block = data.pop(0)
com = block[0]
if com == 'err':
print(block[1], end='')
ret = True
elif com == 'out':
print(str(block[1]), end='')
elif com == 'wait':
data = restore(block[1])
else:
print(block)
print()
return ret
def main_check_cio(args):
filename = get_filename(args)
mission = args.mission[0]
domain_data = conf.default_domain_data
if main_run(args):
return
print()
print('Start checking...')
print()
mission_info = get_mission_info(mission)
mission_id = mission_info['id']
with open(filename, encoding="utf-8") as fh:
data = check_solution(code_for_check(fh.read()), mission_id)
while data:
block = data.pop(0)
com = block[0]
if com == 'start_in':
print('*** ' + block[1] + ' ***')
elif com == 'in':
print('->' + str(block[1]))
elif com == 'out':
print('<-' + str(block[1]))
elif com == 'ext':
res = block[1]
if not res['result']:
print('!!' + str(res['answer']))
elif com == 'check':
if block[1]:
print()
print('!!! Congratulation !!!')
print()
print(
'Link for checking solution of other users: {}/mission/{}/publications/'
.format(domain_data['url_main'], mission))
print()
print(
'Link for sharing solution: {}/mission/{}/publications/add/'
.format(domain_data['url_main'], mission))
print()
else:
print('!! Failed !!')
elif com == 'wait':
print('Waiting for the next piece of data...')
time.sleep(block[2])
print('Restore checking...')
data = restore(block[1])
else:
print(block)
async def eoc_websocket_run(args):
import websockets
import json
mission = args.mission[0]
filename = get_filename(args)
if filename is None:
filename = sync_single_mission(mission)
if filename is None:
print('Mission "{}" not found'.format(mission))
return
domain_data = conf.default_domain_data
if 'key' not in domain_data:
print('client is not configured. Please do $ checkio config')
return
logging.info('USING:' + filename)
with open(filename, encoding="utf-8") as fh:
code = code_for_check(fh.read())
async with websockets.connect(domain_data['ws_url'],
extra_headers=websockets.http.Headers({
'Cookie':
'apiKey=' + domain_data['key']
})) as websocket:
greeting = await websocket.recv()
# print('greeting', greeting)
await websocket.send(
json.dumps({
'action': 'run',
'data': {
'code': code,
'env_name': domain_data['interpreter'],
'mission': mission,
'process': randint(10**5, 10**6)
}
}))
while True:
resp = await websocket.recv()
resp = json.loads(resp)
if resp['action'] == 'run':
stream = resp['data'][
'type'] == 'stdout' and sys.stdout or sys.stderr
print(resp['data']['result'], end='', file=stream, flush=True)
elif resp['action'] == 'runDone':
print('SUCCESS', resp['data']['success'])
break
async def eoc_websocket_check(args):
import websockets
mission = args.mission[0]
filename = get_filename(args)
if filename is None:
filename = sync_single_mission(mission)
if filename is None:
print('Mission "{}" not found'.format(mission))
return
domain_data = conf.default_domain_data
if 'key' not in domain_data:
print('client is not configured. Please do $ checkio config')
return
logging.info('USING: %s', filename)
with open(filename, encoding="utf-8") as fh:
code = code_for_check(fh.read())
async with websockets.connect(domain_data['ws_url'],
extra_headers=websockets.http.Headers({
'Cookie':
'apiKey=' + domain_data['key']
})) as websocket:
greeting = await websocket.recv()
#print('greeting', greeting)
await websocket.send(
json.dumps({
'action': 'check',
'data': {
'code': code,
'env_name': domain_data['interpreter'],
'mission': mission,
'process': mission
}
}))
while True:
resp = await websocket.recv()
resp = json.loads(resp)
if resp['action'] == 'check':
# stream = resp['data']['type'] == 'stdout' and sys.stdout or sys.stderr
# print(resp['data']['result'], end='', file=stream, flush=True)
c_data = resp['data']
if c_data['type'] == 'pre_test':
print('\n - CALL', c_data['representation'])
elif c_data['type'] == 'post_test':
if c_data['test_passed']:
print('\n - EXPECTED', c_data['expected_result'])
else:
print('\n - ACTUAL', c_data['actual_result'])
print('\n - EXPECTED', c_data['expected_result'])
elif c_data['type'] == 'stdout':
print(c_data['result'],
file=sys.stdout,
flush=True,
end='')
elif c_data['type'] == 'stderr':
print(c_data['result'],
file=sys.stderr,
flush=True,
end='')
else:
print('UNKNOW', c_data)
elif resp['action'] == 'checkDone':
print('SUCCESS', resp['data']['success'])
break
def main_check_cio(args):
filename = get_filename(args)
mission = args.mission[0].replace('_', '-')
domain_data = conf.default_domain_data
if main_run(args):
return
print()
print('Start checking...')
print()
mission_info = get_mission_info(mission)
mission_id = mission_info['id']
with open(filename, encoding="utf-8") as fh:
data = check_solution(code_for_check(fh.read()), mission_id)
while data:
block = data.pop(0)
com = block[0]
if com == 'start_in':
print('*** ' + block[1] + ' ***' )
elif com == 'in':
print('->' + str(block[1]))
elif com == 'out':
print('<-' + str(block[1]))
elif com == 'ext':
res = block[1]
if not res['result']:
print('!!' + str(res['answer']))
elif com == 'check':
if block[1]:
system_data = {
'info': 'passed',
'solutions_link': '{}/mission/{}/publications/'.format(
domain_data['url_main'], mission
),
'add_link': '{}/mission/{}/publications/add/'.format(
domain_data['url_main'], mission
)
}
print()
print('!!! Congratulation !!!')
print()
print('Link for checking solution of other users: ' + system_data['solutions_link'])
print()
print('Link for sharing solution: ' + system_data['add_link'])
print()
if args.sysinfo:
print(SYSTEM_BLOCK_START)
print(json.dumps(system_data, indent=1))
print(SYSTEM_BLOCK_END)
else:
print('!! Failed !!')
elif com == 'wait':
print('Waiting for the next piece of data...')
time.sleep(block[2])
print('Restore checking...')
data = restore(block[1])
else:
print(block)