Przeglądaj źródła

Merge branch 'pdf_fix'

zwxi 7 miesięcy temu
rodzic
commit
08f5f96544
1 zmienionych plików z 12 dodań i 8 usunięć
  1. 12 8
      services/smart_report.go

+ 12 - 8
services/smart_report.go

@@ -142,23 +142,27 @@ import asyncio
 from pyppeteer import launch
 
 @asyncio.coroutine
-def main():
+async def main():
     # 异步代码
-    browser = yield from launch({
+    browser = await launch({
         'executablePath': '%s',
         'headless': True,
         'args': ['--disable-infobars', '--no-sandbox']
     })
-    page = yield from browser.newPage()
-    yield from page.setViewport({
+    page = await browser.newPage()
+    await page.setViewport({
         'width': 1920,
         'height': 1080,
     })
-    yield from page.goto('%s', {
+    await page.goto('%s', {
         'waitUntil': 'networkidle0',
         'timeout': 1000000  # 设置超时时间为 100 秒
     })
-    yield from page.pdf({
+
+    # 在生成PDF之前等待2秒
+    await asyncio.sleep(5)
+
+    await page.pdf({
         'path': "%s",
         'printBackground': True,
         'format': "A2",
@@ -169,7 +173,7 @@ def main():
             'right': '10mm'
         }
     })
-    yield from browser.close()
+    await browser.close()
 
 # 创建事件循环
 loop = asyncio.get_event_loop()
@@ -230,7 +234,7 @@ async def main():
         # Customizing footer for page numbers starting from page 2
 
         # 在这里添加两秒的等待
-        await asyncio.sleep(2)
+        await asyncio.sleep(5)
 
         await page.screenshot({
             'path': "%s",