Mara Pipelines

轻量级意见化 ETL 框架,介于普通脚本和 Apache Airflow 之间。「A lightweight opinionated ETL framework, halfway between plain scripts and Apache Airflow」

Github星跟踪图

Mara Pipelines

Build & Test
PyPI - License
PyPI version
Slack Status

This package contains a lightweight data transformation framework with a focus on transparency and complexity reduction. It has a number of baked-in assumptions/ principles:

  • Data integration pipelines as code: pipelines, tasks and commands are created using declarative Python code.

  • PostgreSQL as a data processing engine.

  • Extensive web ui. The web browser as the main tool for inspecting, running and debugging pipelines.

  • GNU make semantics. Nodes depend on the completion of upstream nodes. No data dependencies or data flows.

  • No in-app data processing: command line tools as the main tool for interacting with databases and data.

  • Single machine pipeline execution based on Python's multiprocessing. No need for distributed task queues. Easy debugging and output logging.

  • Cost based priority queues: nodes with higher cost (based on recorded run times) are run first.

 

Installation

To use the library directly, use pip:

pip install mara-pipelines

or

pip install git+https://github.com/mara/mara-pipelines.git

For an example of an integration into a flask application, have a look at the mara example project 1 and mara example project 2.

Due to the heavy use of forking, Mara Pipelines does not run natively on Windows. If you want to run it on Windows, then please use Docker or the Windows Subsystem for Linux.

 

Example

Here is a pipeline "demo" consisting of three nodes that depend on each other: the task ping_localhost, the pipeline sub_pipeline and the task sleep:

from mara_pipelines.commands.bash import RunBash
from mara_pipelines.pipelines import Pipeline, Task
from mara_pipelines.cli import run_pipeline, run_interactively

pipeline = Pipeline(
    id='demo',
    description='A small pipeline that demonstrates the interplay between pipelines, tasks and commands')

pipeline.add(Task(id='ping_localhost', description='Pings localhost',
                  commands=[RunBash('ping -c 3 localhost')]))

sub_pipeline = Pipeline(id='sub_pipeline', description='Pings a number of hosts')

for host in ['google', 'amazon', 'facebook']:
    sub_pipeline.add(Task(id=f'ping_{host}', description=f'Pings {host}',
                          commands=[RunBash(f'ping -c 3 {host}.com')]))

sub_pipeline.add_dependency('ping_amazon', 'ping_facebook')
sub_pipeline.add(Task(id='ping_foo', description='Pings foo',
                      commands=[RunBash('ping foo')]), ['ping_amazon'])

pipeline.add(sub_pipeline, ['ping_localhost'])

pipeline.add(Task(id='sleep', description='Sleeps for 2 seconds',
                  commands=[RunBash('sleep 2')]), ['sub_pipeline'])

Tasks contain lists of commands, which do the actual work (in this case running bash commands that ping various hosts).

 

In order to run the pipeline, a PostgreSQL database is recommended to be configured for storing run-time information, run output and status of incremental processing:

import mara_db.auto_migration
import mara_db.config
import mara_db.dbs

mara_db.config.databases \
    = lambda: {'mara': mara_db.dbs.PostgreSQLDB(host='localhost', user='root', database='example_etl_mara')}

mara_db.auto_migration.auto_discover_models_and_migrate()

Given that PostgresSQL is running and the credentials work, the output looks like this (a database with a number of tables is created):

Created database "postgresql+psycopg2://root@localhost/example_etl_mara"

CREATE TABLE data_integration_file_dependency (
    node_path TEXT[] NOT NULL,
    dependency_type VARCHAR NOT NULL,
    hash VARCHAR,
    timestamp TIMESTAMP WITHOUT TIME ZONE,
    PRIMARY KEY (node_path, dependency_type)
);

.. more tables

CLI UI

This runs a pipeline with output to stdout:

from mara_pipelines.cli import run_pipeline

run_pipeline(pipeline)

Example run cli 1

 

And this runs a single node of pipeline sub_pipeline together with all the nodes that it depends on:

run_pipeline(sub_pipeline, nodes=[sub_pipeline.nodes['ping_amazon']], with_upstreams=True)

Example run cli 2

 

And finally, there is some sort of menu based on pythondialog that allows to navigate and run pipelines like this:

from mara_pipelines.cli import run_interactively

run_interactively()

Example run cli 3

Web UI

More importantly, this package provides an extensive web interface. It can be easily integrated into any Flask based app and the mara example project demonstrates how to do this using mara-app.

For each pipeline, there is a page that shows

  • a graph of all child nodes and the dependencies between them
  • a chart of the overal run time of the pipeline and it's most expensive nodes over the last 30 days (configurable)
  • a table of all the pipeline's nodes with their average run times and the resulting queuing priority
  • output and timeline for the last runs of the pipeline

Mara pipelines web ui 1

For each task, there is a page showing

  • the upstreams and downstreams of the task in the pipeline
  • the run times of the task in the last 30 days
  • all commands of the task
  • output of the last runs of the task

Mara pipelines web ui 2

Pipelines and tasks can be run from the web ui directly, which is probably one of the main features of this package:

Example run web ui

 

Getting started

Documentation is currently work in progress. Please use the mara example project 1 and mara example project 2 as a reference for getting started.

主要指标

概览
名称与所有者mara/mara-pipelines
主编程语言Python
编程语言Python (语言数: 5)
平台
许可证MIT License
所有者活动
创建于2018-03-31 20:37:22
推送于2023-12-15 16:14:47
最后一次提交2022-04-01 10:20:01
发布数40
最新版本名称3.5.0 (发布于 )
第一版名称1.0.0 (发布于 )
用户参与
星数2.1k
关注者数53
派生数100
提交数172
已启用问题?
问题数34
打开的问题数18
拉请求数56
打开的拉请求数8
关闭的拉请求数10
项目设置
已启用Wiki?
已存档?
是复刻?
已锁定?
是镜像?
是私有?