/pyfil

Python one-liners in the spirit of Perl and AWK

Primary LanguagePythonBSD 2-Clause "Simplified" LicenseBSD-2-Clause

pyfil

Python one-liners in the spirit of Perl and AWK.

pyfil gives you the rep command. This is because when I initially posted it in #python IRC channel, user [Tritium] (that ray of sunshine) said I had recreated the REP of the python REPL (read evaluate print loop). That is more or less the case. rep reads python expressions at the command line, evaluates them and prints them to stdout. It might be interesting as a quick calculator or to test something, like the Python REPL, but it also has some special flags for iterating on stdin, which make it useful as a filter for shell one-liners or scripts (like Perl).

As a more modern touch, if the return value is a container type, python will attempt to serialize it as json before printing, so you can pipe output into other tools that deal with json, store it to a file for later use, or send it over http. This, combined with the abilitiy to read json from stdin (with --json) make a good translator between the web, which tends to speak json these days, and the posix environment, which tends to think about data in terms of lines in a file (frequently with multiple fields per line).

pyfil is in pypi (i.e. you can get it easily with pip, if you want)

note:
pyfil has only been tested with python3, and only has wheels available for python3

pyfil ain't the first project to try something like this. Here are some other cracks at this problem:

Don't worry. I've stolen some of their best ideas already, and I will go on stealing as long as it takes!

rep [-h] [-l] [-x] [-q] [-j] [-o] [-b PRE] [-e POST] [-s] [-F PATTERN]
    [-n STRING] [-R] [-S] [-H EXCEPTION_HANDLER]
    expression [expression ...]
positional arguments:
expression expression(s) to be executed. If multiple expression arguments are given, and --exec is not used, the value of the previous expression is available as 'x' in the following expression. if --exec is used, all assignment must be explicit.
optional arguments:
-h, --help show this help message and exit
-l, --loop for n, i in enumerate(stdin): expressions
-x, --exec use exec instead of eval. statements are allowed, but automatic printing is lost. doesn't affect --post
-q, --quiet suppress automatic printing. doesn't affect --post
-j, --json load stdin as json into object 'j'; If used with --loop, treat each line of stdin as a new object
-o, --force-oneline-json
 outside of loops and iterators, objects serialzed to json print with two-space indent. this forces this forces all json objects to print on a single line.
-b PRE, --pre PRE
 statement to evaluate before expression args. multiple statements may be combined with ';'. no automatic printing
-e POST, --post POST
 expression to evaluate after the loop. always handeled by eval, even if --exec, and always prints return value, even if --quiet. implies --loop
-s, --split split lines from stdin on whitespace into list 'f'. implies --loop
-F PATTERN, --field-sep PATTERN
 regex used to split lines from stdin into list 'f'. implies --loop
-n STRING, --join STRING
 join items in iterables with STRING
-R, --raise-errors
 raise errors in evaluation and stop execution (default: print message to stderr and continue)
-S, --silence-errors
 suppress error messages
-H EXCEPTION_HANDLER, --exception-handler EXCEPTION_HANDLER
 specify exception handler with the format 'Exception: alternative expression to eval'

rep automatically imports any modules used in expressions.

If you'd like to create any other objects to use in the execution environment ~/.config/pyfil-env.py and put things in it.

default objects:

  • l = []
  • d = {}

These are empty containers you might wish to add items to during iteration, for example.

  • x is always the return value of the previous expression unless --exec.

The execution environment also has a special object for stdin, creatively named stdin. This differs from sys.stdin in that it rstrips (aka chomps) all the lines when you iterate over it, and it has a property, stdin.l, which returns a list of the (rstripped) lines. pyfil is quite bullish about using rstrip because python's print function will supply an additional newline, and if you just want the value of the text in the line, you almost never want the newline character. If you do want the newlines, access sys.stdin directly.

stdin inherits the rest of its methods from sys.stdin, so you can use stdin.read() to get a string of all lines, if that's what you need.

Certain other flags; --loop (or anything that implies --loop), --json, --split or --field_sep; may create additional objects. Check the flag descriptions for further details.

