routes.py 59 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873
  1. from flask import jsonify, request, make_response, Blueprint, current_app, send_file
  2. from app.api.data_parse import bp
  3. from app.core.data_parse.parse_system import (
  4. update_business_card,
  5. get_business_cards,
  6. update_business_card_status,
  7. create_talent_tag,
  8. get_talent_tag_list,
  9. update_talent_tag,
  10. delete_talent_tag,
  11. query_neo4j_graph,
  12. talent_get_tags,
  13. talent_update_tags,
  14. get_business_card,
  15. search_business_cards_by_mobile,
  16. get_duplicate_records,
  17. process_duplicate_record,
  18. get_duplicate_record_detail,
  19. fix_broken_duplicate_records
  20. )
  21. # 导入解析任务相关函数
  22. from app.core.data_parse.parse_task import (
  23. get_parse_tasks,
  24. get_parse_task_detail,
  25. add_parse_task,
  26. add_parsed_talents
  27. )
  28. # 导入酒店管理相关函数
  29. from app.core.data_parse.hotel_management import (
  30. get_hotel_positions_list,
  31. add_hotel_positions,
  32. update_hotel_positions,
  33. query_hotel_positions,
  34. delete_hotel_positions,
  35. get_hotel_group_brands_list,
  36. add_hotel_group_brands,
  37. update_hotel_group_brands,
  38. query_hotel_group_brands,
  39. delete_hotel_group_brands
  40. )
  41. # 导入名片处理函数
  42. from app.core.data_parse.parse_card import delete_business_card, batch_process_business_card_images
  43. # 导入网页文本解析函数
  44. from app.core.data_parse.parse_web import batch_process_md
  45. # 导入简历解析函数
  46. from app.core.data_parse.parse_resume import batch_parse_resumes
  47. # 导入门墩儿数据处理函数
  48. from app.core.data_parse.parse_menduner import batch_process_menduner_data
  49. # 导入图片批量处理函数
  50. from app.core.data_parse.parse_pic import batch_process_images
  51. from app.config.config import DevelopmentConfig, ProductionConfig
  52. import logging
  53. import boto3
  54. from botocore.config import Config
  55. from botocore.exceptions import ClientError
  56. from io import BytesIO
  57. import base64
  58. import os
  59. import urllib.parse
  60. from minio import Minio
  61. # Define logger
  62. logger = logging.getLogger(__name__)
  63. # For failure responses
  64. def failed(message, code=500):
  65. return {
  66. 'success': False,
  67. 'message': message,
  68. 'data': None
  69. }, code
  70. # 根据环境选择配置
  71. if os.environ.get('FLASK_ENV') == 'production':
  72. config = ProductionConfig()
  73. else:
  74. config = DevelopmentConfig()
  75. # 使用配置变量
  76. minio_url = f"{'https' if config.MINIO_SECURE else 'http'}://{config.MINIO_HOST}"
  77. minio_access_key = config.MINIO_USER
  78. minio_secret_key = config.MINIO_PASSWORD
  79. minio_bucket = config.MINIO_BUCKET
  80. use_ssl = config.MINIO_SECURE
  81. def get_minio_client():
  82. """获取 MinIO 客户端实例"""
  83. return Minio(
  84. '192.168.3.143:9000',
  85. access_key=config.MINIO_USER,
  86. secret_key=config.MINIO_PASSWORD,
  87. secure=config.MINIO_SECURE
  88. )
  89. # 更新名片信息接口
  90. @bp.route('/business-cards/<int:card_id>', methods=['PUT'])
  91. def update_business_card_route(card_id):
  92. """
  93. 更新名片信息的API接口
  94. 路径参数:
  95. - card_id: 名片记录ID
  96. 请求参数:
  97. - JSON格式的名片信息
  98. 返回:
  99. - JSON: 包含更新后的名片信息和处理状态
  100. """
  101. # 获取请求数据
  102. data = request.json
  103. if not data:
  104. return jsonify({
  105. 'success': False,
  106. 'message': '请求数据为空',
  107. 'data': None
  108. }), 400
  109. # 调用业务逻辑函数处理更新
  110. result = update_business_card(card_id, data)
  111. # 根据处理结果设置HTTP状态码
  112. status_code = 200 if result['success'] else 500
  113. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  114. status_code = 404
  115. return jsonify(result), status_code
  116. # 获取所有名片记录的API接口
  117. @bp.route('/get-business-cards', methods=['GET'])
  118. def get_business_cards_route():
  119. """
  120. 获取所有名片记录的API接口
  121. 返回:
  122. - JSON: 包含名片记录列表和处理状态
  123. """
  124. # 调用业务逻辑函数获取名片列表
  125. result = get_business_cards()
  126. # 根据处理结果设置HTTP状态码
  127. status_code = 200 if result['success'] else 500
  128. return jsonify(result), status_code
  129. @bp.route('/update-business-cards/<int:card_id>/status', methods=['PUT'])
  130. def update_business_card_status_route(card_id):
  131. """
  132. 更新名片状态的API接口
  133. 路径参数:
  134. - card_id: 名片记录ID
  135. 请求参数:
  136. - JSON格式,包含status字段
  137. 返回:
  138. - JSON: 包含更新后的名片信息和处理状态
  139. """
  140. # 获取请求数据
  141. data = request.json
  142. if not data or 'status' not in data:
  143. return jsonify({
  144. 'success': False,
  145. 'message': '请求数据为空或缺少status字段',
  146. 'data': None
  147. }), 400
  148. status = data['status']
  149. # 调用业务逻辑函数处理状态更新
  150. result = update_business_card_status(card_id, status)
  151. # 根据处理结果设置HTTP状态码
  152. status_code = 200 if result['success'] else 500
  153. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  154. status_code = 404
  155. return jsonify(result), status_code
  156. # 从MinIO获取名片图片的API接口
  157. @bp.route('/business-cards/image/<path:image_path>', methods=['GET'])
  158. def get_business_card_image(image_path):
  159. """
  160. 从MinIO获取名片图片的API接口
  161. 路径参数:
  162. - image_path: MinIO中的图片路径
  163. 返回:
  164. - 图片数据流
  165. """
  166. try:
  167. # 记录下载请求信息,便于调试
  168. logger.info(f"获取名片图片请求: {image_path}")
  169. # 获取 MinIO 客户端
  170. minio_client = get_minio_client()
  171. if not minio_client:
  172. return jsonify(failed("MinIO客户端初始化失败")), 500
  173. try:
  174. # 使用正确的MinIO客户端方法
  175. data = minio_client.get_object(minio_bucket, image_path)
  176. # 创建内存文件流
  177. file_stream = BytesIO(data.read())
  178. # 获取文件名
  179. file_name = image_path.split('/')[-1]
  180. # 返回文件
  181. return send_file(
  182. file_stream,
  183. as_attachment=False, # 设置为False,让浏览器直接显示图片
  184. download_name=file_name,
  185. mimetype='image/jpeg' # 根据实际图片类型设置
  186. )
  187. except Exception as e:
  188. logger.error(f"MinIO获取文件失败: {str(e)}")
  189. return jsonify(failed(f"文件获取失败: {str(e)}")), 404
  190. except Exception as e:
  191. logger.error(f"文件下载失败: {str(e)}")
  192. return jsonify(failed(str(e))), 500
  193. finally:
  194. # 确保关闭数据流
  195. if 'data' in locals():
  196. data.close()
  197. # 创建人才标签接口
  198. @bp.route('/create-talent-tag', methods=['POST'])
  199. def create_talent_tag_route():
  200. """
  201. 创建人才标签的API接口
  202. 请求参数:
  203. - JSON格式,包含以下字段:
  204. - name: 标签名称
  205. - category: 标签分类
  206. - description: 标签描述
  207. - status: 启用状态,默认为'active'
  208. 返回:
  209. - JSON: 包含创建结果和标签信息
  210. """
  211. try:
  212. # 获取请求数据
  213. data = request.get_json()
  214. if not data:
  215. return jsonify({
  216. 'success': False,
  217. 'message': '请求数据为空',
  218. 'data': None
  219. }), 400
  220. # 验证必要字段
  221. if 'name' not in data or not data['name']:
  222. return jsonify({
  223. 'success': False,
  224. 'message': '标签名称不能为空',
  225. 'data': None
  226. }), 400
  227. # 处理分类字段,如果未提供则设置默认值
  228. if 'category' not in data or not data['category']:
  229. data['category'] = '未分类'
  230. # 调用业务逻辑函数处理创建
  231. result = create_talent_tag(data)
  232. # 根据处理结果设置HTTP状态码
  233. status_code = 200 if result['success'] else 500
  234. return jsonify(result), status_code
  235. except Exception as e:
  236. logger.error(f"创建人才标签失败: {str(e)}")
  237. return jsonify({
  238. 'success': False,
  239. 'message': f'创建人才标签失败: {str(e)}',
  240. 'data': None
  241. }), 500
  242. # 获取人才标签列表接口
  243. @bp.route('/get-talent-tag-list', methods=['GET'])
  244. def get_talent_tag_list_route():
  245. """
  246. 获取人才标签列表的API接口
  247. 返回:
  248. - JSON: 包含人才标签列表和处理状态
  249. """
  250. try:
  251. # 调用业务逻辑函数获取人才标签列表
  252. result = get_talent_tag_list()
  253. # 根据处理结果设置HTTP状态码
  254. status_code = 200 if result['success'] else 500
  255. return jsonify(result), status_code
  256. except Exception as e:
  257. logger.error(f"获取人才标签列表失败: {str(e)}")
  258. return jsonify({
  259. 'success': False,
  260. 'message': f'获取人才标签列表失败: {str(e)}',
  261. 'data': []
  262. }), 500
  263. # 更新人才标签接口
  264. @bp.route('/update-talent-tag/<int:tag_id>', methods=['PUT'])
  265. def update_talent_tag_route(tag_id):
  266. """
  267. 更新人才标签的API接口
  268. 路径参数:
  269. - tag_id: 标签节点ID
  270. 请求参数:
  271. - JSON格式,可能包含以下字段:
  272. - name: 标签名称
  273. - category: 标签分类
  274. - description: 标签描述
  275. - status: 启用状态
  276. 返回:
  277. - JSON: 包含更新结果和标签信息
  278. """
  279. try:
  280. # 获取请求数据
  281. data = request.get_json()
  282. if not data:
  283. return jsonify({
  284. 'success': False,
  285. 'message': '请求数据为空',
  286. 'data': None
  287. }), 400
  288. # 调用业务逻辑函数处理更新
  289. result = update_talent_tag(tag_id, data)
  290. # 根据处理结果设置HTTP状态码
  291. if not result['success']:
  292. if result['code'] == 404:
  293. status_code = 404
  294. elif result['code'] == 400:
  295. status_code = 400
  296. else:
  297. status_code = 500
  298. else:
  299. status_code = 200
  300. return jsonify(result), status_code
  301. except Exception as e:
  302. logger.error(f"更新人才标签失败: {str(e)}")
  303. return jsonify({
  304. 'success': False,
  305. 'message': f'更新人才标签失败: {str(e)}',
  306. 'data': None
  307. }), 500
  308. # 删除人才标签接口
  309. @bp.route('/delete-talent-tag/<int:tag_id>', methods=['DELETE'])
  310. def delete_talent_tag_route(tag_id):
  311. """
  312. 删除人才标签的API接口
  313. 路径参数:
  314. - tag_id: 标签节点ID
  315. 返回:
  316. - JSON: 包含删除结果和被删除的标签信息
  317. """
  318. try:
  319. # 调用业务逻辑函数执行删除
  320. result = delete_talent_tag(tag_id)
  321. # 根据处理结果设置HTTP状态码
  322. if not result['success']:
  323. if result['code'] == 404:
  324. status_code = 404
  325. else:
  326. status_code = 500
  327. else:
  328. status_code = 200
  329. return jsonify(result), status_code
  330. except Exception as e:
  331. logger.error(f"删除人才标签失败: {str(e)}")
  332. return jsonify({
  333. 'success': False,
  334. 'message': f'删除人才标签失败: {str(e)}',
  335. 'data': None
  336. }), 500
  337. @bp.route('/query-kg', methods=['POST'])
  338. def query_kg():
  339. """
  340. 查询知识图谱API接口
  341. 请求参数:
  342. - query_requirement: 查询需求描述(JSON格式)
  343. 返回:
  344. - JSON: 包含查询结果和处理状态
  345. """
  346. try:
  347. # 获取请求数据
  348. data = request.json
  349. if not data or 'query_requirement' not in data:
  350. return jsonify({
  351. 'code': 400,
  352. 'success': False,
  353. 'message': '请求数据为空或缺少query_requirement字段',
  354. 'data': []
  355. }), 400
  356. query_requirement = data['query_requirement']
  357. # 调用业务逻辑函数执行查询
  358. result = query_neo4j_graph(query_requirement)
  359. # 根据处理结果设置HTTP状态码
  360. status_code = 200 if result['success'] else 500
  361. return jsonify(result), status_code
  362. except Exception as e:
  363. logger.error(f"查询知识图谱失败: {str(e)}")
  364. return jsonify({
  365. 'code': 500,
  366. 'success': False,
  367. 'message': f"查询知识图谱失败: {str(e)}",
  368. 'data': []
  369. }), 500
  370. @bp.route('/talent-get-tags/<int:talent_id>', methods=['GET'])
  371. def talent_get_tags_route(talent_id):
  372. """
  373. 获取人才标签的API接口
  374. 路径参数:
  375. - talent_id: 人才节点ID
  376. 返回:
  377. - JSON: 包含人才关联的标签列表和处理状态
  378. """
  379. try:
  380. # 调用业务逻辑函数获取人才标签
  381. result = talent_get_tags(talent_id)
  382. # 根据处理结果设置HTTP状态码
  383. status_code = 200 if result['success'] else 500
  384. return jsonify(result), status_code
  385. except Exception as e:
  386. logger.error(f"获取人才标签失败: {str(e)}")
  387. return jsonify({
  388. 'code': 500,
  389. 'success': False,
  390. 'message': f"获取人才标签失败: {str(e)}",
  391. 'data': []
  392. }), 500
  393. @bp.route('/talent-update-tags', methods=['POST'])
  394. def talent_update_tags_route():
  395. """
  396. 更新人才标签关系的API接口
  397. 请求参数:
  398. - JSON数组,包含talent和tag字段的对象列表
  399. 例如: [
  400. {"talent": 12345, "tag": "市场营销"},
  401. {"talent": 12345, "tag": "酒店管理"}
  402. ]
  403. 返回:
  404. - JSON: 包含更新结果的状态信息
  405. """
  406. try:
  407. # 获取请求数据
  408. data = request.json
  409. if not data:
  410. return jsonify({
  411. 'code': 400,
  412. 'success': False,
  413. 'message': '请求数据为空',
  414. 'data': None
  415. }), 400
  416. # 调用业务逻辑函数处理标签关系更新
  417. result = talent_update_tags(data)
  418. # 根据处理结果设置HTTP状态码
  419. if result['code'] == 200:
  420. status_code = 200
  421. elif result['code'] == 206:
  422. status_code = 206 # Partial Content
  423. elif result['code'] == 400:
  424. status_code = 400 # Bad Request
  425. elif result['code'] == 404:
  426. status_code = 404 # Not Found
  427. else:
  428. status_code = 500 # Internal Server Error
  429. return jsonify(result), status_code
  430. except Exception as e:
  431. logger.error(f"更新人才标签关系失败: {str(e)}")
  432. return jsonify({
  433. 'code': 500,
  434. 'success': False,
  435. 'message': f"更新人才标签关系失败: {str(e)}",
  436. 'data': None
  437. }), 500
  438. # 获取单个名片记录的API接口
  439. @bp.route('/get-business-card/<int:card_id>', methods=['GET'])
  440. def get_business_card_route(card_id):
  441. """
  442. 获取单个名片记录的API接口
  443. 路径参数:
  444. - card_id: 名片记录ID
  445. 返回:
  446. - JSON: 包含名片记录信息和处理状态
  447. """
  448. # 调用业务逻辑函数获取名片记录
  449. result = get_business_card(card_id)
  450. # 根据处理结果设置HTTP状态码
  451. if not result['success']:
  452. if result['code'] == 404:
  453. status_code = 404
  454. else:
  455. status_code = 500
  456. else:
  457. status_code = 200
  458. return jsonify(result), status_code
  459. @bp.route('/search-business-cards-by-mobile', methods=['GET'])
  460. def search_business_cards_by_mobile_route():
  461. """
  462. 根据手机号码搜索名片记录的API接口
  463. 查询参数:
  464. - mobile: 要搜索的手机号码
  465. 返回:
  466. - JSON: 包含搜索到的名片记录列表和处理状态
  467. 示例:
  468. GET /search-business-cards-by-mobile?mobile=13800138000
  469. """
  470. try:
  471. # 获取查询参数
  472. mobile_number = request.args.get('mobile', '').strip()
  473. if not mobile_number:
  474. return jsonify({
  475. 'success': False,
  476. 'message': '请提供要搜索的手机号码',
  477. 'data': []
  478. }), 400
  479. # 调用业务逻辑函数搜索名片记录
  480. result = search_business_cards_by_mobile(mobile_number)
  481. # 根据处理结果设置HTTP状态码
  482. if result['code'] == 200:
  483. status_code = 200
  484. elif result['code'] == 400:
  485. status_code = 400
  486. else:
  487. status_code = 500
  488. return jsonify(result), status_code
  489. except Exception as e:
  490. # 处理未预期的异常
  491. error_msg = f"根据手机号码搜索名片时发生错误: {str(e)}"
  492. logger.error(error_msg, exc_info=True)
  493. return jsonify({
  494. 'success': False,
  495. 'message': error_msg,
  496. 'data': []
  497. }), 500
  498. @bp.route('/get-hotel-positions-list', methods=['GET'])
  499. def get_hotel_positions_list_route():
  500. """
  501. 获取酒店职位数据表全部记录的API接口
  502. 返回:
  503. - JSON: 包含酒店职位记录列表和处理状态
  504. """
  505. try:
  506. # 调用业务逻辑函数获取酒店职位列表
  507. result = get_hotel_positions_list()
  508. # 根据处理结果设置HTTP状态码
  509. status_code = 200 if result['success'] else 500
  510. return jsonify(result), status_code
  511. except Exception as e:
  512. # 处理未预期的异常
  513. error_msg = f"获取酒店职位列表时发生错误: {str(e)}"
  514. logger.error(error_msg, exc_info=True)
  515. return jsonify({
  516. 'success': False,
  517. 'message': error_msg,
  518. 'data': [],
  519. 'count': 0
  520. }), 500
  521. @bp.route('/add-hotel-positions', methods=['POST'])
  522. def add_hotel_positions_route():
  523. """
  524. 新增酒店职位数据表记录的API接口
  525. 请求参数:
  526. - JSON格式,包含以下字段:
  527. - department_zh: 部门中文名称 (必填)
  528. - department_en: 部门英文名称 (必填)
  529. - position_zh: 职位中文名称 (必填)
  530. - position_en: 职位英文名称 (必填)
  531. - position_abbr: 职位英文缩写 (可选)
  532. - level_zh: 职级中文名称 (必填)
  533. - level_en: 职级英文名称 (必填)
  534. - created_by: 创建者 (可选)
  535. - updated_by: 更新者 (可选)
  536. - status: 状态 (可选)
  537. 返回:
  538. - JSON: 包含创建结果和职位信息
  539. """
  540. try:
  541. # 获取请求数据
  542. data = request.get_json()
  543. if not data:
  544. return jsonify({
  545. 'success': False,
  546. 'message': '请求数据为空',
  547. 'data': None
  548. }), 400
  549. # 调用业务逻辑函数处理创建
  550. result = add_hotel_positions(data)
  551. # 根据处理结果设置HTTP状态码
  552. if result['code'] == 200:
  553. status_code = 201 # Created
  554. elif result['code'] == 400:
  555. status_code = 400 # Bad Request
  556. elif result['code'] == 409:
  557. status_code = 409 # Conflict
  558. else:
  559. status_code = 500 # Internal Server Error
  560. return jsonify(result), status_code
  561. except Exception as e:
  562. # 处理未预期的异常
  563. error_msg = f"创建酒店职位记录时发生错误: {str(e)}"
  564. logger.error(error_msg, exc_info=True)
  565. return jsonify({
  566. 'success': False,
  567. 'message': error_msg,
  568. 'data': None
  569. }), 500
  570. @bp.route('/update-hotel-positions/<int:position_id>', methods=['PUT'])
  571. def update_hotel_positions_route(position_id):
  572. """
  573. 修改酒店职位数据表记录的API接口
  574. 路径参数:
  575. - position_id: 职位记录ID
  576. 请求参数:
  577. - JSON格式,可能包含以下字段:
  578. - department_zh: 部门中文名称
  579. - department_en: 部门英文名称
  580. - position_zh: 职位中文名称
  581. - position_en: 职位英文名称
  582. - position_abbr: 职位英文缩写
  583. - level_zh: 职级中文名称
  584. - level_en: 职级英文名称
  585. - updated_by: 更新者
  586. - status: 状态
  587. 返回:
  588. - JSON: 包含更新结果和职位信息
  589. """
  590. try:
  591. # 获取请求数据
  592. data = request.get_json()
  593. if not data:
  594. return jsonify({
  595. 'success': False,
  596. 'message': '请求数据为空',
  597. 'data': None
  598. }), 400
  599. # 调用业务逻辑函数处理更新
  600. result = update_hotel_positions(position_id, data)
  601. # 根据处理结果设置HTTP状态码
  602. if result['code'] == 200:
  603. status_code = 200 # OK
  604. elif result['code'] == 400:
  605. status_code = 400 # Bad Request
  606. elif result['code'] == 404:
  607. status_code = 404 # Not Found
  608. elif result['code'] == 409:
  609. status_code = 409 # Conflict
  610. else:
  611. status_code = 500 # Internal Server Error
  612. return jsonify(result), status_code
  613. except Exception as e:
  614. # 处理未预期的异常
  615. error_msg = f"更新酒店职位记录时发生错误: {str(e)}"
  616. logger.error(error_msg, exc_info=True)
  617. return jsonify({
  618. 'success': False,
  619. 'message': error_msg,
  620. 'data': None
  621. }), 500
  622. @bp.route('/query-hotel-positions/<int:position_id>', methods=['GET'])
  623. def query_hotel_positions_route(position_id):
  624. """
  625. 查找指定ID的酒店职位数据表记录的API接口
  626. 路径参数:
  627. - position_id: 职位记录ID
  628. 返回:
  629. - JSON: 包含查找结果和职位信息
  630. """
  631. try:
  632. # 调用业务逻辑函数查找职位记录
  633. result = query_hotel_positions(position_id)
  634. # 根据处理结果设置HTTP状态码
  635. if result['code'] == 200:
  636. status_code = 200 # OK
  637. elif result['code'] == 404:
  638. status_code = 404 # Not Found
  639. else:
  640. status_code = 500 # Internal Server Error
  641. return jsonify(result), status_code
  642. except Exception as e:
  643. # 处理未预期的异常
  644. error_msg = f"查找酒店职位记录时发生错误: {str(e)}"
  645. logger.error(error_msg, exc_info=True)
  646. return jsonify({
  647. 'success': False,
  648. 'message': error_msg,
  649. 'data': None
  650. }), 500
  651. @bp.route('/delete-hotel-positions/<int:position_id>', methods=['DELETE'])
  652. def delete_hotel_positions_route(position_id):
  653. """
  654. 删除指定ID的酒店职位数据表记录的API接口
  655. 路径参数:
  656. - position_id: 职位记录ID
  657. 返回:
  658. - JSON: 包含删除结果和被删除的职位信息
  659. """
  660. try:
  661. # 调用业务逻辑函数删除职位记录
  662. result = delete_hotel_positions(position_id)
  663. # 根据处理结果设置HTTP状态码
  664. if result['code'] == 200:
  665. status_code = 200 # OK
  666. elif result['code'] == 404:
  667. status_code = 404 # Not Found
  668. else:
  669. status_code = 500 # Internal Server Error
  670. return jsonify(result), status_code
  671. except Exception as e:
  672. # 处理未预期的异常
  673. error_msg = f"删除酒店职位记录时发生错误: {str(e)}"
  674. logger.error(error_msg, exc_info=True)
  675. return jsonify({
  676. 'success': False,
  677. 'message': error_msg,
  678. 'data': None
  679. }), 500
  680. @bp.route('/get-hotel-group-brands-list', methods=['GET'])
  681. def get_hotel_group_brands_list_route():
  682. """
  683. 获取酒店集团子品牌数据表全部记录的API接口
  684. 返回:
  685. - JSON: 包含酒店集团品牌记录列表和处理状态
  686. """
  687. try:
  688. # 调用业务逻辑函数获取酒店集团品牌列表
  689. result = get_hotel_group_brands_list()
  690. # 根据处理结果设置HTTP状态码
  691. status_code = 200 if result['success'] else 500
  692. return jsonify(result), status_code
  693. except Exception as e:
  694. # 处理未预期的异常
  695. error_msg = f"获取酒店集团品牌列表时发生错误: {str(e)}"
  696. logger.error(error_msg, exc_info=True)
  697. return jsonify({
  698. 'success': False,
  699. 'message': error_msg,
  700. 'data': [],
  701. 'count': 0
  702. }), 500
  703. @bp.route('/add-hotel-group-brands', methods=['POST'])
  704. def add_hotel_group_brands_route():
  705. """
  706. 新增酒店集团子品牌数据表记录的API接口
  707. 请求参数:
  708. - JSON格式,包含以下字段:
  709. - group_name_en: 集团英文名称 (必填)
  710. - group_name_zh: 集团中文名称 (必填)
  711. - brand_name_en: 品牌英文名称 (必填)
  712. - brand_name_zh: 品牌中文名称 (必填)
  713. - positioning_level_en: 定位级别英文名称 (必填)
  714. - positioning_level_zh: 定位级别中文名称 (必填)
  715. - created_by: 创建者 (可选)
  716. - updated_by: 更新者 (可选)
  717. - status: 状态 (可选)
  718. 返回:
  719. - JSON: 包含创建结果和品牌信息
  720. """
  721. try:
  722. # 获取请求数据
  723. data = request.get_json()
  724. if not data:
  725. return jsonify({
  726. 'success': False,
  727. 'message': '请求数据为空',
  728. 'data': None
  729. }), 400
  730. # 调用业务逻辑函数处理创建
  731. result = add_hotel_group_brands(data)
  732. # 根据处理结果设置HTTP状态码
  733. if result['code'] == 200:
  734. status_code = 201 # Created
  735. elif result['code'] == 400:
  736. status_code = 400 # Bad Request
  737. elif result['code'] == 409:
  738. status_code = 409 # Conflict
  739. else:
  740. status_code = 500 # Internal Server Error
  741. return jsonify(result), status_code
  742. except Exception as e:
  743. # 处理未预期的异常
  744. error_msg = f"创建酒店集团品牌记录时发生错误: {str(e)}"
  745. logger.error(error_msg, exc_info=True)
  746. return jsonify({
  747. 'success': False,
  748. 'message': error_msg,
  749. 'data': None
  750. }), 500
  751. @bp.route('/update-hotel-group-brands/<int:brand_id>', methods=['PUT'])
  752. def update_hotel_group_brands_route(brand_id):
  753. """
  754. 修改酒店集团子品牌数据表记录的API接口
  755. 路径参数:
  756. - brand_id: 品牌记录ID
  757. 请求参数:
  758. - JSON格式,可能包含以下字段:
  759. - group_name_en: 集团英文名称
  760. - group_name_zh: 集团中文名称
  761. - brand_name_en: 品牌英文名称
  762. - brand_name_zh: 品牌中文名称
  763. - positioning_level_en: 定位级别英文名称
  764. - positioning_level_zh: 定位级别中文名称
  765. - updated_by: 更新者
  766. - status: 状态
  767. 返回:
  768. - JSON: 包含更新结果和品牌信息
  769. """
  770. try:
  771. # 获取请求数据
  772. data = request.get_json()
  773. if not data:
  774. return jsonify({
  775. 'success': False,
  776. 'message': '请求数据为空',
  777. 'data': None
  778. }), 400
  779. # 调用业务逻辑函数处理更新
  780. result = update_hotel_group_brands(brand_id, data)
  781. # 根据处理结果设置HTTP状态码
  782. if result['code'] == 200:
  783. status_code = 200 # OK
  784. elif result['code'] == 400:
  785. status_code = 400 # Bad Request
  786. elif result['code'] == 404:
  787. status_code = 404 # Not Found
  788. elif result['code'] == 409:
  789. status_code = 409 # Conflict
  790. else:
  791. status_code = 500 # Internal Server Error
  792. return jsonify(result), status_code
  793. except Exception as e:
  794. # 处理未预期的异常
  795. error_msg = f"更新酒店集团品牌记录时发生错误: {str(e)}"
  796. logger.error(error_msg, exc_info=True)
  797. return jsonify({
  798. 'success': False,
  799. 'message': error_msg,
  800. 'data': None
  801. }), 500
  802. @bp.route('/query-hotel-group-brands/<int:brand_id>', methods=['GET'])
  803. def query_hotel_group_brands_route(brand_id):
  804. """
  805. 查找指定ID的酒店集团子品牌数据表记录的API接口
  806. 路径参数:
  807. - brand_id: 品牌记录ID
  808. 返回:
  809. - JSON: 包含查找结果和品牌信息
  810. """
  811. try:
  812. # 调用业务逻辑函数查找品牌记录
  813. result = query_hotel_group_brands(brand_id)
  814. # 根据处理结果设置HTTP状态码
  815. if result['code'] == 200:
  816. status_code = 200 # OK
  817. elif result['code'] == 404:
  818. status_code = 404 # Not Found
  819. else:
  820. status_code = 500 # Internal Server Error
  821. return jsonify(result), status_code
  822. except Exception as e:
  823. # 处理未预期的异常
  824. error_msg = f"查找酒店集团品牌记录时发生错误: {str(e)}"
  825. logger.error(error_msg, exc_info=True)
  826. return jsonify({
  827. 'success': False,
  828. 'message': error_msg,
  829. 'data': None
  830. }), 500
  831. @bp.route('/delete-hotel-group-brands/<int:brand_id>', methods=['DELETE'])
  832. def delete_hotel_group_brands_route(brand_id):
  833. """
  834. 删除指定ID的酒店集团子品牌数据表记录的API接口
  835. 路径参数:
  836. - brand_id: 品牌记录ID
  837. 返回:
  838. - JSON: 包含删除结果和被删除的品牌信息
  839. """
  840. try:
  841. # 调用业务逻辑函数删除品牌记录
  842. result = delete_hotel_group_brands(brand_id)
  843. # 根据处理结果设置HTTP状态码
  844. if result['code'] == 200:
  845. status_code = 200 # OK
  846. elif result['code'] == 404:
  847. status_code = 404 # Not Found
  848. else:
  849. status_code = 500 # Internal Server Error
  850. return jsonify(result), status_code
  851. except Exception as e:
  852. # 处理未预期的异常
  853. error_msg = f"删除酒店集团品牌记录时发生错误: {str(e)}"
  854. logger.error(error_msg, exc_info=True)
  855. return jsonify({
  856. 'success': False,
  857. 'message': error_msg,
  858. 'data': None
  859. }), 500
  860. # ==================================
  861. # 重复记录处理API接口
  862. # ==================================
  863. @bp.route('/get-duplicate-records', methods=['GET'])
  864. def get_duplicate_records_route():
  865. """
  866. 获取重复记录列表的API接口
  867. 查询参数:
  868. - status: 可选,筛选特定状态的记录 ('pending', 'processed', 'ignored')
  869. 返回:
  870. - JSON: 包含重复记录列表和处理状态
  871. """
  872. try:
  873. # 获取查询参数
  874. status = request.args.get('status', None)
  875. # 验证status参数的有效性
  876. if status and status not in ['pending', 'processed', 'ignored']:
  877. return jsonify({
  878. 'success': False,
  879. 'message': 'status参数无效,必须为 pending、processed 或 ignored',
  880. 'data': None
  881. }), 400
  882. # 调用业务逻辑函数获取重复记录列表
  883. result = get_duplicate_records(status)
  884. # 根据处理结果设置HTTP状态码
  885. status_code = 200 if result['success'] else 500
  886. return jsonify(result), status_code
  887. except Exception as e:
  888. # 处理未预期的异常
  889. error_msg = f"获取重复记录列表时发生错误: {str(e)}"
  890. logger.error(error_msg, exc_info=True)
  891. return jsonify({
  892. 'success': False,
  893. 'message': error_msg,
  894. 'data': [],
  895. 'count': 0
  896. }), 500
  897. @bp.route('/process-duplicate-record/<int:duplicate_id>', methods=['POST'])
  898. def process_duplicate_record_route(duplicate_id):
  899. """
  900. 处理重复记录的API接口
  901. 路径参数:
  902. - duplicate_id: 重复记录ID
  903. 请求参数:
  904. - JSON格式,包含以下字段:
  905. - action: 处理动作 (必填) ('merge_to_suspected', 'keep_main', 'ignore')
  906. - selected_duplicate_id: 当action为'merge_to_suspected'时,选择的疑似重复记录ID (可选)
  907. - processed_by: 处理人 (可选)
  908. - notes: 处理备注 (可选)
  909. 返回:
  910. - JSON: 包含处理结果和状态信息
  911. """
  912. try:
  913. # 获取请求数据
  914. data = request.get_json()
  915. if not data:
  916. return jsonify({
  917. 'success': False,
  918. 'message': '请求数据为空',
  919. 'data': None
  920. }), 400
  921. # 验证必填字段
  922. action = data.get('action')
  923. if not action:
  924. return jsonify({
  925. 'success': False,
  926. 'message': '缺少必填字段: action',
  927. 'data': None
  928. }), 400
  929. # 验证action参数的有效性
  930. if action not in ['merge_to_suspected', 'keep_main', 'ignore']:
  931. return jsonify({
  932. 'success': False,
  933. 'message': 'action参数无效,必须为 merge_to_suspected、keep_main 或 ignore',
  934. 'data': None
  935. }), 400
  936. # 提取其他参数
  937. selected_duplicate_id = data.get('selected_duplicate_id')
  938. processed_by = data.get('processed_by')
  939. notes = data.get('notes')
  940. # 特殊验证:如果action为merge_to_suspected,必须提供selected_duplicate_id
  941. if action == 'merge_to_suspected' and not selected_duplicate_id:
  942. return jsonify({
  943. 'success': False,
  944. 'message': '执行merge_to_suspected操作时必须提供selected_duplicate_id',
  945. 'data': None
  946. }), 400
  947. # 调用业务逻辑函数处理重复记录
  948. result = process_duplicate_record(
  949. duplicate_id=duplicate_id,
  950. action=action,
  951. selected_duplicate_id=selected_duplicate_id,
  952. processed_by=processed_by,
  953. notes=notes
  954. )
  955. # 根据处理结果设置HTTP状态码
  956. if result['code'] == 200:
  957. status_code = 200 # OK
  958. elif result['code'] == 400:
  959. status_code = 400 # Bad Request
  960. elif result['code'] == 404:
  961. status_code = 404 # Not Found
  962. else:
  963. status_code = 500 # Internal Server Error
  964. return jsonify(result), status_code
  965. except Exception as e:
  966. # 处理未预期的异常
  967. error_msg = f"处理重复记录时发生错误: {str(e)}"
  968. logger.error(error_msg, exc_info=True)
  969. return jsonify({
  970. 'success': False,
  971. 'message': error_msg,
  972. 'data': None
  973. }), 500
  974. @bp.route('/get-duplicate-record-detail/<int:duplicate_id>', methods=['GET'])
  975. def get_duplicate_record_detail_route(duplicate_id):
  976. """
  977. 获取指定重复记录详细信息的API接口
  978. 路径参数:
  979. - duplicate_id: 重复记录ID
  980. 返回:
  981. - JSON: 包含重复记录详细信息
  982. """
  983. try:
  984. # 调用业务逻辑函数获取重复记录详情
  985. result = get_duplicate_record_detail(duplicate_id)
  986. # 根据处理结果设置HTTP状态码
  987. if result['code'] == 200:
  988. status_code = 200 # OK
  989. elif result['code'] == 404:
  990. status_code = 404 # Not Found
  991. else:
  992. status_code = 500 # Internal Server Error
  993. return jsonify(result), status_code
  994. except Exception as e:
  995. # 处理未预期的异常
  996. error_msg = f"获取重复记录详情时发生错误: {str(e)}"
  997. logger.error(error_msg, exc_info=True)
  998. return jsonify({
  999. 'success': False,
  1000. 'message': error_msg,
  1001. 'data': None
  1002. }), 500
  1003. # 删除名片记录接口
  1004. @bp.route('/delete-business-card/<int:card_id>', methods=['DELETE'])
  1005. def delete_business_card_route(card_id):
  1006. """
  1007. 删除名片记录的API接口
  1008. 路径参数:
  1009. - card_id: 名片记录ID (必填)
  1010. 功能说明:
  1011. - 删除PostgreSQL数据库中business_cards表的指定记录
  1012. - 删除PostgreSQL数据库中duplicate_business_cards表的相关记录
  1013. - 删除MinIO存储中的名片图片文件
  1014. - 删除Neo4j图数据库中talent节点及其关联关系
  1015. 返回:
  1016. - JSON: 包含删除操作的结果状态和被删除的记录信息
  1017. 状态码:
  1018. - 200: 完全成功删除所有相关数据
  1019. - 206: 部分成功 (PostgreSQL删除成功,但Neo4j删除失败)
  1020. - 404: 未找到指定ID的名片记录
  1021. - 500: 删除操作失败
  1022. """
  1023. try:
  1024. # 验证card_id参数
  1025. if not card_id or card_id <= 0:
  1026. return jsonify({
  1027. 'success': False,
  1028. 'message': '无效的名片记录ID',
  1029. 'data': None
  1030. }), 400
  1031. # 调用删除函数
  1032. result = delete_business_card(card_id)
  1033. # 根据处理结果设置HTTP状态码和返回响应
  1034. if result['success']:
  1035. if result['code'] == 200:
  1036. status_code = 200 # 完全成功
  1037. elif result['code'] == 206:
  1038. status_code = 206 # 部分成功
  1039. else:
  1040. status_code = 200 # 默认成功
  1041. else:
  1042. if result['code'] == 404:
  1043. status_code = 404 # 未找到记录
  1044. elif result['code'] == 400:
  1045. status_code = 400 # 参数错误
  1046. else:
  1047. status_code = 500 # 服务器错误
  1048. return jsonify(result), status_code
  1049. except Exception as e:
  1050. logger.error(f"删除名片记录失败: {str(e)}")
  1051. return jsonify({
  1052. 'success': False,
  1053. 'message': f'删除名片记录失败: {str(e)}',
  1054. 'data': None
  1055. }), 500
  1056. # 修复损坏的重复记录接口
  1057. @bp.route('/fix-broken-duplicate-records', methods=['POST'])
  1058. def fix_broken_duplicate_records_route():
  1059. """
  1060. 修复duplicate_business_cards表中main_card_id为null的损坏记录
  1061. 功能说明:
  1062. - 查找所有main_card_id为null的损坏记录
  1063. - 删除这些损坏的记录以维护数据完整性
  1064. - 返回修复操作的详细结果
  1065. 返回:
  1066. - JSON: 包含修复操作的结果和被删除记录的信息
  1067. 状态码:
  1068. - 200: 修复成功
  1069. - 500: 修复失败
  1070. 注意:
  1071. - 此操作会永久删除损坏的记录
  1072. - 建议在系统维护时执行此操作
  1073. """
  1074. try:
  1075. # 调用修复函数
  1076. result = fix_broken_duplicate_records()
  1077. # 根据结果设置状态码
  1078. if result['success']:
  1079. status_code = 200
  1080. else:
  1081. status_code = 500
  1082. return jsonify(result), status_code
  1083. except Exception as e:
  1084. logger.error(f"修复损坏记录接口调用失败: {str(e)}")
  1085. return jsonify({
  1086. 'success': False,
  1087. 'message': f'修复损坏记录接口调用失败: {str(e)}',
  1088. 'data': None
  1089. }), 500
  1090. # 获取解析任务列表接口
  1091. @bp.route('/get-parse-tasks', methods=['GET'])
  1092. def get_parse_tasks_route():
  1093. """
  1094. 获取解析任务列表的API接口,支持分页
  1095. 查询参数:
  1096. - page: 页码,从1开始,默认为1
  1097. - per_page: 每页记录数,默认为10,最大100
  1098. - task_type: 任务类型过滤,可选
  1099. - task_status: 任务状态过滤,可选
  1100. 返回:
  1101. - JSON: 包含解析任务列表和分页信息
  1102. 功能说明:
  1103. - 支持分页查询,每页默认10条记录
  1104. - 支持按任务类型和状态过滤
  1105. - 按创建时间倒序排列
  1106. - 返回总记录数和分页信息
  1107. 状态码:
  1108. - 200: 查询成功
  1109. - 400: 请求参数错误
  1110. - 500: 查询失败
  1111. """
  1112. try:
  1113. # 获取查询参数
  1114. page = request.args.get('page', 1, type=int)
  1115. per_page = request.args.get('per_page', 10, type=int)
  1116. task_type = request.args.get('task_type', type=str)
  1117. task_status = request.args.get('task_status', type=str)
  1118. # 记录请求日志
  1119. logger.info(f"获取解析任务列表请求: page={page}, per_page={per_page}, task_type={task_type}, task_status={task_status}")
  1120. # 调用核心业务逻辑
  1121. result = get_parse_tasks(page, per_page, task_type, task_status)
  1122. # 返回结果
  1123. return jsonify({
  1124. 'success': result['success'],
  1125. 'message': result['message'],
  1126. 'data': result['data']
  1127. }), result['code']
  1128. except Exception as e:
  1129. # 记录错误日志
  1130. error_msg = f"获取解析任务列表接口失败: {str(e)}"
  1131. logger.error(error_msg, exc_info=True)
  1132. # 返回错误响应
  1133. return jsonify({
  1134. 'success': False,
  1135. 'message': error_msg,
  1136. 'data': None
  1137. }), 500
  1138. # 获取解析任务详情接口
  1139. @bp.route('/get-parse-task-detail', methods=['GET'])
  1140. def get_parse_task_detail_route():
  1141. """
  1142. 获取解析任务详情的API接口
  1143. 查询参数:
  1144. - task_name: 任务名称,必填
  1145. 返回:
  1146. - JSON: 包含任务详细信息
  1147. 功能说明:
  1148. - 根据任务名称查询指定任务的详细信息
  1149. - 返回任务的所有字段信息
  1150. - 包含解析结果的完整数据
  1151. 状态码:
  1152. - 200: 查询成功
  1153. - 400: 请求参数错误
  1154. - 404: 任务不存在
  1155. - 500: 查询失败
  1156. """
  1157. try:
  1158. # 获取查询参数
  1159. task_name = request.args.get('task_name', type=str)
  1160. # 参数验证
  1161. if not task_name:
  1162. return jsonify({
  1163. 'success': False,
  1164. 'message': '任务名称参数不能为空',
  1165. 'data': None
  1166. }), 400
  1167. # 记录请求日志
  1168. logger.info(f"获取解析任务详情请求: task_name={task_name}")
  1169. # 调用核心业务逻辑
  1170. result = get_parse_task_detail(task_name)
  1171. # 返回结果
  1172. return jsonify({
  1173. 'success': result['success'],
  1174. 'message': result['message'],
  1175. 'data': result['data']
  1176. }), result['code']
  1177. except Exception as e:
  1178. # 记录错误日志
  1179. error_msg = f"获取解析任务详情接口失败: {str(e)}"
  1180. logger.error(error_msg, exc_info=True)
  1181. # 返回错误响应
  1182. return jsonify({
  1183. 'success': False,
  1184. 'message': error_msg,
  1185. 'data': None
  1186. }), 500
  1187. # 新增解析任务接口
  1188. @bp.route('/add-parse-task', methods=['POST'])
  1189. def add_parse_task_route():
  1190. """
  1191. 新增解析任务的API接口
  1192. 请求参数:
  1193. - task_type: 任务类型 (form-data字段,必填)
  1194. 可选值:'名片', '简历', '新任命', '招聘', '杂项'
  1195. - files: 文件数组 (multipart/form-data,对于招聘类型可选)
  1196. - created_by: 创建者 (可选,form-data字段)
  1197. 返回:
  1198. - JSON: 包含任务创建结果和上传摘要
  1199. 功能说明:
  1200. - 根据任务类型处理不同格式的文件
  1201. - 名片任务:JPG/PNG格式图片 → talent_photos目录
  1202. - 简历任务:PDF格式文件 → resume_files目录
  1203. - 新任命任务:MD格式文件 → appointment_files目录
  1204. - 招聘任务:数据库记录处理,无需文件上传
  1205. - 杂项任务:任意格式文件 → misc_files目录
  1206. - 使用timestamp+uuid自动生成文件名
  1207. - 在parse_task_repository表中创建待解析任务记录
  1208. 状态码:
  1209. - 200: 所有文件上传成功,任务创建成功
  1210. - 206: 部分文件上传成功,任务创建成功
  1211. - 400: 请求参数错误
  1212. - 500: 服务器内部错误
  1213. """
  1214. try:
  1215. # 获取任务类型参数
  1216. task_type = request.form.get('task_type')
  1217. # 验证任务类型
  1218. if not task_type:
  1219. return jsonify({
  1220. 'success': False,
  1221. 'message': '缺少task_type参数',
  1222. 'data': None
  1223. }), 400
  1224. if task_type not in ['名片', '简历', '新任命', '招聘', '杂项']:
  1225. return jsonify({
  1226. 'success': False,
  1227. 'message': 'task_type参数必须是以下值之一:名片、简历、新任命、招聘、杂项',
  1228. 'data': None
  1229. }), 400
  1230. # 获取创建者信息(可选参数)
  1231. created_by = request.form.get('created_by', 'api_user')
  1232. # 对于招聘类型,不需要文件上传
  1233. if task_type == '招聘':
  1234. # 检查是否误传了文件
  1235. if 'files' in request.files and request.files.getlist('files'):
  1236. return jsonify({
  1237. 'success': False,
  1238. 'message': '招聘类型任务不需要上传文件',
  1239. 'data': None
  1240. }), 400
  1241. # 记录请求日志
  1242. logger.info(f"新增招聘任务请求: 创建者={created_by}")
  1243. # 调用核心业务逻辑
  1244. result = add_parse_task(None, task_type, created_by)
  1245. else:
  1246. # 其他类型需要文件上传
  1247. if 'files' not in request.files:
  1248. return jsonify({
  1249. 'success': False,
  1250. 'message': f'{task_type}任务需要上传文件,请使用files字段上传文件',
  1251. 'data': None
  1252. }), 400
  1253. # 获取上传的文件列表
  1254. uploaded_files = request.files.getlist('files')
  1255. # 检查文件列表是否为空
  1256. if not uploaded_files or len(uploaded_files) == 0:
  1257. return jsonify({
  1258. 'success': False,
  1259. 'message': '文件数组不能为空',
  1260. 'data': None
  1261. }), 400
  1262. # 验证所有文件
  1263. valid_files = []
  1264. for i, file in enumerate(uploaded_files):
  1265. # 检查文件是否为空
  1266. if not file or file.filename == '':
  1267. return jsonify({
  1268. 'success': False,
  1269. 'message': f'第{i+1}个文件为空或未选择',
  1270. 'data': None
  1271. }), 400
  1272. valid_files.append(file)
  1273. # 记录请求日志
  1274. logger.info(f"新增{task_type}任务请求: 文件数量={len(valid_files)}, 创建者={created_by}")
  1275. # 调用核心业务逻辑
  1276. result = add_parse_task(valid_files, task_type, created_by)
  1277. # 根据处理结果设置HTTP状态码
  1278. if result['success']:
  1279. if result['code'] == 200:
  1280. status_code = 200
  1281. elif result['code'] == 206:
  1282. status_code = 206
  1283. else:
  1284. status_code = 200
  1285. else:
  1286. if result['code'] == 400:
  1287. status_code = 400
  1288. else:
  1289. status_code = 500
  1290. # 返回结果
  1291. return jsonify({
  1292. 'success': result['success'],
  1293. 'message': result['message'],
  1294. 'data': result['data']
  1295. }), status_code
  1296. except Exception as e:
  1297. # 记录错误日志
  1298. error_msg = f"新增解析任务接口失败: {str(e)}"
  1299. logger.error(error_msg, exc_info=True)
  1300. # 返回错误响应
  1301. return jsonify({
  1302. 'success': False,
  1303. 'message': error_msg,
  1304. 'data': None
  1305. }), 500
  1306. @bp.route('/execute-parse-task', methods=['POST'])
  1307. def execute_parse_task():
  1308. """
  1309. 执行解析任务接口
  1310. 根据task_type参数调用相应的批量处理函数:
  1311. - 名片: batch_process_business_card_images
  1312. - 简历: batch_parse_resumes
  1313. - 新任命: batch_process_md
  1314. - 招聘: batch_process_menduner_data
  1315. - 杂项: batch_process_images
  1316. 请求参数:
  1317. - task_type (str): 任务类型,可选值:'名片', '简历', '新任命', '招聘', '杂项'
  1318. - data (list): 数据列表,根据task_type不同,数据格式不同
  1319. - publish_time (str, optional): 发布时间,仅新任命任务需要
  1320. """
  1321. try:
  1322. # 获取请求数据
  1323. data = request.get_json()
  1324. if not data:
  1325. return jsonify({
  1326. 'success': False,
  1327. 'message': '请求数据不能为空',
  1328. 'data': None
  1329. }), 400
  1330. # 获取任务类型
  1331. task_type = data.get('task_type', '').strip()
  1332. if not task_type:
  1333. return jsonify({
  1334. 'success': False,
  1335. 'message': 'task_type参数不能为空',
  1336. 'data': None
  1337. }), 400
  1338. # 获取数据列表
  1339. task_data = data.get('data')
  1340. if not task_data:
  1341. return jsonify({
  1342. 'success': False,
  1343. 'message': 'data参数不能为空',
  1344. 'data': None
  1345. }), 400
  1346. # 根据任务类型执行相应的处理函数
  1347. try:
  1348. if task_type == '名片':
  1349. # 调用名片批量处理函数
  1350. result = batch_process_business_card_images(task_data)
  1351. elif task_type == '简历':
  1352. # 调用简历批量处理函数
  1353. result = batch_parse_resumes(task_data)
  1354. elif task_type == '新任命':
  1355. # 获取发布时间参数
  1356. publish_time = data.get('publish_time', '')
  1357. if not publish_time:
  1358. return jsonify({
  1359. 'success': False,
  1360. 'message': '新任命任务需要提供publish_time参数',
  1361. 'data': None
  1362. }), 400
  1363. # 调用新任命批量处理函数
  1364. result = batch_process_md(task_data, publish_time)
  1365. elif task_type == '招聘':
  1366. # 调用招聘数据批量处理函数
  1367. result = batch_process_menduner_data(task_data)
  1368. elif task_type == '杂项':
  1369. # 调用图片批量处理函数(表格类型)
  1370. process_type = data.get('process_type', 'table')
  1371. result = batch_process_images(task_data, process_type)
  1372. else:
  1373. return jsonify({
  1374. 'success': False,
  1375. 'message': f'不支持的任务类型: {task_type},支持的类型:名片、简历、新任命、招聘、杂项',
  1376. 'data': None
  1377. }), 400
  1378. # 记录处理结果日志
  1379. if result.get('success'):
  1380. logging.info(f"执行{task_type}解析任务成功: {result.get('message', '')}")
  1381. # ===== 精简:只根据id字段唯一定位任务记录 =====
  1382. from app.core.data_parse.parse_system import db, ParseTaskRepository
  1383. task_id = data.get('id')
  1384. if task_id:
  1385. task_obj = ParseTaskRepository.query.filter_by(id=task_id).first()
  1386. if task_obj:
  1387. task_obj.task_status = '成功'
  1388. task_obj.parse_result = result.get('data')
  1389. db.session.commit()
  1390. logging.info(f"已更新解析任务记录: id={getattr(task_obj, 'id', None)}, 状态=成功")
  1391. else:
  1392. logging.error(f"执行{task_type}解析任务失败: {result.get('message', '')}")
  1393. # 确定HTTP状态码
  1394. if result.get('success'):
  1395. # 检查是否有部分成功的情况
  1396. if 'code' in result:
  1397. status_code = result['code']
  1398. elif 'summary' in result.get('data', {}):
  1399. # 检查处理摘要
  1400. summary = result['data']['summary']
  1401. if summary.get('failed_count', 0) > 0 and summary.get('success_count', 0) > 0:
  1402. status_code = 206 # 部分成功
  1403. else:
  1404. status_code = 200 # 完全成功
  1405. else:
  1406. status_code = 200
  1407. else:
  1408. status_code = 500
  1409. return jsonify({
  1410. 'success': result.get('success', False),
  1411. 'message': result.get('message', '处理完成'),
  1412. 'data': result.get('data')
  1413. }), status_code
  1414. except Exception as process_error:
  1415. error_msg = f"执行{task_type}解析任务时发生错误: {str(process_error)}"
  1416. logging.error(error_msg, exc_info=True)
  1417. return jsonify({
  1418. 'success': False,
  1419. 'message': error_msg,
  1420. 'data': None
  1421. }), 500
  1422. except Exception as e:
  1423. # 记录错误日志
  1424. error_msg = f"执行解析任务接口失败: {str(e)}"
  1425. logging.error(error_msg, exc_info=True)
  1426. # 返回错误响应
  1427. return jsonify({
  1428. 'success': False,
  1429. 'message': error_msg,
  1430. 'data': None
  1431. }), 500
  1432. @bp.route('/add-parsed-talents', methods=['POST'])
  1433. def add_parsed_talents_route():
  1434. """
  1435. 处理解析任务响应数据并写入人才信息接口
  1436. 请求参数:
  1437. - api_response_data: execute-parse-task API的完整返回数据 (JSON格式)
  1438. 请求体示例:
  1439. {
  1440. "success": true,
  1441. "message": "处理完成",
  1442. "data": {
  1443. "summary": {
  1444. "total_files": 5,
  1445. "success_count": 4,
  1446. "failed_count": 1,
  1447. "success_rate": 80.0
  1448. },
  1449. "results": [
  1450. {
  1451. "index": 0,
  1452. "success": true,
  1453. "data": {
  1454. "name_zh": "张三",
  1455. "title_zh": "经理",
  1456. "hotel_zh": "某酒店"
  1457. }
  1458. }
  1459. ],
  1460. "processed_time": "2025-01-18T10:30:00"
  1461. }
  1462. }
  1463. 返回:
  1464. - JSON: 包含批量处理结果和状态信息
  1465. 功能说明:
  1466. - 接收 execute-parse-task API 的完整返回数据
  1467. - 自动识别和处理不同格式的人才数据(单人/批量)
  1468. - 调用 add_single_talent 函数将人才信息写入 business_cards 表
  1469. - 支持新任命等包含多个人员信息的批量数据
  1470. - 提供详细的处理统计和结果追踪
  1471. - 保留原始API响应数据用于调试
  1472. 状态码:
  1473. - 200: 全部处理成功
  1474. - 206: 部分处理成功
  1475. - 400: 请求参数错误
  1476. - 500: 服务器内部错误
  1477. """
  1478. try:
  1479. # 检查请求是否为 JSON 格式
  1480. if not request.is_json:
  1481. return jsonify({
  1482. 'success': False,
  1483. 'message': '请求必须是 JSON 格式',
  1484. 'data': None
  1485. }), 400
  1486. # 获取请求数据
  1487. api_response_data = request.get_json()
  1488. # 基本参数验证
  1489. if not api_response_data:
  1490. return jsonify({
  1491. 'success': False,
  1492. 'message': '请求数据不能为空',
  1493. 'data': None
  1494. }), 400
  1495. # 验证数据格式
  1496. if not isinstance(api_response_data, dict):
  1497. return jsonify({
  1498. 'success': False,
  1499. 'message': '请求数据必须是JSON对象格式',
  1500. 'data': None
  1501. }), 400
  1502. # 记录请求日志
  1503. total_results = 0
  1504. if api_response_data.get('data') and api_response_data['data'].get('results'):
  1505. total_results = len(api_response_data['data']['results'])
  1506. logger.info(f"收到处理解析任务响应数据请求,包含 {total_results} 条结果记录")
  1507. # 调用核心业务逻辑
  1508. result = add_parsed_talents(api_response_data)
  1509. # 根据处理结果设置HTTP状态码
  1510. if result.get('success', False):
  1511. if result.get('code') == 200:
  1512. status_code = 200 # 全部成功
  1513. elif result.get('code') == 206:
  1514. status_code = 206 # 部分成功
  1515. else:
  1516. status_code = 200 # 默认成功
  1517. else:
  1518. if result.get('code') == 400:
  1519. status_code = 400 # 参数错误
  1520. else:
  1521. status_code = 500 # 服务器错误
  1522. # 记录处理结果日志
  1523. if result.get('success'):
  1524. data_summary = result.get('data', {}).get('summary', {})
  1525. success_count = data_summary.get('success_count', 0)
  1526. failed_count = data_summary.get('failed_count', 0)
  1527. logger.info(f"处理解析任务响应数据完成: 成功 {success_count} 条,失败 {failed_count} 条")
  1528. else:
  1529. logger.error(f"处理解析任务响应数据失败: {result.get('message', '未知错误')}")
  1530. # 返回结果
  1531. return jsonify({
  1532. 'success': result.get('success', False),
  1533. 'message': result.get('message', '处理完成'),
  1534. 'data': result.get('data')
  1535. }), status_code
  1536. except Exception as e:
  1537. # 记录错误日志
  1538. error_msg = f"处理解析任务响应数据接口失败: {str(e)}"
  1539. logger.error(error_msg, exc_info=True)
  1540. # 返回错误响应
  1541. return jsonify({
  1542. 'success': False,
  1543. 'message': error_msg,
  1544. 'data': None
  1545. }), 500