routes.py 91 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782
  1. from flask import jsonify, request, make_response, Blueprint, current_app, send_file
  2. from datetime import datetime
  3. import json
  4. from app.core.data_parse.time_utils import get_east_asia_time_naive, get_east_asia_time_str, get_east_asia_timestamp, get_east_asia_isoformat, get_east_asia_date_str
  5. from app.api.data_parse import bp
  6. from app.core.data_parse.parse_system import (
  7. update_business_card,
  8. get_business_cards,
  9. update_business_card_status,
  10. create_talent_tag,
  11. get_talent_tag_list,
  12. update_talent_tag,
  13. delete_talent_tag,
  14. query_neo4j_graph,
  15. talent_get_tags,
  16. talent_update_tags,
  17. get_business_card,
  18. search_business_cards_by_mobile,
  19. get_duplicate_records,
  20. process_duplicate_record,
  21. get_duplicate_record_detail,
  22. fix_broken_duplicate_records
  23. )
  24. # 导入解析任务相关函数
  25. from app.core.data_parse.parse_task import (
  26. get_parse_tasks,
  27. get_parse_task_detail,
  28. add_parse_task,
  29. add_parsed_talents,
  30. web_url_crawl
  31. )
  32. # 导入酒店管理相关函数
  33. from app.core.data_parse.hotel_management import (
  34. get_hotel_positions_list,
  35. add_hotel_positions,
  36. update_hotel_positions,
  37. query_hotel_positions,
  38. delete_hotel_positions,
  39. get_hotel_group_brands_list,
  40. add_hotel_group_brands,
  41. update_hotel_group_brands,
  42. query_hotel_group_brands,
  43. delete_hotel_group_brands
  44. )
  45. # 导入名片处理函数
  46. from app.core.data_parse.parse_card import delete_business_card, batch_process_business_card_images
  47. # 导入网页文本解析函数
  48. from app.core.data_parse.parse_web import batch_process_md
  49. # 导入简历解析函数
  50. from app.core.data_parse.parse_resume import batch_parse_resumes
  51. # 导入门墩儿数据处理函数
  52. from app.core.data_parse.parse_menduner import batch_process_menduner_data
  53. # 导入图片批量处理函数
  54. from app.core.data_parse.parse_pic import batch_process_images
  55. # 导入日历相关函数
  56. from app.core.data_parse.calendar import get_calendar_by_date
  57. # 导入微信认证相关函数
  58. from app.core.data_parse.calendar import (
  59. register_wechat_user,
  60. login_wechat_user,
  61. logout_wechat_user,
  62. get_wechat_user_info,
  63. update_wechat_user_info
  64. )
  65. from app.config.config import DevelopmentConfig, ProductionConfig
  66. import logging
  67. import boto3
  68. from botocore.config import Config
  69. from botocore.exceptions import ClientError
  70. from io import BytesIO
  71. import base64
  72. import os
  73. import urllib.parse
  74. from minio import Minio
  75. from app.models.parse_models import ParseTaskRepository
  76. from app.core.data_parse.parse_system import db
  77. # Define logger
  78. logger = logging.getLogger(__name__)
  79. # For failure responses
  80. def failed(message, code=500):
  81. return {
  82. 'success': False,
  83. 'message': message,
  84. 'data': None
  85. }, code
  86. # 根据环境选择配置
  87. if os.environ.get('FLASK_ENV') == 'production':
  88. config = ProductionConfig()
  89. else:
  90. config = DevelopmentConfig()
  91. # 使用配置变量
  92. minio_url = f"{'https' if config.MINIO_SECURE else 'http'}://{config.MINIO_HOST}"
  93. minio_access_key = config.MINIO_USER
  94. minio_secret_key = config.MINIO_PASSWORD
  95. minio_bucket = config.MINIO_BUCKET
  96. use_ssl = config.MINIO_SECURE
  97. def get_minio_client():
  98. """获取 MinIO 客户端实例"""
  99. return Minio(
  100. '192.168.3.143:9000',
  101. access_key=config.MINIO_USER,
  102. secret_key=config.MINIO_PASSWORD,
  103. secure=config.MINIO_SECURE
  104. )
  105. # 更新名片信息接口
  106. @bp.route('/business-cards/<int:card_id>', methods=['PUT'])
  107. def update_business_card_route(card_id):
  108. """
  109. 更新名片信息的API接口
  110. 路径参数:
  111. - card_id: 名片记录ID
  112. 请求参数:
  113. - JSON格式的名片信息
  114. 返回:
  115. - JSON: 包含更新后的名片信息和处理状态
  116. """
  117. # 获取请求数据
  118. data = request.json
  119. if not data:
  120. return jsonify({
  121. 'success': False,
  122. 'message': '请求数据为空',
  123. 'data': None
  124. }), 400
  125. # 调用业务逻辑函数处理更新
  126. result = update_business_card(card_id, data)
  127. # 根据处理结果设置HTTP状态码
  128. status_code = 200 if result['success'] else 500
  129. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  130. status_code = 404
  131. return jsonify(result), status_code
  132. # 获取所有名片记录的API接口
  133. @bp.route('/get-business-cards', methods=['GET'])
  134. def get_business_cards_route():
  135. """
  136. 获取所有名片记录的API接口
  137. 返回:
  138. - JSON: 包含名片记录列表和处理状态
  139. """
  140. # 调用业务逻辑函数获取名片列表
  141. result = get_business_cards()
  142. # 根据处理结果设置HTTP状态码
  143. status_code = 200 if result['success'] else 500
  144. return jsonify(result), status_code
  145. @bp.route('/update-business-cards/<int:card_id>/status', methods=['PUT'])
  146. def update_business_card_status_route(card_id):
  147. """
  148. 更新名片状态的API接口
  149. 路径参数:
  150. - card_id: 名片记录ID
  151. 请求参数:
  152. - JSON格式,包含status字段
  153. 返回:
  154. - JSON: 包含更新后的名片信息和处理状态
  155. """
  156. # 获取请求数据
  157. data = request.json
  158. if not data or 'status' not in data:
  159. return jsonify({
  160. 'success': False,
  161. 'message': '请求数据为空或缺少status字段',
  162. 'data': None
  163. }), 400
  164. status = data['status']
  165. # 调用业务逻辑函数处理状态更新
  166. result = update_business_card_status(card_id, status)
  167. # 根据处理结果设置HTTP状态码
  168. status_code = 200 if result['success'] else 500
  169. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  170. status_code = 404
  171. return jsonify(result), status_code
  172. # 从MinIO获取名片图片的API接口
  173. @bp.route('/business-cards/image/<path:image_path>', methods=['GET'])
  174. def get_business_card_image(image_path):
  175. """
  176. 从MinIO获取名片图片的API接口
  177. 路径参数:
  178. - image_path: MinIO中的图片路径
  179. 返回:
  180. - 图片数据流
  181. """
  182. try:
  183. # 记录下载请求信息,便于调试
  184. logger.info(f"获取名片图片请求: {image_path}")
  185. # 获取 MinIO 客户端
  186. minio_client = get_minio_client()
  187. if not minio_client:
  188. return jsonify(failed("MinIO客户端初始化失败")), 500
  189. try:
  190. # 使用正确的MinIO客户端方法
  191. data = minio_client.get_object(minio_bucket, image_path)
  192. # 创建内存文件流
  193. file_stream = BytesIO(data.read())
  194. # 获取文件名
  195. file_name = image_path.split('/')[-1]
  196. # 返回文件
  197. return send_file(
  198. file_stream,
  199. as_attachment=False, # 设置为False,让浏览器直接显示图片
  200. download_name=file_name,
  201. mimetype='image/jpeg' # 根据实际图片类型设置
  202. )
  203. except Exception as e:
  204. logger.error(f"MinIO获取文件失败: {str(e)}")
  205. return jsonify(failed(f"文件获取失败: {str(e)}")), 404
  206. except Exception as e:
  207. logger.error(f"文件下载失败: {str(e)}")
  208. return jsonify(failed(str(e))), 500
  209. finally:
  210. # 确保关闭数据流
  211. if 'data' in locals():
  212. data.close()
  213. # 创建人才标签接口
  214. @bp.route('/create-talent-tag', methods=['POST'])
  215. def create_talent_tag_route():
  216. """
  217. 创建人才标签的API接口
  218. 请求参数:
  219. - JSON格式,包含以下字段:
  220. - name_zh: 标签名称
  221. - category: 标签分类
  222. - description: 标签描述
  223. - status: 启用状态,默认为'active'
  224. 返回:
  225. - JSON: 包含创建结果和标签信息
  226. """
  227. try:
  228. # 获取请求数据
  229. data = request.get_json()
  230. if not data:
  231. return jsonify({
  232. 'success': False,
  233. 'message': '请求数据为空',
  234. 'data': None
  235. }), 400
  236. # 验证必要字段
  237. if 'name_zh' not in data or not data['name_zh']:
  238. return jsonify({
  239. 'success': False,
  240. 'message': '标签名称不能为空',
  241. 'data': None
  242. }), 400
  243. # 处理分类字段,如果未提供则设置默认值
  244. if 'category' not in data or not data['category']:
  245. data['category'] = '未分类'
  246. # 调用业务逻辑函数处理创建
  247. result = create_talent_tag(data)
  248. # 根据处理结果设置HTTP状态码
  249. status_code = 200 if result['success'] else 500
  250. return jsonify(result), status_code
  251. except Exception as e:
  252. logger.error(f"创建人才标签失败: {str(e)}")
  253. return jsonify({
  254. 'success': False,
  255. 'message': f'创建人才标签失败: {str(e)}',
  256. 'data': None
  257. }), 500
  258. # 获取人才标签列表接口
  259. @bp.route('/get-talent-tag-list', methods=['GET'])
  260. def get_talent_tag_list_route():
  261. """
  262. 获取人才标签列表的API接口
  263. 返回:
  264. - JSON: 包含人才标签列表和处理状态
  265. """
  266. try:
  267. # 调用业务逻辑函数获取人才标签列表
  268. result = get_talent_tag_list()
  269. # 根据处理结果设置HTTP状态码
  270. status_code = 200 if result['success'] else 500
  271. return jsonify(result), status_code
  272. except Exception as e:
  273. logger.error(f"获取人才标签列表失败: {str(e)}")
  274. return jsonify({
  275. 'success': False,
  276. 'message': f'获取人才标签列表失败: {str(e)}',
  277. 'data': []
  278. }), 500
  279. # 更新人才标签接口
  280. @bp.route('/update-talent-tag/<int:tag_id>', methods=['PUT'])
  281. def update_talent_tag_route(tag_id):
  282. """
  283. 更新人才标签的API接口
  284. 路径参数:
  285. - tag_id: 标签节点ID
  286. 请求参数:
  287. - JSON格式,可能包含以下字段:
  288. - name_zh: 标签名称
  289. - category: 标签分类
  290. - description: 标签描述
  291. - status: 启用状态
  292. 返回:
  293. - JSON: 包含更新结果和标签信息
  294. """
  295. try:
  296. # 获取请求数据
  297. data = request.get_json()
  298. if not data:
  299. return jsonify({
  300. 'success': False,
  301. 'message': '请求数据为空',
  302. 'data': None
  303. }), 400
  304. # 调用业务逻辑函数处理更新
  305. result = update_talent_tag(tag_id, data)
  306. # 根据处理结果设置HTTP状态码
  307. if not result['success']:
  308. if result['code'] == 404:
  309. status_code = 404
  310. elif result['code'] == 400:
  311. status_code = 400
  312. else:
  313. status_code = 500
  314. else:
  315. status_code = 200
  316. return jsonify(result), status_code
  317. except Exception as e:
  318. logger.error(f"更新人才标签失败: {str(e)}")
  319. return jsonify({
  320. 'success': False,
  321. 'message': f'更新人才标签失败: {str(e)}',
  322. 'data': None
  323. }), 500
  324. # 删除人才标签接口
  325. @bp.route('/delete-talent-tag/<int:tag_id>', methods=['DELETE'])
  326. def delete_talent_tag_route(tag_id):
  327. """
  328. 删除人才标签的API接口
  329. 路径参数:
  330. - tag_id: 标签节点ID
  331. 返回:
  332. - JSON: 包含删除结果和被删除的标签信息
  333. """
  334. try:
  335. # 调用业务逻辑函数执行删除
  336. result = delete_talent_tag(tag_id)
  337. # 根据处理结果设置HTTP状态码
  338. if not result['success']:
  339. if result['code'] == 404:
  340. status_code = 404
  341. else:
  342. status_code = 500
  343. else:
  344. status_code = 200
  345. return jsonify(result), status_code
  346. except Exception as e:
  347. logger.error(f"删除人才标签失败: {str(e)}")
  348. return jsonify({
  349. 'success': False,
  350. 'message': f'删除人才标签失败: {str(e)}',
  351. 'data': None
  352. }), 500
  353. @bp.route('/query-kg', methods=['POST'])
  354. def query_kg():
  355. """
  356. 查询知识图谱API接口
  357. 请求参数:
  358. - query_requirement: 查询需求描述(JSON格式)
  359. 返回:
  360. - JSON: 包含查询结果和处理状态
  361. """
  362. try:
  363. # 获取请求数据
  364. data = request.json
  365. if not data or 'query_requirement' not in data:
  366. return jsonify({
  367. 'code': 400,
  368. 'success': False,
  369. 'message': '请求数据为空或缺少query_requirement字段',
  370. 'data': []
  371. }), 400
  372. query_requirement = data['query_requirement']
  373. # 调用业务逻辑函数执行查询
  374. result = query_neo4j_graph(query_requirement)
  375. # 根据处理结果设置HTTP状态码
  376. status_code = 200 if result['success'] else 500
  377. return jsonify(result), status_code
  378. except Exception as e:
  379. logger.error(f"查询知识图谱失败: {str(e)}")
  380. return jsonify({
  381. 'code': 500,
  382. 'success': False,
  383. 'message': f"查询知识图谱失败: {str(e)}",
  384. 'data': []
  385. }), 500
  386. @bp.route('/talent-get-tags/<int:talent_id>', methods=['GET'])
  387. def talent_get_tags_route(talent_id):
  388. """
  389. 获取人才标签的API接口
  390. 路径参数:
  391. - talent_id: 人才节点ID
  392. 返回:
  393. - JSON: 包含人才关联的标签列表和处理状态
  394. """
  395. try:
  396. # 调用业务逻辑函数获取人才标签
  397. result = talent_get_tags(talent_id)
  398. # 根据处理结果设置HTTP状态码
  399. status_code = 200 if result['success'] else 500
  400. return jsonify(result), status_code
  401. except Exception as e:
  402. logger.error(f"获取人才标签失败: {str(e)}")
  403. return jsonify({
  404. 'code': 500,
  405. 'success': False,
  406. 'message': f"获取人才标签失败: {str(e)}",
  407. 'data': []
  408. }), 500
  409. @bp.route('/talent-update-tags', methods=['POST'])
  410. def talent_update_tags_route():
  411. """
  412. 更新人才标签关系的API接口
  413. 请求参数:
  414. - JSON数组,包含talent和tag字段的对象列表
  415. 例如: [
  416. {"talent": 12345, "tag": "市场营销"},
  417. {"talent": 12345, "tag": "酒店管理"}
  418. ]
  419. 返回:
  420. - JSON: 包含更新结果的状态信息
  421. """
  422. try:
  423. # 获取请求数据
  424. data = request.json
  425. if not data:
  426. return jsonify({
  427. 'code': 400,
  428. 'success': False,
  429. 'message': '请求数据为空',
  430. 'data': None
  431. }), 400
  432. # 调用业务逻辑函数处理标签关系更新
  433. result = talent_update_tags(data)
  434. # 根据处理结果设置HTTP状态码
  435. if result['code'] == 200:
  436. status_code = 200
  437. elif result['code'] == 206:
  438. status_code = 206 # Partial Content
  439. elif result['code'] == 400:
  440. status_code = 400 # Bad Request
  441. elif result['code'] == 404:
  442. status_code = 404 # Not Found
  443. else:
  444. status_code = 500 # Internal Server Error
  445. return jsonify(result), status_code
  446. except Exception as e:
  447. logger.error(f"更新人才标签关系失败: {str(e)}")
  448. return jsonify({
  449. 'code': 500,
  450. 'success': False,
  451. 'message': f"更新人才标签关系失败: {str(e)}",
  452. 'data': None
  453. }), 500
  454. # 获取单个名片记录的API接口
  455. @bp.route('/get-business-card/<int:card_id>', methods=['GET'])
  456. def get_business_card_route(card_id):
  457. """
  458. 获取单个名片记录的API接口
  459. 路径参数:
  460. - card_id: 名片记录ID
  461. 返回:
  462. - JSON: 包含名片记录信息和处理状态
  463. """
  464. # 调用业务逻辑函数获取名片记录
  465. result = get_business_card(card_id)
  466. # 根据处理结果设置HTTP状态码
  467. if not result['success']:
  468. if result['code'] == 404:
  469. status_code = 404
  470. else:
  471. status_code = 500
  472. else:
  473. status_code = 200
  474. return jsonify(result), status_code
  475. @bp.route('/search-business-cards-by-mobile', methods=['GET'])
  476. def search_business_cards_by_mobile_route():
  477. """
  478. 根据手机号码搜索名片记录的API接口
  479. 查询参数:
  480. - mobile: 要搜索的手机号码
  481. 返回:
  482. - JSON: 包含搜索到的名片记录列表和处理状态
  483. 示例:
  484. GET /search-business-cards-by-mobile?mobile=13800138000
  485. """
  486. try:
  487. # 获取查询参数
  488. mobile_number = request.args.get('mobile', '').strip()
  489. if not mobile_number:
  490. return jsonify({
  491. 'success': False,
  492. 'message': '请提供要搜索的手机号码',
  493. 'data': []
  494. }), 400
  495. # 调用业务逻辑函数搜索名片记录
  496. result = search_business_cards_by_mobile(mobile_number)
  497. # 根据处理结果设置HTTP状态码
  498. if result['code'] == 200:
  499. status_code = 200
  500. elif result['code'] == 400:
  501. status_code = 400
  502. else:
  503. status_code = 500
  504. return jsonify(result), status_code
  505. except Exception as e:
  506. # 处理未预期的异常
  507. error_msg = f"根据手机号码搜索名片时发生错误: {str(e)}"
  508. logger.error(error_msg, exc_info=True)
  509. return jsonify({
  510. 'success': False,
  511. 'message': error_msg,
  512. 'data': []
  513. }), 500
  514. @bp.route('/get-hotel-positions-list', methods=['GET'])
  515. def get_hotel_positions_list_route():
  516. """
  517. 获取酒店职位数据表全部记录的API接口
  518. 返回:
  519. - JSON: 包含酒店职位记录列表和处理状态
  520. """
  521. try:
  522. # 调用业务逻辑函数获取酒店职位列表
  523. result = get_hotel_positions_list()
  524. # 根据处理结果设置HTTP状态码
  525. status_code = 200 if result['success'] else 500
  526. return jsonify(result), status_code
  527. except Exception as e:
  528. # 处理未预期的异常
  529. error_msg = f"获取酒店职位列表时发生错误: {str(e)}"
  530. logger.error(error_msg, exc_info=True)
  531. return jsonify({
  532. 'success': False,
  533. 'message': error_msg,
  534. 'data': [],
  535. 'count': 0
  536. }), 500
  537. @bp.route('/add-hotel-positions', methods=['POST'])
  538. def add_hotel_positions_route():
  539. """
  540. 新增酒店职位数据表记录的API接口
  541. 请求参数:
  542. - JSON格式,包含以下字段:
  543. - department_zh: 部门中文名称 (必填)
  544. - department_en: 部门英文名称 (必填)
  545. - position_zh: 职位中文名称 (必填)
  546. - position_en: 职位英文名称 (必填)
  547. - position_abbr: 职位英文缩写 (可选)
  548. - level_zh: 职级中文名称 (必填)
  549. - level_en: 职级英文名称 (必填)
  550. - created_by: 创建者 (可选)
  551. - updated_by: 更新者 (可选)
  552. - status: 状态 (可选)
  553. 返回:
  554. - JSON: 包含创建结果和职位信息
  555. """
  556. try:
  557. # 获取请求数据
  558. data = request.get_json()
  559. if not data:
  560. return jsonify({
  561. 'success': False,
  562. 'message': '请求数据为空',
  563. 'data': None
  564. }), 400
  565. # 调用业务逻辑函数处理创建
  566. result = add_hotel_positions(data)
  567. # 根据处理结果设置HTTP状态码
  568. if result['code'] == 200:
  569. status_code = 201 # Created
  570. elif result['code'] == 400:
  571. status_code = 400 # Bad Request
  572. elif result['code'] == 409:
  573. status_code = 409 # Conflict
  574. else:
  575. status_code = 500 # Internal Server Error
  576. return jsonify(result), status_code
  577. except Exception as e:
  578. # 处理未预期的异常
  579. error_msg = f"创建酒店职位记录时发生错误: {str(e)}"
  580. logger.error(error_msg, exc_info=True)
  581. return jsonify({
  582. 'success': False,
  583. 'message': error_msg,
  584. 'data': None
  585. }), 500
  586. @bp.route('/update-hotel-positions/<int:position_id>', methods=['PUT'])
  587. def update_hotel_positions_route(position_id):
  588. """
  589. 修改酒店职位数据表记录的API接口
  590. 路径参数:
  591. - position_id: 职位记录ID
  592. 请求参数:
  593. - JSON格式,可能包含以下字段:
  594. - department_zh: 部门中文名称
  595. - department_en: 部门英文名称
  596. - position_zh: 职位中文名称
  597. - position_en: 职位英文名称
  598. - position_abbr: 职位英文缩写
  599. - level_zh: 职级中文名称
  600. - level_en: 职级英文名称
  601. - updated_by: 更新者
  602. - status: 状态
  603. 返回:
  604. - JSON: 包含更新结果和职位信息
  605. """
  606. try:
  607. # 获取请求数据
  608. data = request.get_json()
  609. if not data:
  610. return jsonify({
  611. 'success': False,
  612. 'message': '请求数据为空',
  613. 'data': None
  614. }), 400
  615. # 调用业务逻辑函数处理更新
  616. result = update_hotel_positions(position_id, data)
  617. # 根据处理结果设置HTTP状态码
  618. if result['code'] == 200:
  619. status_code = 200 # OK
  620. elif result['code'] == 400:
  621. status_code = 400 # Bad Request
  622. elif result['code'] == 404:
  623. status_code = 404 # Not Found
  624. elif result['code'] == 409:
  625. status_code = 409 # Conflict
  626. else:
  627. status_code = 500 # Internal Server Error
  628. return jsonify(result), status_code
  629. except Exception as e:
  630. # 处理未预期的异常
  631. error_msg = f"更新酒店职位记录时发生错误: {str(e)}"
  632. logger.error(error_msg, exc_info=True)
  633. return jsonify({
  634. 'success': False,
  635. 'message': error_msg,
  636. 'data': None
  637. }), 500
  638. @bp.route('/query-hotel-positions/<int:position_id>', methods=['GET'])
  639. def query_hotel_positions_route(position_id):
  640. """
  641. 查找指定ID的酒店职位数据表记录的API接口
  642. 路径参数:
  643. - position_id: 职位记录ID
  644. 返回:
  645. - JSON: 包含查找结果和职位信息
  646. """
  647. try:
  648. # 调用业务逻辑函数查找职位记录
  649. result = query_hotel_positions(position_id)
  650. # 根据处理结果设置HTTP状态码
  651. if result['code'] == 200:
  652. status_code = 200 # OK
  653. elif result['code'] == 404:
  654. status_code = 404 # Not Found
  655. else:
  656. status_code = 500 # Internal Server Error
  657. return jsonify(result), status_code
  658. except Exception as e:
  659. # 处理未预期的异常
  660. error_msg = f"查找酒店职位记录时发生错误: {str(e)}"
  661. logger.error(error_msg, exc_info=True)
  662. return jsonify({
  663. 'success': False,
  664. 'message': error_msg,
  665. 'data': None
  666. }), 500
  667. @bp.route('/delete-hotel-positions/<int:position_id>', methods=['DELETE'])
  668. def delete_hotel_positions_route(position_id):
  669. """
  670. 删除指定ID的酒店职位数据表记录的API接口
  671. 路径参数:
  672. - position_id: 职位记录ID
  673. 返回:
  674. - JSON: 包含删除结果和被删除的职位信息
  675. """
  676. try:
  677. # 调用业务逻辑函数删除职位记录
  678. result = delete_hotel_positions(position_id)
  679. # 根据处理结果设置HTTP状态码
  680. if result['code'] == 200:
  681. status_code = 200 # OK
  682. elif result['code'] == 404:
  683. status_code = 404 # Not Found
  684. else:
  685. status_code = 500 # Internal Server Error
  686. return jsonify(result), status_code
  687. except Exception as e:
  688. # 处理未预期的异常
  689. error_msg = f"删除酒店职位记录时发生错误: {str(e)}"
  690. logger.error(error_msg, exc_info=True)
  691. return jsonify({
  692. 'success': False,
  693. 'message': error_msg,
  694. 'data': None
  695. }), 500
  696. @bp.route('/get-hotel-group-brands-list', methods=['GET'])
  697. def get_hotel_group_brands_list_route():
  698. """
  699. 获取酒店集团子品牌数据表全部记录的API接口
  700. 返回:
  701. - JSON: 包含酒店集团品牌记录列表和处理状态
  702. """
  703. try:
  704. # 调用业务逻辑函数获取酒店集团品牌列表
  705. result = get_hotel_group_brands_list()
  706. # 根据处理结果设置HTTP状态码
  707. status_code = 200 if result['success'] else 500
  708. return jsonify(result), status_code
  709. except Exception as e:
  710. # 处理未预期的异常
  711. error_msg = f"获取酒店集团品牌列表时发生错误: {str(e)}"
  712. logger.error(error_msg, exc_info=True)
  713. return jsonify({
  714. 'success': False,
  715. 'message': error_msg,
  716. 'data': [],
  717. 'count': 0
  718. }), 500
  719. @bp.route('/add-hotel-group-brands', methods=['POST'])
  720. def add_hotel_group_brands_route():
  721. """
  722. 新增酒店集团子品牌数据表记录的API接口
  723. 请求参数:
  724. - JSON格式,包含以下字段:
  725. - group_name_en: 集团英文名称 (必填)
  726. - group_name_zh: 集团中文名称 (必填)
  727. - brand_name_en: 品牌英文名称 (必填)
  728. - brand_name_zh: 品牌中文名称 (必填)
  729. - positioning_level_en: 定位级别英文名称 (必填)
  730. - positioning_level_zh: 定位级别中文名称 (必填)
  731. - created_by: 创建者 (可选)
  732. - updated_by: 更新者 (可选)
  733. - status: 状态 (可选)
  734. 返回:
  735. - JSON: 包含创建结果和品牌信息
  736. """
  737. try:
  738. # 获取请求数据
  739. data = request.get_json()
  740. if not data:
  741. return jsonify({
  742. 'success': False,
  743. 'message': '请求数据为空',
  744. 'data': None
  745. }), 400
  746. # 调用业务逻辑函数处理创建
  747. result = add_hotel_group_brands(data)
  748. # 根据处理结果设置HTTP状态码
  749. if result['code'] == 200:
  750. status_code = 201 # Created
  751. elif result['code'] == 400:
  752. status_code = 400 # Bad Request
  753. elif result['code'] == 409:
  754. status_code = 409 # Conflict
  755. else:
  756. status_code = 500 # Internal Server Error
  757. return jsonify(result), status_code
  758. except Exception as e:
  759. # 处理未预期的异常
  760. error_msg = f"创建酒店集团品牌记录时发生错误: {str(e)}"
  761. logger.error(error_msg, exc_info=True)
  762. return jsonify({
  763. 'success': False,
  764. 'message': error_msg,
  765. 'data': None
  766. }), 500
  767. @bp.route('/update-hotel-group-brands/<int:brand_id>', methods=['PUT'])
  768. def update_hotel_group_brands_route(brand_id):
  769. """
  770. 修改酒店集团子品牌数据表记录的API接口
  771. 路径参数:
  772. - brand_id: 品牌记录ID
  773. 请求参数:
  774. - JSON格式,可能包含以下字段:
  775. - group_name_en: 集团英文名称
  776. - group_name_zh: 集团中文名称
  777. - brand_name_en: 品牌英文名称
  778. - brand_name_zh: 品牌中文名称
  779. - positioning_level_en: 定位级别英文名称
  780. - positioning_level_zh: 定位级别中文名称
  781. - updated_by: 更新者
  782. - status: 状态
  783. 返回:
  784. - JSON: 包含更新结果和品牌信息
  785. """
  786. try:
  787. # 获取请求数据
  788. data = request.get_json()
  789. if not data:
  790. return jsonify({
  791. 'success': False,
  792. 'message': '请求数据为空',
  793. 'data': None
  794. }), 400
  795. # 调用业务逻辑函数处理更新
  796. result = update_hotel_group_brands(brand_id, data)
  797. # 根据处理结果设置HTTP状态码
  798. if result['code'] == 200:
  799. status_code = 200 # OK
  800. elif result['code'] == 400:
  801. status_code = 400 # Bad Request
  802. elif result['code'] == 404:
  803. status_code = 404 # Not Found
  804. elif result['code'] == 409:
  805. status_code = 409 # Conflict
  806. else:
  807. status_code = 500 # Internal Server Error
  808. return jsonify(result), status_code
  809. except Exception as e:
  810. # 处理未预期的异常
  811. error_msg = f"更新酒店集团品牌记录时发生错误: {str(e)}"
  812. logger.error(error_msg, exc_info=True)
  813. return jsonify({
  814. 'success': False,
  815. 'message': error_msg,
  816. 'data': None
  817. }), 500
  818. @bp.route('/query-hotel-group-brands/<int:brand_id>', methods=['GET'])
  819. def query_hotel_group_brands_route(brand_id):
  820. """
  821. 查找指定ID的酒店集团子品牌数据表记录的API接口
  822. 路径参数:
  823. - brand_id: 品牌记录ID
  824. 返回:
  825. - JSON: 包含查找结果和品牌信息
  826. """
  827. try:
  828. # 调用业务逻辑函数查找品牌记录
  829. result = query_hotel_group_brands(brand_id)
  830. # 根据处理结果设置HTTP状态码
  831. if result['code'] == 200:
  832. status_code = 200 # OK
  833. elif result['code'] == 404:
  834. status_code = 404 # Not Found
  835. else:
  836. status_code = 500 # Internal Server Error
  837. return jsonify(result), status_code
  838. except Exception as e:
  839. # 处理未预期的异常
  840. error_msg = f"查找酒店集团品牌记录时发生错误: {str(e)}"
  841. logger.error(error_msg, exc_info=True)
  842. return jsonify({
  843. 'success': False,
  844. 'message': error_msg,
  845. 'data': None
  846. }), 500
  847. @bp.route('/delete-hotel-group-brands/<int:brand_id>', methods=['DELETE'])
  848. def delete_hotel_group_brands_route(brand_id):
  849. """
  850. 删除指定ID的酒店集团子品牌数据表记录的API接口
  851. 路径参数:
  852. - brand_id: 品牌记录ID
  853. 返回:
  854. - JSON: 包含删除结果和被删除的品牌信息
  855. """
  856. try:
  857. # 调用业务逻辑函数删除品牌记录
  858. result = delete_hotel_group_brands(brand_id)
  859. # 根据处理结果设置HTTP状态码
  860. if result['code'] == 200:
  861. status_code = 200 # OK
  862. elif result['code'] == 404:
  863. status_code = 404 # Not Found
  864. else:
  865. status_code = 500 # Internal Server Error
  866. return jsonify(result), status_code
  867. except Exception as e:
  868. # 处理未预期的异常
  869. error_msg = f"删除酒店集团品牌记录时发生错误: {str(e)}"
  870. logger.error(error_msg, exc_info=True)
  871. return jsonify({
  872. 'success': False,
  873. 'message': error_msg,
  874. 'data': None
  875. }), 500
  876. # ==================================
  877. # 重复记录处理API接口
  878. # ==================================
  879. @bp.route('/get-duplicate-records', methods=['GET'])
  880. def get_duplicate_records_route():
  881. """
  882. 获取重复记录列表的API接口
  883. 查询参数:
  884. - status: 可选,筛选特定状态的记录 ('pending', 'processed', 'ignored')
  885. 返回:
  886. - JSON: 包含重复记录列表和处理状态
  887. """
  888. try:
  889. # 获取查询参数
  890. status = request.args.get('status', None)
  891. # 验证status参数的有效性
  892. if status and status not in ['pending', 'processed', 'ignored']:
  893. return jsonify({
  894. 'success': False,
  895. 'message': 'status参数无效,必须为 pending、processed 或 ignored',
  896. 'data': None
  897. }), 400
  898. # 调用业务逻辑函数获取重复记录列表
  899. result = get_duplicate_records(status)
  900. # 根据处理结果设置HTTP状态码
  901. status_code = 200 if result['success'] else 500
  902. return jsonify(result), status_code
  903. except Exception as e:
  904. # 处理未预期的异常
  905. error_msg = f"获取重复记录列表时发生错误: {str(e)}"
  906. logger.error(error_msg, exc_info=True)
  907. return jsonify({
  908. 'success': False,
  909. 'message': error_msg,
  910. 'data': [],
  911. 'count': 0
  912. }), 500
  913. @bp.route('/process-duplicate-record/<int:duplicate_id>', methods=['POST'])
  914. def process_duplicate_record_route(duplicate_id):
  915. """
  916. 处理重复记录的API接口
  917. 路径参数:
  918. - duplicate_id: 重复记录ID
  919. 请求参数:
  920. - JSON格式,包含以下字段:
  921. - action: 处理动作 (必填) ('merge_to_suspected', 'keep_main', 'ignore')
  922. - selected_duplicate_id: 当action为'merge_to_suspected'时,选择的疑似重复记录ID (可选)
  923. - processed_by: 处理人 (可选)
  924. - notes: 处理备注 (可选)
  925. 返回:
  926. - JSON: 包含处理结果和状态信息
  927. 返回格式:
  928. {
  929. 'success': true/false,
  930. 'message': '处理结果描述'
  931. }
  932. 功能说明:
  933. - 接收包含人才数据的请求体
  934. - 严格按照样例格式处理 results 数组中的人才数据
  935. - 调用 add_single_talent 函数将人才信息写入 business_cards 表
  936. - 提供详细的处理统计和结果追踪
  937. """
  938. try:
  939. # 获取请求数据
  940. data = request.get_json()
  941. if not data:
  942. return jsonify({
  943. 'success': False,
  944. 'message': '请求数据为空',
  945. 'data': None
  946. }), 400
  947. # 验证必填字段
  948. action = data.get('action')
  949. if not action:
  950. return jsonify({
  951. 'success': False,
  952. 'message': '缺少必填字段: action',
  953. 'data': None
  954. }), 400
  955. # 验证action参数的有效性
  956. if action not in ['merge_to_suspected', 'keep_main', 'ignore']:
  957. return jsonify({
  958. 'success': False,
  959. 'message': 'action参数无效,必须为 merge_to_suspected、keep_main 或 ignore',
  960. 'data': None
  961. }), 400
  962. # 提取其他参数
  963. selected_duplicate_id = data.get('selected_duplicate_id')
  964. processed_by = data.get('processed_by')
  965. notes = data.get('notes')
  966. # 特殊验证:如果action为merge_to_suspected,必须提供selected_duplicate_id
  967. if action == 'merge_to_suspected' and not selected_duplicate_id:
  968. return jsonify({
  969. 'success': False,
  970. 'message': '执行merge_to_suspected操作时必须提供selected_duplicate_id',
  971. 'data': None
  972. }), 400
  973. # 调用业务逻辑函数处理重复记录
  974. result = process_duplicate_record(
  975. duplicate_id=duplicate_id,
  976. action=action,
  977. selected_duplicate_id=selected_duplicate_id,
  978. processed_by=processed_by,
  979. notes=notes
  980. )
  981. # 根据处理结果设置HTTP状态码
  982. if result['code'] == 200:
  983. status_code = 200 # OK
  984. elif result['code'] == 400:
  985. status_code = 400 # Bad Request
  986. elif result['code'] == 404:
  987. status_code = 404 # Not Found
  988. else:
  989. status_code = 500 # Internal Server Error
  990. return jsonify({
  991. 'success': result['success'],
  992. 'message': result['message']
  993. }), 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. @bp.route('/get-duplicate-record-detail/<int:duplicate_id>', methods=['GET'])
  1004. def get_duplicate_record_detail_route(duplicate_id):
  1005. """
  1006. 获取指定重复记录详细信息的API接口
  1007. 路径参数:
  1008. - duplicate_id: 重复记录ID
  1009. 返回:
  1010. - JSON: 包含重复记录详细信息
  1011. """
  1012. try:
  1013. # 调用业务逻辑函数获取重复记录详情
  1014. result = get_duplicate_record_detail(duplicate_id)
  1015. # 根据处理结果设置HTTP状态码
  1016. if result['code'] == 200:
  1017. status_code = 200 # OK
  1018. elif result['code'] == 404:
  1019. status_code = 404 # Not Found
  1020. else:
  1021. status_code = 500 # Internal Server Error
  1022. return jsonify(result), status_code
  1023. except Exception as e:
  1024. # 处理未预期的异常
  1025. error_msg = f"获取重复记录详情时发生错误: {str(e)}"
  1026. logger.error(error_msg, exc_info=True)
  1027. return jsonify({
  1028. 'success': False,
  1029. 'message': error_msg,
  1030. 'data': None
  1031. }), 500
  1032. # 删除名片记录接口
  1033. @bp.route('/delete-business-card/<int:card_id>', methods=['DELETE'])
  1034. def delete_business_card_route(card_id):
  1035. """
  1036. 删除名片记录的API接口
  1037. 路径参数:
  1038. - card_id: 名片记录ID (必填)
  1039. 功能说明:
  1040. - 删除PostgreSQL数据库中business_cards表的指定记录
  1041. - 删除PostgreSQL数据库中duplicate_business_cards表的相关记录
  1042. - 删除MinIO存储中的名片图片文件
  1043. - 删除Neo4j图数据库中talent节点及其关联关系
  1044. 返回:
  1045. - JSON: 包含删除操作的结果状态和被删除的记录信息
  1046. 状态码:
  1047. - 200: 完全成功删除所有相关数据
  1048. - 206: 部分成功 (PostgreSQL删除成功,但Neo4j删除失败)
  1049. - 404: 未找到指定ID的名片记录
  1050. - 500: 删除操作失败
  1051. """
  1052. try:
  1053. # 验证card_id参数
  1054. if not card_id or card_id <= 0:
  1055. return jsonify({
  1056. 'success': False,
  1057. 'message': '无效的名片记录ID',
  1058. 'data': None
  1059. }), 400
  1060. # 调用删除函数
  1061. result = delete_business_card(card_id)
  1062. # 根据处理结果设置HTTP状态码和返回响应
  1063. if result['success']:
  1064. if result['code'] == 200:
  1065. status_code = 200 # 完全成功
  1066. elif result['code'] == 206:
  1067. status_code = 206 # 部分成功
  1068. else:
  1069. status_code = 200 # 默认成功
  1070. else:
  1071. if result['code'] == 404:
  1072. status_code = 404 # 未找到记录
  1073. elif result['code'] == 400:
  1074. status_code = 400 # 参数错误
  1075. else:
  1076. status_code = 500 # 服务器错误
  1077. return jsonify(result), status_code
  1078. except Exception as e:
  1079. logger.error(f"删除名片记录失败: {str(e)}")
  1080. return jsonify({
  1081. 'success': False,
  1082. 'message': f'删除名片记录失败: {str(e)}',
  1083. 'data': None
  1084. }), 500
  1085. # 修复损坏的重复记录接口
  1086. @bp.route('/fix-broken-duplicate-records', methods=['POST'])
  1087. def fix_broken_duplicate_records_route():
  1088. """
  1089. 修复duplicate_business_cards表中main_card_id为null的损坏记录
  1090. 功能说明:
  1091. - 查找所有main_card_id为null的损坏记录
  1092. - 删除这些损坏的记录以维护数据完整性
  1093. - 返回修复操作的详细结果
  1094. 返回:
  1095. - JSON: 包含修复操作的结果和被删除记录的信息
  1096. 状态码:
  1097. - 200: 修复成功
  1098. - 500: 修复失败
  1099. 注意:
  1100. - 此操作会永久删除损坏的记录
  1101. - 建议在系统维护时执行此操作
  1102. """
  1103. try:
  1104. # 调用修复函数
  1105. result = fix_broken_duplicate_records()
  1106. # 根据结果设置状态码
  1107. if result['success']:
  1108. status_code = 200
  1109. else:
  1110. status_code = 500
  1111. return jsonify(result), status_code
  1112. except Exception as e:
  1113. logger.error(f"修复损坏记录接口调用失败: {str(e)}")
  1114. return jsonify({
  1115. 'success': False,
  1116. 'message': f'修复损坏记录接口调用失败: {str(e)}',
  1117. 'data': None
  1118. }), 500
  1119. # 获取解析任务列表接口
  1120. @bp.route('/get-parse-tasks', methods=['GET'])
  1121. def get_parse_tasks_route():
  1122. """
  1123. 获取解析任务列表的API接口,支持分页
  1124. 查询参数:
  1125. - page: 页码,从1开始,默认为1
  1126. - per_page: 每页记录数,默认为10,最大100
  1127. - task_type: 任务类型过滤,可选
  1128. - task_status: 任务状态过滤,可选
  1129. 返回:
  1130. - JSON: 包含解析任务列表和分页信息
  1131. 功能说明:
  1132. - 支持分页查询,每页默认10条记录
  1133. - 支持按任务类型和状态过滤
  1134. - 按创建时间倒序排列
  1135. - 返回总记录数和分页信息
  1136. 状态码:
  1137. - 200: 查询成功
  1138. - 400: 请求参数错误
  1139. - 500: 查询失败
  1140. """
  1141. try:
  1142. # 获取查询参数
  1143. page = request.args.get('page', 1, type=int)
  1144. per_page = request.args.get('per_page', 10, type=int)
  1145. task_type = request.args.get('task_type', type=str)
  1146. task_status = request.args.get('task_status', type=str)
  1147. # 记录请求日志
  1148. logger.info(f"获取解析任务列表请求: page={page}, per_page={per_page}, task_type={task_type}, task_status={task_status}")
  1149. # 调用核心业务逻辑
  1150. result = get_parse_tasks(page, per_page, task_type, task_status)
  1151. # 返回结果
  1152. return jsonify({
  1153. 'success': result['success'],
  1154. 'message': result['message'],
  1155. 'data': result['data']
  1156. }), result['code']
  1157. except Exception as e:
  1158. # 记录错误日志
  1159. error_msg = f"获取解析任务列表接口失败: {str(e)}"
  1160. logger.error(error_msg, exc_info=True)
  1161. # 返回错误响应
  1162. return jsonify({
  1163. 'success': False,
  1164. 'message': error_msg,
  1165. 'data': None
  1166. }), 500
  1167. # 获取解析任务详情接口
  1168. @bp.route('/get-parse-task-detail', methods=['GET'])
  1169. def get_parse_task_detail_route():
  1170. """
  1171. 获取解析任务详情的API接口
  1172. 查询参数:
  1173. - task_name: 任务名称,必填
  1174. 返回:
  1175. - JSON: 包含任务详细信息
  1176. 功能说明:
  1177. - 根据任务名称查询指定任务的详细信息
  1178. - 返回任务的所有字段信息
  1179. - 包含解析结果的完整数据
  1180. 状态码:
  1181. - 200: 查询成功
  1182. - 400: 请求参数错误
  1183. - 404: 任务不存在
  1184. - 500: 查询失败
  1185. """
  1186. try:
  1187. # 获取查询参数
  1188. task_name = request.args.get('task_name', type=str)
  1189. # 参数验证
  1190. if not task_name:
  1191. return jsonify({
  1192. 'success': False,
  1193. 'message': '任务名称参数不能为空',
  1194. 'data': None
  1195. }), 400
  1196. # 记录请求日志
  1197. logger.info(f"获取解析任务详情请求: task_name={task_name}")
  1198. # 调用核心业务逻辑
  1199. result = get_parse_task_detail(task_name)
  1200. # 返回结果
  1201. return jsonify({
  1202. 'success': result['success'],
  1203. 'message': result['message'],
  1204. 'data': result['data']
  1205. }), result['code']
  1206. except Exception as e:
  1207. # 记录错误日志
  1208. error_msg = f"获取解析任务详情接口失败: {str(e)}"
  1209. logger.error(error_msg, exc_info=True)
  1210. # 返回错误响应
  1211. return jsonify({
  1212. 'success': False,
  1213. 'message': error_msg,
  1214. 'data': None
  1215. }), 500
  1216. # 新增解析任务接口
  1217. @bp.route('/add-parse-task', methods=['POST'])
  1218. def add_parse_task_route():
  1219. """
  1220. 新增解析任务的API接口
  1221. 请求参数:
  1222. - task_type: 任务类型 (form-data字段,必填)
  1223. 可选值:'名片', '简历', '新任命', '招聘', '杂项'
  1224. - files: 文件数组 (multipart/form-data,对于招聘类型可选)
  1225. - created_by: 创建者 (可选,form-data字段)
  1226. - data: 数据内容 (form-data字段,招聘类型必填)
  1227. - publish_time: 发布时间 (form-data字段,新任命类型必填)
  1228. 返回:
  1229. - JSON: 包含任务创建结果和状态信息
  1230. 返回格式:
  1231. {
  1232. 'success': true/false,
  1233. 'message': '处理结果描述'
  1234. }
  1235. 功能说明:
  1236. - 根据任务类型处理不同格式的文件
  1237. - 名片任务:JPG/PNG格式图片 → talent_photos目录
  1238. - 简历任务:PDF格式文件 → resume_files目录
  1239. - 新任命任务:MD格式文件 → appointment_files目录
  1240. - 招聘任务:数据库记录处理,无需文件上传,创建任务后立即执行解析
  1241. - 杂项任务:任意格式文件 → misc_files目录
  1242. - 使用timestamp+uuid自动生成文件名
  1243. - 在parse_task_repository表中创建待解析任务记录
  1244. 状态码:
  1245. - 200: 所有文件上传成功,任务创建成功
  1246. - 206: 部分文件上传成功,任务创建成功
  1247. - 400: 请求参数错误
  1248. - 500: 服务器内部错误
  1249. """
  1250. try:
  1251. # 获取任务类型参数
  1252. task_type = request.form.get('task_type')
  1253. # 验证任务类型
  1254. if not task_type:
  1255. return jsonify({
  1256. 'success': False,
  1257. 'message': '缺少task_type参数'
  1258. }), 400
  1259. if task_type not in ['名片', '简历', '新任命', '招聘', '杂项']:
  1260. return jsonify({
  1261. 'success': False,
  1262. 'message': 'task_type参数必须是以下值之一:名片、简历、新任命、招聘、杂项'
  1263. }), 400
  1264. # 获取创建者信息(可选参数)
  1265. created_by = request.form.get('created_by', 'api_user')
  1266. # 获取数据内容和发布时间参数
  1267. data = request.form.get('data')
  1268. publish_time = request.form.get('publish_time')
  1269. # 对于招聘类型,不需要文件上传
  1270. if task_type == '招聘':
  1271. # 检查是否误传了文件
  1272. if 'files' in request.files and request.files.getlist('files'):
  1273. return jsonify({
  1274. 'success': False,
  1275. 'message': '招聘类型任务不需要上传文件'
  1276. }), 400
  1277. # 检查data参数是否有内容
  1278. if not data:
  1279. return jsonify({
  1280. 'success': False,
  1281. 'message': '招聘类型任务需要提供data参数'
  1282. }), 400
  1283. # 记录请求日志
  1284. logger.info(f"新增招聘任务请求: 创建者={created_by}, data长度={len(data) if data else 0}")
  1285. # 调用核心业务逻辑
  1286. result = add_parse_task(None, task_type, created_by, data, publish_time)
  1287. # 如果任务创建成功,继续执行批量处理
  1288. if result['success']:
  1289. # 招聘任务创建成功,不需要进一步处理
  1290. logger.info(f"招聘任务创建成功")
  1291. else:
  1292. logger.error(f"招聘任务创建失败: {result.get('message', '未知错误')}")
  1293. else:
  1294. # 其他类型需要文件上传
  1295. if 'files' not in request.files:
  1296. return jsonify({
  1297. 'success': False,
  1298. 'message': f'{task_type}任务需要上传文件,请使用files字段上传文件'
  1299. }), 400
  1300. # 获取上传的文件列表
  1301. uploaded_files = request.files.getlist('files')
  1302. # 检查文件列表是否为空
  1303. if not uploaded_files or len(uploaded_files) == 0:
  1304. return jsonify({
  1305. 'success': False,
  1306. 'message': '文件数组不能为空'
  1307. }), 400
  1308. # 验证所有文件
  1309. valid_files = []
  1310. for i, file in enumerate(uploaded_files):
  1311. # 检查文件是否为空
  1312. if not file or file.filename == '':
  1313. return jsonify({
  1314. 'success': False,
  1315. 'message': f'第{i+1}个文件为空或未选择'
  1316. }), 400
  1317. valid_files.append(file)
  1318. # 对于新任命类型,检查publish_time参数
  1319. if task_type == '新任命':
  1320. if not publish_time:
  1321. return jsonify({
  1322. 'success': False,
  1323. 'message': '新任命类型任务需要提供publish_time参数'
  1324. }), 400
  1325. # 记录请求日志
  1326. logger.info(f"新增{task_type}任务请求: 文件数量={len(valid_files)}, 创建者={created_by}")
  1327. # 调用核心业务逻辑
  1328. result = add_parse_task(valid_files, task_type, created_by, data, publish_time)
  1329. # 根据处理结果设置HTTP状态码
  1330. if result['success']:
  1331. if result['code'] == 200:
  1332. status_code = 200
  1333. elif result['code'] == 206:
  1334. status_code = 206
  1335. else:
  1336. status_code = 200
  1337. else:
  1338. if result['code'] == 400:
  1339. status_code = 400
  1340. else:
  1341. status_code = 500
  1342. # 返回结果
  1343. return jsonify({
  1344. 'success': result['success'],
  1345. 'message': result['message']
  1346. }), status_code
  1347. except Exception as e:
  1348. # 记录错误日志
  1349. error_msg = f"新增解析任务接口失败: {str(e)}"
  1350. logger.error(error_msg, exc_info=True)
  1351. # 返回错误响应
  1352. return jsonify({
  1353. 'success': False,
  1354. 'message': error_msg
  1355. }), 500
  1356. @bp.route('/execute-parse-task', methods=['POST'])
  1357. def execute_parse_task():
  1358. """
  1359. 执行解析任务接口
  1360. 根据task_type参数调用相应的批量处理函数:
  1361. - 名片: batch_process_business_card_images
  1362. - 简历: batch_parse_resumes
  1363. - 新任命: batch_process_md
  1364. - 招聘: 已在add-parse-task接口中自动处理,此处不再支持
  1365. - 杂项: batch_process_images
  1366. 请求参数:
  1367. - data (dict): 包含完整任务信息的对象,格式如下:
  1368. {
  1369. "id": 123,
  1370. "task_name": "parse_task_20241201_a1b2c3d4",
  1371. "task_status": "待解析",
  1372. "task_type": "名片",
  1373. "task_source": [
  1374. {
  1375. "original_filename": "张三名片.jpg",
  1376. "minio_path": "https://192.168.3.143:9000/dataops-platform/talent_photos/20241201_001234_张三名片.jpg",
  1377. "status": "正常"
  1378. }
  1379. ],
  1380. "collection_count": 2,
  1381. "parse_count": 0,
  1382. "parse_result": null,
  1383. "created_at": "2024-12-01 10:30:45",
  1384. "created_by": "api_user",
  1385. "updated_at": "2024-12-01 10:30:45",
  1386. "updated_by": "api_user"
  1387. }
  1388. 对于新任命类型,task_source中的每个对象还需要包含publish_time字段:
  1389. {
  1390. "publish_time": "20250731",
  1391. "original_filename": "张三任命.md",
  1392. "minio_path": "https://192.168.3.143:9000/dataops-platform/appointment_files/20241201_001234_张三任命.md",
  1393. "status": "正常"
  1394. }
  1395. """
  1396. try:
  1397. # 获取请求数据
  1398. request_data = request.get_json()
  1399. if not request_data:
  1400. return jsonify({
  1401. 'success': False,
  1402. 'message': '请求数据不能为空',
  1403. 'data': None
  1404. }), 400
  1405. # 验证请求数据格式
  1406. if not isinstance(request_data, dict) or 'data' not in request_data:
  1407. return jsonify({
  1408. 'success': False,
  1409. 'message': '请求数据格式错误,必须包含data字段',
  1410. 'data': None
  1411. }), 400
  1412. # 获取任务数据
  1413. task_data = request_data.get('data')
  1414. if not task_data:
  1415. return jsonify({
  1416. 'success': False,
  1417. 'message': '任务数据不能为空',
  1418. 'data': None
  1419. }), 400
  1420. # 验证任务数据格式
  1421. if not isinstance(task_data, dict):
  1422. return jsonify({
  1423. 'success': False,
  1424. 'message': '任务数据必须是对象格式',
  1425. 'data': None
  1426. }), 400
  1427. # 获取任务类型
  1428. task_type = task_data.get('task_type', '').strip()
  1429. if not task_type:
  1430. return jsonify({
  1431. 'success': False,
  1432. 'message': '任务类型不能为空',
  1433. 'data': None
  1434. }), 400
  1435. # 获取任务源数据
  1436. task_source = task_data.get('task_source', [])
  1437. if not task_source:
  1438. return jsonify({
  1439. 'success': False,
  1440. 'message': '任务源数据不能为空',
  1441. 'data': None
  1442. }), 400
  1443. # 验证任务源数据格式
  1444. if not isinstance(task_source, list):
  1445. return jsonify({
  1446. 'success': False,
  1447. 'message': '任务源数据必须是数组格式',
  1448. 'data': None
  1449. }), 400
  1450. # 获取任务ID
  1451. task_id = task_data.get('id')
  1452. # 更新parse_task_repository数据库表中的task_source
  1453. if task_id:
  1454. try:
  1455. from app.models.parse_models import ParseTaskRepository
  1456. from app.core.data_parse.parse_system import db
  1457. task_record = ParseTaskRepository.query.get(task_id)
  1458. if task_record:
  1459. task_record.task_source = task_source
  1460. task_record.updated_at = get_east_asia_time_naive()
  1461. task_record.updated_by = 'admin'
  1462. db.session.commit()
  1463. logging.info(f"已更新task_id为{task_id}的任务记录的task_source")
  1464. else:
  1465. logging.warning(f"未找到task_id为{task_id}的任务记录")
  1466. except Exception as update_error:
  1467. logging.error(f"更新任务记录失败: {str(update_error)}")
  1468. db.session.rollback()
  1469. # 根据任务类型执行相应的处理函数
  1470. try:
  1471. if task_type == '名片':
  1472. # 调用名片批量处理函数
  1473. result = batch_process_business_card_images(task_source, task_id, task_type)
  1474. elif task_type == '简历':
  1475. # 调用简历批量处理函数
  1476. result = batch_parse_resumes(task_source, task_id, task_type)
  1477. elif task_type == '新任命':
  1478. # 验证新任命任务的publish_time字段
  1479. for source_item in task_source:
  1480. if not isinstance(source_item, dict) or 'publish_time' not in source_item:
  1481. return jsonify({
  1482. 'success': False,
  1483. 'message': '新任命任务的每个源数据必须包含publish_time字段',
  1484. 'data': None
  1485. }), 400
  1486. # 调用新任命批量处理函数
  1487. result = batch_process_md(task_source, task_id=task_id, task_type=task_type)
  1488. elif task_type == '招聘':
  1489. result = batch_process_menduner_data(task_source, task_id, task_type)
  1490. elif task_type == '杂项':
  1491. # 调用图片批量处理函数(表格类型)
  1492. process_type = request_data.get('process_type', 'table')
  1493. result = batch_process_images(task_source, process_type, task_id, task_type)
  1494. else:
  1495. return jsonify({
  1496. 'success': False,
  1497. 'message': f'不支持的任务类型: {task_type},支持的类型:名片、简历、新任命、招聘、杂项',
  1498. 'data': None
  1499. }), 400
  1500. # 记录处理结果日志并更新任务状态
  1501. from app.models.parse_models import ParseTaskRepository
  1502. from app.core.data_parse.parse_system import db
  1503. task_obj = None
  1504. if task_id:
  1505. task_obj = ParseTaskRepository.query.filter_by(id=task_id).first()
  1506. # 根据解析结果确定任务状态和返回数据
  1507. if result.get('success'):
  1508. logging.info(f"执行{task_type}解析任务成功: {result.get('message', '')}")
  1509. # 获取解析结果数据
  1510. result_data = result.get('data', {})
  1511. success_count = result_data.get('success_count', 0)
  1512. failed_count = result_data.get('failed_count', 0)
  1513. # 对于新任命类型,parsed_record_ids在process_single_markdown_file中已经处理
  1514. parsed_record_ids = result_data.get('parsed_record_ids', [])
  1515. # 确定任务状态
  1516. if failed_count == 0:
  1517. task_status = '解析成功'
  1518. elif success_count > 0:
  1519. task_status = '部分解析成功'
  1520. else:
  1521. task_status = '不成功'
  1522. # 更新任务记录
  1523. if task_obj:
  1524. task_obj.task_status = task_status
  1525. task_obj.parse_count = success_count
  1526. # 对于新任命类型,需要从数据库中查询实际的记录ID
  1527. if task_type == '新任命':
  1528. try:
  1529. from app.core.data_parse.parse_system import ParsedTalent
  1530. # 查询该任务相关的所有记录
  1531. parsed_records = ParsedTalent.query.filter_by(task_id=str(task_id), task_type=task_type).all()
  1532. record_ids = [str(record.id) for record in parsed_records]
  1533. task_obj.parse_result = ','.join(record_ids) if record_ids else ''
  1534. except Exception as e:
  1535. logging.error(f"查询新任命记录ID失败: {str(e)}")
  1536. task_obj.parse_result = ''
  1537. else:
  1538. task_obj.parse_result = ','.join(parsed_record_ids) if parsed_record_ids else ''
  1539. task_obj.updated_at = get_east_asia_time_naive()
  1540. task_obj.updated_by = 'admin'
  1541. db.session.commit()
  1542. logging.info(f"已更新解析任务记录: id={getattr(task_obj, 'id', None)}, 状态={task_obj.task_status}")
  1543. # 构建返回数据,按照请求参数格式返回
  1544. return_data = task_data.copy() if task_data else {}
  1545. # 对于新任命类型,需要从数据库中查询实际的记录ID
  1546. if task_type == '新任命':
  1547. try:
  1548. from app.core.data_parse.parse_system import ParsedTalent
  1549. # 查询该任务相关的所有记录
  1550. parsed_records = ParsedTalent.query.filter_by(task_id=str(task_id), task_type=task_type).all()
  1551. record_ids = [str(record.id) for record in parsed_records]
  1552. parse_result = ','.join(record_ids) if record_ids else ''
  1553. except Exception as e:
  1554. logging.error(f"查询新任命记录ID失败: {str(e)}")
  1555. parse_result = ''
  1556. else:
  1557. parse_result = ','.join(parsed_record_ids) if parsed_record_ids else ''
  1558. return_data.update({
  1559. 'task_status': task_status,
  1560. 'parse_count': success_count,
  1561. 'parse_result': parse_result,
  1562. 'updated_at': get_east_asia_isoformat(),
  1563. 'updated_by': 'admin'
  1564. })
  1565. # 确定HTTP状态码
  1566. if failed_count == 0:
  1567. status_code = 200 # 完全成功
  1568. elif success_count > 0:
  1569. status_code = 206 # 部分成功
  1570. else:
  1571. status_code = 500 # 完全失败
  1572. return jsonify({
  1573. 'success': True,
  1574. 'message': result.get('message', '解析完成'),
  1575. 'data': return_data
  1576. }), status_code
  1577. else:
  1578. logging.error(f"执行{task_type}解析任务失败: {result.get('message', '')}")
  1579. # 设置任务状态为不成功
  1580. if task_obj:
  1581. task_obj.task_status = '不成功'
  1582. task_obj.parse_count = 0
  1583. task_obj.parse_result = ''
  1584. task_obj.updated_at = get_east_asia_time_naive()
  1585. task_obj.updated_by = 'admin'
  1586. db.session.commit()
  1587. logging.info(f"已更新解析任务记录: id={getattr(task_obj, 'id', None)}, 状态=不成功")
  1588. # 构建返回数据,按照请求参数格式返回
  1589. return_data = task_data.copy() if task_data else {}
  1590. return_data.update({
  1591. 'task_status': '不成功',
  1592. 'parse_count': 0,
  1593. 'parse_result': '',
  1594. 'updated_at': get_east_asia_isoformat(),
  1595. 'updated_by': 'admin'
  1596. })
  1597. return jsonify({
  1598. 'success': False,
  1599. 'message': result.get('message', '解析失败'),
  1600. 'data': return_data
  1601. }), 500
  1602. except Exception as process_error:
  1603. error_msg = f"执行{task_type}解析任务时发生错误: {str(process_error)}"
  1604. logging.error(error_msg, exc_info=True)
  1605. return jsonify({
  1606. 'success': False,
  1607. 'message': error_msg,
  1608. 'data': None
  1609. }), 500
  1610. except Exception as e:
  1611. # 记录错误日志
  1612. error_msg = f"执行解析任务接口失败: {str(e)}"
  1613. logging.error(error_msg, exc_info=True)
  1614. # 返回错误响应
  1615. return jsonify({
  1616. 'success': False,
  1617. 'message': error_msg,
  1618. 'data': None
  1619. }), 500
  1620. @bp.route('/add-parsed-talents', methods=['POST'])
  1621. def add_parsed_talents_route():
  1622. """
  1623. 处理解析任务响应数据并写入人才信息接口
  1624. 请求参数:
  1625. - 请求体: 包含任务ID和人才数据的JSON对象 (JSON格式)
  1626. - task_id: 任务ID,用于更新任务状态(可选)
  1627. - task_type: 任务类型(可选)
  1628. - data: 包含人才解析结果的数据对象
  1629. 请求体格式(严格按照样例格式):
  1630. {
  1631. "task_id": "119",
  1632. "task_type": "名片",
  1633. "data": {
  1634. "results": [
  1635. {
  1636. "name_zh": "王仁",
  1637. "name_en": "Owen Wang",
  1638. "title_zh": "总经理",
  1639. "title_en": "General Manager",
  1640. "mobile": "+86 138 1685 0647",
  1641. "phone": null,
  1642. "email": "rwang5@urcove-hotels.com",
  1643. "hotel_zh": "上海静安逸扉酒店",
  1644. "hotel_en": "UrCove by HYATT Shanghai Jing'an",
  1645. "brand_zh": null,
  1646. "brand_en": null,
  1647. "affiliation_zh": null,
  1648. "affiliation_en": null,
  1649. "brand_group": "UrCove, HYATT",
  1650. "address_zh": "中国上海市静安区武定西路1185号",
  1651. "address_en": "No.1185 West Wuding Road, Jing'an District",
  1652. "postal_code_zh": "200042",
  1653. "postal_code_en": "200042",
  1654. "birthday": null,
  1655. "residence": null,
  1656. "age": 0,
  1657. "native_place": null,
  1658. "image_path": "",
  1659. "talent_profile": "测试用名片",
  1660. "career_path": [
  1661. {
  1662. "date": "2025-08-01",
  1663. "hotel_en": "UrCove by HYATT Shanghai Jing'an",
  1664. "hotel_zh": "上海静安逸扉酒店",
  1665. "image_path": "",
  1666. "source": "business_card_creation",
  1667. "title_en": "General Manager",
  1668. "title_zh": "总经理"
  1669. }
  1670. ],
  1671. "origin_source": [
  1672. {
  1673. "task_type": "招聘",
  1674. "minio_path": "http://example.com/path/to/image.jpg",
  1675. "source_date": "2025-08-01"
  1676. }
  1677. ],
  1678. "minio_path": "http://example.com/path/to/image.jpg" // 可选字段
  1679. }
  1680. ]
  1681. }
  1682. }
  1683. 返回:
  1684. - JSON: 包含批量处理结果和状态信息
  1685. 功能说明:
  1686. - 接收包含人才数据的请求体
  1687. - 严格按照样例格式处理 results 数组中的人才数据
  1688. - 调用 add_single_talent 函数将人才信息写入 business_cards 表
  1689. - 提供详细的处理统计和结果追踪
  1690. 状态码:
  1691. - 200: 全部处理成功
  1692. - 206: 部分处理成功
  1693. - 400: 请求参数错误
  1694. - 500: 服务器内部错误
  1695. """
  1696. try:
  1697. # 检查请求是否为 JSON 格式
  1698. if not request.is_json:
  1699. return jsonify({
  1700. 'success': False,
  1701. 'message': '请求必须是 JSON 格式'
  1702. }), 400
  1703. # 获取请求数据
  1704. api_response_data = request.get_json()
  1705. # 基本参数验证
  1706. if not api_response_data:
  1707. return jsonify({
  1708. 'success': False,
  1709. 'message': '请求数据不能为空'
  1710. }), 400
  1711. # 验证数据格式
  1712. if not isinstance(api_response_data, dict):
  1713. return jsonify({
  1714. 'success': False,
  1715. 'message': '请求数据必须是JSON对象格式'
  1716. }), 400
  1717. # 记录请求日志
  1718. total_results = 0
  1719. if api_response_data.get('data') and api_response_data['data'].get('results'):
  1720. total_results = len(api_response_data['data']['results'])
  1721. logger.info(f"收到处理人才数据请求,包含 {total_results} 条结果记录")
  1722. # 调用核心业务逻辑
  1723. result = add_parsed_talents(api_response_data)
  1724. # 根据处理结果设置HTTP状态码
  1725. if result.get('success', False):
  1726. if result.get('code') == 200:
  1727. status_code = 200 # 全部成功
  1728. elif result.get('code') == 206:
  1729. status_code = 206 # 部分成功
  1730. else:
  1731. status_code = 200 # 默认成功
  1732. else:
  1733. if result.get('code') == 400:
  1734. status_code = 400 # 参数错误
  1735. else:
  1736. status_code = 500 # 服务器错误
  1737. # 记录处理结果日志
  1738. if result.get('success'):
  1739. data_summary = result.get('data', {}).get('summary', {})
  1740. success_count = data_summary.get('success_count', 0)
  1741. failed_count = data_summary.get('failed_count', 0)
  1742. logger.info(f"处理人才数据完成: 成功 {success_count} 条,失败 {failed_count} 条")
  1743. else:
  1744. logger.error(f"处理人才数据失败: {result.get('message', '未知错误')}")
  1745. # 返回结果
  1746. return jsonify({
  1747. 'success': result.get('success', False),
  1748. 'message': result.get('message', '处理完成')
  1749. }), status_code
  1750. except Exception as e:
  1751. # 记录错误日志
  1752. error_msg = f"处理人才数据接口失败: {str(e)}"
  1753. logger.error(error_msg, exc_info=True)
  1754. # 返回错误响应
  1755. return jsonify({
  1756. 'success': False,
  1757. 'message': error_msg
  1758. }), 500
  1759. @bp.route('/get-parsed-talents', methods=['GET'])
  1760. def get_parsed_talents_route():
  1761. """
  1762. 获取解析人才记录列表接口
  1763. 请求参数:
  1764. - status (str, optional): 状态过滤参数,如果为空则查询所有记录
  1765. 请求示例:
  1766. GET /get-parsed-talents?status=待审核
  1767. GET /get-parsed-talents (查询所有记录)
  1768. 返回:
  1769. - JSON: 包含人才记录列表和状态信息
  1770. - 200: 成功获取数据
  1771. - 500: 服务器内部错误
  1772. """
  1773. try:
  1774. # 获取查询参数
  1775. status = request.args.get('status', '').strip()
  1776. # 调用核心业务逻辑
  1777. from app.core.data_parse.parse_system import get_parsed_talents
  1778. result = get_parsed_talents(status)
  1779. # 根据处理结果设置HTTP状态码
  1780. if result.get('success', False):
  1781. status_code = result.get('code', 200)
  1782. else:
  1783. status_code = result.get('code', 500)
  1784. # 记录处理结果日志
  1785. if result.get('success'):
  1786. count = result.get('count', 0)
  1787. if status:
  1788. logging.info(f"成功获取状态为 '{status}' 的解析人才记录: {count} 条")
  1789. else:
  1790. logging.info(f"成功获取所有解析人才记录: {count} 条")
  1791. else:
  1792. logging.error(f"获取解析人才记录失败: {result.get('message', '未知错误')}")
  1793. # 返回结果
  1794. return jsonify({
  1795. 'success': result.get('success', False),
  1796. 'message': result.get('message', '处理完成'),
  1797. 'data': result.get('data', []),
  1798. 'count': result.get('count', 0)
  1799. }), status_code
  1800. except Exception as e:
  1801. # 记录错误日志
  1802. error_msg = f"获取解析人才记录接口失败: {str(e)}"
  1803. logging.error(error_msg, exc_info=True)
  1804. # 返回错误响应
  1805. return jsonify({
  1806. 'success': False,
  1807. 'message': error_msg,
  1808. 'data': [],
  1809. 'count': 0
  1810. }), 500
  1811. @bp.route('/process-urls', methods=['POST'])
  1812. def process_urls_route():
  1813. """
  1814. 处理网页URL爬取接口
  1815. 请求参数:
  1816. - JSON格式,包含以下字段:
  1817. - urlArr: 字符串数组,每个元素为一个网页URL地址
  1818. 请求示例:
  1819. POST /process-urls
  1820. Content-Type: application/json
  1821. {
  1822. "urlArr": [
  1823. "https://example.com/page1",
  1824. "https://example.com/page2",
  1825. "https://mp.weixin.qq.com/s/4yz-kNAWAlF36aeQ_cgQQg"
  1826. ]
  1827. }
  1828. 返回:
  1829. - JSON: 包含网页爬取结果的字典
  1830. 返回格式:
  1831. {
  1832. "success": true/false,
  1833. "message": "处理结果描述",
  1834. "data": {
  1835. "total_urls": 总URL数量,
  1836. "success_count": 成功爬取的URL数量,
  1837. "failed_count": 失败的URL数量,
  1838. "contents": [
  1839. {
  1840. "url": "URL地址",
  1841. "data": "网页内容",
  1842. "status": "success",
  1843. "content_length": 内容长度,
  1844. "original_length": 原始内容长度,
  1845. "status_code": HTTP状态码,
  1846. "encoding": 编码格式
  1847. }
  1848. ],
  1849. "failed_items": [
  1850. {
  1851. "url": "URL地址",
  1852. "error": "错误信息",
  1853. "status": "failed"
  1854. }
  1855. ]
  1856. }
  1857. }
  1858. 功能说明:
  1859. - 接收包含URL数组的POST请求
  1860. - 调用web_url_crawl函数进行网页内容爬取
  1861. - 返回结构化的爬取结果
  1862. - 支持批量处理多个URL
  1863. - 提供详细的成功/失败统计信息
  1864. 状态码:
  1865. - 200: 完全成功,所有URL都成功爬取
  1866. - 206: 部分成功,部分URL成功爬取
  1867. - 400: 请求参数错误
  1868. - 500: 服务器内部错误
  1869. """
  1870. try:
  1871. # 检查请求是否为JSON格式
  1872. if not request.is_json:
  1873. return jsonify({
  1874. 'success': False,
  1875. 'message': '请求必须是JSON格式'
  1876. }), 400
  1877. # 获取请求数据
  1878. request_data = request.get_json()
  1879. # 基本参数验证
  1880. if not request_data:
  1881. return jsonify({
  1882. 'success': False,
  1883. 'message': '请求数据不能为空'
  1884. }), 400
  1885. # 验证urlArr字段
  1886. if 'urlArr' not in request_data:
  1887. return jsonify({
  1888. 'success': False,
  1889. 'message': '缺少必填字段: urlArr'
  1890. }), 400
  1891. url_arr = request_data.get('urlArr')
  1892. # 验证urlArr是否为数组
  1893. if not isinstance(url_arr, list):
  1894. return jsonify({
  1895. 'success': False,
  1896. 'message': 'urlArr字段必须是数组格式'
  1897. }), 400
  1898. # 验证urlArr是否为空
  1899. if len(url_arr) == 0:
  1900. return jsonify({
  1901. 'success': False,
  1902. 'message': 'urlArr数组不能为空'
  1903. }), 400
  1904. # 验证每个URL是否为字符串
  1905. for i, url in enumerate(url_arr):
  1906. if not isinstance(url, str):
  1907. return jsonify({
  1908. 'success': False,
  1909. 'message': f'urlArr[{i}]必须是字符串格式,当前类型: {type(url).__name__}'
  1910. }), 400
  1911. # 记录请求日志
  1912. logger.info(f"收到网页URL爬取请求,包含 {len(url_arr)} 个URL")
  1913. # 调用核心业务逻辑 - web_url_crawl函数
  1914. result = web_url_crawl(url_arr)
  1915. # 根据处理结果设置HTTP状态码
  1916. if result.get('success', False):
  1917. success_count = result.get('data', {}).get('success_count', 0)
  1918. failed_count = result.get('data', {}).get('failed_count', 0)
  1919. if failed_count == 0:
  1920. status_code = 200 # 完全成功
  1921. elif success_count > 0:
  1922. status_code = 206 # 部分成功
  1923. else:
  1924. status_code = 500 # 完全失败
  1925. else:
  1926. status_code = 500 # 服务器错误
  1927. # 记录处理结果日志
  1928. if result.get('success'):
  1929. data = result.get('data', {})
  1930. success_count = data.get('success_count', 0)
  1931. failed_count = data.get('failed_count', 0)
  1932. total_urls = data.get('total_urls', 0)
  1933. if failed_count == 0:
  1934. logger.info(f"网页URL爬取完全成功: 共 {total_urls} 个URL,全部成功")
  1935. else:
  1936. logger.info(f"网页URL爬取部分成功: 共 {total_urls} 个URL,成功 {success_count} 个,失败 {failed_count} 个")
  1937. else:
  1938. logger.error(f"网页URL爬取失败: {result.get('message', '未知错误')}")
  1939. # 返回结果
  1940. return jsonify({
  1941. 'success': result.get('success', False),
  1942. 'message': result.get('message', '处理完成'),
  1943. 'data': result.get('data', {})
  1944. }), status_code
  1945. except Exception as e:
  1946. # 记录错误日志
  1947. error_msg = f"网页URL爬取接口失败: {str(e)}"
  1948. logger.error(error_msg, exc_info=True)
  1949. # 返回错误响应
  1950. return jsonify({
  1951. 'success': False,
  1952. 'message': error_msg,
  1953. 'data': {
  1954. 'total_urls': 0,
  1955. 'success_count': 0,
  1956. 'failed_count': 0,
  1957. 'contents': [],
  1958. 'failed_items': []
  1959. }
  1960. }), 500
  1961. @bp.route('/get-calendar-info', methods=['GET'])
  1962. def get_calendar_info_api():
  1963. """
  1964. 获取指定日期的黄历信息
  1965. GET /api/data_parse/get-calendar-info?date=YYYY-MM-DD
  1966. Args:
  1967. date (str): 查询日期,格式为YYYY-MM-DD
  1968. Returns:
  1969. JSON: 包含黄历信息的响应数据
  1970. """
  1971. try:
  1972. # 获取查询参数
  1973. date_param = request.args.get('date')
  1974. # 验证日期参数
  1975. if not date_param:
  1976. return jsonify({
  1977. 'reason': 'failed',
  1978. 'return_code': 400,
  1979. 'result': None,
  1980. 'error': '缺少必填参数: date'
  1981. }), 400
  1982. # 验证日期格式
  1983. if not isinstance(date_param, str) or len(date_param) != 10:
  1984. return jsonify({
  1985. 'reason': 'failed',
  1986. 'return_code': 400,
  1987. 'result': None,
  1988. 'error': '日期格式错误,请使用YYYY-MM-DD格式'
  1989. }), 400
  1990. # 记录请求日志
  1991. logger.info(f"收到黄历信息查询请求,日期: {date_param}")
  1992. # 调用核心业务逻辑 - get_calendar_by_date函数
  1993. result = get_calendar_by_date(date_param)
  1994. # 根据返回结果设置HTTP状态码
  1995. status_code = result.get('return_code', 500)
  1996. # 记录处理结果日志
  1997. if result.get('return_code') == 200:
  1998. logger.info(f"黄历信息查询成功,日期: {date_param}")
  1999. else:
  2000. error_msg = result.get('error', '未知错误')
  2001. logger.warning(f"黄历信息查询失败,日期: {date_param},错误: {error_msg}")
  2002. # 返回结果
  2003. return jsonify(result), status_code
  2004. except Exception as e:
  2005. # 记录错误日志
  2006. error_msg = f"黄历信息查询接口失败: {str(e)}"
  2007. logger.error(error_msg, exc_info=True)
  2008. # 返回错误响应
  2009. return jsonify({
  2010. 'reason': 'failed',
  2011. 'return_code': 500,
  2012. 'result': None,
  2013. 'error': error_msg
  2014. }), 500
  2015. # ================================
  2016. # 微信认证相关API路由
  2017. # ================================
  2018. @bp.route('/wechat-register', methods=['POST'])
  2019. def wechat_register_api():
  2020. """
  2021. 微信用户注册接口
  2022. POST /api/parse/wechat-register
  2023. Request Body:
  2024. {
  2025. "wechat_code": "wx_openid_123", // 必填:微信授权码/openid
  2026. "phone_number": "13800138000", // 可选:手机号码
  2027. "id_card_number": "110101199001011234" // 可选:身份证号码
  2028. }
  2029. Returns:
  2030. JSON: 包含注册结果的响应数据
  2031. """
  2032. try:
  2033. # 获取请求数据
  2034. data = request.get_json()
  2035. # 验证请求数据
  2036. if not data:
  2037. return jsonify({
  2038. 'reason': 'failed',
  2039. 'return_code': 400,
  2040. 'result': None,
  2041. 'error': '请求体不能为空'
  2042. }), 400
  2043. # 验证必填参数
  2044. wechat_code = data.get('wechat_code')
  2045. if not wechat_code:
  2046. return jsonify({
  2047. 'reason': 'failed',
  2048. 'return_code': 400,
  2049. 'result': None,
  2050. 'error': '缺少必填参数: wechat_code'
  2051. }), 400
  2052. # 获取可选参数
  2053. phone_number = data.get('phone_number')
  2054. id_card_number = data.get('id_card_number')
  2055. # 记录请求日志
  2056. logger.info(f"收到微信用户注册请求,wechat_code: {wechat_code}")
  2057. # 调用核心业务逻辑
  2058. result = register_wechat_user(wechat_code, phone_number, id_card_number)
  2059. # 根据返回结果设置HTTP状态码
  2060. status_code = result.get('return_code', 500)
  2061. # 记录处理结果日志
  2062. if result.get('return_code') == 201:
  2063. logger.info(f"微信用户注册成功,wechat_code: {wechat_code}")
  2064. else:
  2065. error_msg = result.get('error', '未知错误')
  2066. logger.warning(f"微信用户注册失败,wechat_code: {wechat_code},错误: {error_msg}")
  2067. # 返回结果
  2068. return jsonify(result), status_code
  2069. except Exception as e:
  2070. # 记录错误日志
  2071. error_msg = f"微信用户注册接口失败: {str(e)}"
  2072. logger.error(error_msg, exc_info=True)
  2073. # 返回错误响应
  2074. return jsonify({
  2075. 'reason': 'failed',
  2076. 'return_code': 500,
  2077. 'result': None,
  2078. 'error': error_msg
  2079. }), 500
  2080. @bp.route('/wechat-login', methods=['POST'])
  2081. def wechat_login_api():
  2082. """
  2083. 微信用户登录接口
  2084. POST /api/parse/wechat-login
  2085. Request Body:
  2086. {
  2087. "wechat_code": "wx_openid_123" // 必填:微信授权码/openid
  2088. }
  2089. Returns:
  2090. JSON: 包含登录结果的响应数据
  2091. """
  2092. try:
  2093. # 获取请求数据
  2094. data = request.get_json()
  2095. # 验证请求数据
  2096. if not data:
  2097. return jsonify({
  2098. 'reason': 'failed',
  2099. 'return_code': 400,
  2100. 'result': None,
  2101. 'error': '请求体不能为空'
  2102. }), 400
  2103. # 验证必填参数
  2104. wechat_code = data.get('wechat_code')
  2105. if not wechat_code:
  2106. return jsonify({
  2107. 'reason': 'failed',
  2108. 'return_code': 400,
  2109. 'result': None,
  2110. 'error': '缺少必填参数: wechat_code'
  2111. }), 400
  2112. # 记录请求日志
  2113. logger.info(f"收到微信用户登录请求,wechat_code: {wechat_code}")
  2114. # 调用核心业务逻辑
  2115. result = login_wechat_user(wechat_code)
  2116. # 根据返回结果设置HTTP状态码
  2117. status_code = result.get('return_code', 500)
  2118. # 记录处理结果日志
  2119. if result.get('return_code') == 200:
  2120. logger.info(f"微信用户登录成功,wechat_code: {wechat_code}")
  2121. else:
  2122. error_msg = result.get('error', '未知错误')
  2123. logger.warning(f"微信用户登录失败,wechat_code: {wechat_code},错误: {error_msg}")
  2124. # 返回结果
  2125. return jsonify(result), status_code
  2126. except Exception as e:
  2127. # 记录错误日志
  2128. error_msg = f"微信用户登录接口失败: {str(e)}"
  2129. logger.error(error_msg, exc_info=True)
  2130. # 返回错误响应
  2131. return jsonify({
  2132. 'reason': 'failed',
  2133. 'return_code': 500,
  2134. 'result': None,
  2135. 'error': error_msg
  2136. }), 500
  2137. @bp.route('/wechat-logout', methods=['POST'])
  2138. def wechat_logout_api():
  2139. """
  2140. 微信用户登出接口
  2141. POST /api/parse/wechat-logout
  2142. Request Body:
  2143. {
  2144. "wechat_code": "wx_openid_123" // 必填:微信授权码/openid
  2145. }
  2146. Returns:
  2147. JSON: 包含登出结果的响应数据
  2148. """
  2149. try:
  2150. # 获取请求数据
  2151. data = request.get_json()
  2152. # 验证请求数据
  2153. if not data:
  2154. return jsonify({
  2155. 'reason': 'failed',
  2156. 'return_code': 400,
  2157. 'result': None,
  2158. 'error': '请求体不能为空'
  2159. }), 400
  2160. # 验证必填参数
  2161. wechat_code = data.get('wechat_code')
  2162. if not wechat_code:
  2163. return jsonify({
  2164. 'reason': 'failed',
  2165. 'return_code': 400,
  2166. 'result': None,
  2167. 'error': '缺少必填参数: wechat_code'
  2168. }), 400
  2169. # 记录请求日志
  2170. logger.info(f"收到微信用户登出请求,wechat_code: {wechat_code}")
  2171. # 调用核心业务逻辑
  2172. result = logout_wechat_user(wechat_code)
  2173. # 根据返回结果设置HTTP状态码
  2174. status_code = result.get('return_code', 500)
  2175. # 记录处理结果日志
  2176. if result.get('return_code') == 200:
  2177. logger.info(f"微信用户登出成功,wechat_code: {wechat_code}")
  2178. else:
  2179. error_msg = result.get('error', '未知错误')
  2180. logger.warning(f"微信用户登出失败,wechat_code: {wechat_code},错误: {error_msg}")
  2181. # 返回结果
  2182. return jsonify(result), status_code
  2183. except Exception as e:
  2184. # 记录错误日志
  2185. error_msg = f"微信用户登出接口失败: {str(e)}"
  2186. logger.error(error_msg, exc_info=True)
  2187. # 返回错误响应
  2188. return jsonify({
  2189. 'reason': 'failed',
  2190. 'return_code': 500,
  2191. 'result': None,
  2192. 'error': error_msg
  2193. }), 500
  2194. @bp.route('/wechat-user', methods=['GET'])
  2195. def wechat_get_user_info_api():
  2196. """
  2197. 获取微信用户信息接口
  2198. GET /api/parse/wechat-user?wechat_code=wx_openid_123
  2199. Args:
  2200. wechat_code (str): 微信授权码/openid,作为查询参数
  2201. Returns:
  2202. JSON: 包含用户信息的响应数据
  2203. """
  2204. try:
  2205. # 获取查询参数
  2206. wechat_code = request.args.get('wechat_code')
  2207. # 验证必填参数
  2208. if not wechat_code:
  2209. return jsonify({
  2210. 'reason': 'failed',
  2211. 'return_code': 400,
  2212. 'result': None,
  2213. 'error': '缺少必填参数: wechat_code'
  2214. }), 400
  2215. # 记录请求日志
  2216. logger.info(f"收到获取微信用户信息请求,wechat_code: {wechat_code}")
  2217. # 调用核心业务逻辑
  2218. result = get_wechat_user_info(wechat_code)
  2219. # 根据返回结果设置HTTP状态码
  2220. status_code = result.get('return_code', 500)
  2221. # 记录处理结果日志
  2222. if result.get('return_code') == 200:
  2223. logger.info(f"获取微信用户信息成功,wechat_code: {wechat_code}")
  2224. else:
  2225. error_msg = result.get('error', '未知错误')
  2226. logger.warning(f"获取微信用户信息失败,wechat_code: {wechat_code},错误: {error_msg}")
  2227. # 返回结果
  2228. return jsonify(result), status_code
  2229. except Exception as e:
  2230. # 记录错误日志
  2231. error_msg = f"获取微信用户信息接口失败: {str(e)}"
  2232. logger.error(error_msg, exc_info=True)
  2233. # 返回错误响应
  2234. return jsonify({
  2235. 'reason': 'failed',
  2236. 'return_code': 500,
  2237. 'result': None,
  2238. 'error': error_msg
  2239. }), 500
  2240. @bp.route('/wechat-user', methods=['PUT'])
  2241. def wechat_update_user_info_api():
  2242. """
  2243. 更新微信用户信息接口
  2244. PUT /api/parse/wechat-user
  2245. Request Body:
  2246. {
  2247. "wechat_code": "wx_openid_123", // 必填:微信授权码/openid
  2248. "phone_number": "13900139000", // 可选:要更新的手机号码
  2249. "id_card_number": "110101199001011234" // 可选:要更新的身份证号码
  2250. }
  2251. Returns:
  2252. JSON: 包含更新结果的响应数据
  2253. """
  2254. try:
  2255. # 获取请求数据
  2256. data = request.get_json()
  2257. # 验证请求数据
  2258. if not data:
  2259. return jsonify({
  2260. 'reason': 'failed',
  2261. 'return_code': 400,
  2262. 'result': None,
  2263. 'error': '请求体不能为空'
  2264. }), 400
  2265. # 验证必填参数
  2266. wechat_code = data.get('wechat_code')
  2267. if not wechat_code:
  2268. return jsonify({
  2269. 'reason': 'failed',
  2270. 'return_code': 400,
  2271. 'result': None,
  2272. 'error': '缺少必填参数: wechat_code'
  2273. }), 400
  2274. # 构建更新数据,排除wechat_code
  2275. update_data = {}
  2276. if 'phone_number' in data:
  2277. update_data['phone_number'] = data['phone_number']
  2278. if 'id_card_number' in data:
  2279. update_data['id_card_number'] = data['id_card_number']
  2280. # 检查是否有要更新的数据
  2281. if not update_data:
  2282. return jsonify({
  2283. 'reason': 'failed',
  2284. 'return_code': 400,
  2285. 'result': None,
  2286. 'error': '没有提供要更新的数据'
  2287. }), 400
  2288. # 记录请求日志
  2289. logger.info(f"收到更新微信用户信息请求,wechat_code: {wechat_code}, 更新字段: {list(update_data.keys())}")
  2290. # 调用核心业务逻辑
  2291. result = update_wechat_user_info(wechat_code, update_data)
  2292. # 根据返回结果设置HTTP状态码
  2293. status_code = result.get('return_code', 500)
  2294. # 记录处理结果日志
  2295. if result.get('return_code') == 200:
  2296. logger.info(f"更新微信用户信息成功,wechat_code: {wechat_code}")
  2297. else:
  2298. error_msg = result.get('error', '未知错误')
  2299. logger.warning(f"更新微信用户信息失败,wechat_code: {wechat_code},错误: {error_msg}")
  2300. # 返回结果
  2301. return jsonify(result), status_code
  2302. except Exception as e:
  2303. # 记录错误日志
  2304. error_msg = f"更新微信用户信息接口失败: {str(e)}"
  2305. logger.error(error_msg, exc_info=True)
  2306. # 返回错误响应
  2307. return jsonify({
  2308. 'reason': 'failed',
  2309. 'return_code': 500,
  2310. 'result': None,
  2311. 'error': error_msg
  2312. }), 500