我正在使用Celery来自动化一些屏幕抓取.我正在使用Selenium打开Chrome webdriver,操作页面,保存一些数据,然后转到队列中的下一页.问题在于它为队列中的每个任务构建和分解Web驱动程序,这非常耗时且占用大量资源.
如何在通话中保留对象?我已经阅读了有关Celery中连接池的一些内容,但我不清楚这是如何工作的 – 我在哪里构建webdriver – 在任务文件或主排队文件中?如果是后者,工人们如何知道使用哪个webdriver?
例:
scrape.py:
for page in list: scrape.delay(str(row['product_id']),str(row['pg_code']))
tasks.py:
def scrape: # do some stuff
解决方法
由于每个worker将任务实例化为单例,因此您可以将Web驱动程序缓存在任务对象中.文档特别提出了这种方法.
http://docs.celeryproject.org/en/latest/userguide/tasks.html#instantiation