Combining node.js and Python

后端 未结 7 1878
迷失自我
迷失自我 2020-11-30 16:06

Node.js is a perfect match for our web project, but there are few computational tasks for which we would prefer Python. We also already have a Python code for them. We are h

7条回答
  •  误落风尘
    2020-11-30 16:32

    Update 2019

    There are several ways to achieve this and here is the list in increasing order of complexity

    1. Python Shell, you will write streams to the python console and it will write back to you
    2. Redis Pub Sub, you can have a channel listening in Python while your node js publisher pushes data
    3. Websocket connection where Node acts as the client and Python acts as the server or vice-versa
    4. API connection with Express/Flask/Tornado etc working separately with an API endpoint exposed for the other to query

    Approach 1 Python Shell Simplest approach

    source.js file

    const ps = require('python-shell')
    // very important to add -u option since our python script runs infinitely
    var options = {
        pythonPath: '/Users/zup/.local/share/virtualenvs/python_shell_test-TJN5lQez/bin/python',
        pythonOptions: ['-u'], // get print results in real-time
        // make sure you use an absolute path for scriptPath
        scriptPath: "./subscriber/",
        // args: ['value1', 'value2', 'value3'],
        mode: 'json'
    };
    
    const shell = new ps.PythonShell("destination.py", options);
    
    function generateArray() {
        const list = []
        for (let i = 0; i < 1000; i++) {
            list.push(Math.random() * 1000)
        }
        return list
    }
    
    setInterval(() => {
        shell.send(generateArray())
    }, 1000);
    
    shell.on("message", message => {
        console.log(message);
    })
    

    destination.py file

    import datetime
    import sys
    import time
    import numpy
    import talib
    import timeit
    import json
    import logging
    logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.INFO)
    
    size = 1000
    p = 100
    o = numpy.random.random(size)
    h = numpy.random.random(size)
    l = numpy.random.random(size)
    c = numpy.random.random(size)
    v = numpy.random.random(size)
    
    def get_indicators(values):
        # Return the RSI of the values sent from node.js
        numpy_values = numpy.array(values, dtype=numpy.double) 
        return talib.func.RSI(numpy_values, 14)
    
    for line in sys.stdin:
        l = json.loads(line)
        print(get_indicators(l))
        # Without this step the output may not be immediately available in node
        sys.stdout.flush()
    

    Notes: Make a folder called subscriber which is at the same level as source.js file and put destination.py inside it. Dont forget to change your virtualenv environment

提交回复
热议问题