By default, pyfil prints the return value of expressions. Different types of objects use different printing conventions.

  • None does not print (as in the REPL)
  • strings are sent directly to to print()
  • iterators (not other iterables) print each item on a new line.
  • other objects are serialized as json. If an object cannot be serialized as json, it is sent directly to print().
  • all of these are overridden by --join

Iterators will also try to serialize each returned object as json if they are not strings. json objects will be indented if only one is being printed. If --loop is set or an number of objects is being serialzed from an iterator, it will be one object per-line. --force-oneline-json extends this policy to printing single json objects as well.

examples:

$ # None gets skipped
$ rep None
$ # strings and numbers just print
$ rep sys.platfrom
linux
$ rep math.pi
3.141592653589793
$ # objects try to print as json
$ rep sys.path
[
  "/home/ninjaaron/.local/bin",
  "/usr/lib/python35.zip",
  "/usr/lib/python3.5",
  "/usr/lib/python3.5/plat-linux",
  "/usr/lib/python3.5/lib-dynload",
  "/home/ninjaaron/.local/lib/python3.5/site-packages",
  "/usr/lib/python3.5/site-packages"
]
$ rep '{i: n for n, i in enumerate(sys.path)}'
{
  "/usr/lib/python3.5/plat-linux": 3,
  "/usr/lib/python35.zip": 1,
  "/usr/lib/python3.5": 2,
  "/usr/lib/python3.5/lib-dynload": 4,
  "/usr/lib/python3.5/site-packages": 6,
  "/home/ninjaaron/.local/lib/python3.5/site-packages": 5,
  "/home/ninjaaron/.local/bin": 0
}
$ # unless they can't
$ rep '[list, print, re]'
[<class 'list'>, <built-in function print>, <module 're' from '/usr/lib/python3.5/re.py'>]
$ # iterators print each item on a new line, applying the same conventions
$ rep 'iter(sys.path)'
/home/ninjaaron/src/py/pyfil/venv/bin
/home/ninjaaron/src/py/pyfil
/usr/lib/python35.zip
/usr/lib/python3.5
/usr/lib/python3.5/plat-linux
/usr/lib/python3.5/lib-dynload
/home/ninjaaron/src/py/pyfil/venv/lib/python3.5/site-package
$ rep '(i.split('/')[1:] for i in sys.path)'
["home", "ninjaaron", "src", "py", "pyfil", "venv", "bin"]
["home", "ninjaaron", "src", "py", "pyfil"]
["usr", "lib", "python35.zip"]
["usr", "lib", "python3.5"]
["usr", "lib", "python3.5", "plat-linux"]
["usr", "lib", "python3.5", "lib-dynload"]
["home", "ninjaaron", "src", "py", "pyfil", "venv", "lib", "python3.5", "site-packages"]

Most JSON is also valid Python, but be aware that you may occasionally see null instead of None along with true and false instead of True and False, and your tuples will look like list. I guess that's a risk I'm willing to take. (The rational for this is that pyfil, despite what the name of the rep command may indicate, is more about composability in the shell than printing valid Python literals. JSON is the defacto standard for serialization, or should be, if only people would stop using XML for that...)

Because these defaults use eval() internally to get value of expressions, statements may not be used. exec() supports statements, but it does not return the value of expressions when they are evaluated. When the -x/--exec flag is used, automatic printing is suppressed, and expressions are evaluated with exec, so statements, such as assignments, may be used. Values may still be printed explicitly.

--quite suppresses automatic printing, but eval is still used.

The --post option is immune from --quiet and --exec. It will always be evaluated with eval(), and it will always try to print. The only difference is that if --quiet or --exec was used, json will be printed with indentation unless --force-oneline-json is used.

rep doesn't have any parameters for input and output files. Instead, use redirection.

rep -s 'i.upper' > output.txt < input.txt

rep can take as many expressions as desired as arguments. When used with --exec, this works pretty much as expected, and assignment must be done manually.

Without --exec, the return value of each expression is assigned to the variable x, which can be used in the next expression. The final value of x is what is ultimately printed, not any intermediate values.

$ rep 'reversed("abcd")' '(i.upper() for i in x)'
D
C
B
A

one can do simple loops with a generator expression. (note that any expression that evaluates to an iterator will print each item on a new line unless the --join option is specified.)

