Skip to content

Latest commit

 

History

History
249 lines (181 loc) · 7.57 KB

README.ts.md

File metadata and controls

249 lines (181 loc) · 7.57 KB

python-bridge Build Status Build Status

Most robust and simple Python bridge. Features, and comparisons to other Python bridges below, supports Windows.

API for TypeScript

View documentation with JavaScript examples.

npm install python-bridge
import assert from 'assert';
import { pythonBridge } from 'python-bridge';

async function main() {
    const python = pythonBridge();

    await python.ex`import math`;
    const x = await python`math.sqrt(9)`;
    assert.equal(x, 3);

    const list = [3, 4, 2, 1];
    const sorted = await python`sorted(${list})`;
    assert.deepEqual(sorted, list.sort());

    await python.end();
}

main().catch(console.error);

var python = pythonBridge(options)

Spawns a Python interpreter, exposing a bridge to the running processing. Configurable via options.

  • options.python - Python interpreter, defaults to python

Also inherits the following from child_process.spawn([options]).

  • options.cwd - String Current working directory of the child process
  • options.env - Object Environment key-value pairs
  • options.stdio - Array Child's stdio configuration. Defaults to ['pipe', process.stdout, process.stderr]
  • options.uid - Number Sets the user identity of the process.
  • options.gid - Number Sets the group identity of the process.
const python = pythonBridge({
    python: 'python3',
    env: {PYTHONPATH: '/foo/bar'}
});

python`expression(args...)`.then(...)

Evaluates Python code, returning the value back to Node.

// Interpolates arguments using JSON serialization.
assert.deepEqual([1, 3, 4, 6], await python`sorted(${[6, 4, 1, 3]})`);

// Passing key-value arguments
const obj = {hello: 'world', foo: 'bar'};
assert.deepEqual(
    {baz: 123, hello: 'world', foo: 'bar'},
    await python`dict(baz=123, **${obj})`
);

python.ex`statement`.then(...)

Execute Python statements.

const a = 123, b = 321;
python.ex`
    def hello(a, b):
        return a + b
`;
assert.equal(a + b, await python`hello(${a}, ${b})`);

python.lock(...).then(...)

Locks access to the Python interpreter so code can be executed atomically. If possible, it's recommend to define a function in Python to handle atomicity.

const x: number = await python.lock(async python =>{
    await python.ex`hello = 123`;
    return await python`hello + 321`;
});
assert.equal(x, 444);

// Recommended to define function in Python
await python.ex`
    def atomic():
        hello = 123
        return hello + 321
`;
assert.equal(444, await python`atomic()`);

python.stdin, python.stdout, python.stderr

Pipes going into the Python process, separate from execution & evaluation. This can be used to stream data between processes, without buffering.

import { delay, promisifyAll } from 'bluebird';
const { createWriteStream, readFileAsync } = promisifyAll(require('fs'));

const fileWriter = createWriteStream('hello.txt');
python.stdout.pipe(fileWriter);

// listen on Python process's stdout
const stdinToStdout = python.ex`
    import sys
    for line in sys.stdin:
        sys.stdout.write(line)
        sys.stdout.flush()
`;

// write to Python process's stdin
python.stdin.write('hello\n');
await delay(10);
python.stdin.write('world\n');

// close python's stdin, and wait for python to finish writing
python.stdin.end();
await stdinToStdout;

// assert file contents is the same as what was written
const fileContents = await readFileAsync('hello.txt', {encoding: 'utf8'});
assert.equal(fileContents.replace(/\r/g, ''), 'hello\nworld\n');

python.end()

Stops accepting new Python commands, and waits for queue to finish then gracefully closes the Python process.

python.disconnect()

Alias to python.end()

python.kill([signal])

Send signal to Python process, same as child_process child.kill.

import { TimeoutError } from 'bluebird';

let python = pythonBridge();

try {
    await python.ex`
        from time import sleep
        sleep(9000)
    `.timeout(100);
    assert.ok(false); // should not reach this
} catch (e) {
    if (e instanceof TimeoutError) {
        python.kill('SIGKILL');
        python = pythonBridge();
    } else {
        throw e;
    }
}
python.end();

Handling Exceptions

We can use Bluebird's promise.catch(...) catch handler in combination with Python's typed Exceptions to make exception handling easy.

python.Exception

Catch any raised Python exception.

python.ex`
    hello = 123
    print(hello + world)
    world = 321
`.catch(python.Exception, () => console.log('Woops!  `world` was used before it was defined.'));

python.isException(name)

Catch a Python exception matching the passed name.

import { isPythonException } from 'python-bridge';

async function pyDivide(numerator, denominator) {
    try {
        await python`${numerator} / ${denominator}`;
    } catch (e) {
        if (isPythonException('ZeroDivisionError', e)) {
            return Infinity;
        }
        throw e;
    }
}

async function main() {
    assert.equal(Infinity, await pyDivide(1, 0));
    assert.equal(1 / 0, await pyDivide(1, 0));
}

main().catch(console.error);

pythonBridge.PythonException

Alias to python.Exception, this is useful if you want to import the function to at the root of the module.

pythonBridge.isPythonException

Alias to python.isException, this is useful if you want to import the function to at the root of the module.


Features

  • Does not affect Python's stdin, stdout, or stderr pipes.
  • Exception stack traces forwarded to Node for easy debugging.
  • Python 2 & 3 support, end-to-end tested.
  • Windows support, end-to-end tested.
  • Command queueing, with promises.
  • Long running Python sessions.
  • ES6 template tags for easy interpolation & multiline code.

Comparisons

After evaluating of the existing landscape of Python bridges, the following issues are why python-bridge was built.

  • python-shell — No promises for queued requests; broken evaluation parser; conflates evaluation and stdout; complex configuration.
  • python — Broken evaluation parsing; no exception handling; conflates evaluation, stdout, and stderr.
  • node-python — Complects execution protocol with incomplete Python embedded DSL.
  • python-runner — No long running sessions; child_process.spawn wrapper with unintuitive API; no serialization.
  • python.js — Embeds specific version of CPython; requires compiler and CPython dev packages; incomplete Python embedded DSL.
  • cpython — Complects execution protocol with incomplete Python embedded DSL.
  • eval.py — Can only evaluate single line expressions.
  • py.js — For setting up virtualenvs only.

License

MIT