Skip to main content

Toolkit for developing command-line utilities in Python

Project description

Overview

Cogs is a toolkit for developing command-line utilities in Python. It handles common operations such as parsing command-line parameters, dispatching commands and loading configuration files. It is targeted to developers, sysadmins, testers, or anyone who needs to script their routine tasks.

Cogs is a free software licensed under MIT license. Cogs is written by Kirill Simonov from Prometheus Research, LLC.

Getting Started

You can install Cogs using PIP package manager:

# pip install Cogs

This operation installs a command-line utility cogs and a Python package of the same name. The cogs utility is a dispatcher which which lets you select and execute your scripts (called tasks). Let us show how to do it with a simple example.

In the current directory, create a file cogs.local.py with the following content:

from cogs import task
import os

@task
def Hello(name=None):
    """greet someone (if not specified, the current user)"""
    if name is None:
        name = os.getlogin()
    print "Hello, %s!" % name.capitalize()

Now run:

$ cogs hello world
Hello, World!

$ cogs hello
Hello, Xi!

$ cogs help hello
HELLO - greet someone (if not specified, the current user)
Usage: cogs hello [<name>]

Cogs converts function Hello() into a command-line script with parameters inferred from the function signature, so then when you execute a command cogs hello world, you invoke a function Hello('world').

Loading Extensions

In this section, we describe how Cogs finds and loads extensions.

Cogs loads extensions from two places:

  • cogs.local.py or cogs.local/__init__.py module in the current directory;

  • all Python packages under cogs.extensions entry point.

The easiest way to add an extension is to create a cogs.local.py module and add your tasks there. If you want to split the module into multiple files, create a cogs.local subdirectory with an __init__.py module. cogs.local.py or cogs.local/__init__.py are executed by Cogs on startup. The modules must be owned by the same user who runs the cogs script, or by root; otherwise they are ignored.

If you need to package and distribute your Cogs extensions, using cogs.local may be inconvenient. In this case, you may package your Cogs extensions as a regular Python distribution.

Suppose we want to pack the hello task as a separate package. Create a directory tree with the following structure:

Cogs-Hello/
    src/
        cogs/
            __init__.py
            hello.py
    setup.py

The file cogs/hello.py contains the definition of the hello task and has the same content as cogs.local.py in our previous example.

The file cogs/__init__.py contains just one line:

__import__('pkg_resources').declare_namespace(__name__)

The file setup.py contains the meta-data of the package and may look like this:

from setuptools import setup

setup(
    name='Cogs-Hello',
    version='0.1',
    description="""A Cogs task to greet somebody""",
    packages=['cogs'],
    namespace_packages=['cogs'],
    package_dir={'': 'src'},
    install_requires=['Cogs'],
    entry_points={ 'cogs.extensions': ['Hello = cogs.hello'] },
)

Note the parameter entry_points in setup() invocation; it adds an entry point cogs.extensions named Hello that refers to module cogs.hello. On startup, Cogs finds and loads all packages defined for the entry point cogs.extensions.

Defining Tasks

A task can be created from a function or a class by augmenting it with the task decorator:

from cogs import task, argument
from cogs.log import log, fail

@task
def Factorial(n):
    """calculate n!

    This task calculates the value of the factorial of the given
    positive number `n`.  Factorial of n, also known as n!, is
    defined by the formula:

        n! = 1*2*...*(n-1)*n
    """
    try:
        n = int(n)
    except ValueError:
        raise fail("n must be an integer")
    if n < 1:
        raise fail("n must be positive")
    f = 1
    for k in range(2, n+1):
        f *= k
    log("{}! = `{}`", n, f)

@task
class Fibonacci:
    """calculate the n-th Fibonacci number

    The n-th Fibonacci number `F_n` is defined by:

        F_0 = 0
        F_1 = 1
        F_n = F_{n-1}+F_{n-2} (n>1)
    """

    n = argument(int)

    def __init__(self, n):
        if n < 0:
            raise ValueError("n must be non-negative")
        self.n = n

    def __call__(self):
        p, q = 0, 1
        for k in range(self.n):
            p, q = p+q, p
        log("F_{} = `{}`", self.n, p)

You can now execute the tasks by running:

$ cogs factorial 10
10! = 3628800

$ cogs fibonacci 10
F_10 = 55

Cogs uses the name of the function or the class as the task identifier. The name is normalized: it is converted to lower case and has all underscore characters converted to the dash symbol.

If the task is derived from a function, the task arguments are inferred from the function signature. Cogs executes such task by calling the function with the parsed command-line parameters.

If the task is derived from a class, the task arguments and options must be defined using argument() and option() descriptors. To execute a task, Cogs creates an instance of the class passing the task parameters as the constructor arguments. Then Cogs invokes the __call__ method of the instance. Thus the call of:

$ cogs factorial 10

is translated to:

Factorial('10')

and the call of:

$ cogs fibonacci 10

is translated to:

t = Fibonacci(10)
t()

The docstring of the function or the class becomes the task description:

$ cogs help factorial
FACTORIAL - calculate n!
Usage: cogs factorial <n>

This task calculates the value of the factorial of the given
positive number n.  Factorial of n, also known as n!, is
defined by the formula:

    n! = 1*2*...*(n-1)*n

$ cogs help fibonacci
Usage: cogs fibonacci <n>

The n-th Fibonacci number F_n is defined by:

    F_0 = 0
    F_1 = 1
    F_n = F_{n-1}+F_{n-2} (n>1)

A task derived from a function cannot have options. To add an option to a task derived from a class, use the option() descriptor. For example:

from cogs import task, argument, option
import sys, os

@task
class Write_Hello:

    name = argument(default=None)
    output = option(key='o', default=None)

    def __init__(self, name, output):
        if name is None:
            name = os.getlogin()
        self.name = name
        if output is None:
            self.file = sys.stdout
        else:
            self.file = open(output, 'w')

    def __call__(self):
        self.file.write("Hello, %s!\n" % self.name.capitalize())

You can execute this task with option --output or -o to redirect the output to a file:

$ cogs write-hello world -o hello.txt

Configuration and Environment

Cogs allows you to define custom configuration parameters. For example:

from cogs import env, task, setting
import os

@setting
def Default_Name(name=None):
    """the name to use for greetings (if not set: login name)"""
    if name is None or name == '':
        name = os.getlogin()
    if not isinstance(name, str):
        raise ValueError("a string value is expected")
    env.add(default_name=name)

@task
def Hello_With_Configuration(name=None):
    if name is None:
        name = env.default_name
    print "Hello, %s!" % name.capitalize()

Now you could specify the name as a configuration parameter default-name. One way to do it is to use global option --default-name:

$ cogs --default-name=world hello-with-configuration

You could also pass a configuration parameter using an environment variable:

$ COGS_DEFAULT_NAME=world cogs hello-with-configuration

Alternatively, you can put parameters to a configuration file. In the current directory, create a file cogs.conf with the following content:

default-name: world

Now run:

$ cogs hello-with-configuration

Cogs reads configuration from the following locations:

  • /etc/cogs.conf

  • $PREFIX/etc/cogs.conf

  • $HOME/.cogs/cogs.conf

  • ./cogs.conf

  • program environment

  • command-line parameters

If you’d like to specify the usage of a configuration file that is in a different location than the standard locations listed above, you can use the global option --config as follows:

$ cogs --config=alternate-cogs.conf hello-with-configuration

To create a new configuration parameter, wrap a function named after the parameter with the @setting decorator. The function must accept zero or one argument: the function is called without arguments if the parameter is not specified explicitly, and is called with the value of the parameter is it was set using one of the methods described above.

Cogs does not impose any rules on what to do with the parameter value, but we recommend to store the value in the global env variable. The call of env.add(default_name=name) adds a new parameter default_name which could then be accessed as env.default_name.

API Reference

cogs.core

Classes and functions defined in cogs.core are also importable from the cogs package.

@task

The @task decorator converts the wrapped function or class into a task. Task properties are inferred from the wrapped object as follows:

name

Generated from the function or the class name. The name is converted to lower case and all underscores are replaced with dashes.