$ ls / | rep '(i.upper() for i in stdin)'
BIN@
BOOT/
DEV/
ETC/
HOME/
...

However, the -l/--loop flag rep loops over stdin in a context like this:

for n, i in enumerate(stdin):
    expressions

Therefore, the above loop can also be written thusly:

$ ls / | rep -l 'i.upper()'

--pre and --post (-b and -e) options can be used to specify actions to run before or after the loop. Note that the --pre option is run with exec instead of eval, and therefore output is never printed, and statements may be used. This is for things like initializing container types. --post is automatically printed and statements are not allowed (unless --exec is used). --loop is implied if --post is used. --pre can be used without a --loop to import additional modules (or whatever else you may want to do with a statement).

Using -s/--split or -F/--field-sep for doing awk things also implies --loop. The resulting list is named f in the execution environment, in quazi-Perl fashion. (oh, and that list is actually a subclass of collections.UserList that returns an empty string if the index doesn't exist, so it acts more like awk with empty fields, rather than throwing and error and interrupting iteration).

rep can parse json objects from stdin with the -j/--json flag. They are passed into the environment as the j object. combining with the --loop flag will treat stdin as one json object per line.

It's probably obvious that the most powerful way to format strings is with Python's str.format method and the -F or -s options.

$ ls -l /|rep -s '"{0}\t{2}\t{8}".format(*f)'
Error: tuple index out of range
lrwxrwxrwx    root    bin
drwxr-xr-x    root    boot/
drwxr-xr-x    root    dev/
drwxr-xr-x    root    etc/
drwxr-xr-x    root    home/
lrwxrwxrwx    root    lib
...

However, you will note that using string.format(*f) produces an error and does not print anything to stdout (error message is sent to stderr; see error handling for more options) for lines without enough fields, which may not be the desired behavior when dealing with lines containing arbitrary numbers of fields.

For simpler cases, you may wish to use the -n/--join option, which will join any iterables with the specified string before printing, and, in the case of the f list, will replace any none-existent fields with an empty string.

$ ls -l /|rep -sn '\t' 'f[0], f[2], f[8]'
total
lrwxrwxrwx    root    bin
drwxr-xr-x    root    boot/
drwxr-xr-x    root    dev/
drwxr-xr-x    root    etc/
drwxr-xr-x    root    home/
lrwxrwxrwx    root    lib

In this case, the first line of ls -l / provides values for all available fields.

Technical note:
The separator specified with the --join option is implemented internally as ast.literal_eval("'''"+STRING.replace("'", r"\'")+"'''"). If one works hard at it, it is possible to pass values which will cause pyfil to crash; i.e. patterns ending with a backslash. Keep in mind rules about escape sequences in the shell and in python if you absolutely must have a pattern that terminates with a backslash. (The reason it is implemented this way is to allow the use of escape sequences that are meaningful to the python, but not the shell, such as \n, \t, \x, \u, etc.)

I realize that it's much better to do most of these things with the original utility. This is just to give some ideas of how to use `rep`

replace wc -l:

$ ls / | rep 'len(stdin.l)'
20

replace fgrep:

$ ls / | rep '(i for i in stdin if "v" in i)'
$ ls / | rep -l 'i if "v" in i else None'

replace grep:

$ ls / | rep 'filter(lambda x: re.search("^m", x), stdin)'
$ ls / | rep -lS 're.search("^m", i).string)'
$ # using the -S option to suppress a ton of error messages

replace sed 's/...:

$ ls / | rep -l 're.sub("^([^aeiou][aeiou][^aeiou]\W)", lambda m: m.group(0).upper(), i)'
BIN@
boot/
data/
DEV/
etc/
...

This example illustrates that, while you might normally prefer sed for replacement tasks, the ability to define a replacement function with re.sub does offer some interesting possibilities. Indeed, someone familiar with coreutils should never prefer to do something they already comfortable doing the traditional way with rep (coreutils are heavily optimized). Python is interesting for this use-case because it offers great logic, anonymous functions and all kinds of other goodies that only full-fledged, modern programming language can offer. Use coreutiles for the jobs they were designed to excel in. Use rep to do whatever they can't... and seriously, how will coreutils do this?:

$ wget -qO- http://pypi.python.org/pypi/pyfil/json/ | rep -j 'j["urls"][0]["filename"]'
pyfil-0.5-py3-none-any.whl
$ ls -l | rep -qSs \
"d.update({f[8]: {'permissions': f[0], 'user': f[2], 'group': f[3],
                  'size': int(f[4]), 'timestamp': ' '.join(f[5:8])}})" \
--post 'd'
{
  "README.rst": {
    "group": "users",
    "user": "ninjaaron",
    "permissions": "-rw-r--r--",
    "timestamp": "Sep 6 20:55",
    "size": 18498
  },
  "pyfil/": {
    "group": "users",
    "user": "ninjaaron",
    "permissions": "drwxr-xr-x",
    "timestamp": "Sep 6 20:20",
    "size": 16
  },
  "setup.py": {
    "group": "users",
    "user": "ninjaaron",
    "permissions": "-rw-r--r--",
    "timestamp": "Sep 6 20:30",
    "size": 705
  },
  "LICENSE": {
    "group": "users",
    "user": "ninjaaron",
    "permissions": "-rw-r--r--",
    "timestamp": "Sep 3 13:32",
    "size": 1306
  }
}

