从脚本而不是从Flask路由调用时,Asyncio函数有效

时间:2019-08-23 15:25:19

标签: python python-3.x flask python-asyncio

我是Python和这些库/模块的新手。我正在编写一个简单的ping测试网络扫描仪作为一个学习项目。

我首先使用asyncio开发了一个脚本来ping网络上的地址

#ip_test.py
import asyncio
import ipaddress

async def ping(addr):
    proc = await asyncio.create_subprocess_exec(
        'ping','-W','1','-c','3',addr,
        stdout=asyncio.subprocess.PIPE
    )
    await proc.wait()
    return proc.returncode

async def pingMain(net):
    #hosts() returns list of Ipv4Address objects
    result = await asyncio.gather(*(ping(str(addr)) for addr in net.hosts()))
    return result

def getHosts(net_): #net_ is an Ipv4Network object
    return asyncio.run(pingMain(net_))
    #Returns list of response codes which I then zip with the list of searched ips

当我打开python并运行以下命令时,它会按预期工作:

import ip_test as iptest
import ipaddress
print(iptest.getHosts(ipaddress.ip_network('192.168.1.0/29')))
#prints: [0, 0, 0, 1, 1, 1] as expected on this network

但是,最终目标是通过表单输入从用户那里获取输入(结果记录到数据库中,这是出于说明目的的简化示例)。我通过烧瓶路径收集输入:

@app.route("/newscan",methods=['POST'])
def newScan():
    form = request.form
    networkstring = form.get('network') + "/" + form.get('mask')
    result = iptest.getHosts(ipaddress.ip_network(networkstring))
    return result

以这种方式调用模块时,出现错误:Runtime Error: Cannot add child handler, the child watcher does not have a loop attached.

为什么当我从命令行导入模块并运行该函数时却无法正常工作,而当我从烧瓶路径中使用相同的输入调用它时却为什么不起作用?

编辑:回溯:

Traceback (most recent call last):
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 2463, in __call__
  return self.wsgi_app(environ, start_response)
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 2449, in wsgi_app
  response = self.handle_exception(e)
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 1866, in handle_exception
  reraise(exc_type, exc_value, tb)
File "/usr/local/lib/python3.7/site-packages/flask/_compat.py", line 39, in reraise
  raise value
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 2446, in wsgi_app
  response = self.full_dispatch_request()
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 1951, in full_dispatch_request
  rv = self.handle_user_exception(e)
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 1820, in handle_user_exception
  reraise(exc_type, exc_value, tb)
File "/usr/local/lib/python3.7/site-packages/flask/_compat.py", line 39, in reraise
  raise value
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 1949, in full_dispatch_request
  rv = self.dispatch_request()
File "/usr/local/lib/python3.7/site-packages/flask/app.py", line 1935, in dispatch_request
  return self.view_functions[rule.endpoint](**req.view_args)
File "/app/app.py", line 41, in newScan
  result = iptest.getHosts(ipaddress.ip_network(networkstring))
File "/app/ip_test.py", line 22, in getHosts
  res = asyncio.run(pingMain(net_))
File "/usr/local/lib/python3.7/asyncio/runners.py", line 43, in run
  return loop.run_until_complete(main)
File "/usr/local/lib/python3.7/asyncio/base_events.py", line 579, in run_until_complete
  return future.result()
File "/app/ip_test.py", line 15, in pingMain
  result = await asyncio.gather(*(ping(str(addr)) for addr in net.hosts()))
File "/app/ip_test.py", line 7, in ping
  stdout=asyncio.subprocess.PIPE
File "/usr/local/lib/python3.7/asyncio/subprocess.py", line 217, in create_subprocess_exec
  stderr=stderr, **kwds)
File "/usr/local/lib/python3.7/asyncio/base_events.py", line 1529, in subprocess_exec
  bufsize, **kwargs)
File "/usr/local/lib/python3.7/asyncio/unix_events.py", line 193, in _make_subprocess_transport
  self._child_watcher_callback, transp)
File "/usr/local/lib/python3.7/asyncio/unix_events.py", line 930, in add_child_handler
  "Cannot add child handler, "
RuntimeError: Cannot add child handler, the child watcher does not have a loop attached

1 个答案:

答案 0 :(得分:2)

您正在尝试从主线程以外的其他线程运行异步子进程。这需要从主线程进行一些初始设置,请参见asyncio Subprocesses 文档的Subprocesses and Threads section

  

标准asyncio事件循环支持从不同线程运行子进程,但存在局限性:

     
      
  • 事件循环必须在主线程中运行。
  •   
  • 在从其他线程执行子进程之前,必须在主线程中实例化子监视程序。在主线程中调用get_child_watcher()函数以实例化儿童观察者。
  •   

这里发生的是您的WSGI服务器正在使用多个线程来处理传入的请求,因此请求处理程序不在 main 线程上运行。但是您的代码使用asyncio.run()来启动新的事件循环,因此您的asyncio.create_subprocess_exec()调用将失败,因为主线程上没有子监视程序。

您必须从主线程开始循环(而不是停止循环),并在该线程上调用asyncio.get_child_watcher(),以使代码不会失败:

# to be run on the main thread, set up a subprocess child watcher
assert threading.current_thread() is threading.main_thread()
asyncio.get_event_loop()
asyncio.get_child_watcher()

注意:此限制仅适用于Python 3.7以下的Python版本,即限制has been lifted in Python 3.8

但是,仅运行一堆子流程并等待它们完成,使用asyncio实在是太过分了;您的操作系统可以并行运行子进程。只需使用subprocess.Popen()并通过Popen.poll() method检查每个过程:

import subprocess

def ping_proc(addr):
    return subprocess.Popen(
        ['ping', '-W', '1', '-c', '3', addr],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL
    )

def get_hosts(net):
    # hosts() returns list of Ipv4Address objects
    procs = [ping_proc(str(addr)) for addr in net.hosts()]
    while any(p.poll() is None for p in procs):
        time.sleep(0.1)
    return [p.returncode for p in procs]

Popen.poll()不会阻止;如果尚未设置Popen.returncode,它将使用waitpid([pid], WNOHANG)在操作系统中检查进程状态,如果进程仍在运行,则返回None或现在可用的returncode值。上面的代码只是在循环中检查这些状态,并且之间要短暂睡眠,以免发生抖动。

asyncio子进程包装器(至少在POSIX上)使用SIGCHLD信号处理程序来通知子进程退出,或者(在Python 3.8中)为每个子进程使用单独的线程来使用在创建的每个子流程上阻止waitpid()调用。您可以实现相同的信号处理程序,但是要考虑到信号处理程序只能在主线程上注册,因此您必须跳过多个步骤,以将传入的SIGCHLD信号信息传递到正确的线程。 / p>