Skip to content
This repository has been archived by the owner on Dec 27, 2022. It is now read-only.
/ pyautomate Public archive

Automation/make tool where you describe a state machine in a config file and use the CLI to go to a certain state.

License

Notifications You must be signed in to change notification settings

timdiels/pyautomate

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

73 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A more mature DAG based build system with better performance and more features exists, see tup.

pyautomate is a CLI tool to automate any set of tasks. You describe a state machine in a config file and use the CLI to go to a certain state. pyautomate will automatically find out how to get there using the config file you provided.

To get a good idea of what pyautomate can do for you I suggest you have a quick scroll through the usage section.

Note: links are broken due to a rst bug in github.

pyautomate is free software, available under the GPLv3 license.

pyautomate is not under any active development at the moment (it has all the features I need and there are no requests for more). The documentation is complete, pyautomate is tested with automated tests with good test coverage. pyautomate is stable.

pyautomate is distributed using 0install, and thus requires no installation, you can just run it.

The 0install feed (with instructions): http://timdiels.be/feeds/pyautomate.xml

pyautomate's state machine is a non-deterministic finite state machine (NFA) with the concept of guards added to its transitions, i.e. the transition can only be taken if the guard is true. If that made no sense to you, read on.

pyautomate's state machine is a system based on states and transitions between those states. The machine starts in one or more initial states and can be asked to try to get to a desired state. A transition consists of a source state, multiple target states and the action to execute in order to make the transition.

The state machine can be represented as a state diagram, e.g.:

https://github.com/limyreth/pyautomate/raw/master/readme_files/usage_diagram.png

The state machine can be in multiple states at the same time (as it is non-deterministic). E.g. when executing 'test_server()' with 'server started' as current state, the machine will switch to both 'server started' and 'server passed tests'. After that, it can still make a transition to 'server stopped' and 'server passed tests' by executing 'stop_server()'.

The state machine can make its transitions conditional with guards. There's a special state called the anonymous state, the state machine is always in the anonymous state.

If you're not familiar with state machines, you should read the above first.

I'll explain pyautomate with a simple example: developing a daemon. I want to start, stop the daemon and run tests. If I want to run tests, the daemon should obviously be running.

You usually want to start out by drawing your system on paper, for this case I resulted with this state diagram:

https://github.com/limyreth/pyautomate/raw/master/readme_files/usage_diagram.png

This represents your state machine.

The first step is to write an auto.py file. When running pyautomate, it looks for this file in the working directory. You can also use the --file parameter to point to a different file.

The auto.py file contains:

  • the definition of your state machine (a non-deterministic finite state machine to be exact)
  • a function which returns the initial state of the state machine
  • definitions of any transition functions you used
  • optionally, weights of transitions

Transform your diagram into YAML code that describes each state:

states = '''
- name: server stopped

  transitions:

      - action: start_server()
        to: server started

- name: server started

  transitions:

      - action: stop_server()
        to: server stopped

      - action: test_server()
        to:
            - server started
            - server passed tests
'''

The action of each transition is python code that will be executed when the transition is followed.

Note that the source state only needs to match the current state partially. For example, when the current state is ('server started', 'server passed tests'), it is clever enough to notice that it can get to ('server stopped', 'server passed tests') by looking at the transitions of ('server started',).

Tip: you can put the YAML in a separate file like so:

with open('states.yaml') as f:
    states = f.read()

In some cases you may want to use guards to limit when a transition can be done. Here's an example:

https://github.com/limyreth/pyautomate/raw/master/readme_files/guard_diagram.png

In this case you would only want to release a new version when both the client and the server tests succeed, so we'll use a guard for that, which is specified as:

states = '''

- name: not released

  transitions:

      - action: release()
        to: released last version
        guard:
            state contains:
                - server passed tests
                - client passed tests

- name: server stopped

  transitions:

      - action: start_server()
        to: server started

- name: server started

  transitions:

      - action: stop_server()
        to: server stopped

      - action: test_server()
        to:
            - server started
            - server passed tests

- name: client untested

  transitions:

      - action: test_client()
        to: client passed tests

'''

I.e. not_released will only run when the state machine's current state partially matches ('server passed tests', 'client passed tests')

The anonymous state is a nameless state. The current state always contains the anonymous state. This allows you to use it as a starting point, e.g. you can rewrite the guards example as follows:

states = '''

- transitions:

      - action: release()
        to: released last version
        guard:
            state contains:
                - server passed tests
                - client passed tests

      - action: test_client()
        to: client passed tests

- name: server stopped

  transitions:

      - action: start_server()
        to: server started

- name: server started

  transitions:

      - action: stop_server()
        to: server stopped

      - action: test_server()
        to:
            - server started
            - server passed tests

'''

This is shorter to write and is still easy to understand.

pyautomate also needs a way to figure out the start state, so we have to provide it with a get_initial_state function:

def get_initial_state():
    # omitted code that finds out whether server is stopped/started
    return ('server stopped',)

This function returns a tuple of states in which the system starts (remember that the state machine can be in multiple states at the same time).

When returning a single state, you may also return a string:

def get_initial_state():
    # omitted code that finds out whether server is stopped/started
    return 'server stopped'

Note that when using the anonymous state, you can return an empty tuple. This way the machine starts in the anonymous state:

def get_initial_state():
    return ()

