diff options
Diffstat (limited to 'prequeue.py')
-rw-r--r-- | prequeue.py | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/prequeue.py b/prequeue.py index 00bcee36..d9b37e03 100644 --- a/prequeue.py +++ b/prequeue.py @@ -10,13 +10,14 @@ import sys import preprocessor -def main(): - processor = preprocessor.setup_processor() +def preprocess_queue(filenames=sys.argv[1:]): stdin = sys.stdin stdout = sys.stdout - for source in sys.argv[1:]: + processor = preprocessor.setup_processor() + + for source in filenames: dest = os.path.splitext(source)[0] + '.tx' sys.stdin = open(source, 'r') sys.stdout = open(dest, 'w') @@ -27,6 +28,13 @@ def main(): sys.stdin = stdin sys.stdout = stdout +def main(): + filenames = list(set(sys.argv[1:])) + if filenames: + num_files = len(filenames) + s = '' if num_files == 1 else 's' + sys.stdout.write('Preprocessing {0} file{1}...\n'.format(num_files, s)) + preprocess_queue(filenames) if __name__ == '__main__': main() |