Flatten main loop

Flatten main loop

Add _maybe_open which acts as a context manager for either a file
object (if a filename is provided) or sys.stdout (if '-' is provided).

This is a little weird, but makes it much clearer what's going on in
main().