routes.py 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425
  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 import process_business_card, update_business_card, get_business_cards, update_business_card_status, create_talent_tag, get_talent_tag_list, update_talent_tag, delete_talent_tag, query_neo4j_graph, talent_get_tags, talent_update_tags, get_business_card, get_hotel_positions_list, add_hotel_positions, update_hotel_positions, query_hotel_positions, delete_hotel_positions, get_hotel_group_brands_list, add_hotel_group_brands, update_hotel_group_brands, query_hotel_group_brands, delete_hotel_group_brands, get_duplicate_records, process_duplicate_record, get_duplicate_record_detail, fix_broken_duplicate_records
  4. # 导入新的名片图片解析函数和添加名片函数
  5. from app.core.data_parse.parse_card import process_business_card_image, add_business_card, delete_business_card
  6. from app.config.config import DevelopmentConfig, ProductionConfig
  7. import logging
  8. import boto3
  9. from botocore.config import Config
  10. from botocore.exceptions import ClientError
  11. from io import BytesIO
  12. import base64
  13. import os
  14. import urllib.parse
  15. from minio import Minio
  16. # Define logger
  17. logger = logging.getLogger(__name__)
  18. # For failure responses
  19. def failed(message, code=500):
  20. return {
  21. 'success': False,
  22. 'message': message,
  23. 'data': None
  24. }, code
  25. # 根据环境选择配置
  26. if os.environ.get('FLASK_ENV') == 'production':
  27. config = ProductionConfig()
  28. else:
  29. config = DevelopmentConfig()
  30. # 使用配置变量
  31. minio_url = f"{'https' if config.MINIO_SECURE else 'http'}://{config.MINIO_HOST}"
  32. minio_access_key = config.MINIO_USER
  33. minio_secret_key = config.MINIO_PASSWORD
  34. minio_bucket = config.MINIO_BUCKET
  35. use_ssl = config.MINIO_SECURE
  36. def get_minio_client():
  37. """获取 MinIO 客户端实例"""
  38. return Minio(
  39. '192.168.3.143:9000',
  40. access_key=config.MINIO_USER,
  41. secret_key=config.MINIO_PASSWORD,
  42. secure=config.MINIO_SECURE
  43. )
  44. # 名片解析接口
  45. @bp.route('/business-card-parse', methods=['POST'])
  46. def parse_business_card_route():
  47. """
  48. 解析名片图片并提取信息的API接口(仅解析,不保存到数据库)
  49. 请求参数:
  50. - image: 名片图片文件 (multipart/form-data)
  51. 返回:
  52. - JSON: 包含提取的名片信息和处理状态
  53. 注意:此接口仅负责图片解析和信息提取,不会将数据保存到数据库
  54. """
  55. # 检查是否上传了文件
  56. if 'image' not in request.files:
  57. return jsonify({
  58. 'success': False,
  59. 'message': '未上传图片',
  60. 'data': None
  61. }), 400
  62. image_file = request.files['image']
  63. # 检查文件是否为空
  64. if image_file.filename == '':
  65. return jsonify({
  66. 'success': False,
  67. 'message': '未选择文件',
  68. 'data': None
  69. }), 400
  70. # 检查文件类型是否为图片
  71. if not image_file.content_type.startswith('image/'):
  72. return jsonify({
  73. 'success': False,
  74. 'message': '上传的文件不是图片',
  75. 'data': None
  76. }), 400
  77. # 处理名片图片
  78. result = process_business_card_image(image_file)
  79. if result['success']:
  80. return jsonify(result), 200
  81. else:
  82. return jsonify(result), 500
  83. # 添加名片记录接口
  84. @bp.route('/add-business-card', methods=['POST'])
  85. def add_business_card_route():
  86. """
  87. 添加名片记录的API接口(解析图片并保存到数据库)
  88. 请求参数:
  89. - card_data: 名片信息数据 (JSON格式,可以通过form-data或JSON body传递)
  90. - image: 名片图片文件 (multipart/form-data,可选)
  91. 返回:
  92. - JSON: 包含保存结果和处理状态
  93. 注意:此接口负责业务逻辑处理,包括重复检查、MinIO上传和数据库保存
  94. """
  95. try:
  96. # 获取名片数据 - 支持两种方式
  97. card_data = None
  98. # 方式1:通过JSON body传递
  99. if request.is_json:
  100. card_data = request.get_json()
  101. # 方式2:通过form-data传递card_data字段
  102. elif 'card_data' in request.form:
  103. import json
  104. try:
  105. card_data = json.loads(request.form['card_data'])
  106. except json.JSONDecodeError:
  107. return jsonify({
  108. 'success': False,
  109. 'message': 'card_data格式错误,必须是有效的JSON字符串',
  110. 'data': None
  111. }), 400
  112. # 检查是否提供了名片数据
  113. if not card_data:
  114. return jsonify({
  115. 'success': False,
  116. 'message': '未提供名片数据,请通过JSON body或form-data的card_data字段传递',
  117. 'data': None
  118. }), 400
  119. # 获取可选的图片文件
  120. image_file = None
  121. if 'image' in request.files:
  122. image_file = request.files['image']
  123. # 检查文件是否为空
  124. if image_file.filename == '':
  125. image_file = None
  126. # 检查文件类型是否为图片
  127. elif not image_file.content_type.startswith('image/'):
  128. return jsonify({
  129. 'success': False,
  130. 'message': '上传的文件不是图片',
  131. 'data': None
  132. }), 400
  133. # 调用业务逻辑函数处理名片数据
  134. result = add_business_card(card_data, image_file)
  135. # 根据处理结果设置HTTP状态码
  136. if result['success']:
  137. if result['code'] == 200:
  138. status_code = 200
  139. elif result['code'] == 202:
  140. status_code = 202 # Accepted - 创建成功但有疑似重复记录
  141. else:
  142. status_code = 200
  143. else:
  144. if result['code'] == 400:
  145. status_code = 400
  146. else:
  147. status_code = 500
  148. return jsonify(result), status_code
  149. except Exception as e:
  150. logger.error(f"添加名片记录失败: {str(e)}")
  151. return jsonify({
  152. 'success': False,
  153. 'message': f'添加名片记录失败: {str(e)}',
  154. 'data': None
  155. }), 500
  156. # 更新名片信息接口
  157. @bp.route('/business-cards/<int:card_id>', methods=['PUT'])
  158. def update_business_card_route(card_id):
  159. """
  160. 更新名片信息的API接口
  161. 路径参数:
  162. - card_id: 名片记录ID
  163. 请求参数:
  164. - JSON格式的名片信息
  165. 返回:
  166. - JSON: 包含更新后的名片信息和处理状态
  167. """
  168. # 获取请求数据
  169. data = request.json
  170. if not data:
  171. return jsonify({
  172. 'success': False,
  173. 'message': '请求数据为空',
  174. 'data': None
  175. }), 400
  176. # 调用业务逻辑函数处理更新
  177. result = update_business_card(card_id, data)
  178. # 根据处理结果设置HTTP状态码
  179. status_code = 200 if result['success'] else 500
  180. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  181. status_code = 404
  182. return jsonify(result), status_code
  183. # 获取所有名片记录的API接口
  184. @bp.route('/get-business-cards', methods=['GET'])
  185. def get_business_cards_route():
  186. """
  187. 获取所有名片记录的API接口
  188. 返回:
  189. - JSON: 包含名片记录列表和处理状态
  190. """
  191. # 调用业务逻辑函数获取名片列表
  192. result = get_business_cards()
  193. # 根据处理结果设置HTTP状态码
  194. status_code = 200 if result['success'] else 500
  195. return jsonify(result), status_code
  196. @bp.route('/update-business-cards/<int:card_id>/status', methods=['PUT'])
  197. def update_business_card_status_route(card_id):
  198. """
  199. 更新名片状态的API接口
  200. 路径参数:
  201. - card_id: 名片记录ID
  202. 请求参数:
  203. - JSON格式,包含status字段
  204. 返回:
  205. - JSON: 包含更新后的名片信息和处理状态
  206. """
  207. # 获取请求数据
  208. data = request.json
  209. if not data or 'status' not in data:
  210. return jsonify({
  211. 'success': False,
  212. 'message': '请求数据为空或缺少status字段',
  213. 'data': None
  214. }), 400
  215. status = data['status']
  216. # 调用业务逻辑函数处理状态更新
  217. result = update_business_card_status(card_id, status)
  218. # 根据处理结果设置HTTP状态码
  219. status_code = 200 if result['success'] else 500
  220. if 'not found' in result.get('message', '').lower() or '未找到' in result.get('message', ''):
  221. status_code = 404
  222. return jsonify(result), status_code
  223. # 从MinIO获取名片图片的API接口
  224. @bp.route('/business-cards/image/<path:image_path>', methods=['GET'])
  225. def get_business_card_image(image_path):
  226. """
  227. 从MinIO获取名片图片的API接口
  228. 路径参数:
  229. - image_path: MinIO中的图片路径
  230. 返回:
  231. - 图片数据流
  232. """
  233. try:
  234. # 记录下载请求信息,便于调试
  235. logger.info(f"获取名片图片请求: {image_path}")
  236. # 获取 MinIO 客户端
  237. minio_client = get_minio_client()
  238. if not minio_client:
  239. return jsonify(failed("MinIO客户端初始化失败")), 500
  240. try:
  241. # 使用正确的MinIO客户端方法
  242. data = minio_client.get_object(minio_bucket, image_path)
  243. # 创建内存文件流
  244. file_stream = BytesIO(data.read())
  245. # 获取文件名
  246. file_name = image_path.split('/')[-1]
  247. # 返回文件
  248. return send_file(
  249. file_stream,
  250. as_attachment=False, # 设置为False,让浏览器直接显示图片
  251. download_name=file_name,
  252. mimetype='image/jpeg' # 根据实际图片类型设置
  253. )
  254. except Exception as e:
  255. logger.error(f"MinIO获取文件失败: {str(e)}")
  256. return jsonify(failed(f"文件获取失败: {str(e)}")), 404
  257. except Exception as e:
  258. logger.error(f"文件下载失败: {str(e)}")
  259. return jsonify(failed(str(e))), 500
  260. finally:
  261. # 确保关闭数据流
  262. if 'data' in locals():
  263. data.close()
  264. # 创建人才标签接口
  265. @bp.route('/create-talent-tag', methods=['POST'])
  266. def create_talent_tag_route():
  267. """
  268. 创建人才标签的API接口
  269. 请求参数:
  270. - JSON格式,包含以下字段:
  271. - name: 标签名称
  272. - category: 标签分类
  273. - description: 标签描述
  274. - status: 启用状态,默认为'active'
  275. 返回:
  276. - JSON: 包含创建结果和标签信息
  277. """
  278. try:
  279. # 获取请求数据
  280. data = request.get_json()
  281. if not data:
  282. return jsonify({
  283. 'success': False,
  284. 'message': '请求数据为空',
  285. 'data': None
  286. }), 400
  287. # 验证必要字段
  288. if 'name' not in data or not data['name']:
  289. return jsonify({
  290. 'success': False,
  291. 'message': '标签名称不能为空',
  292. 'data': None
  293. }), 400
  294. # 处理分类字段,如果未提供则设置默认值
  295. if 'category' not in data or not data['category']:
  296. data['category'] = '未分类'
  297. # 调用业务逻辑函数处理创建
  298. result = create_talent_tag(data)
  299. # 根据处理结果设置HTTP状态码
  300. status_code = 200 if result['success'] else 500
  301. return jsonify(result), status_code
  302. except Exception as e:
  303. logger.error(f"创建人才标签失败: {str(e)}")
  304. return jsonify({
  305. 'success': False,
  306. 'message': f'创建人才标签失败: {str(e)}',
  307. 'data': None
  308. }), 500
  309. # 获取人才标签列表接口
  310. @bp.route('/get-talent-tag-list', methods=['GET'])
  311. def get_talent_tag_list_route():
  312. """
  313. 获取人才标签列表的API接口
  314. 返回:
  315. - JSON: 包含人才标签列表和处理状态
  316. """
  317. try:
  318. # 调用业务逻辑函数获取人才标签列表
  319. result = get_talent_tag_list()
  320. # 根据处理结果设置HTTP状态码
  321. status_code = 200 if result['success'] else 500
  322. return jsonify(result), status_code
  323. except Exception as e:
  324. logger.error(f"获取人才标签列表失败: {str(e)}")
  325. return jsonify({
  326. 'success': False,
  327. 'message': f'获取人才标签列表失败: {str(e)}',
  328. 'data': []
  329. }), 500
  330. # 更新人才标签接口
  331. @bp.route('/update-talent-tag/<int:tag_id>', methods=['PUT'])
  332. def update_talent_tag_route(tag_id):
  333. """
  334. 更新人才标签的API接口
  335. 路径参数:
  336. - tag_id: 标签节点ID
  337. 请求参数:
  338. - JSON格式,可能包含以下字段:
  339. - name: 标签名称
  340. - category: 标签分类
  341. - description: 标签描述
  342. - status: 启用状态
  343. 返回:
  344. - JSON: 包含更新结果和标签信息
  345. """
  346. try:
  347. # 获取请求数据
  348. data = request.get_json()
  349. if not data:
  350. return jsonify({
  351. 'success': False,
  352. 'message': '请求数据为空',
  353. 'data': None
  354. }), 400
  355. # 调用业务逻辑函数处理更新
  356. result = update_talent_tag(tag_id, data)
  357. # 根据处理结果设置HTTP状态码
  358. if not result['success']:
  359. if result['code'] == 404:
  360. status_code = 404
  361. elif result['code'] == 400:
  362. status_code = 400
  363. else:
  364. status_code = 500
  365. else:
  366. status_code = 200
  367. return jsonify(result), status_code
  368. except Exception as e:
  369. logger.error(f"更新人才标签失败: {str(e)}")
  370. return jsonify({
  371. 'success': False,
  372. 'message': f'更新人才标签失败: {str(e)}',
  373. 'data': None
  374. }), 500
  375. # 删除人才标签接口
  376. @bp.route('/delete-talent-tag/<int:tag_id>', methods=['DELETE'])
  377. def delete_talent_tag_route(tag_id):
  378. """
  379. 删除人才标签的API接口
  380. 路径参数:
  381. - tag_id: 标签节点ID
  382. 返回:
  383. - JSON: 包含删除结果和被删除的标签信息
  384. """
  385. try:
  386. # 调用业务逻辑函数执行删除
  387. result = delete_talent_tag(tag_id)
  388. # 根据处理结果设置HTTP状态码
  389. if not result['success']:
  390. if result['code'] == 404:
  391. status_code = 404
  392. else:
  393. status_code = 500
  394. else:
  395. status_code = 200
  396. return jsonify(result), status_code
  397. except Exception as e:
  398. logger.error(f"删除人才标签失败: {str(e)}")
  399. return jsonify({
  400. 'success': False,
  401. 'message': f'删除人才标签失败: {str(e)}',
  402. 'data': None
  403. }), 500
  404. @bp.route('/query-kg', methods=['POST'])
  405. def query_kg():
  406. """
  407. 查询知识图谱API接口
  408. 请求参数:
  409. - query_requirement: 查询需求描述(JSON格式)
  410. 返回:
  411. - JSON: 包含查询结果和处理状态
  412. """
  413. try:
  414. # 获取请求数据
  415. data = request.json
  416. if not data or 'query_requirement' not in data:
  417. return jsonify({
  418. 'code': 400,
  419. 'success': False,
  420. 'message': '请求数据为空或缺少query_requirement字段',
  421. 'data': []
  422. }), 400
  423. query_requirement = data['query_requirement']
  424. # 调用业务逻辑函数执行查询
  425. result = query_neo4j_graph(query_requirement)
  426. # 根据处理结果设置HTTP状态码
  427. status_code = 200 if result['success'] else 500
  428. return jsonify(result), status_code
  429. except Exception as e:
  430. logger.error(f"查询知识图谱失败: {str(e)}")
  431. return jsonify({
  432. 'code': 500,
  433. 'success': False,
  434. 'message': f"查询知识图谱失败: {str(e)}",
  435. 'data': []
  436. }), 500
  437. @bp.route('/talent-get-tags/<int:talent_id>', methods=['GET'])
  438. def talent_get_tags_route(talent_id):
  439. """
  440. 获取人才标签的API接口
  441. 路径参数:
  442. - talent_id: 人才节点ID
  443. 返回:
  444. - JSON: 包含人才关联的标签列表和处理状态
  445. """
  446. try:
  447. # 调用业务逻辑函数获取人才标签
  448. result = talent_get_tags(talent_id)
  449. # 根据处理结果设置HTTP状态码
  450. status_code = 200 if result['success'] else 500
  451. return jsonify(result), status_code
  452. except Exception as e:
  453. logger.error(f"获取人才标签失败: {str(e)}")
  454. return jsonify({
  455. 'code': 500,
  456. 'success': False,
  457. 'message': f"获取人才标签失败: {str(e)}",
  458. 'data': []
  459. }), 500
  460. @bp.route('/talent-update-tags', methods=['POST'])
  461. def talent_update_tags_route():
  462. """
  463. 更新人才标签关系的API接口
  464. 请求参数:
  465. - JSON数组,包含talent和tag字段的对象列表
  466. 例如: [
  467. {"talent": 12345, "tag": "市场营销"},
  468. {"talent": 12345, "tag": "酒店管理"}
  469. ]
  470. 返回:
  471. - JSON: 包含更新结果的状态信息
  472. """
  473. try:
  474. # 获取请求数据
  475. data = request.json
  476. if not data:
  477. return jsonify({
  478. 'code': 400,
  479. 'success': False,
  480. 'message': '请求数据为空',
  481. 'data': None
  482. }), 400
  483. # 调用业务逻辑函数处理标签关系更新
  484. result = talent_update_tags(data)
  485. # 根据处理结果设置HTTP状态码
  486. if result['code'] == 200:
  487. status_code = 200
  488. elif result['code'] == 206:
  489. status_code = 206 # Partial Content
  490. elif result['code'] == 400:
  491. status_code = 400 # Bad Request
  492. elif result['code'] == 404:
  493. status_code = 404 # Not Found
  494. else:
  495. status_code = 500 # Internal Server Error
  496. return jsonify(result), status_code
  497. except Exception as e:
  498. logger.error(f"更新人才标签关系失败: {str(e)}")
  499. return jsonify({
  500. 'code': 500,
  501. 'success': False,
  502. 'message': f"更新人才标签关系失败: {str(e)}",
  503. 'data': None
  504. }), 500
  505. # 测试MinIO连接
  506. def test_minio_connection():
  507. """测试MinIO连接是否正常"""
  508. try:
  509. client = get_minio_client()
  510. if client.bucket_exists(minio_bucket):
  511. return {
  512. 'success': True,
  513. 'message': f'连接MinIO服务器成功,存储桶 {minio_bucket} 存在',
  514. 'config': {
  515. 'host': config.MINIO_HOST,
  516. 'bucket': minio_bucket,
  517. 'secure': use_ssl
  518. }
  519. }
  520. else:
  521. return {
  522. 'success': False,
  523. 'message': f'连接MinIO服务器成功,但存储桶 {minio_bucket} 不存在',
  524. 'config': {
  525. 'host': config.MINIO_HOST,
  526. 'bucket': minio_bucket,
  527. 'secure': use_ssl
  528. }
  529. }
  530. except Exception as e:
  531. return {
  532. 'success': False,
  533. 'message': f'连接MinIO服务器失败: {str(e)}',
  534. 'config': {
  535. 'host': config.MINIO_HOST,
  536. 'bucket': minio_bucket,
  537. 'secure': use_ssl
  538. }
  539. }
  540. # MinIO测试接口
  541. @bp.route('/test-minio-connection', methods=['GET'])
  542. def test_minio_connection_route():
  543. """
  544. 测试MinIO连接的API接口
  545. 返回:
  546. - JSON: 包含连接测试结果
  547. """
  548. try:
  549. result = test_minio_connection()
  550. status_code = 200 if result['success'] else 500
  551. return jsonify(result), status_code
  552. except Exception as e:
  553. logger.error(f"测试MinIO连接失败: {str(e)}")
  554. return jsonify({
  555. 'success': False,
  556. 'message': f'测试MinIO连接失败: {str(e)}',
  557. 'config': {
  558. 'host': config.MINIO_HOST,
  559. 'bucket': minio_bucket,
  560. 'secure': use_ssl
  561. }
  562. }), 500
  563. # 获取单个名片记录的API接口
  564. @bp.route('/get-business-card/<int:card_id>', methods=['GET'])
  565. def get_business_card_route(card_id):
  566. """
  567. 获取单个名片记录的API接口
  568. 路径参数:
  569. - card_id: 名片记录ID
  570. 返回:
  571. - JSON: 包含名片记录信息和处理状态
  572. """
  573. # 调用业务逻辑函数获取名片记录
  574. result = get_business_card(card_id)
  575. # 根据处理结果设置HTTP状态码
  576. if not result['success']:
  577. if result['code'] == 404:
  578. status_code = 404
  579. else:
  580. status_code = 500
  581. else:
  582. status_code = 200
  583. return jsonify(result), status_code
  584. @bp.route('/get-hotel-positions-list', methods=['GET'])
  585. def get_hotel_positions_list_route():
  586. """
  587. 获取酒店职位数据表全部记录的API接口
  588. 返回:
  589. - JSON: 包含酒店职位记录列表和处理状态
  590. """
  591. try:
  592. # 调用业务逻辑函数获取酒店职位列表
  593. result = get_hotel_positions_list()
  594. # 根据处理结果设置HTTP状态码
  595. status_code = 200 if result['success'] else 500
  596. return jsonify(result), status_code
  597. except Exception as e:
  598. # 处理未预期的异常
  599. error_msg = f"获取酒店职位列表时发生错误: {str(e)}"
  600. logger.error(error_msg, exc_info=True)
  601. return jsonify({
  602. 'success': False,
  603. 'message': error_msg,
  604. 'data': [],
  605. 'count': 0
  606. }), 500
  607. @bp.route('/add-hotel-positions', methods=['POST'])
  608. def add_hotel_positions_route():
  609. """
  610. 新增酒店职位数据表记录的API接口
  611. 请求参数:
  612. - JSON格式,包含以下字段:
  613. - department_zh: 部门中文名称 (必填)
  614. - department_en: 部门英文名称 (必填)
  615. - position_zh: 职位中文名称 (必填)
  616. - position_en: 职位英文名称 (必填)
  617. - position_abbr: 职位英文缩写 (可选)
  618. - level_zh: 职级中文名称 (必填)
  619. - level_en: 职级英文名称 (必填)
  620. - created_by: 创建者 (可选)
  621. - updated_by: 更新者 (可选)
  622. - status: 状态 (可选)
  623. 返回:
  624. - JSON: 包含创建结果和职位信息
  625. """
  626. try:
  627. # 获取请求数据
  628. data = request.get_json()
  629. if not data:
  630. return jsonify({
  631. 'success': False,
  632. 'message': '请求数据为空',
  633. 'data': None
  634. }), 400
  635. # 调用业务逻辑函数处理创建
  636. result = add_hotel_positions(data)
  637. # 根据处理结果设置HTTP状态码
  638. if result['code'] == 200:
  639. status_code = 201 # Created
  640. elif result['code'] == 400:
  641. status_code = 400 # Bad Request
  642. elif result['code'] == 409:
  643. status_code = 409 # Conflict
  644. else:
  645. status_code = 500 # Internal Server Error
  646. return jsonify(result), status_code
  647. except Exception as e:
  648. # 处理未预期的异常
  649. error_msg = f"创建酒店职位记录时发生错误: {str(e)}"
  650. logger.error(error_msg, exc_info=True)
  651. return jsonify({
  652. 'success': False,
  653. 'message': error_msg,
  654. 'data': None
  655. }), 500
  656. @bp.route('/update-hotel-positions/<int:position_id>', methods=['PUT'])
  657. def update_hotel_positions_route(position_id):
  658. """
  659. 修改酒店职位数据表记录的API接口
  660. 路径参数:
  661. - position_id: 职位记录ID
  662. 请求参数:
  663. - JSON格式,可能包含以下字段:
  664. - department_zh: 部门中文名称
  665. - department_en: 部门英文名称
  666. - position_zh: 职位中文名称
  667. - position_en: 职位英文名称
  668. - position_abbr: 职位英文缩写
  669. - level_zh: 职级中文名称
  670. - level_en: 职级英文名称
  671. - updated_by: 更新者
  672. - status: 状态
  673. 返回:
  674. - JSON: 包含更新结果和职位信息
  675. """
  676. try:
  677. # 获取请求数据
  678. data = request.get_json()
  679. if not data:
  680. return jsonify({
  681. 'success': False,
  682. 'message': '请求数据为空',
  683. 'data': None
  684. }), 400
  685. # 调用业务逻辑函数处理更新
  686. result = update_hotel_positions(position_id, data)
  687. # 根据处理结果设置HTTP状态码
  688. if result['code'] == 200:
  689. status_code = 200 # OK
  690. elif result['code'] == 400:
  691. status_code = 400 # Bad Request
  692. elif result['code'] == 404:
  693. status_code = 404 # Not Found
  694. elif result['code'] == 409:
  695. status_code = 409 # Conflict
  696. else:
  697. status_code = 500 # Internal Server Error
  698. return jsonify(result), status_code
  699. except Exception as e:
  700. # 处理未预期的异常
  701. error_msg = f"更新酒店职位记录时发生错误: {str(e)}"
  702. logger.error(error_msg, exc_info=True)
  703. return jsonify({
  704. 'success': False,
  705. 'message': error_msg,
  706. 'data': None
  707. }), 500
  708. @bp.route('/query-hotel-positions/<int:position_id>', methods=['GET'])
  709. def query_hotel_positions_route(position_id):
  710. """
  711. 查找指定ID的酒店职位数据表记录的API接口
  712. 路径参数:
  713. - position_id: 职位记录ID
  714. 返回:
  715. - JSON: 包含查找结果和职位信息
  716. """
  717. try:
  718. # 调用业务逻辑函数查找职位记录
  719. result = query_hotel_positions(position_id)
  720. # 根据处理结果设置HTTP状态码
  721. if result['code'] == 200:
  722. status_code = 200 # OK
  723. elif result['code'] == 404:
  724. status_code = 404 # Not Found
  725. else:
  726. status_code = 500 # Internal Server Error
  727. return jsonify(result), status_code
  728. except Exception as e:
  729. # 处理未预期的异常
  730. error_msg = f"查找酒店职位记录时发生错误: {str(e)}"
  731. logger.error(error_msg, exc_info=True)
  732. return jsonify({
  733. 'success': False,
  734. 'message': error_msg,
  735. 'data': None
  736. }), 500
  737. @bp.route('/delete-hotel-positions/<int:position_id>', methods=['DELETE'])
  738. def delete_hotel_positions_route(position_id):
  739. """
  740. 删除指定ID的酒店职位数据表记录的API接口
  741. 路径参数:
  742. - position_id: 职位记录ID
  743. 返回:
  744. - JSON: 包含删除结果和被删除的职位信息
  745. """
  746. try:
  747. # 调用业务逻辑函数删除职位记录
  748. result = delete_hotel_positions(position_id)
  749. # 根据处理结果设置HTTP状态码
  750. if result['code'] == 200:
  751. status_code = 200 # OK
  752. elif result['code'] == 404:
  753. status_code = 404 # Not Found
  754. else:
  755. status_code = 500 # Internal Server Error
  756. return jsonify(result), status_code
  757. except Exception as e:
  758. # 处理未预期的异常
  759. error_msg = f"删除酒店职位记录时发生错误: {str(e)}"
  760. logger.error(error_msg, exc_info=True)
  761. return jsonify({
  762. 'success': False,
  763. 'message': error_msg,
  764. 'data': None
  765. }), 500
  766. @bp.route('/get-hotel-group-brands-list', methods=['GET'])
  767. def get_hotel_group_brands_list_route():
  768. """
  769. 获取酒店集团子品牌数据表全部记录的API接口
  770. 返回:
  771. - JSON: 包含酒店集团品牌记录列表和处理状态
  772. """
  773. try:
  774. # 调用业务逻辑函数获取酒店集团品牌列表
  775. result = get_hotel_group_brands_list()
  776. # 根据处理结果设置HTTP状态码
  777. status_code = 200 if result['success'] else 500
  778. return jsonify(result), status_code
  779. except Exception as e:
  780. # 处理未预期的异常
  781. error_msg = f"获取酒店集团品牌列表时发生错误: {str(e)}"
  782. logger.error(error_msg, exc_info=True)
  783. return jsonify({
  784. 'success': False,
  785. 'message': error_msg,
  786. 'data': [],
  787. 'count': 0
  788. }), 500
  789. @bp.route('/add-hotel-group-brands', methods=['POST'])
  790. def add_hotel_group_brands_route():
  791. """
  792. 新增酒店集团子品牌数据表记录的API接口
  793. 请求参数:
  794. - JSON格式,包含以下字段:
  795. - group_name_en: 集团英文名称 (必填)
  796. - group_name_zh: 集团中文名称 (必填)
  797. - brand_name_en: 品牌英文名称 (必填)
  798. - brand_name_zh: 品牌中文名称 (必填)
  799. - positioning_level_en: 定位级别英文名称 (必填)
  800. - positioning_level_zh: 定位级别中文名称 (必填)
  801. - created_by: 创建者 (可选)
  802. - updated_by: 更新者 (可选)
  803. - status: 状态 (可选)
  804. 返回:
  805. - JSON: 包含创建结果和品牌信息
  806. """
  807. try:
  808. # 获取请求数据
  809. data = request.get_json()
  810. if not data:
  811. return jsonify({
  812. 'success': False,
  813. 'message': '请求数据为空',
  814. 'data': None
  815. }), 400
  816. # 调用业务逻辑函数处理创建
  817. result = add_hotel_group_brands(data)
  818. # 根据处理结果设置HTTP状态码
  819. if result['code'] == 200:
  820. status_code = 201 # Created
  821. elif result['code'] == 400:
  822. status_code = 400 # Bad Request
  823. elif result['code'] == 409:
  824. status_code = 409 # Conflict
  825. else:
  826. status_code = 500 # Internal Server Error
  827. return jsonify(result), status_code
  828. except Exception as e:
  829. # 处理未预期的异常
  830. error_msg = f"创建酒店集团品牌记录时发生错误: {str(e)}"
  831. logger.error(error_msg, exc_info=True)
  832. return jsonify({
  833. 'success': False,
  834. 'message': error_msg,
  835. 'data': None
  836. }), 500
  837. @bp.route('/update-hotel-group-brands/<int:brand_id>', methods=['PUT'])
  838. def update_hotel_group_brands_route(brand_id):
  839. """
  840. 修改酒店集团子品牌数据表记录的API接口
  841. 路径参数:
  842. - brand_id: 品牌记录ID
  843. 请求参数:
  844. - JSON格式,可能包含以下字段:
  845. - group_name_en: 集团英文名称
  846. - group_name_zh: 集团中文名称
  847. - brand_name_en: 品牌英文名称
  848. - brand_name_zh: 品牌中文名称
  849. - positioning_level_en: 定位级别英文名称
  850. - positioning_level_zh: 定位级别中文名称
  851. - updated_by: 更新者
  852. - status: 状态
  853. 返回:
  854. - JSON: 包含更新结果和品牌信息
  855. """
  856. try:
  857. # 获取请求数据
  858. data = request.get_json()
  859. if not data:
  860. return jsonify({
  861. 'success': False,
  862. 'message': '请求数据为空',
  863. 'data': None
  864. }), 400
  865. # 调用业务逻辑函数处理更新
  866. result = update_hotel_group_brands(brand_id, data)
  867. # 根据处理结果设置HTTP状态码
  868. if result['code'] == 200:
  869. status_code = 200 # OK
  870. elif result['code'] == 400:
  871. status_code = 400 # Bad Request
  872. elif result['code'] == 404:
  873. status_code = 404 # Not Found
  874. elif result['code'] == 409:
  875. status_code = 409 # Conflict
  876. else:
  877. status_code = 500 # Internal Server Error
  878. return jsonify(result), status_code
  879. except Exception as e:
  880. # 处理未预期的异常
  881. error_msg = f"更新酒店集团品牌记录时发生错误: {str(e)}"
  882. logger.error(error_msg, exc_info=True)
  883. return jsonify({
  884. 'success': False,
  885. 'message': error_msg,
  886. 'data': None
  887. }), 500
  888. @bp.route('/query-hotel-group-brands/<int:brand_id>', methods=['GET'])
  889. def query_hotel_group_brands_route(brand_id):
  890. """
  891. 查找指定ID的酒店集团子品牌数据表记录的API接口
  892. 路径参数:
  893. - brand_id: 品牌记录ID
  894. 返回:
  895. - JSON: 包含查找结果和品牌信息
  896. """
  897. try:
  898. # 调用业务逻辑函数查找品牌记录
  899. result = query_hotel_group_brands(brand_id)
  900. # 根据处理结果设置HTTP状态码
  901. if result['code'] == 200:
  902. status_code = 200 # OK
  903. elif result['code'] == 404:
  904. status_code = 404 # Not Found
  905. else:
  906. status_code = 500 # Internal Server Error
  907. return jsonify(result), status_code
  908. except Exception as e:
  909. # 处理未预期的异常
  910. error_msg = f"查找酒店集团品牌记录时发生错误: {str(e)}"
  911. logger.error(error_msg, exc_info=True)
  912. return jsonify({
  913. 'success': False,
  914. 'message': error_msg,
  915. 'data': None
  916. }), 500
  917. @bp.route('/delete-hotel-group-brands/<int:brand_id>', methods=['DELETE'])
  918. def delete_hotel_group_brands_route(brand_id):
  919. """
  920. 删除指定ID的酒店集团子品牌数据表记录的API接口
  921. 路径参数:
  922. - brand_id: 品牌记录ID
  923. 返回:
  924. - JSON: 包含删除结果和被删除的品牌信息
  925. """
  926. try:
  927. # 调用业务逻辑函数删除品牌记录
  928. result = delete_hotel_group_brands(brand_id)
  929. # 根据处理结果设置HTTP状态码
  930. if result['code'] == 200:
  931. status_code = 200 # OK
  932. elif result['code'] == 404:
  933. status_code = 404 # Not Found
  934. else:
  935. status_code = 500 # Internal Server Error
  936. return jsonify(result), status_code
  937. except Exception as e:
  938. # 处理未预期的异常
  939. error_msg = f"删除酒店集团品牌记录时发生错误: {str(e)}"
  940. logger.error(error_msg, exc_info=True)
  941. return jsonify({
  942. 'success': False,
  943. 'message': error_msg,
  944. 'data': None
  945. }), 500
  946. # ==================================
  947. # 重复记录处理API接口
  948. # ==================================
  949. @bp.route('/get-duplicate-records', methods=['GET'])
  950. def get_duplicate_records_route():
  951. """
  952. 获取重复记录列表的API接口
  953. 查询参数:
  954. - status: 可选,筛选特定状态的记录 ('pending', 'processed', 'ignored')
  955. 返回:
  956. - JSON: 包含重复记录列表和处理状态
  957. """
  958. try:
  959. # 获取查询参数
  960. status = request.args.get('status', None)
  961. # 验证status参数的有效性
  962. if status and status not in ['pending', 'processed', 'ignored']:
  963. return jsonify({
  964. 'success': False,
  965. 'message': 'status参数无效,必须为 pending、processed 或 ignored',
  966. 'data': None
  967. }), 400
  968. # 调用业务逻辑函数获取重复记录列表
  969. result = get_duplicate_records(status)
  970. # 根据处理结果设置HTTP状态码
  971. status_code = 200 if result['success'] else 500
  972. return jsonify(result), status_code
  973. except Exception as e:
  974. # 处理未预期的异常
  975. error_msg = f"获取重复记录列表时发生错误: {str(e)}"
  976. logger.error(error_msg, exc_info=True)
  977. return jsonify({
  978. 'success': False,
  979. 'message': error_msg,
  980. 'data': [],
  981. 'count': 0
  982. }), 500
  983. @bp.route('/process-duplicate-record/<int:duplicate_id>', methods=['POST'])
  984. def process_duplicate_record_route(duplicate_id):
  985. """
  986. 处理重复记录的API接口
  987. 路径参数:
  988. - duplicate_id: 重复记录ID
  989. 请求参数:
  990. - JSON格式,包含以下字段:
  991. - action: 处理动作 (必填) ('merge_to_suspected', 'keep_main', 'ignore')
  992. - selected_duplicate_id: 当action为'merge_to_suspected'时,选择的疑似重复记录ID (可选)
  993. - processed_by: 处理人 (可选)
  994. - notes: 处理备注 (可选)
  995. 返回:
  996. - JSON: 包含处理结果和状态信息
  997. """
  998. try:
  999. # 获取请求数据
  1000. data = request.get_json()
  1001. if not data:
  1002. return jsonify({
  1003. 'success': False,
  1004. 'message': '请求数据为空',
  1005. 'data': None
  1006. }), 400
  1007. # 验证必填字段
  1008. action = data.get('action')
  1009. if not action:
  1010. return jsonify({
  1011. 'success': False,
  1012. 'message': '缺少必填字段: action',
  1013. 'data': None
  1014. }), 400
  1015. # 验证action参数的有效性
  1016. if action not in ['merge_to_suspected', 'keep_main', 'ignore']:
  1017. return jsonify({
  1018. 'success': False,
  1019. 'message': 'action参数无效,必须为 merge_to_suspected、keep_main 或 ignore',
  1020. 'data': None
  1021. }), 400
  1022. # 提取其他参数
  1023. selected_duplicate_id = data.get('selected_duplicate_id')
  1024. processed_by = data.get('processed_by')
  1025. notes = data.get('notes')
  1026. # 特殊验证:如果action为merge_to_suspected,必须提供selected_duplicate_id
  1027. if action == 'merge_to_suspected' and not selected_duplicate_id:
  1028. return jsonify({
  1029. 'success': False,
  1030. 'message': '执行merge_to_suspected操作时必须提供selected_duplicate_id',
  1031. 'data': None
  1032. }), 400
  1033. # 调用业务逻辑函数处理重复记录
  1034. result = process_duplicate_record(
  1035. duplicate_id=duplicate_id,
  1036. action=action,
  1037. selected_duplicate_id=selected_duplicate_id,
  1038. processed_by=processed_by,
  1039. notes=notes
  1040. )
  1041. # 根据处理结果设置HTTP状态码
  1042. if result['code'] == 200:
  1043. status_code = 200 # OK
  1044. elif result['code'] == 400:
  1045. status_code = 400 # Bad Request
  1046. elif result['code'] == 404:
  1047. status_code = 404 # Not Found
  1048. else:
  1049. status_code = 500 # Internal Server Error
  1050. return jsonify(result), status_code
  1051. except Exception as e:
  1052. # 处理未预期的异常
  1053. error_msg = f"处理重复记录时发生错误: {str(e)}"
  1054. logger.error(error_msg, exc_info=True)
  1055. return jsonify({
  1056. 'success': False,
  1057. 'message': error_msg,
  1058. 'data': None
  1059. }), 500
  1060. @bp.route('/get-duplicate-record-detail/<int:duplicate_id>', methods=['GET'])
  1061. def get_duplicate_record_detail_route(duplicate_id):
  1062. """
  1063. 获取指定重复记录详细信息的API接口
  1064. 路径参数:
  1065. - duplicate_id: 重复记录ID
  1066. 返回:
  1067. - JSON: 包含重复记录详细信息
  1068. """
  1069. try:
  1070. # 调用业务逻辑函数获取重复记录详情
  1071. result = get_duplicate_record_detail(duplicate_id)
  1072. # 根据处理结果设置HTTP状态码
  1073. if result['code'] == 200:
  1074. status_code = 200 # OK
  1075. elif result['code'] == 404:
  1076. status_code = 404 # Not Found
  1077. else:
  1078. status_code = 500 # Internal Server Error
  1079. return jsonify(result), status_code
  1080. except Exception as e:
  1081. # 处理未预期的异常
  1082. error_msg = f"获取重复记录详情时发生错误: {str(e)}"
  1083. logger.error(error_msg, exc_info=True)
  1084. return jsonify({
  1085. 'success': False,
  1086. 'message': error_msg,
  1087. 'data': None
  1088. }), 500
  1089. # 删除名片记录接口
  1090. @bp.route('/delete-business-card/<int:card_id>', methods=['DELETE'])
  1091. def delete_business_card_route(card_id):
  1092. """
  1093. 删除名片记录的API接口
  1094. 路径参数:
  1095. - card_id: 名片记录ID (必填)
  1096. 功能说明:
  1097. - 删除PostgreSQL数据库中business_cards表的指定记录
  1098. - 删除PostgreSQL数据库中duplicate_business_cards表的相关记录
  1099. - 删除MinIO存储中的名片图片文件
  1100. - 删除Neo4j图数据库中talent节点及其关联关系
  1101. 返回:
  1102. - JSON: 包含删除操作的结果状态和被删除的记录信息
  1103. 状态码:
  1104. - 200: 完全成功删除所有相关数据
  1105. - 206: 部分成功 (PostgreSQL删除成功,但Neo4j删除失败)
  1106. - 404: 未找到指定ID的名片记录
  1107. - 500: 删除操作失败
  1108. """
  1109. try:
  1110. # 验证card_id参数
  1111. if not card_id or card_id <= 0:
  1112. return jsonify({
  1113. 'success': False,
  1114. 'message': '无效的名片记录ID',
  1115. 'data': None
  1116. }), 400
  1117. # 调用删除函数
  1118. result = delete_business_card(card_id)
  1119. # 根据处理结果设置HTTP状态码和返回响应
  1120. if result['success']:
  1121. if result['code'] == 200:
  1122. status_code = 200 # 完全成功
  1123. elif result['code'] == 206:
  1124. status_code = 206 # 部分成功
  1125. else:
  1126. status_code = 200 # 默认成功
  1127. else:
  1128. if result['code'] == 404:
  1129. status_code = 404 # 未找到记录
  1130. elif result['code'] == 400:
  1131. status_code = 400 # 参数错误
  1132. else:
  1133. status_code = 500 # 服务器错误
  1134. return jsonify(result), status_code
  1135. except Exception as e:
  1136. logger.error(f"删除名片记录失败: {str(e)}")
  1137. return jsonify({
  1138. 'success': False,
  1139. 'message': f'删除名片记录失败: {str(e)}',
  1140. 'data': None
  1141. }), 500
  1142. # 修复损坏的重复记录接口
  1143. @bp.route('/fix-broken-duplicate-records', methods=['POST'])
  1144. def fix_broken_duplicate_records_route():
  1145. """
  1146. 修复duplicate_business_cards表中main_card_id为null的损坏记录
  1147. 功能说明:
  1148. - 查找所有main_card_id为null的损坏记录
  1149. - 删除这些损坏的记录以维护数据完整性
  1150. - 返回修复操作的详细结果
  1151. 返回:
  1152. - JSON: 包含修复操作的结果和被删除记录的信息
  1153. 状态码:
  1154. - 200: 修复成功
  1155. - 500: 修复失败
  1156. 注意:
  1157. - 此操作会永久删除损坏的记录
  1158. - 建议在系统维护时执行此操作
  1159. """
  1160. try:
  1161. # 调用修复函数
  1162. result = fix_broken_duplicate_records()
  1163. # 根据结果设置状态码
  1164. if result['success']:
  1165. status_code = 200
  1166. else:
  1167. status_code = 500
  1168. return jsonify(result), status_code
  1169. except Exception as e:
  1170. logger.error(f"修复损坏记录接口调用失败: {str(e)}")
  1171. return jsonify({
  1172. 'success': False,
  1173. 'message': f'修复损坏记录接口调用失败: {str(e)}',
  1174. 'data': None
  1175. }), 500