Python 异步调用命令行工具
gemky
8年前
<p>当你在自己的 Python 程序中采用了基于事件循环的异步编程方法之后,你就会发现自己不自觉地被其牢牢吸引住,并不是说这一方法多么棒,而是因为你不得不想办法保证程序中的任意环节都不能是阻塞的!</p> <p>例如当前的场景是希望从 MongoDB 中读取每一条未处理过的数据,下载并保存其中的图片信息,然后更新数据库的内容。Python 常用的 MongoDB 异步驱动是 Motor :</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/8f8ccd817acbd2f0a6f289ad44b59465.jpg"></p> <p>结合 asyncio 使用方法如下:</p> <pre> <code class="language-python">importmotor.motor_asyncio importasyncio client = motor.motor_asyncio.AsyncIOMotorClient() db = client.test_database asyncdefrun(): asyncfor mmin db.test_database.find({"status": 0}): print(mm['img_src']) # Download Image Here # dl_img(mm['img_src']) awaitdb.test_database.update({"_id": mm['_id']}, {"$set": {"status":1}}) loop = asyncio.get_event_loop() loop.run_until_complete(run()) </code></pre> <p>此时如果 dl_img() 处的操作是阻塞的,那么异步处理就没有意义了。当然这里依然可以借助异步网络请求库 aiohttp 来实现图片下载:</p> <pre> <code class="language-python">asyncwithsession.get(img) as resp: withopen(img.split("/")[-1], 'wb') as fd: while True: chunk = awaitresp.content.read(1024) if not chunk: break fd.write(chunk) </code></pre> <p>当然也可以不需要自己动手下载,直接调用系统命令行工具(例如 wget )来完成下载任务。Python 通过 <a href="/misc/goto?guid=4959726165325410122" rel="nofollow,noindex"> subprocess </a> 标准库实现系统命令调用(取代旧的 os.system(cmd) ),执行下载任务只需要:</p> <pre> <code class="language-python">importsubprocessas sb sb.run(['wget', img], shell=True) </code></pre> <p>但是这种调用方式是无法直接在 asyncio 的事件循环中使用的,但是 asyncio 提供了对应的 subprocess 接口 :</p> <pre> <code class="language-python">asyncio.create_subprocess_exec(*args, ...) asyncio.create_subprocess_shell(cmd, ...) </code></pre> <p>这两个方法均返回一个 asyncio.subprocess.Process 实例,而它的接口设计完全模仿了 subprocess.Popen (上面提到 subprocess.run() 的底层实现),因此很容易将其用法移植到事件循环中:</p> <pre> <code class="language-python">asyncdefdl_img(src): dl = awaitasyncio.create_subprocess_shell('wget {} -O {}'.format(src, src.split("/")[-1]) awaitdl.wait() </code></pre> <p>除了上面场景中的用法,也可以直接将命令行的执行作为任务放入事件循环:</p> <pre> <code class="language-python">loop = asyncio.get_event_loop() sb = asyncio.create_subprocess_shell('exit 7', loop=loop) proc = loop.run_until_complete(sb) exitcode = loop.run_until_complete(proc.wait()) </code></pre> <h3><strong>小结</strong></h3> <p>在 Python 异步编程的意义就在于不要让 CPU 堵在 IO 上,因此需要在每一处涉及到阻塞的操作都需要注意使用正确的异步方法,而一旦这些操作被封装成异步的 Task 之后,其后续的调度执行就无需再顾虑了。</p> <p> </p> <p> </p> <p> </p> <p>来自:http://python.jobbole.com/86829/</p> <p> </p>