Skip to main content

Library for writing command line programs

Project description

This module contains a lightweight library for creating command line programs, conz.

conz has following features:

  • Manages signals (SIGINT and SIGPIPE)

  • Simplifies working with pipes

  • Supports output colorization

  • Provides tools for working with long-running tasks

  • Controls output in interactive and non-interactive scenarios

Installing

You can install conz from PyPI using pip or easy_install:

pip install conz

easy_install conz

Quick tour

A quick tour example can be found in examples/quicktour.py:

import conz

someval = True

cn = conz.Console(verbose=True)

cn.pstd('This goes to STDOUT')
cn.perr('This goes to STDERR')
cn.pverr(someval, 'This message is related to somevar')

cn.pstd(cn.color.green('This message is green'))

with cn.progress('Some long operation'):
    import time
    time.sleep(2)

data = cn.read('Type something in:')
cn.pstd('You typed in {}'.format(cn.color.yellow(data, style='italic')))

Usage

The conz package includes a class Console which is the only class you will even need to work with. Simply import and instantiate it at the top of your program.

import conz
cn = conz.Console()

Working with output

The Console() class is, for the most part, a wrapper around the print() function (not print statement, so not compatible with versions of Python that do not support this). It controls how print() is invoked and takes care of some of the edge cases where it may malfunction.

The print() method on a Console object is a very simple wrapper around Python’s print() whic does nothing except pass it’s positional and keywrod arguments to the print() function. We will never use it directly, though, as there are shortcuts for doing specific things with print().

To output to STDOUT, we use the pstd() method. It takes the same arguments as print() function, with the exception of file keyword argument which is set by this method and cannot be overridden.

cn.pstd('This always goes to STDOUT', end='...')
out: This always goes to STDOUT

To output to STDERR, we use the perr() method. As with pstd(), it overrides the file argument for us.

cn.perr('Mayday, mayday!')
err: Mayday, mayday!

The main difference between regular print() and pstd()/perr() methods is that the latter will flush the STDOUT/STDERR after writing to it. This can prvent weird issues in some edge cases.

There is a variant of perr() which prints a more structured message to STDERR. The pverr() method takes a value and a message, and prints then in VALUE: Message format.

path = '/foo/bar/baz.txt'
cn.verr(path, 'not found')
err: /foo/bar/baz.txt: not found

A variant of pstd() is pverb(). It is exactly the same as pstd(), except that it only outputs when verbose flag on the Console object is True. This is useful for programs that need to differentiate between interactive and non-interactive use (e.g., using in pipe vs invoking directly) or wish to have a --verbose switch, etc.

cn.verbose = True
cn.pverb("I'm a talkative program")
out: I'm a talkative program

cn.verbose = False
cn.pverb("I'm a talkative program")
out:

The verbose flag can be set either as an argument during instantiation, or simply by setting the attribute as in the previous example.

The Console object also provides a outterm property which is False when program is outputting to a pipe rather than the terminal:

if cn.outterm:
    # give full output to the user
else:
    # give a short output that can be parsed by a machine, etc

Colorizing

Before we start, note that this implementation is not cross-platform. If you need something with a bit more punch, you should look at colorama.

To colorize the output, both the conz module and Console class have a color attribute, which provides methods for output colorization. Each piece of text can have the following attributes:

  • foreground color

  • style

  • background color

Foreground and background colors can be:

  • black

  • red

  • green

  • yellow

  • blue

  • purple (magenta)

  • cyan

  • white

Styles can be:

  • bold

  • italic

  • underline

  • blink

  • reverse (inverts foreground and background colors)

Each of these colors have a method on the color attribute. Each color method takes style and bg keyword arguments which set the style and background color respectively. The color() method can be used to specify colors dynamically. Here are some examples:

cn.color.red('This is red text')
cn.color.color('This is red text', color='red')
cn.color.blue('This is blue underlined text', style='underline')
cn.color.color('This is green on yellow', color='green', bg='yellow')

You can find an example script in examples/colors.py which prings all possible combinations of various colors, styles, and backgrounds.

Working with input

There are two types of input you can work with: interactive user input, and pipes.

To read the user input, use read() method. This method takes two optional arguments. One is the prompt argument, which we use to set the prompt. It is an empty string by default. The other argument is a data-cleaning function. When you pass the clean argument, user input is passed through the function before it is retuned. For example:

cn.read('Exit? [y/N] ', clean=lambda x: x.lower()[:1] == 'y')
out: Exit? [y/N] _
in: y
==> True

Note that this method uses raw_input() on Python 2.7.x and input() on Python 3.x.

To work with pipes, we use the readpipe() method. This method reads from the STDIN pipe one line at a time and returns an iterator that allows us to iterate over the lines.

