需求解析功能完善

This commit is contained in:
2025-04-30 17:44:52 +08:00
parent 4faed52de5
commit 9a716bb730
6 changed files with 255 additions and 112 deletions

View File

@@ -234,7 +234,7 @@ class UploadController(ControllerBase):
# 上传需求规格说明.docx进行解析
@route.post("/upload_xq_docx/", url_name='dut-xq-docx')
def upload_xq_docx(self, dut_key: str, project_id: int, file: File[UploadedFile]):
def upload_xq_docx(self, parseChapter: str, file: File[UploadedFile]):
# 构建临时目录
with tempfile.TemporaryDirectory() as tmp_dir:
# 保存到临时目录
@@ -242,5 +242,5 @@ class UploadController(ControllerBase):
with open(docx_path, 'wb') as f:
for chunk in file.chunks():
f.write(chunk)
extractor = DocxChapterExtractor(docx_path)
extractor.main('需求')
extracter = DocxChapterExtractor(docx_path)
return extracter.main(parseChapter)