fix enqueque
Showing
2 changed files
with
2 additions
and
3 deletions
... | @@ -89,7 +89,6 @@ class Command(BaseCommand, LoggerMixin): | ... | @@ -89,7 +89,6 @@ class Command(BaseCommand, LoggerMixin): |
89 | base64_data = base64.b64encode(f.read()) | 89 | base64_data = base64.b64encode(f.read()) |
90 | return {'imgBase64': base64_data.decode('utf-8')} | 90 | return {'imgBase64': base64_data.decode('utf-8')} |
91 | 91 | ||
92 | |||
93 | async def fetch_ocr_result(self, img_path): | 92 | async def fetch_ocr_result(self, img_path): |
94 | async with aiohttp.ClientSession( | 93 | async with aiohttp.ClientSession( |
95 | headers=self.ocr_header, connector=aiohttp.TCPConnector(ssl=False) | 94 | headers=self.ocr_header, connector=aiohttp.TCPConnector(ssl=False) | ... | ... |
... | @@ -186,12 +186,12 @@ class DocView(GenericView, DocHandler): | ... | @@ -186,12 +186,12 @@ class DocView(GenericView, DocHandler): |
186 | data_source=random.choice(['POS', 'EAPP', 'Econtract']), | 186 | data_source=random.choice(['POS', 'EAPP', 'Econtract']), |
187 | upload_finish_time=timezone.now(), | 187 | upload_finish_time=timezone.now(), |
188 | ) | 188 | ) |
189 | rh.enqueue(doc.id) | 189 | enqueue_res = rh.enqueue(doc.id) |
190 | 190 | ||
191 | pdf_file = args.get('pdf_file') | 191 | pdf_file = args.get('pdf_file') |
192 | save_dir_path = os.path.join(conf.DATA_DIR, str(doc.id)) | 192 | save_dir_path = os.path.join(conf.DATA_DIR, str(doc.id)) |
193 | save_file_path = os.path.join(save_dir_path, '{0}.pdf'.format(doc.id)) | 193 | save_file_path = os.path.join(save_dir_path, '{0}.pdf'.format(doc.id)) |
194 | os.makedirs(save_dir_path, exist_ok=True) | 194 | os.makedirs(save_dir_path, exist_ok=True) |
195 | file_write(pdf_file, save_file_path) | 195 | file_write(pdf_file, save_file_path) |
196 | self.running_log.info('[mock doc upload success] [doc_id={0}]'.format(doc.id)) | 196 | self.running_log.info('[mock doc upload success] [doc_id={0}] [enqueue_res={1}]'.format(doc.id, enqueue_res)) |
197 | return response.ok() | 197 | return response.ok() | ... | ... |
-
Please register or sign in to post a comment