for l in cn.readpipe():
    l = l.strip()
    cn.pstd('Received: {}'.format())

Note that line-feed characters are not stripped from the output so it is up to us to strip it away.

When working with a large number of lines coming down the pipe, we may sometimes need to work in batches, rather than one line at a time. The chunk argument can be set to an integer value that specifies the number of lines we want buffered before they are returned to us. When using chunks, the lines are returned as a list of strings, rather than strings. The following example will return pipe input in groups of 500:

for lines in cn.readpipe(500):
    # do something with 500 lines

If we need to know whether input will come from a pipe or not, we can use the interm property.

if cn.interm:
    # possibly interactive version
else:
    # we are on the receiving end of a pipe

Working with progress

Progress is a more complex construct that we use to notify user of some activity that may take a while. Each progress has a start banner, which is printed before we begin, and two end banners, one for success and one for failure.

Before we can use the progress context manager, we must enable verbose mode.

cn.verbose = True

A progress is started using the progress() method, which is a context manager.

with cn.progress("Let's get this show on the road"):
    # do something

This is the simplest form. When an exception of any kind is triggered inside the context, it is trapped, the failure banner is printed, and the conz.ProgressAbrt exception is raised. (This exception is also available as an attribute on Console objects for convenience.) If everything goes well, then the success banner will be printed. With the previous code snippet, sucess output may look like this:

Let's get this show on the road...DONE

And failure would look like this:

Let's get this show on the road...FAIL

The end banners can be customized by using the end and abrt arguments:

with cn.progress('Almost there', end='finally!', abrt='awww, bummer'):
    # do something

The outputs would look like this:

Almost there...finally!

Almost there...awww, bummer

The elipsis (three dots) can be customized using the sep argument:

with cn.progress('File check', sep=': '):
    # do something

This results in:

File check: DONE

File check: FAIL

By default, the progress context manager will trap any exception. This may or may not make sense for a particular situation. This behavior can therefore be customized using the excs argument, which takes a tuple of exception classes that we are expecting. Passing exceptions explicitly like this allows the context manager to propagate unhandled exceptions and reval subtle flaws in our logic.

We can also specify a callback that runs each time an exception (other than ProgressAbrt and ProgressOK are raised inside the context. This callback is specified using onerror argument, and defaults to an error handler that prints ‘Program error: ERROR MESSAGE’ to STDERR. For convenience, the Console object has a error() method which creates such handlers.

To create a handler, we call the error() method like so:

handler = cn.error('Ouch!', exit=1)
with cn.progress('Ouch progress', onerror=handler):
    raise RuntimeError()

The above results in:

Outch progress...FAIL
Ouch!

The message may have a {err} placeholder, which gets replaced by the string representation of the exception that was raised in the block.

To completely suppress the error handler, simply pass it a function that does nothing.

with cn.progress('No ouch', onerror: lambda exc: None):
    raise RuntimeError()

This results in:

No ouch...FAIL

The progress context manager returns a Progress object, which provides methods for explicitly terminating the progress, and printing the progress indicator. This object has end() and abrt() methods, which are called to terminate with success and error status respectively. For example:

with cn.progress('Something') as prg:
    if not success:
        prg.abrt()
    prg.end()

The end() and abrt() methods raise ProgressOK and ProgressAbrt exceptions repsectively. We can suppress raising of the exceptions using noraise argument and setting it to True. Both of the methods will use the default end banners. We can also use any banner we want by passing it as the first positional argument. This can be useful in cases where the end banner should indicate different outcomes.

The ProgressOK exception is not meant to be handled by us in any way, and it’s simply there to facilitate flow control. ProgressAbrt is, by default, reraised so that code outside the context manager can handle it. Therefore, we normally wrap the context block in a try-except:

try:
    with cn.progress('Something'):
        # do something
except cn.ProgressAbrt:
    # something went wrong

This reraising of the ProgressAbrt exception can be suppressed by using the reraise argument which can be True or False. Setting this flag to False silences the ProgressAbrt exception. At that point, we are still able to do error handling using the onerror callback.

You can find a script in examples/progress.py which demonstrates a few typical cases.

Quitting

To quit the program, we call the quit() method on the Console object. This method works the same way as sys.exit() (except that it takes one less import to use it).

Signal handling

The default implementation of Console class automatically takes care of SIGINT (keyboard interrupt) and SIGPIPE (broken pipe) signals. You can customize the way those are handled by overloading the onint() and onpipe() methods. You can also customize the registration of signals themselves by overloading the register_signals() method.

Reporting bugs

Please report any bugs or feature requests to the issue tracker.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

conz-0.1.zip (29.6 kB view hashes)

Uploaded Source

conz-0.1.tar.gz (22.9 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page