Skip to main content

Simple, Event-driven Automation in Python using Elasticsearch data to inform automation decisions

Project description

Codeship Status for jgericke/automaton Docs License: MIT Code style codecov

Automaton Engine

Simple, Event-driven Automation in Python

Automaton Engine makes use of Elasticsearch as a general data aggregation layer. You feed your Automaton(s) queries via environment variables (formatted in JSON). Based on the query result, you can chain actions together which Automaton Engine will act on, passing query metadata off to action handlers and generally having a good time.

Basically Automaton Engine is all about the following:

  1. Query Elasticsearch on a defined schedule
  2. Upon receiving a query response, execute defined actions and wait for a defined backoff period
  3. Return to querying
  4. ???
  5. Profit!

Current Actions

  • notify - Send a Rocket Chat webhook.
  • awx - Calls an Ansible AWX API endpoint.

Installation

Install via pip:

$ pip install automaton-engine

Build from source:

$ git clone https://github.com/jgericke/automaton_engine.git
$ python setup.py install

Run tests:

$ python setup.py test

Example Workflow

1. Create an Elasticsearch query to read the number of orders being processed (assuming you have some sort of order system thats sending metrics to Elasticsearch...)
2. Create an Ansible AWX job template to scale up pods running in Kubernetes based (using say, extra_parameters or something)
3. Create an Automaton Engine instance that runs the above query, with an action to call the AWX job template based on query result
4. Good gravy, your pods are scaling on a transactional metric!

Automaton Features

  • Runs on python 3.7
  • Fully asynchronous.
  • Extendable (actions are functions created under automaton/actions).
  • Not your Grandad's Automaton Engine.

Adding Actions to Automaton Engine

  • Create a new action file: automaton_engine/actions/my_action.py
  • Actions are defined as a list within your automaton environment variable, and can be chained, for example the below action fires off a rocket chat webhook (this forms part of your AUTOMATON_ENGINE_CONFIG environment variable):
"actions": [
    {
        "name": "notify.rocketchat_webhook",
        "backoff_seconds": 300, 
        "parameters": {
            "rocketchat_webhook": "https://my.rocketchat/hooks/my_webhook_id",
            "rocketchat_message": "@here Things are happening!\nView dashboard: https://my.kibana/goto/my_dashboard_id",
            "rocketchat_timeout": 20
        }
    }
]
  • When adding or extending an action, you need to make Automaton aware of the new action by adding it to automaton/automaton.py within the action_dispatcher (This will be simplified in the future). The below represents the action file and function name for actions/notify.py.
action_dispatcher = {
    "notify.rocketchat_webhook": notify.rocketchat_webhook 
}

Creating An Automaton

You can launch Automaton Engine by defining the AUTOMATON_ENGINE_CONFIG environment variable, which has an expected structure as outlined below. When you're good to go, you can execute samples/auto.py which will gather all of the automatons you've defined and run them through the execution chain.

Have a look at samples provided (samples/) to get a feel for scaffolding Automatons

export AUTOMATON_ENGINE_LOGLEVEL='DEBUG'
export AUTOMATON_ENGINE_CONFIG='{ 
    "automatons": [
        {
            "name": "my_neat_automaton",
            "enabled": True,
            "runonce": False,
            "elasticsearch": {
                "url": "https://http://my.elasticsearch:9200",
                "timeout": 10,
                "auth": {
                    "username": "es_user",
                    "password": "es_pass",
                },
            },
            "elasticsearch_query": {
                "query_interval": 5,
                "query_endpoint": "/_search",
                "query_type": "aggregations",
                "query_name": "my_automaton_query",
                "query_payload": {
						"size": 0,
						"query": {
						    "range": {
                                "@timestamp": {
                                    "gte": "now-1m"
                                }
                            }
						},
						"aggregations": {
						    "bad_actors": {
                                "terms": {
                                    "field": "my_automaton_query.keyword",
                                    "min_doc_count": 1000
                                }
						    }
						}
                },
                "query_response_mapping": { 
                        "key": "go_automaton_go", 
                        "doc_count": "hits"
                }
            },
            "actions": [
                {
                    "name": "notify.rocketchat_webhook",
                    "backoff_seconds": 60,
                    "parameters": {
                        "rocketchat_webhook": "https://my.rocketchat/hooks/my_webhook_id",
                        "rocketchat_message": "@here Things are happening, Imma do stuff!",
                        "rocketchat_timeout": 10
                    }
                },
                {
                    "name": "awx.api_call",
                    "backoff_seconds": 60,
                    "parameters": {
                        "awx_url": "https://my.awx.url",
                        "awx_context": "/api/v2/job_templates/1/launch/",
                        "awx_timeout": 20,
                        "awx_auth": {
                            "username": "YXV0b21hdG9u",
                            "password": "ZDY5VG5vZUdZYjdU"
                        }
                    }
                }
            ]
        }
    ]
}'

Elasticsearch Authentication

Automaton now supports basic authentication for elasticsearch, to configure it add an "auth" section to your automaton configuration as per below:

            "elasticsearch": {
                "url": "https://http://my.elasticsearch:9200",
                "timeout": 10,
                "auth": {
                    "username": "es_user",
                    "password": "es_pass",
                },
            },

No elasticsearch authentication? No problem! Simply remove the auth stanza:

            "elasticsearch": {
                "url": "https://http://my.elasticsearch:9200",
                "timeout": 10,
            },

Original Author(s)

Julian Gericke
(c) LSD
julian@lsd.co.za
Checkout LSD, your friendly OSS DevOps company!

Contributors

  • Maybe you?!

Project details


Download files

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

Files for automaton-engine, version 1.0.2
Filename, size File type Python version Upload date Hashes
Filename, size automaton_engine-1.0.2-py2.py3-none-any.whl (10.8 kB) File type Wheel Python version py2.py3 Upload date Hashes View hashes
Filename, size automaton_engine-1.0.2.tar.gz (9.6 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page