documentation

Generated from the docstring. The first line of the docstring produces a one-line hint string, the rest of the docstring produces a multi-line help string.

arguments

When the task is inferred from a function, the arguments are generated from the function signature. Each function parameter becomes a task argument, those which have default values are optional.

If the task is inferred from a class, the arguments must be specified using the argument() descriptor.

options

A task inferred from a function has no options. A task inferred from a class may have options specified using the option() descriptor.

When a task is executed, the wrapped object is invoked according to the following rules:

  • If the task is inferred from a function, parsed command-line parameters are passed as the function arguments.

  • If the task is inferred from a class, command-line parameters are passed to the class constructor, then the __call__ method is called on the instance.

@setting

The @setting decorator converts the wrapped function to a configuration parameter, which properties are inferred from the function attributes.

The setting name is generated from the function name. The name is converted to lower case and has all underscores replaced with dashes.

The setting documentation is generated from the function docstring.

The function must be able to accept zero and one parameter. The function is called at startup with no parameters if the setting is not explicitly set by the user; otherwise it is called with the value of the setting. The function is responsible for storing the value in the env object.

argument(check, default, plural=False)

Describes a task argument.

check

A function which is called to check and/or transform the argument value. The function must return the transformed value or raise ValueError exception on error.

default

The default value to be used if the argument is optional and not specified. If this parameter is not set, the argument is mandatory.

plural

If set, the argument consumes all the remaining command-line parameters. Must be the last argument specified.

option(key, check, default, plural=False, value_name=None, hint=None)

Describes a task option.

key

A one-character shorthand.

check

A function called to check and transform the value of the option. The function must return the transformed value or raise ValueError exception on error.

default

The default value used when the option is not specified. If this parameter is not set, the option does not accept a value. Such an option is treated is a toggle and takes a value True if set and False if not set.

plural

If set, indicates that the option could be specified more than once.

value_name

The preferred name for the option value; used for the task description.

hint

A one-line description of the option; used for the task description.

env

A global object that keeps values of configuration parameters and other properties.

env.add(**keywords)

Add new parameters.

env.set(**keywords)

Set values for existing parameters.

env.push(**keywords)

Save the current state and set new values for existing parameters.

env.pop()

Restore a previously saved state of parameters and values.

env(**keywords)

A context manager for with statement. On entering, saves the current state and sets new parameter values. On exiting, restores the saved state.

cogs.log

Printing utilities.

log(msg="", *args, **kwds)

Print the message to the standard output.

If extra positional or keyword arguments are given, they are formatted with format() function using msg as the template.

log() supports output coloring: a substring of the form:

`...`

or:

:fmt:`...`

is colorized when displayed on a color terminal. The supported formats are: default (white), debug (dark grey), warning (red), success (green).

debug(msg, *args, **kwds)

Print the message when the env.debug parameter is set. We recommend to accompany any permanent change to the filesystem or other system state with a respective debug() call.

Add command-line parameter --debug or set environment variable COGS_DEBUG=1 to see debug output.

warn(msg, *args, **kwds)

Display a warning. warn() should be used for reporting error conditions which do not prevent the script from continuing the job.

fail(msg, *args, **kwds)

Display an error message and return an exception object. It should be used in the following manner:

raise fail("no more beer in the refrigerator")

cogs.fs

File and system utilities.

cp(src, dst)

Copy a file or a directory tree.

mv(src, dst)

Move a file or a directory tree.

rm(path)

Remove a file.

mktree(path)

Create all directories in the path.

rmtree(path)

Remove a directory tree.

exe(cmd, cd=None, environ=None)

Replace the current process with the given shell command.

If cd is given, changes the directory to the specified path before executing the command.

If environ is given, adds the given parameters to the environment before executing the command.

sh(cmd, data=None, cd=None, environ=None)

Execute a shell command with the given input and working directory.

pipe(cmd, data=None, cd=None, environ=None)

Execute a shell command with the given input and working directory; return the command output.

Project details


Download files

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

Source Distribution

Cogs-0.4.0.tar.gz (17.8 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