Простое сетевое UDP-прослушивание во фляге или пирамиде

Question

Мне нужно создать веб-приложение, которое отображает данные, предоставляемые через периодические входящие пакеты UDP. Сайт, вероятно, будет находиться во Flask (возможно, в Pyramid), развернутом под Nginx. Как я могу создать очень простую фоновую задачу (в основном просто socket.recv()), чтобы прослушивать любые входящие пакеты и помещать данные в глобально доступные списки?

Могу ли я просто создать поток из main() для этого или мне нужно использовать что-то вроде Celery или PyRes?

Спасибо за любое руководство.

Answer 1

Вам придется использовать сельдерей, но вам повезло, потому что уже есть расширение колбы, которое интегрирует сельдерей. вам придется pip install flask, pip install flask-celery, pip install redis и вам понадобится сервер Redis в вашей системе.

import socket, select, Queue

from flask import Flask
from celery import Celery


def make_celery(app):
    celery = Celery(app.import_name, broker=app.config['CELERY_BROKER_URL'])
    celery.conf.update(app.config)
    TaskBase = celery.Task
    class ContextTask(TaskBase):
        abstract = True
        def __call__(self, *args, **kwargs):
            with app.app_context():
                return TaskBase.__call__(self, *args, **kwargs)
    celery.Task = ContextTask
    return celery

app = Flask(__name__)
app.config.update(
    CELERY_BROKER_URL='redis://localhost:6379',
    CELERY_RESULT_BACKEND='redis://localhost:6379'
)
celery = make_celery(app)
socket_queue = Queue.Queue()


@celery.task()
def listen_to_udp():
    """
    This code was taken from 
    https://stackru.com/questions/9969259/python-raw-socket-listening-for-udp-packets-only-half-of-the-packets-received
    """
    s1 = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s1.bind(('0.0.0.0', 1337))
    s2 = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_UDP)
    s2.bind(('0.0.0.0', 1337))
    while True:
        r, w, x = select.select([s1, s2], [], [])
        for i in r:
            socket_queue.put((i, i.recvfrom(131072)))

@app.route("/")
def test_home():
    listen_to_udp.delay()
    print(socket_queue.get())

if __name__ == "__main__":
    #run install.py to install dependencies and create the database
    app.run(host="0.0.0.0", port=5000, debug=True)