Other things which might be difficult with coreutils:

$ ls / | rep -n '  ' 'reversed(stdin.l)'
var/  usr/  tmp/  sys/  srv/  sbin@  run/  root/  proc/  opt/  ...
$ # ^^ also, `ls /|rep -n '  ' 'stdin.l[::-1]'

If pyfil encounters an exception while evaluating user input the default is to print the error message to stderr and continue (if looping over stdin), as we saw in the section on formatting output. However, errors can also be silenced entirely with the -S/--silence-errors option. In the below example, the first line produces an error, but we don't hear about it.

$ ls -l /|rep -sS '"{0}\t{2}\t{8}".format(*f)'
lrwxrwxrwx    root    bin
drwxr-xr-x    root    boot/
drwxr-xr-x    root    dev/
drwxr-xr-x    root    etc/
drwxr-xr-x    root    home/
lrwxrwxrwx    root    lib
...

Alternatively, errors may be raised when encountered, which will stop execution and give a (fairly useless, in this case) traceback. This is done with the -R/--raise-errors flag.

$ ls -l /|rep -sR '"{0}\t{2}\t{8}".format(*f)'
Traceback (most recent call last):
  File "/home/ninjaaron/src/py/pyfil/venv/bin/rep", line 9, in <module>
    load_entry_point('pyfil', 'console_scripts', 'rep')()
  File "/home/ninjaaron/src/py/pyfil/pyfil/pyfil.py", line 242, in main
    run(expressions, a, namespace)
  File "/home/ninjaaron/src/py/pyfil/pyfil/pyfil.py", line 164, in run
    handle_errors(e, args)
  File "/home/ninjaaron/src/py/pyfil/pyfil/pyfil.py", line 134, in handle_errors
    raise exception
  File "/home/ninjaaron/src/py/pyfil/pyfil/pyfil.py", line 162, in run
    value = func(expr, namespace)
  File "<string>", line 1, in <module>
IndexError: tuple index out of range

In addition to these two handlers, it is possible to specify a rudimentary custom handler with the -H/--exception-handler flags. The syntax is -H 'Exception: expression', where Exception can be any builtin exception class (including Exception, to catch all errors), and expression is the alternative expression to evaluate (and print, if not --quiet).

$ ls -l /|rep -sH 'IndexError: i' '"{0}\t{2}\t{8}".format(*f)'
total 32
lrwxrwxrwx    root    bin
drwxr-xr-x    root    boot/
drwxr-xr-x    root    dev/
drwxr-xr-x    root    etc/
drwxr-xr-x    root    home/
lrwxrwxrwx    root    lib
...

In this case, we've chosen to print line without any additional formatting. If other errors are encountered, it will fall back to other handlers (-S, -R, or the default). For more sophisticated error handling... Write a real Python script, where you can handle to your heart's content.

Also note that this case is possible to handle with a test instead of an exception handler because f is a special list that will return an empty string instead of throw an index error if the index is out of range:

ls -l / | rep -s '"{0}\t{2}\t{8}".format(*f) if f[2] else i'

Easy-peasy.