在Windows下作为子进程运行Flask

2024-06-06 03:53:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用python编写一个与远程PowerShell代理交互的工具(基本上是PowerShell帝国,但用于psv1)。我想使用HTTP(S)作为代理和用户界面之间的命令和输出的传输媒介。你知道吗

为此,我让Flask作为子进程运行,然后在主线程中向用户显示一个简单的命令行。这在Linux上运行良好,但在Windows下,我在Process上调用start时遇到以下错误:

TypeError("can't pickle _thread.lock objects")

我相信这与Windows如何处理Process内的线程有关。有人能建议一种方法让它在Windows上工作吗?我的代码如下:

from flask import Flask, request
from multiprocessing import Process, Queue
import logging

port = 8080 # TODO: set this from cmdline args

cmds = Queue()
quitting = Queue()

app = Flask(__name__)
@app.route('/', methods=['GET'])
def get_command():
    if cmds.empty():
        return ''
    else:
        return cmds.get()

@app.route('/', methods=['POST'])
def get_output():
    if request.data.decode() == "quitting":
        quitting.put(True)
    print("{} ".format(request.data.decode()),end='\n> ')
    return ''

if __name__ == "__main__":
    log = logging.getLogger('werkzeug')
    log.setLevel(logging.ERROR)

    t = Process(target=app.run, kwargs={'host': '0.0.0.0'})
    t.start()

    running = True
    while running:
        cmd = input("> ")
        if cmd == "kill" or cmd =="quit":
            print("Waiting for agent to close...")
            cmds.put("quit")
            while quitting.empty():
                pass
            t.terminate()
            t.join()
            running = False
        else:
            cmds.put(cmd)

Tags: fromimportcmdappflaskgetreturnif