Now we'll define functions for anything we used as an action:

# they don't really have to be defined here, they just have to be available
# in this namespace
from myproject.server import start_server, stop_server
from myproject.tests import runner

def test_server():
    tests_succeeded = runner.run_tests()
    if not tests_succeeded:
        raise Exception('tests failed')

Execution of actions stops at the first action that throws an exception.

You can influence pyautomation's path finding by attaching weights to actions. The greater the weight the less likely an action is to be executed/followed.

Although not useful in this example, we could hint to pyautomate that stop_server is inexpensive to execute by giving it a low weight:

weights = {
    'test_server()' : 500
}

Note that the omitted actions are assigned a default weight of 1000.

Now that auto.py is written, you can get it to run actions for you.

I like to make the following alias for pyautomate:

0alias auto http://timdiels.be/feeds/pyautomate.xml

Some examples:

# all the examples are executed from the same directory as the auto.py file
# (use --file param if you want to run elsewhere, note that pyautomate will
cd to the directory where the file is located before executing it)

# tests the server, and makes sure it's stopped afterwards
auto 'server passed tests' 'server stopped'

# you can use underscores instead of spaces
auto server_passed_tests server_stopped

# note that you don't have to specify the exact state a partial state will
# do, e.g. if we don't care if the server is running or not but just want it
# tested
auto server_passed_tests

# if you really do want an exact match, you can specify --exact to force this
auto --exact server_passed_tests server_stopped

This section documents functions that will help you write auto.py files.

You'll often want to track changes to your environment in order to find out in which state the machine currently is.

For example, you might want to know whether or not the current version has been released or not. pyautomate provides you with the trackers dict-like object for this purpose:

from pyautomate import trackers

trackers['last released version'] = get_version

def get_initial_state():
    if trackers['last released version'].has_changed:
        return 'not released'
    else:
        return 'released'

def get_version():
    # call some git commands or ...

def release()
    # omitted actual release code
    trackers['last released version'].save()

You assign a callable to a key in trackers. This callable is used to get the current value. You store the current value with save on the tracker object returned by the trackers object. When you read has_changed on the tracker, it will compare the saved value of the tracker with the current value. Saved tracker values are persisted between runs.

Often you'll want to track changes to files and directories, you can do this by combining trackers with the hash_ function:

from pyautomate import hash_, trackers

src_files = 'main.cpp folder_with_more_source'.split()

trackers['last compiled source'] = lambda: hash_(*src_files)

def get_initial_state():
    if trackers['last compiled source'].has_changed:
        return 'binaries outdated'
    return 'binaries up to date'

def make():
    # omitted compile commands
    trackers['last compiled source'].save()

hash_ hashes files and directories and returns the resulting (sha256) digest.

The above example does not take into account missing binaries, we can fix this by using files_exist:

from pyautomate import files_exist, hash_, trackers

src_files = 'main.cpp folder_with_more_source'.split()

trackers['last compiled source'] = lambda: hash_(*src_files)

def get_initial_state():
    binaries_exist = files_exist(*src_files)
    if trackers['last compiled source'].has_changed or not binaries_exist:
        return 'binaries outdated'
    return 'binaries up to date'

Note that files_exist takes both files and directories.

For a complete example of tracking file system changes see publishing a document.

When calling other programs or shell commands you may be tempted to use os.system or the like, but you shouldn't. In fact, the most convenient way to call other applications and shell commands is using subprocess.check_call:

subprocess.check_call(["ls", "-l"])

check_call will throw an exception when the subprocess' return code is not 0, this allows pyautomate to detect that the action has failed so that it can stop execution.

If you need to save data between pyautomate runs, you can use pyautomate.persisted like so:

from pyautomate import persisted

def release():
    persisted['key'] = value
    print(persisted['key'])

Keys mustn't start with '#', these are reserved for pyautomate. The data is saved in .pyautomate in the same directory as the auto.py file.

This example shows how to automate converting rst to html, and upload it to a server. It is clever enough to notice missing html, out of date html and remember if it still needs to upload.

auto.py:

from subprocess import check_call
from pyautomate import files_exist, hash_, trackers

states = '''
- name: rst
  transitions:
    - action: make()
      to: html

- name: html
  transitions:
    - action: upload()
      to: uploaded
'''

files = 'browser_based productlisting'.split()
rst_files = [file + '.rst' for file in files]
html_files = [file + '.html' for file in files]

trackers['last converted rst'] = lambda: hash_(*rst_files)
trackers['last uploaded html'] = lambda: hash_(*html_files)

def get_initial_state():
    html_exists = files_exist(*html_files)
    if trackers['last converted rst'].has_changed or not html_exists:
        return 'rst'

    if trackers['last uploaded html'].has_changed:
        return 'html'

    return 'uploaded'

def make():
    for name in files:
        check_call([
            'rst2html',
            '--stylesheet=http://timdiels.be/style.css ',
            '--link-stylesheet',
            name + '.rst',
            name + '.html'])
    trackers['last converted rst'].save()

def upload():
    args = ['scp']
    args.extend([file + '.html' for file in files])
    args.append('sin.khk.be:public_html/')
    check_call(args)
    trackers['last uploaded html'].save()

About

Automation/make tool where you describe a state machine in a config file and use the CLI to go to a certain state.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages