Skip to main content

Simple boilerplate for cli scripts

Project description

Create a command line interface with minimal setup.

PyPI Python versions PyPI license Build status Dependencies

clima - command line interface with a schema

Table of contents

Briefly

Features

Clima handles loading and parsing command line arguments with some off-the-shelf features including:

  • a global configuration object
    • quick definition of defaults
    • defining defaults doubles as description for help on the command line
    • type handling with annotations
  • definitions with configuration files
  • env variables
    • loading .env files
  • secrets stored with pass
  • post_init hook

Cli definition

Creating a cli:

  1. Import all necessary parts from the package clima
  2. (optional) Define configuration i.e. Schema
  3. Define the command line commands i.e. Cli-class:

example ascii

Example: to setup a configuration and a command line interface ready to go.

from clima import c

@c
class Cli:
    def say_hi(self):
        print('oh hi - whatever this is..')

The command line usage form could be as simple as:

 my_tool say_hi

Configuration object in a spiffy

from clima import c

# Defining the settings (configuration object)
class S(Schema):
    place = 'world'
    
@c
class Cli:
    def say_hi(self):
        # using configuration object 'c'
        print(f'oh hi - {c.place}')

The command line usage form could be as simple as:

 my_tool say_hi
 my_tool say_hi --place 'other world'

See the examples folder and other sections for more examples. For example the folder includes something that resembles the example above.

Installing

pip install --user clima

toc

Usage

See the example file in examples/script_example.py. Here's a run down of the individual parts in such a script (adapted from another example at module example).

First import the required components:

from clima import c, Schema

In your code define the Schema subclass:

class Configuration(Schema):
    a: str = 'A'  # a description
    x: int = 1  # x description

Here "Configuration" is an arbitrary name, no magic there. The inherited Schema class defines the attributes (i.e. a and x in this example).

Note the specific formatting of the Schema subclass:

    # attribute[: type] = default value  [# Description for the --help]
    a: str = 'A'  # a description

a is the attribute which can be called in the code later with c.a. In this example, it has a type of 'str' and a default value of 'A'. The comment after is not redundant in the sense that it is parsed for the command line help. The values in square brackets [] are optional.

All of these parts will be parsed for the '--help' for the subcommands of the cli, for example:

./script.py foo -h

Will now produce:

 Usage:       script.py foo [ARGS]
 
 Description: Args:
     --a (str): a description (Default is 'A')
     --x (int): x description (Default is 1)    

The example in the readme can be found at examples/readme_example.py.

The subcommands - or commands of the script - should be defined somewhat as follows:

@c
class Cli:
    def subcommand_foo(self):
        """This will be shown in --help for subcommand-foo"""
        print('foo')
        print(c.a)
        print(c.x)

    def subcommand_bar(self):
        """This will be shown in --help for subcommand-bar"""
        print('bar')

The methods are parsed as subcommands and their respective doc strings will show in the subcommands' help printout. Note the double usage of the c - first as a decorated and later as the parsed configuration inside the method:

...
    ...
    print(c.a)
    print(c.x)

toc

Examples and platforms

Tried and used on linux, macos and windows. However, python packaging and dependency management is sometimes hairy and your mileage may vary.

More examples in the examples directory with printouts of the defined subcommands and helps.

Testing the examples

The examples can be tried out by cloning the repo and running from repo directory root (on linux and the like):

git clone https://github.com/d3rp/clima.git 
cd clima
PYTHONPATH=$PWD python ./examples/readme_example.py foo -h

Running the examples that wrap a module:

PYTHONPATH=$PWD python ./examples/module_example/__main__.py -h
PYTHONPATH=$PWD python ./examples/module_example/__main__.py subcommand-foo -h
PYTHONPATH=$PWD python ./examples/module_example/__main__.py subcommand-bar
...

