重构简历优化

2026-06-23 18:21:30 +08:00
parent 1edc28e332
commit 2f38c80207
11 changed files with 99 additions and 113 deletions
@@ -0,0 +1,114 @@
+"""简历 Service
+
+上传简历文件 → 解析为纯文本 → AI 两阶段并行结构化 → 写入数据库。
+依赖：file_parser（文件解析工具）、resume_extractor（AI两阶段并行提取）
+使用表：bg_user_resume（主表）、bg_user_resume_education/work/internship/project/competition（5张子表）
+"""
+
+import asyncio
+
+import shortuuid
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.ai.resume_extractor.extractor import extract_all
+from app.ai.resume_polisher.polisher import polish_paragraphs
+from app.core.logger import log
+from app.models.user_resume import UserResume
+from app.models.user_resume_competition import UserResumeCompetition
+from app.models.user_resume_education import UserResumeEducation
+from app.models.user_resume_internship import UserResumeInternship
+from app.models.user_resume_project import UserResumeProject
+from app.models.user_resume_work import UserResumeWork
+from app.tool.file_parser import parse_to_text
+from app.tool.snowflake import next_id
+
+
+class ResumeService:
+
+    async def parse_and_extract(self, filename: str, content: bytes) -> dict:
+        """文件解析 + AI 两阶段并行结构化，不涉及数据库操作"""
+        log.info(f"开始解析简历文件: {filename}")
+        text = await asyncio.to_thread(parse_to_text, filename, content)
+        if not text or not text.strip():
+            raise ValueError("文件内容为空，无法解析")
+        log.info(f"文件解析完成，文本长度: {len(text)}")
+
+        log.info("开始AI两阶段并行结构化提取")
+        parsed = await extract_all(text)
+        log.info("AI两阶段并行结构化提取完成")
+        return parsed
+
+    async def polish_paragraphs(self, content: list[str]) -> list[str]:
+        """对简历段落做表达层面的润色（格式/错字/表达），不涉及数据库操作"""
+        log.info(f"开始简历段落润色, 段落数={len(content)}")
+        result = await polish_paragraphs(content)
+        log.info("简历段落润色完成")
+        return result
+
+    async def save_resume(self, session: AsyncSession, user_id: int, filename: str, parsed: dict) -> int:
+        """将解析结果写入主表 + 5张子表，返回简历ID"""
+        resume_id = next_id()
+
+        session.add(UserResume(
+            id=resume_id, user_id=user_id,
+            resume_name=filename.rsplit(".", 1)[0],
+            target_position=None, is_default=0, sort_order=0,
+            name=parsed.get("name"), email=parsed.get("email"),
+            mobile_number=parsed.get("mobileNumber"), city=parsed.get("city"),
+            wechat_number=parsed.get("wechatNumber"), portfolio_url=parsed.get("portfolioUrl"),
+            skills=parsed.get("skills") or [], certificates=parsed.get("certificates") or [],
+            summary=parsed.get("summary"),
+        ))
+
+        for i, edu in enumerate(parsed.get("education") or []):
+            session.add(UserResumeEducation(
+                id=next_id(), resume_id=resume_id, user_id=user_id,
+                school=edu.get("school"), major=edu.get("major"),
+                degree=edu.get("degree"), study_type=edu.get("studyType"),
+                start_date=edu.get("startDate"), end_date=edu.get("endDate"),
+                description=_to_paragraphs(edu.get("description")), sort_order=i,
+            ))
+
+        for i, work in enumerate(parsed.get("work") or []):
+            session.add(UserResumeWork(
+                id=next_id(), resume_id=resume_id, user_id=user_id,
+                company_name=work.get("companyName"), position=work.get("position"),
+                start_date=work.get("startDate"), end_date=work.get("endDate"),
+                description=_to_paragraphs(work.get("description")), sort_order=i,
+            ))
+
+        for i, intern in enumerate(parsed.get("internship") or []):
+            session.add(UserResumeInternship(
+                id=next_id(), resume_id=resume_id, user_id=user_id,
+                company_name=intern.get("companyName"), position=intern.get("position"),
+                start_date=intern.get("startDate"), end_date=intern.get("endDate"),
+                description=_to_paragraphs(intern.get("description")), sort_order=i,
+            ))
+
+        for i, proj in enumerate(parsed.get("project") or []):
+            session.add(UserResumeProject(
+                id=next_id(), resume_id=resume_id, user_id=user_id,
+                company_name=proj.get("companyName"), project_name=proj.get("projectName"),
+                role=proj.get("role"),
+                start_date=proj.get("startDate"), end_date=proj.get("endDate"),
+                description=_to_paragraphs(proj.get("description")), sort_order=i,
+            ))
+
+        for i, comp in enumerate(parsed.get("competition") or []):
+            session.add(UserResumeCompetition(
+                id=next_id(), resume_id=resume_id, user_id=user_id,
+                competition_name=comp.get("competitionName"), award=comp.get("award"),
+                award_date=comp.get("awardDate"),
+                description=_to_paragraphs(comp.get("description")), sort_order=i,
+            ))
+
+        await session.flush()
+        log.info(f"简历保存完成，resumeId: {resume_id}")
+        return resume_id
+
+
+def _to_paragraphs(texts: list[str] | None) -> list[dict] | None:
+    """将字符串数组转为 [{id, text}] 格式的描述段落"""
+    if not texts:
+        return None
+    return [{"id": shortuuid.ShortUUID().random(length=8), "text": t} for t in texts if t]