Guido van Rossum | f06ee5f | 1996-11-27 19:52:01 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 2 | |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 3 | # Emulate some Perl command line options. |
| 4 | # Usage: pp [-a] [-c] [-d] [-e scriptline] [-F fieldsep] [-n] [-p] [file] ... |
| 5 | # Where the options mean the following: |
| 6 | # -a : together with -n or -p, splits each line into list F |
| 7 | # -c : check syntax only, do not execute any code |
| 8 | # -d : run the script under the debugger, pdb |
| 9 | # -e scriptline : gives one line of the Python script; may be repeated |
| 10 | # -F fieldsep : sets the field separator for the -a option [not in Perl] |
| 11 | # -n : runs the script for each line of input |
| 12 | # -p : prints the line after the script has run |
| 13 | # When no script lines have been passed, the first file argument |
| 14 | # contains the script. With -n or -p, the remaining arguments are |
| 15 | # read as input to the script, line by line. If a file is '-' |
| 16 | # or missing, standard input is read. |
| 17 | |
| 18 | # XXX To do: |
| 19 | # - add -i extension option (change files in place) |
| 20 | # - make a single loop over the files and lines (changes effect of 'break')? |
| 21 | # - add an option to specify the record separator |
| 22 | # - except for -n/-p, run directly from the file if at all possible |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 23 | |
| 24 | import sys |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 25 | import getopt |
| 26 | |
| 27 | FS = '' |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 28 | SCRIPT = [] |
| 29 | AFLAG = 0 |
| 30 | CFLAG = 0 |
| 31 | DFLAG = 0 |
| 32 | NFLAG = 0 |
| 33 | PFLAG = 0 |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 34 | |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 35 | try: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 36 | optlist, ARGS = getopt.getopt(sys.argv[1:], 'acde:F:np') |
Guido van Rossum | b940e11 | 2007-01-10 16:19:56 +0000 | [diff] [blame] | 37 | except getopt.error as msg: |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 38 | sys.stderr.write('%s: %s\n' % (sys.argv[0], msg)) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 39 | sys.exit(2) |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 40 | |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 41 | for option, optarg in optlist: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 42 | if option == '-a': |
| 43 | AFLAG = 1 |
| 44 | elif option == '-c': |
| 45 | CFLAG = 1 |
| 46 | elif option == '-d': |
| 47 | DFLAG = 1 |
| 48 | elif option == '-e': |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 49 | for line in optarg.split('\n'): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 50 | SCRIPT.append(line) |
| 51 | elif option == '-F': |
| 52 | FS = optarg |
| 53 | elif option == '-n': |
| 54 | NFLAG = 1 |
| 55 | PFLAG = 0 |
| 56 | elif option == '-p': |
| 57 | NFLAG = 1 |
| 58 | PFLAG = 1 |
| 59 | else: |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame] | 60 | print(option, 'not recognized???') |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 61 | |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 62 | if not ARGS: ARGS.append('-') |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 63 | |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 64 | if not SCRIPT: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 65 | if ARGS[0] == '-': |
| 66 | fp = sys.stdin |
| 67 | else: |
| 68 | fp = open(ARGS[0], 'r') |
| 69 | while 1: |
| 70 | line = fp.readline() |
| 71 | if not line: break |
| 72 | SCRIPT.append(line[:-1]) |
| 73 | del fp |
| 74 | del ARGS[0] |
| 75 | if not ARGS: ARGS.append('-') |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 76 | |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 77 | if CFLAG: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 78 | prologue = ['if 0:'] |
| 79 | epilogue = [] |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 80 | elif NFLAG: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 81 | # Note that it is on purpose that AFLAG and PFLAG are |
| 82 | # tested dynamically each time through the loop |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 83 | prologue = [ |
| 84 | 'LINECOUNT = 0', |
| 85 | 'for FILE in ARGS:', |
| 86 | ' \tif FILE == \'-\':', |
| 87 | ' \t \tFP = sys.stdin', |
| 88 | ' \telse:', |
| 89 | ' \t \tFP = open(FILE, \'r\')', |
| 90 | ' \tLINENO = 0', |
| 91 | ' \twhile 1:', |
| 92 | ' \t \tLINE = FP.readline()', |
| 93 | ' \t \tif not LINE: break', |
| 94 | ' \t \tLINENO = LINENO + 1', |
| 95 | ' \t \tLINECOUNT = LINECOUNT + 1', |
| 96 | ' \t \tL = LINE[:-1]', |
| 97 | ' \t \taflag = AFLAG', |
| 98 | ' \t \tif aflag:', |
| 99 | ' \t \t \tif FS: F = L.split(FS)', |
| 100 | ' \t \t \telse: F = L.split()' |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 101 | ] |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 102 | epilogue = [ |
| 103 | ' \t \tif not PFLAG: continue', |
| 104 | ' \t \tif aflag:', |
| 105 | ' \t \t \tif FS: print(FS.join(F))', |
| 106 | ' \t \t \telse: print(\' \'.join(F))', |
| 107 | ' \t \telse: print(L)', |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 108 | ] |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 109 | else: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 110 | prologue = ['if 1:'] |
| 111 | epilogue = [] |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 112 | |
| 113 | # Note that we indent using tabs only, so that any indentation style |
| 114 | # used in 'command' will come out right after re-indentation. |
| 115 | |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 116 | program = '\n'.join(prologue) + '\n' |
Guido van Rossum | ff535a1 | 1992-08-10 10:42:36 +0000 | [diff] [blame] | 117 | for line in SCRIPT: |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 118 | program += ' \t \t' + line + '\n' |
| 119 | program += '\n'.join(epilogue) + '\n' |
Guido van Rossum | ca7b213 | 1992-07-07 09:11:53 +0000 | [diff] [blame] | 120 | |
Guido van Rossum | 3b0a329 | 2002-08-09 16:38:32 +0000 | [diff] [blame] | 121 | if DFLAG: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 122 | import pdb |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 123 | pdb.run(program) |
Guido van Rossum | 3b0a329 | 2002-08-09 16:38:32 +0000 | [diff] [blame] | 124 | else: |
Georg Brandl | 22fff43 | 2009-10-27 20:19:02 +0000 | [diff] [blame] | 125 | exec(program) |