Output should resemble this (fire v0.1.3 prints out Args, fire v0.2.1 doesn't (though looks much nicer))

$ tester subcommand-foo -- -h

Type:        method
String form: <bound method Cli.subcommand_foo of <__main__.Cli object at 0x000002995AD74BE0>>
File:        C:\Users\foobar\code\py\clima\tester\__main__.py
Line:        18
Docstring:   This will be shown in --help for subcommand-foo
Args:
    --a (str): a description (Default is 'A')
    --x (int): x description (Default is 1)

Usage:       __main__.py subcommand-foo [--X ...]

All of the example scripts can be run by installing poetry and running the run_examples.bash script:

pip install --user poetry
./run_examples.bash

toc

Version printing

Version printing works via the version subcommand. This is intended for scripts that are packaged as command line tools with poetry. Thus with bumping the version with poetry, clima will handle parsing the current version of your tool so it can be queried with:

my_tool version

The actual version is parsed into the c so overwrite it with post_init or something if you want control over it.

Autocompletion

..in IDEs (wip)

Also, to enable autocompletion in IDEs, this hack suffices:

c: Configuration = c

Put it in the "global space" e.g. just after defining the template. See the examples/script_example.py for a specific example.

When all is complete, the imported c variable should have all the bits and pieces for the configuration. It can be used inside the Cli class as well as imported around the codebase thus encapsulating all the configurations into one container with quick access with attributes c.a, c.x, etc...

..in bash

Run your script with -- --completion arguments:

my_tool -- --completion

This should print an autocompletion definition to include in your bash completions.

TBD: zsh etc. completions

Post init hook

There's two ways to define a post_init hook depending if it is done in the Schema subclass or the Cli definition.

Cli.post_init()

In some occasions it's useful to deduce specific defaults from the given parameters e.g. in a cross platform build allowing only minimal cli arguments. For those cases there's an post_init hook When defining the post_init() in the Cli class, i.e.

@c
class Cli:

    @staticmethod
    def post_init(s):
        if s.platform = 'win':
            self.bin_path = 'c:/Users/foo/bar'
        else:
            s.bin_path = '/Users/mac/sth'
        
    def subcommand(self):
        pass

The method will have access to the cli args, but can not introduce new variables to the schema.

This is arguably the more useful of the two variations of post_inits.

Note: The signature of the post_init() differs depending on which of the stages it is defined in. For the time being it is a @staticmethod

Schema.post_init()

This alternative is for to use post_init-like features positioning the steps so that command line arguments can still override things.

class SoAdvanced(Schema):

    platform: str = 'win'  # a description
    bin_path: pathlib.Path = ''  # x description
    
    def post_init(self, *args):
        if self.platform = 'win':
            self.win_specific_field = 'All your files are locked by us..'

Note: This post_init() does not have access to the cli arguments, but the Schema's post_init can introduce new attributes/properties/fields/arguments to the configuration, which the Cli-class post-init can't. Schema post init hook is run after schema initialization, but BEFORE the cli initialization.

toc

Configuration options

It's tedious to have to write a long list of parameters on the command line, when most of the use cases follow a similar pattern. To facilitate the use of configurations, there's several options to choose from.

The c decorator/configuration chains multiple configuration options together in order of priority (lower number overrides higher number):

  1. command line arguments
  2. Environment variables
  3. .env file
  4. configuration file definitions
  5. decrypted passwords from ~/.password-store if gnugpg is installed
  6. defaults in the subclass inheriting Schema

Configuration file and environment variables

The configuration file should be named with either the postfix .conf or .cfg e.g. foo.conf and have an ini type formatting with a 'Clima' section:

# foo.conf
[Clima]
x = 2

The keys are the same as what you define in the schema. You can define all, some or none of the attributes. Same applies for the env variables.

# linux example
X=2 tester subcommand-foo

A configuration file defined this way can be located in the current working directory or - if your Schema defines a cwd field - there. Clima will try to use the first configuration file it finds, so that might produce some caveats.

class Conf(Schema):
    cwd = ''

# Running ./script.py --cwd <folder> would automatically load the first *.conf file in <folder>

Type casting with configuration definition

The Schema definition can have type annotations, which are used to cast the given arguments. For example

class C(Schema):
    p: Path = ''  # Path to something

Results in c.p's type cast as Path.

Configuration file in the home directory

You can also define the config file in the configuration class (one inheriting Schema) by defining the magic field CFG.

For example, lets say the command my_tool (packaged etc) has a user configuration file at ~/.my_tool.conf. This can now be handled with just adding CFG = Path.home() / '.my_tool.conf to the Schema:

from pathlib import Path

class S(Schema):
    bing = 'bang'
    CFG = Path.home() / '.my_tool.conf'

Then, for example, the configuration file would be written as:

#~/.my_tool.conf
[Clima]
bing = diudiu

Running the command my_tool would produce the value in the configuration file, though the arguments can still be overriden.

my_tool run 
# diudiu

my_tool run --bing bam
# bam

.env file

This is handled by dotenv. In short, all the defaults defined in the Schema subclass can be overridden either by:

<field> = <value>

or

export <field> = <value>

Password unwrapping/decryption with pass

Note: Currently this works only for gpg-keys without password. It's not ideal, but it's better than plain text .env files ;)

