Mercurial > ~astiob > upreckon > hgweb
annotate upreckon/upreckon-vcs @ 210:8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
author | Oleg Oshmyan <chortos@inbox.lv> |
---|---|
date | Thu, 18 Aug 2011 02:51:31 +0300 |
parents | d46bb4495766 |
children | 78be952f6b40 |
rev | line source |
---|---|
21 | 1 #! /usr/bin/env python |
78 | 2 # Copyright (c) 2009-2011 Chortos-2 <chortos@inbox.lv> |
16 | 3 |
4 from __future__ import division, with_statement | |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
5 import optparse |
21 | 6 |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
7 from upreckon import compat |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
8 from upreckon.compat import * |
16 | 9 |
184 | 10 version = '2.03.0 ($$REV$$)' |
45 | 11 parser = optparse.OptionParser(version='Upreckon '+version, epilog='Python 2.5 or newer is required.') |
23 | 12 parser.add_option('-1', dest='legacy', action='store_true', default=False, help='handle configuration files in a way more compatible with test.py 1.x') |
25
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
13 parser.add_option('-p', '--problem', dest='problems', metavar='PROBLEM', action='append', help='test only the PROBLEM (this option can be specified more than once with different problem names, all of which will be tested)') |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
14 parser.add_option('--list-problems', action='store_true', default=False, help='just list all problem names') |
16 | 15 parser.add_option('-m', '--copy-io', dest='copyonly', action='store_true', default=False, help='create a copy of the input/output files of the last test case for manual testing and exit') |
16 parser.add_option('-x', '--auto-exit', dest='pause', action='store_false', default=True, help='do not wait for a key to be pressed after finishing testing') | |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
17 #parser.add_option('-s', '--save-io', dest='erase', action='store_false', default=True, help='do not delete the copies of input/output files after the last test case; create copies of input files and store output in files even if the solution uses standard I/O; delete the stored input/output files if the solution uses standard I/O and the -c/--cleanup option is specified') |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
18 parser.add_option('-s', '--save-io', dest='erase', action='store_false', default=True, help='do not delete the copies of input/output files after the last test case; create copies of input files and store output in files even if the solution uses standard I/O') |
90
1fb319ec33af
Skimming mode added (-k/--skim option)
Oleg Oshmyan <chortos@inbox.lv>
parents:
87
diff
changeset
|
19 parser.add_option('-k', '--skim', action='store_true', default=False, help='skip test groups as soon as one test case is failed') |
21 | 20 parser.add_option('--no-time-limits', dest='no_maxtime', action='store_true', default=False, help='disable all time limits') |
16 | 21 |
22 options, args = parser.parse_args() | |
23 parser.destroy() | |
24 del parser | |
25 | |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
26 from upreckon import config |
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
27 import itertools, os, subprocess, sys, time |
16 | 28 |
70
b9d5857f7b9a
Better emulation of built-ins for testconf
Oleg Oshmyan <chortos@inbox.lv>
parents:
66
diff
changeset
|
29 if options.legacy: |
b9d5857f7b9a
Better emulation of built-ins for testconf
Oleg Oshmyan <chortos@inbox.lv>
parents:
66
diff
changeset
|
30 compat.pseudobuiltins += 'xrange', |
b9d5857f7b9a
Better emulation of built-ins for testconf
Oleg Oshmyan <chortos@inbox.lv>
parents:
66
diff
changeset
|
31 |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
32 if options.list_problems: |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
33 options.pause = False |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
34 |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
35 from upreckon import testcases |
66
34ba0b353fc6
Fixed an issue when import errors would be ignored
Oleg Oshmyan <chortos@inbox.lv>
parents:
64
diff
changeset
|
36 |
34ba0b353fc6
Fixed an issue when import errors would be ignored
Oleg Oshmyan <chortos@inbox.lv>
parents:
64
diff
changeset
|
37 try: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
38 from upreckon.testcases import pause |
22 | 39 except ImportError: |
40 pause = None | |
41 | |
42 try: | |
21 | 43 globalconf = config.load_global() |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
44 |
21 | 45 # Do this check here so that if we have to warn them, we do it as early as possible |
22 | 46 if options.pause and not pause and not hasattr(globalconf, 'pause'): |
25
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
47 if os.name == 'posix': |
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
48 globalconf.pause = 'read -s -n 1' |
45 | 49 say('Warning: configuration variable pause is not defined; it was devised automatically but the choice might be incorrect, so Upreckon might exit immediately after the testing is completed.', file=sys.stderr) |
25
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
50 sys.stderr.flush() |
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
51 elif os.name == 'nt': |
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
52 globalconf.pause = 'pause' |
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
53 else: |
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
54 sys.exit('Error: configuration variable pause is not defined and cannot be devised automatically.') |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
55 |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
56 from upreckon.problem import * |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
57 |
21 | 58 # Support single-problem configurations |
79
ee8a99dcaaed
Renamed configuration variable tasknames to problems
Oleg Oshmyan <chortos@inbox.lv>
parents:
78
diff
changeset
|
59 if globalconf.problems is None: |
21 | 60 shouldprintnames = False |
61 globalconf.multiproblem = False | |
79
ee8a99dcaaed
Renamed configuration variable tasknames to problems
Oleg Oshmyan <chortos@inbox.lv>
parents:
78
diff
changeset
|
62 globalconf.problems = os.path.curdir, |
16 | 63 else: |
21 | 64 globalconf.multiproblem = True |
25
b500e117080e
Bug fixes and overhead reduction
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
65 shouldprintnames = True |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
66 |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
67 if options.list_problems: |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
68 for taskname in globalconf.problems: |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
69 say(taskname) |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
70 sys.exit() |
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
71 |
21 | 72 ntasks = 0 |
73 nfulltasks = 0 | |
74 maxscore = 0 | |
75 realscore = 0 | |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
76 |
202 | 77 for taskname in options.problems or globalconf.problems: |
21 | 78 problem = Problem(taskname) |
79 | |
22 | 80 if ntasks and not options.copyonly: say() |
21 | 81 if shouldprintnames: say(taskname) |
82 | |
83 if options.copyonly: | |
84 problem.copytestdata() | |
85 else: | |
86 real, max = problem.test() | |
87 | |
88 ntasks += 1 | |
22 | 89 nfulltasks += real == max |
21 | 90 realscore += real |
91 maxscore += max | |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
92 |
21 | 93 if options.copyonly: |
94 sys.exit() | |
102
796eb7667fb0
Added the --list-problems command-line option
Oleg Oshmyan <chortos@inbox.lv>
parents:
91
diff
changeset
|
95 |
21 | 96 if ntasks != 1: |
97 say() | |
64 | 98 say('Grand total: %g/%g weighted points; %d/%d problems solved fully' % (realscore, maxscore, nfulltasks, ntasks)) |
21 | 99 except KeyboardInterrupt: |
138
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
100 say('Exiting due to a keyboard interrupt.', end='', file=sys.stderr) |
141 | 101 sys.stderr.flush() |
138
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
102 try: |
146
d5b6708c1955
Distutils support, reorganization and cleaning up
Oleg Oshmyan <chortos@inbox.lv>
parents:
141
diff
changeset
|
103 import signal |
138
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
104 signal.signal(signal.SIGINT, signal.SIG_DFL) |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
105 os.kill(os.getpid(), signal.SIGINT) |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
106 except Exception: |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
107 pass |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
108 # Do this even if we got no exceptions, just in case |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
109 say(file=sys.stderr) |
517dd43f06f8
Upreckon now properly reports exiting due to SIGINT to the calling process
Oleg Oshmyan <chortos@inbox.lv>
parents:
102
diff
changeset
|
110 sys.exit(1) |
16 | 111 |
210
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
112 try: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
113 if options.pause: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
114 say('Press any key to exit...') |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
115 sys.stdout.flush() |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
116 |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
117 if pause: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
118 pause() |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
119 elif callable(globalconf.pause): |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
120 globalconf.pause() |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
121 else: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
122 with open(os.devnull, 'w') as devnull: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
123 subprocess.call(globalconf.pause, shell=True, stdout=devnull, stderr=subprocess.STDOUT) |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
124 except KeyboardInterrupt: |
8c4e92fb32d8
Keyboard interrupts now satisfy the Press any key to exit... prompt
Oleg Oshmyan <chortos@inbox.lv>
parents:
202
diff
changeset
|
125 pass |