Note 2: Leading and trailing whitespace (including \n linefeeds) are stripped, when decrypted.

pass can be used to store passwords as gpg encrypted files under the home directory. Clima uses the default path of ~/.password-store and the files found within. It will then match the arguments with the stored passwords, for example:

 tree -A ~/.password-store                                                                                                                                                                                                                                                                             ✔ | 41s | anaconda3 
 /Users/me/.password-store
 ├── work
 │   ├── ci
 │   │   ├── sign_id.gpg
 │   │   ├── sign_pw.gpg
 ... ... ...

And an according Schema definition:

 class Conf(Schema):
     sign_id: str = ''  # signing id for the CI
     sign_pw: str = ''  # signing pw for the CI

Would accept those arguments as cli arguments, or if omitted, would traverse through the .password-store and decrypt the found sign_id.gpg and sign_pw.gpg placing the values found in the configuration object c.

toc

Additional features via Fire

See the Python Fire's Flags documentation for nice additional features such as:

# e.g. tester.py is our cli program
tester.py subcommand-foo -- --trace
tester.py -- --interactive
tester.py -- --completion

Truncated error printing

Even though I've used python for a few years professionally, I'm still not satisfied with its error printing. Clima truncates the error lists and tries to provide a more readable version of the "first" point of failure. The whole traceback is written into a logfile exception_traceback.log so it can be examined when the truncated output is not enough.

Note: When running the examples, the exception_traceback.log file will be written inside the examples directory

Ways to run the script for the uninitiated

Here's a section to suggest ideas how to wrap scripts using clima.

Linking executable script to ~/.local/bin

Let's say those lines were written in a file named script.py. Command line usage in a terminal would then be e.g.:

python script.py foo
python script.py foo --a 42

Adding this line in script.py

#!/usr/bin/env python

and changing its execution permissions (mac, linux):

chmod +x script.py

Allows for a shorter means of execution:

./script.py foo

Now this could be linked as an adhoc command for example:

ln -s $PWD/script.py ~/.local/bin/my_command

Packaging a module (pip ready)

For a pip-installable package, one could package this as a runnable command - publish in the public or one's private pypi etc - and then approach the convenience factor shown at first.

pip install my_tool
my_command foo -h

To publish with poetry is quite straight forward. First create an account in pypi.org and then:

cd <project directory>
poetry build
poetry publish

You can use version to bump up versions:

poetry version patch

Building/Installing from source

This repo is based on poetry.

git clone https://github.com/d3rp/clima.git 
cd clima
poetry install --no-dev

The --no-dev is for to install the running environment without development tooling.

toc

Long description and background

The subcommands are written as a class encapsulating the "business logic". You can define a simple schema of the configuration that maps to the command line arguments.

In other words, you can use this to wrap your scripts as command line commands without resorting to bash or maintaining argument parsing in python. This removes the need of duplicating comments in order --help to remember what the arguments were and what they did. Sprinkling some decorator magic offers a typical use experience of a cli program (e.g. argument parsing and validation, --help, subcommands, ...).

The implementation is focused on a premise that for a simple script there's usually a script wide global configuration which would be used through out the user code i.e. a context for the program that is refered to in different parts of the code. That configuration is populated with given arguments falling back on defaults in the code and some further complimentary options. Those are then made accessible via a global c variable that can be tossed around the code base with very little additional effort. With a small adjustment this can made to autocomplete in IDEs (as attributes). This helps when the schema of the configuration grows larger as the autocomplete kicks in after typing c. offering those fields in your "schema" as attributes.

Why another cli framework?

Clima is not intended to cater all needs of a feature complete cli framework like the ones enlisted below. This is a package to help with boilerplate to get quick, but reusable tools for your workflow.

Other options for full featured cli experience:

Dependencies

  • dotenv

  • gnugpg - this is pass through though. If it's not installed, the feature is not in use.

  • fire - python-fire from google does the cli wrapping / forked and included into the repo - I wanted to have the version 0.1.x formatting and help output with few hacks of my own

toc

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

clima-0.7.8.tar.gz (46.7 kB view hashes)

Uploaded Source

Built Distribution

clima-0.7.8-py3-none-any.whl (47.8 kB view hashes)

Uploaded Python 3

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