api.py 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445
  1. from re import search
  2. from natsort.natsort import order_by_index
  3. from sqlalchemy import func
  4. from models import *
  5. import requests
  6. import json
  7. from logger import logger
  8. from serial.tools import list_ports
  9. from model import PhotoRecord
  10. import settings, datetime
  11. import pandas as pd
  12. from utils.hlm_http_request import forward_request
  13. from utils.utils_func import check_path
  14. from sockets.socket_client import socket_manager
  15. from mcu.DeviceControl import DeviceControl
  16. import time, shutil, os
  17. from sqlalchemy import and_, asc, desc
  18. from functools import partial
  19. from service.deal_image import DealImage
  20. from databases import DeviceConfig, SysConfigs, SqlQuery, CRUD, select, DeviceConfigTabs
  21. from service.run_main import RunMain
  22. import importlib
  23. from service.auto_deal_pics.upload_pic import UploadPic
  24. from service.OnePicTest import OnePicTest
  25. from service.base import check_move_goods_art_no_folder
  26. import win32api, win32gui, win32con
  27. from win32gui import EnumWindows, GetWindowText
  28. from service.online_request.module_online_data import OnlineDataRequest
  29. from concurrent.futures import ThreadPoolExecutor
  30. from functools import partial
  31. from service.online_request.module_online_data import AIGCDataRequest
  32. import asyncio
  33. from fastapi import BackgroundTasks
  34. import functools
  35. import traceback
  36. import concurrent.futures
  37. def log_exception_with_context(context_message=""):
  38. """装饰器:为函数添加异常日志上下文"""
  39. def decorator(func):
  40. @functools.wraps(func)
  41. def wrapper(*args, **kwargs):
  42. try:
  43. return func(*args, **kwargs)
  44. except Exception as e:
  45. logger.error(f"=== 异常发生在函数: {func.__name__} ===")
  46. if context_message:
  47. logger.error(f"上下文信息: {context_message}")
  48. logger.error(f"函数参数: args={args}, kwargs={kwargs}")
  49. logger.error(f"异常类型: {type(e).__name__}")
  50. logger.error(f"异常信息: {str(e)}")
  51. logger.error("完整堆栈跟踪:")
  52. logger.error(traceback.format_exc())
  53. raise # 重新抛出异常
  54. return wrapper
  55. return decorator
  56. def parserGoodsDict2Aigc(return_data_check_before_detail):
  57. """获取商品组装数据"""
  58. print("获取商品组装数据===========>return_data_check_before_detail",return_data_check_before_detail)
  59. goods_no_dict = return_data_check_before_detail.get("data", {}).get(
  60. "goods_no_dict", {}
  61. )
  62. return goods_no_dict
  63. async def sendAsyncMessage(msg="", goods_arts=[], status="", msg_type="", data=None,progress=None):
  64. """异步发送消息"""
  65. mapping = {"segment_progress":"抠图","scene_progress":"场景图","upper_footer_progress":"模特图","detail_progress":"详情页","upload_goods_progress":"上传第三方商品"}
  66. if progress is not None:
  67. progress["msg_type"] = msg_type
  68. name = mapping.get(msg_type,None)
  69. if name:
  70. progress["name"] = name
  71. data = {
  72. "code": 0,
  73. "msg": msg,
  74. "status": 2,
  75. "data": (
  76. data
  77. if data is not None
  78. else {
  79. "status": status,
  80. "goods_art_nos": goods_arts,
  81. }
  82. ),
  83. "progress": None if progress is None else progress,
  84. "msg_type": msg_type,
  85. }
  86. await message_queue.put(data)
  87. @app.get("/")
  88. async def index():
  89. # await socket_manager.send_message(msg="测试")
  90. return {"message": "Hello World"}
  91. @app.get("/scan_serials", description="扫描可用的设备端口")
  92. async def scanSerials():
  93. """扫描串口"""
  94. ports = list_ports.comports()
  95. print("Scanning", ports)
  96. return {"message": "Hello World"}
  97. @app.api_route(
  98. "/forward_request", methods=["GET", "POST"], description="代理转发hlm项目得请求"
  99. )
  100. async def forwardRequest(request: HlmForwardRequest):
  101. """
  102. 转发HTTP请求到目标URL
  103. :param request: FastAPI Request对象
  104. :return: 目标接口的响应
  105. """
  106. try:
  107. if request.method == "GET":
  108. params = request.query_params
  109. elif request.method == "POST":
  110. params = json.dump(request.query_params)
  111. else:
  112. raise UnicornException("仅支持GET和POST方法")
  113. target_url = request.target_url
  114. method = request.method.upper()
  115. headers = request.headers
  116. if not target_url:
  117. raise UnicornException("目标url地址不能为空")
  118. # 调用 hlm_http_request 中的 forward_request 函数
  119. response = forward_request(
  120. target_url, params=params, method=method, headers=headers
  121. )
  122. return response
  123. except requests.RequestException as e:
  124. raise UnicornException(e)
  125. except Exception as e:
  126. raise UnicornException(e)
  127. def __createExcelGoodsArray(excel_path):
  128. '''创建通过excel形式得货号组数据'''
  129. excel_df = pd.read_excel(excel_path, sheet_name=0, header=0)
  130. if "文件夹名称" not in excel_df.columns:
  131. raise UnicornException("缺失 [文件夹名称] 列")
  132. if "商品货号" not in excel_df.columns:
  133. raise UnicornException("缺失 [商品货号] 列")
  134. if "款号" not in excel_df.columns:
  135. raise UnicornException("缺失 [款号] 列")
  136. goods_art_dirs = excel_df.groupby(excel_df["款号"])
  137. # 抠图时用到的货号列表,与生成详情图有所区别
  138. goods_art_no_arrays = []
  139. # # 详情图生成需要对同款商品进行分组,保证详情图可以生成多个色
  140. goods_art_no_group_arrays = []
  141. for _, goods_row in excel_df.iterrows():
  142. goods_art_no = str(goods_row["商品货号"])
  143. goods_art_no_arrays.append(goods_art_no)
  144. goods_no = str(goods_row["款号"])
  145. a001_df = goods_art_dirs.get_group(goods_no)
  146. goods_art_groups = a001_df["商品货号"].tolist()
  147. if goods_art_groups in goods_art_no_group_arrays:
  148. continue
  149. goods_art_no_group_arrays.append(goods_art_groups)
  150. return goods_art_no_arrays,excel_df,goods_art_no_group_arrays
  151. def group_by_style_number(data):
  152. result = {}
  153. for goods_id, info in data.items():
  154. style_number = info["款号"]
  155. if style_number not in result:
  156. result[style_number] = []
  157. result[style_number].append(goods_id)
  158. return result
  159. @app.post("/handle_detail")
  160. async def handle_detail_background(
  161. request: Request, params: HandlerDetail, background_tasks: BackgroundTasks
  162. ):
  163. goods_art_no_arrays = params.goods_art_no
  164. is_check = params.is_check
  165. obj = None
  166. token = "Bearer " + params.token
  167. uuid = params.uuid
  168. run_main = RunMain(obj, token, uuid)
  169. online_stores = params.online_stores # 上传第三方的店铺名称数组
  170. is_detail = params.is_detail # 上传第三方的店铺名称数组
  171. is_product_scene = params.is_product_scene # 上传第三方的店铺名称数组
  172. is_upper_footer = params.is_upper_footer # 上传第三方的店铺名称数组
  173. # 该数组表示是否需要后面的移动文件夹操作,减少重复抠图,提升抠图时间和速度
  174. if online_stores:
  175. """如果上传第三方,则需要强制生成详情页"""
  176. params.is_detail = 1
  177. is_detail = 1
  178. try:
  179. excel_path = params.excel_path.strip()
  180. print("excel信息打印输出:",excel_path)
  181. if excel_path:
  182. goods_art_no_arrays,excel_df,_ = __createExcelGoodsArray(excel_path)
  183. await _process_excel_mode(
  184. goods_art_no_arrays, excel_df
  185. )
  186. else:
  187. await _process_non_excel_mode(
  188. params, goods_art_no_arrays
  189. )
  190. remote_data = run_main.data_mode_generate_detail.get_goods_art_no_info(goods_art_list=goods_art_no_arrays)
  191. if remote_data == {}:
  192. raise UnicornException("所有货号在商品档案资料中不存在,请检查货号是否正确")
  193. error_goods_art_no = []
  194. for goods_art_no_check in goods_art_no_arrays:
  195. check_item = remote_data.get(goods_art_no_check,None)
  196. if not check_item:
  197. error_goods_art_no.append(goods_art_no_check)
  198. if len(error_goods_art_no) > 0:
  199. raise UnicornException("以下货号在商品档案资料中不存在,请检查货号是否正确:{}".format(error_goods_art_no))
  200. except Exception as e:
  201. raise UnicornException(str(e))
  202. progress = [{"msg_type":"segment_progress","name":"抠图","status":"等待处理","current":0,"total":0,"error":0}]
  203. if is_product_scene == 1:
  204. progress.append({"msg_type":"scene_progress","name":"场景图","status":"等待处理","current":0,"total":0,"error":0})
  205. if is_upper_footer == 1:
  206. progress.append({"msg_type":"upper_footer_progress","name":"模特图","status":"等待处理","current":0,"total":0,"error":0})
  207. if is_detail==1 or (params.excel_path != "" and params.excel_path != None):
  208. progress.append({"msg_type":"detail_progress","name":"详情页","status":"等待处理","current":0,"total":0,"error":0})
  209. if len(online_stores) > 0:
  210. progress.append({"msg_type":"upload_goods_progress","name":"上传第三方商品","status":"等待处理","current":0,"total":0,"error":0})
  211. if is_check == 1:
  212. return {
  213. "code": 0,
  214. "msg": "检测通过",
  215. "data": {"progress":progress},
  216. }
  217. asyncio.create_task(process_handle_detail(request, params))
  218. return {"code": 0, "msg": "任务已提交后台处理", "data": {"status": "processing"}}
  219. # 重构后的简化版本
  220. async def process_handle_detail(request: Request, params: HandlerDetail):
  221. """处理商品详情图生成的主要函数"""
  222. try:
  223. # 初始化基础变量
  224. handler_result = []
  225. handler_result_folder = ""
  226. # 处理参数
  227. obj, token, uuid = None, "Bearer " + params.token, params.uuid
  228. aigc_clazz = AIGCDataRequest(token)
  229. run_main = RunMain(obj, token, uuid)
  230. onlineData = OnlineDataRequest(token)
  231. # 提取参数
  232. goods_art_no_arrays = params.goods_art_no
  233. online_stores = params.online_stores
  234. is_detail = params.is_detail
  235. is_product_scene = params.is_product_scene
  236. is_upper_footer = params.is_upper_footer
  237. upper_footer_params = params.upper_footer_params
  238. product_scene_prompt = params.product_scene_prompt
  239. if params.excel_path and params.excel_path.strip():
  240. # 处理Excel模式
  241. is_detail = 1#强制处理详情页
  242. goods_art_no_arrays,excel_df,_ = __createExcelGoodsArray(params.excel_path)
  243. move_folder_array = await _process_excel_mode(
  244. goods_art_no_arrays, excel_df
  245. )
  246. else:
  247. # 处理非Excel模式
  248. move_folder_array = await _process_non_excel_mode(
  249. params, goods_art_no_arrays
  250. )
  251. # 构建配置数据
  252. config_data = await _build_config_data(params, goods_art_no_arrays)
  253. # 执行抠图处理
  254. handler_result_folder, handler_result = await _process_cutout(
  255. run_main, config_data, goods_art_no_arrays, move_folder_array
  256. )
  257. # 处理场景图和模特图
  258. return_data_check_before_detail = run_main.check_before_detail(config_data)
  259. # 检查处理结果
  260. success_handler = return_data_check_before_detail.get("data", {}).get("config_data", {}).get("success_handler", [])
  261. failed_items = [item for item in success_handler if item.get('success') == False]
  262. if failed_items:
  263. await sendAsyncMessage(
  264. msg="处理结束",
  265. data={"output_folder": handler_result_folder, "list": failed_items},
  266. status="处理结束",
  267. msg_type="detail_result_progress",
  268. )
  269. # 如果全部失败,直接终止
  270. if failed_items and len(failed_items) == len(success_handler):
  271. check_progress = {
  272. "status": "处理失败",
  273. "current": 0,
  274. "total": len(success_handler),
  275. "error": len(failed_items)
  276. }
  277. await sendAsyncMessage(
  278. msg="检查失败",
  279. goods_arts=[],
  280. status="检查失败",
  281. msg_type="segment_progress",
  282. progress=check_progress
  283. )
  284. return True
  285. # 处理场景图
  286. if is_product_scene == 1:
  287. return_data_check_before_detail = await _process_scene_images(
  288. aigc_clazz, run_main, return_data_check_before_detail, product_scene_prompt
  289. )
  290. # 处理模特图
  291. if is_upper_footer == 1:
  292. return_data_check_before_detail = await _process_model_images(
  293. aigc_clazz, run_main, return_data_check_before_detail, upper_footer_params
  294. )
  295. current_day = time.strftime("%Y-%m-%d", time.localtime(time.time()))
  296. # 处理详情页
  297. if is_detail == 1:
  298. handler_result_folder, handler_result= await _process_detail_pages(
  299. run_main, return_data_check_before_detail, onlineData,
  300. online_stores, goods_art_no_arrays, handler_result_folder
  301. )
  302. # 如果需要上传到第三方平台
  303. if online_stores:
  304. await _upload_to_third_party(
  305. onlineData, return_data_check_before_detail,
  306. config_data, online_stores
  307. )
  308. await sendAsyncMessage(
  309. msg="处理结束",
  310. data={"output_folder": f"{handler_result_folder}/{current_day}", "list": handler_result},
  311. status="处理结束",
  312. msg_type="detail_result_progress",
  313. )
  314. else:
  315. await sendAsyncMessage(
  316. msg="处理结束",
  317. data={"output_folder": f"{handler_result_folder}/{current_day}", "list": handler_result},
  318. status="处理结束",
  319. msg_type="detail_result_progress",
  320. )
  321. except UnicornException as e:
  322. await _handle_exception(e.msg, handler_result_folder)
  323. except Exception as e:
  324. await _handle_exception(str(e), handler_result_folder)
  325. return True
  326. async def _process_non_excel_mode(params, goods_art_no_arrays):
  327. """处理非Excel模式的准备工作"""
  328. limit_path = "{}/{}".format(settings.OUTPUT_DIR,time.strftime("%Y-%m-%d", time.localtime(time.time())))
  329. check_path(limit_path)
  330. move_folder_array = check_move_goods_art_no_folder(settings.OUTPUT_DIR, goods_art_no_arrays, limit_path)
  331. for goods_art_no in goods_art_no_arrays:
  332. if not goods_art_no:
  333. raise UnicornException("货号不能为空")
  334. session = SqlQuery()
  335. pr = CRUD(PhotoRecord)
  336. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  337. if not images:
  338. raise UnicornException(
  339. f"商品货号【{goods_art_no}】未查询到拍摄记录,请检查货号是否正确"
  340. )
  341. detail_counts = len(params.template_image_order.split(","))
  342. image_counts = len(images)
  343. if image_counts < detail_counts:
  344. raise UnicornException(
  345. f"货号:[{goods_art_no}],实际照片数量:{image_counts}张,小于详情图要求数量:{detail_counts}张"
  346. )
  347. if move_folder_array.get(goods_art_no) is None:
  348. await _process_image_copy_and_move(goods_art_no, images,False)
  349. return move_folder_array
  350. async def _process_excel_mode(goods_art_no_arrays,excel_df):
  351. """处理非Excel模式的准备工作"""
  352. limit_path = "{}/{}".format(settings.OUTPUT_DIR,time.strftime("%Y-%m-%d", time.localtime(time.time())))
  353. check_path(limit_path)
  354. move_folder_array = check_move_goods_art_no_folder(settings.OUTPUT_DIR, goods_art_no_arrays, limit_path)
  355. for index, row in excel_df.iterrows():
  356. goods_art_no_image_dir = str(row["文件夹名称"])
  357. goods_art_no = str(row["商品货号"])
  358. print("货号数据", goods_art_no)
  359. if not goods_art_no:
  360. raise UnicornException("货号不能为空")
  361. session = SqlQuery()
  362. pr = CRUD(PhotoRecord)
  363. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  364. if not images:
  365. raise UnicornException(
  366. f"商品货号【{goods_art_no}】未查询到拍摄记录,请检查表格中的货号数据列"
  367. )
  368. # 货号目录不存在再去进行移动和创建操作
  369. if move_folder_array.get(goods_art_no) is None:
  370. await _process_image_copy_and_move(goods_art_no_image_dir, images,True)
  371. return move_folder_array
  372. async def _process_image_copy_and_move(goods_art_no, images,is_excel):
  373. """处理图片复制和移动"""
  374. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  375. check_path(image_dir)
  376. for idx, itemImg in enumerate(images):
  377. if not itemImg.image_path:
  378. raise UnicornException(
  379. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  380. )
  381. new_file_name = str(itemImg.goods_art_no) + "_" + str(idx) + ".jpg"
  382. if not os.path.exists(image_dir + "/" + os.path.basename(new_file_name)):
  383. shutil.copy(itemImg.image_path, image_dir + new_file_name)
  384. dealImage = DealImage(image_dir)
  385. resFlag, path = dealImage.dealMoveImage(
  386. image_dir=image_dir, callback_func=None, goods_art_no=goods_art_no
  387. )
  388. if not resFlag:
  389. raise UnicornException(path)
  390. if is_excel:
  391. shutil.rmtree(image_dir)
  392. async def _build_config_data(params, goods_art_no_arrays):
  393. """构建配置数据"""
  394. temp_class = {}
  395. temp_name_list = []
  396. for tempItem in params.temp_list:
  397. temp_class[tempItem.template_id] = tempItem.template_local_classes
  398. temp_name_list.append(tempItem.template_id)
  399. cutOutMode = (
  400. "1"
  401. if settings.getSysConfigs("other_configs", "cutout_mode", "普通抠图") == "普通抠图"
  402. else "2"
  403. )
  404. limit_path = "{}/{}".format(settings.OUTPUT_DIR,time.strftime("%Y-%m-%d", time.localtime(time.time())))
  405. config_data = {
  406. "image_dir": limit_path,
  407. "image_order": (
  408. "俯视,侧视,后跟,鞋底,内里,组合,组合2,组合3,组合4,组合5"
  409. if not params.template_image_order
  410. else params.template_image_order
  411. ),
  412. "goods_art_no": "",
  413. "goods_art_nos": goods_art_no_arrays,
  414. "is_check_number": False,
  415. "resize_image_view": "后跟",
  416. "cutout_mode": cutOutMode,
  417. "logo_path": params.logo_path,
  418. "special_goods_art_no_folder_line": "",
  419. "is_use_excel": (False if params.excel_path == "" else True), # 是否使用excel
  420. "excel_path": params.excel_path, # excel路径
  421. "is_check_color_is_all": False,
  422. "cutout_is_pass": True,
  423. "assigned_page_dict": {},
  424. "detail_is_pass": False,
  425. "upload_is_pass": False,
  426. "upload_is_enable": False,
  427. "is_filter": False,
  428. "temp_class": temp_class,
  429. "temp_name": params.temp_name,
  430. "temp_name_list": temp_name_list,
  431. "target_error_folder": f"{limit_path}/软件-生成详情错误",
  432. "success_handler": [],
  433. }
  434. # 动态导入类
  435. try:
  436. temp_class_dict = {}
  437. for key, class_path in config_data["temp_class"].items():
  438. module_path, class_name = class_path.rsplit(".", 1)
  439. module = importlib.import_module(module_path)
  440. cls = getattr(module, class_name)
  441. temp_class_dict[key] = cls
  442. except:
  443. raise UnicornException("详情页模板不存在或未下载完成,请重启软件后重试")
  444. config_data["temp_class"] = temp_class_dict
  445. return config_data
  446. async def _process_cutout(run_main, config_data, goods_art_no_arrays, move_folder_array):
  447. """处理抠图"""
  448. handler_result = []
  449. handler_result_folder = ""
  450. return_data = run_main.check_before_cutout(config_data)
  451. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  452. if cutout_res:
  453. # sys_path = format(os.getcwd()).replace("\\", "/")
  454. handler_result_folder = f"{config_data['image_dir']}"
  455. for goods_art_item in goods_art_no_arrays:
  456. handler_result.append({
  457. "goods_art_no": goods_art_item,
  458. "success": True,
  459. "info": "处理成功",
  460. })
  461. if len(move_folder_array.keys()) == len(goods_art_no_arrays):
  462. handler_result_folder = handler_result_folder.replace("\\", "/")
  463. success_items = [item for item in handler_result if item.get('success') == True]
  464. cutout_folder = handler_result_folder+"/"+success_items[0].get("goods_art_no")+"/800x800" if len(success_items) > 0 else ""
  465. progress = {
  466. "status": "处理完成",
  467. "current": len(goods_art_no_arrays),
  468. "total": len(goods_art_no_arrays),
  469. "error": 0,
  470. "folder":cutout_folder,
  471. }
  472. await sendAsyncMessage(
  473. msg="抠图完成",
  474. data={"output_folder": handler_result_folder, "list": handler_result},
  475. status="抠图完成",
  476. msg_type="segment_progress",
  477. progress=progress
  478. )
  479. return handler_result_folder, handler_result
  480. async def _process_scene_images(aigc_clazz, run_main, return_data_check_before_detail, product_scene_prompt):
  481. """处理场景图生成"""
  482. # 参数验证
  483. if not product_scene_prompt:
  484. raise UnicornException("请填写场景描述")
  485. goods_dict = parserGoodsDict2Aigc(return_data_check_before_detail)
  486. new_goods_dict = {}
  487. product_scene_total_progress = len(goods_dict.keys())
  488. product_scene_finish_progress = 0
  489. product_scene_error_progress = 0
  490. product_scene_progress = {
  491. "status": "正在处理",
  492. "current": product_scene_finish_progress,
  493. "total": product_scene_total_progress,
  494. "error": product_scene_error_progress
  495. }
  496. await sendAsyncMessage(
  497. msg="开始处理场景图",
  498. goods_arts=[goods_art_no for goods_art_no in goods_dict.keys()],
  499. status="开始处理",
  500. msg_type="scene_progress",
  501. progress=product_scene_progress
  502. )
  503. for goods_art_no in goods_dict.keys():
  504. try:
  505. product_scene_finish_progress += 1
  506. folder = ""
  507. product_scene_progress = {
  508. "status": "正在处理",
  509. "goods_art_no": goods_art_no,
  510. "current": product_scene_finish_progress,
  511. "total": product_scene_total_progress,
  512. "error": product_scene_error_progress,
  513. "goods_art_no": goods_art_no,
  514. "folder": folder,
  515. }
  516. await sendAsyncMessage(
  517. msg="正在处理场景图",
  518. goods_arts=[goods_art_no],
  519. status="正在处理场景图",
  520. msg_type="scene_progress",
  521. progress=product_scene_progress
  522. )
  523. goods_art_dict_info = goods_dict[goods_art_no]
  524. first_goods_art_no_info = goods_art_dict_info.get("货号资料", [])[0]
  525. first_pics = first_goods_art_no_info.get("pics")
  526. ceshi_image_path = first_pics.get("侧视-抠图")
  527. save_root_path = ceshi_image_path.split("阴影图处理")[0]
  528. save_image_path = f"{save_root_path}场景图.jpg"
  529. if os.path.isfile(save_image_path):
  530. goods_art_dict_info["场景图"] = save_image_path
  531. new_goods_dict[goods_art_no] = goods_art_dict_info
  532. product_scene_progress["folder"] = save_image_path.replace("场景图.jpg", "")
  533. await sendAsyncMessage(
  534. msg="场景图处理完成",
  535. goods_arts=[goods_art_no],
  536. status="场景图处理完成",
  537. msg_type="scene_progress",
  538. progress=product_scene_progress
  539. )
  540. continue
  541. aigc_clazz.center_paste_image(ceshi_image_path, save_image_path)
  542. image_path = aigc_clazz.generateProductScene(
  543. save_image_path, product_scene_prompt, save_image_path
  544. )
  545. goods_art_dict_info["场景图"] = image_path.replace("场景图.jpg", "")
  546. new_goods_dict[goods_art_no] = goods_art_dict_info
  547. product_scene_progress["folder"] = save_image_path
  548. await sendAsyncMessage(
  549. msg="场景图处理完成",
  550. goods_arts=[goods_art_no],
  551. status="场景图处理完成",
  552. msg_type="scene_progress",
  553. progress=product_scene_progress
  554. )
  555. except Exception as e:
  556. print("场景图处理异常",e)
  557. os.remove(save_image_path)
  558. product_scene_finish_progress -= 1
  559. product_scene_error_progress += 1
  560. product_scene_progress = {
  561. "status": "处理失败",
  562. "goods_art_no": goods_art_no,
  563. "current": product_scene_finish_progress,
  564. "total": product_scene_total_progress,
  565. "error": product_scene_error_progress
  566. }
  567. await sendAsyncMessage(
  568. msg="场景图处理失败",
  569. goods_arts=[goods_art_no],
  570. status="场景图处理失败",
  571. msg_type="scene_progress",
  572. progress=product_scene_progress
  573. )
  574. status_text = "处理完成" if product_scene_finish_progress > 0 else "处理失败"
  575. product_scene_progress = {
  576. "status": status_text,
  577. "current": product_scene_finish_progress,
  578. "total": product_scene_total_progress,
  579. "error": product_scene_error_progress
  580. }
  581. await sendAsyncMessage(
  582. msg="场景图处理结束",
  583. goods_arts=[],
  584. status="场景图处理结束",
  585. msg_type="scene_progress",
  586. progress=product_scene_progress
  587. )
  588. if new_goods_dict:
  589. return_data_check_before_detail["data"]["goods_no_dict"] = new_goods_dict
  590. return return_data_check_before_detail
  591. async def _process_model_images(aigc_clazz, run_main, return_data_check_before_detail, upper_footer_params):
  592. """处理模特图生成"""
  593. # 参数验证
  594. if not upper_footer_params:
  595. raise UnicornException("请选择模特")
  596. man_id = upper_footer_params.get("man_id")
  597. women_id = upper_footer_params.get("women_id")
  598. if not man_id:
  599. raise UnicornException("请选择男模特")
  600. if not women_id:
  601. raise UnicornException("请选择女模特")
  602. goods_dict = parserGoodsDict2Aigc(return_data_check_before_detail)
  603. new_goods_dict = {}
  604. upper_footer_total_progress = len(goods_dict.keys())
  605. upper_footer_finish_progress = 0
  606. upper_footer_error_progress = 0
  607. upper_footer_progress = {
  608. "status": "正在处理",
  609. "current": upper_footer_finish_progress,
  610. "total": upper_footer_total_progress,
  611. "error": upper_footer_error_progress
  612. }
  613. await sendAsyncMessage(
  614. msg="开始处理模特图",
  615. goods_arts=list(goods_dict.keys()),
  616. status="开始处理模特图",
  617. msg_type="upper_footer_progress",
  618. progress=upper_footer_progress
  619. )
  620. for goods_art_no in goods_dict.keys():
  621. upper_footer_finish_progress += 1
  622. folder = ""
  623. try:
  624. upper_footer_progress = {
  625. "status": "正在处理",
  626. "goods_art_no": goods_art_no,
  627. "current": upper_footer_finish_progress,
  628. "total": upper_footer_total_progress,
  629. "error": upper_footer_error_progress,
  630. "goods_art_no": goods_art_no,
  631. "folder": folder,
  632. }
  633. await sendAsyncMessage(
  634. msg="正在处理模特图",
  635. goods_arts=[goods_art_no],
  636. status="正在处理模特图",
  637. msg_type="upper_footer_progress",
  638. progress=upper_footer_progress
  639. )
  640. goods_art_dict_info = goods_dict[goods_art_no]
  641. first_goods_art_no_info = goods_art_dict_info.get("货号资料", [])[0]
  642. first_pics = first_goods_art_no_info.get("pics")
  643. gender = goods_art_dict_info.get("性别","女")
  644. model_id = man_id if "男" in gender else women_id
  645. ceshi_image_path = first_pics.get("侧视-抠图")
  646. save_root_path = ceshi_image_path.split("阴影图处理")[0]
  647. save_image_path = f"{save_root_path}模特图.jpg"
  648. if os.path.isfile(save_image_path):
  649. goods_art_dict_info["模特图"] = save_image_path
  650. new_goods_dict[goods_art_no] = goods_art_dict_info
  651. upper_footer_progress["folder"] = save_image_path.replace("模特图.jpg", "")
  652. await sendAsyncMessage(
  653. msg="模特图处理成功",
  654. goods_arts=[goods_art_no],
  655. status="模特图处理成功",
  656. msg_type="upper_footer_progress",
  657. progress=upper_footer_progress
  658. )
  659. continue
  660. shutil.copy(ceshi_image_path, save_image_path)
  661. # 添加超时控制
  662. with concurrent.futures.ThreadPoolExecutor() as executor:
  663. future = executor.submit(
  664. aigc_clazz.generateUpperShoes,
  665. save_image_path,
  666. model_id,
  667. save_image_path
  668. )
  669. # 设置70秒超时
  670. image_path = future.result(timeout=70)
  671. goods_art_dict_info["模特图"] = image_path.replace("模特图.jpg", "")
  672. new_goods_dict[goods_art_no] = goods_art_dict_info
  673. # upper_footer_progress = {
  674. # "status": "正在处理",
  675. # "goods_art_no": goods_art_no,
  676. # "current": upper_footer_finish_progress,
  677. # "total": upper_footer_total_progress,
  678. # "error": upper_footer_error_progress
  679. # }
  680. upper_footer_progress["folder"] = save_image_path
  681. await sendAsyncMessage(
  682. msg="模特图处理成功",
  683. goods_arts=[goods_art_no],
  684. status="模特图处理成功",
  685. msg_type="upper_footer_progress",
  686. progress=upper_footer_progress
  687. )
  688. except (concurrent.futures.TimeoutError, Exception) as e:
  689. os.remove(save_image_path)
  690. upper_footer_finish_progress-=1
  691. upper_footer_error_progress += 1
  692. upper_footer_progress = {
  693. "status": "处理失败",
  694. "goods_art_no": goods_art_no,
  695. "current": upper_footer_finish_progress,
  696. "total": upper_footer_total_progress,
  697. "error": upper_footer_error_progress
  698. }
  699. await sendAsyncMessage(
  700. msg="模特图处理失败",
  701. goods_arts=[goods_art_no],
  702. status="模特图处理失败",
  703. msg_type="upper_footer_progress",
  704. progress=upper_footer_progress
  705. )
  706. status_text = "处理完成" if upper_footer_finish_progress > 0 else "处理失败"
  707. upper_footer_progress = {
  708. "status": status_text,
  709. "current": upper_footer_finish_progress,
  710. "total": upper_footer_total_progress,
  711. "error": upper_footer_error_progress
  712. }
  713. await sendAsyncMessage(
  714. msg="模特图处理结束",
  715. goods_arts=[],
  716. status="模特图处理结束",
  717. msg_type="upper_footer_progress",
  718. progress=upper_footer_progress
  719. )
  720. if new_goods_dict:
  721. return_data_check_before_detail["data"]["goods_no_dict"] = new_goods_dict
  722. return return_data_check_before_detail
  723. async def _process_detail_pages(run_main, return_data_check_before_detail, onlineData,
  724. online_stores, goods_art_no_arrays, handler_result_folder):
  725. """处理详情页生成和上传"""
  726. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  727. return_data_check_before_detail
  728. )
  729. print("<======>check_for_detail_first_res<======>",check_for_detail_first_res)
  730. if isinstance(check_for_detail_first_res, partial):
  731. result = check_for_detail_first_res()
  732. try:
  733. config_data = result["config_data"]
  734. except:
  735. config_data = result
  736. else:
  737. config_data = check_for_detail_first_res
  738. out_put_dir = config_data.get("out_put_dir")
  739. success_handler = config_data.get("success_handler")
  740. if not out_put_dir:
  741. handler_result_folder = ""
  742. # 处理失败情况
  743. else:
  744. out_put_dir_path = "{}".format(out_put_dir).replace("\\", "/")
  745. handler_result_folder = os.path.dirname(out_put_dir_path)
  746. if handler_result_folder:
  747. handler_result_folder = os.path.dirname(handler_result_folder)
  748. return handler_result_folder,success_handler
  749. async def _upload_to_third_party(onlineData, return_data_check_before_detail,
  750. config_data, online_stores):
  751. """上传到第三方平台"""
  752. result_goods_no_dict = return_data_check_before_detail["data"]["goods_no_dict"]
  753. for goods_idx, goods_no_dict in enumerate(result_goods_no_dict.keys()):
  754. all_detail_path_list = config_data["all_detail_path_list"]
  755. for detail_path in all_detail_path_list:
  756. if goods_no_dict in detail_path:
  757. detail_path_replace = detail_path.replace("\\", "/")
  758. result_goods_no_dict[goods_no_dict]["detail_path"] = f"{detail_path_replace}/详情页.jpg"
  759. upload_total_progress = len(result_goods_no_dict.keys())
  760. upload_finish_progress = 0
  761. upload_error_progress = 0
  762. upload_progress = {
  763. "status": "开始处理",
  764. "current": upload_finish_progress,
  765. "total": upload_total_progress,
  766. "error": upload_error_progress
  767. }
  768. await sendAsyncMessage(
  769. msg="开始上传商品数据",
  770. goods_arts=[],
  771. status="开始上传商品数据",
  772. msg_type="upload_goods_progress",
  773. progress=upload_progress
  774. )
  775. try:
  776. onlineData.uploadGoods2ThirdParty(
  777. result_goods_no_dict, online_stores=online_stores
  778. )
  779. except Exception as e:
  780. upload_error_progress = upload_total_progress
  781. print(f"上传任务出现错误:{e}")
  782. upload_finish_progress = upload_total_progress
  783. if upload_error_progress > 0:
  784. upload_finish_progress = 0
  785. upload_progress = {
  786. "status": "处理完成",
  787. "current": upload_finish_progress,
  788. "total": upload_total_progress,
  789. "error": upload_error_progress
  790. }
  791. await sendAsyncMessage(
  792. msg="商品上传第三方成功",
  793. goods_arts=[],
  794. status="商品上传第三方成功",
  795. msg_type="upload_goods_progress",
  796. progress=upload_progress
  797. )
  798. async def _handle_exception(error_msg, handler_result_folder):
  799. """处理异常情况"""
  800. handler_result = [{"goods_art_no": "", "success": False, "info": error_msg}]
  801. await sendAsyncMessage(
  802. msg="处理结束",
  803. data={"output_folder": handler_result_folder, "list": handler_result},
  804. status="处理结束",
  805. msg_type="detail_result_progress",
  806. )
  807. @app.get("/get_device_tabs", description="获取可执行程序命令列表")
  808. def get_device_tabs(type: int):
  809. session = SqlQuery()
  810. statement = (
  811. select(DeviceConfigTabs)
  812. .where(DeviceConfigTabs.mode_type == type)
  813. .order_by(asc("id"))
  814. )
  815. result = session.exec(statement).all()
  816. session.close()
  817. sys = CRUD(SysConfigs)
  818. action_configs = sys.read(session, conditions={"key": "action_configs"})
  819. session.close()
  820. return {
  821. "code": 0,
  822. "msg": "",
  823. "data": {"tabs": result, "select_configs": json.loads(action_configs.value)},
  824. }
  825. @app.post("/update_tab_name", description="更改tab名称")
  826. def update_tab_name(params: DeviceConfigTabsReq):
  827. if params.mode_name == "":
  828. return {"code": 1, "msg": "名称不能为空", "data": {}}
  829. session = SqlQuery()
  830. tabModel = CRUD(DeviceConfigTabs)
  831. kwargs = {"mode_name": params.mode_name}
  832. tabModel.updateConditions(session, conditions={"id": params.id}, **kwargs)
  833. session.close()
  834. return {
  835. "code": 0,
  836. "msg": "",
  837. "data": None,
  838. }
  839. @app.post("/get_device_configs", description="获取可执行程序命令列表")
  840. def get_device_configs(params: ModelGetDeviceConfig):
  841. tab_id = params.tab_id
  842. session = SqlQuery()
  843. configModel = CRUD(DeviceConfig)
  844. configList = configModel.read_all(
  845. session,
  846. conditions={"tab_id": tab_id},
  847. order_by="action_index",
  848. ascending=True,
  849. )
  850. return {
  851. "code": 0,
  852. "msg": "",
  853. "data": {"list": configList},
  854. }
  855. @app.post("/device_config_detail", description="获取可执行程序详情")
  856. def device_config_detail(params: ModelGetDeviceConfigDetail):
  857. action_id = params.id
  858. session = SqlQuery()
  859. configModel = CRUD(DeviceConfig)
  860. model = configModel.read(session, conditions={"id": action_id})
  861. if model == None:
  862. return {"code": 1, "msg": "数据不存在", "data": None}
  863. return {"code": 0, "msg": "", "data": model}
  864. @app.post("/device_config_detail_query", description="通过条件获取可执行程序详情")
  865. def device_config_detail_query():
  866. # tab_id = params.tab_id
  867. # action_name = params.action_name
  868. session = SqlQuery()
  869. sys = CRUD(SysConfigs)
  870. action_configs = sys.read(session, conditions={"key": "action_configs"})
  871. action_configs_value = json.loads(action_configs.value)
  872. left_config = action_configs_value.get("left")
  873. configModel = CRUD(DeviceConfig)
  874. model = configModel.read(
  875. session, conditions={"tab_id": left_config, "action_name": "侧视"}
  876. )
  877. if model == None:
  878. model = configModel.read(session, conditions={"tab_id": left_config})
  879. return {"code": 0, "msg": "", "data": model}
  880. @app.post("/remove_config", description="删除一条可执行命令")
  881. def get_device_configs(params: ModelGetDeviceConfigDetail):
  882. action_id = params.id
  883. session = SqlQuery()
  884. configModel = CRUD(DeviceConfig)
  885. model = configModel.read(session, conditions={"id": action_id})
  886. if model == None:
  887. return {"code": 1, "msg": "数据不存在", "data": None}
  888. if model.is_system == True:
  889. return {"code": 1, "msg": "系统配置不允许删除", "data": None}
  890. configArray = configModel.read_all(session, conditions={"tab_id": model.tab_id})
  891. if len(configArray) == 1:
  892. return {"code": 1, "msg": "请至少保留一个配置", "data": None}
  893. configModel.delete(session, obj_id=action_id)
  894. return {"code": 0, "msg": "删除成功", "data": None}
  895. @app.post("/save_device_config", description="创建或修改一条可执行命令")
  896. def save_device_config(params: SaveDeviceConfig):
  897. action_id = params.id
  898. session = SqlQuery()
  899. deviceConfig = CRUD(DeviceConfig)
  900. if action_id == None or action_id == 0:
  901. # 走新增逻辑
  902. params.id = None
  903. save_device_config = deviceConfig.create(session, obj_in=params)
  904. else:
  905. model = deviceConfig.read(session, conditions={"id": action_id})
  906. if model == None:
  907. return {"code": 1, "msg": "数据不存在", "data": None}
  908. # 走编辑逻辑
  909. kwargs = params.__dict__
  910. save_device_config = deviceConfig.update(session, obj_id=action_id, **kwargs)
  911. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  912. @app.post("/reset_config", description="创建或修改一条可执行命令")
  913. def reset_config(params: ModelGetDeviceConfig):
  914. tab_id = params.tab_id
  915. if tab_id == None or tab_id == "":
  916. return {"code": 1, "msg": "参数错误", "data": None}
  917. session = SqlQuery()
  918. deviceConfig = CRUD(DeviceConfig)
  919. first_config = deviceConfig.read(session, conditions={"tab_id": tab_id})
  920. res = deviceConfig.deleteConditions(session, conditions={"tab_id": tab_id})
  921. if res is False:
  922. return {"code": 1, "msg": "操作失败", "data": None}
  923. actions = json.load(open("action.json", encoding="utf-8"))
  924. for data in actions:
  925. data["tab_id"] = tab_id
  926. data["is_system"] = first_config.is_system
  927. device_config = DeviceConfig(**data)
  928. session.add(device_config)
  929. session.commit()
  930. # session.close()
  931. return {"code": 0, "msg": "操作成功", "data": None}
  932. @app.get("/get_photo_records", description="获取拍照记录")
  933. def get_photo_records(page: int = 1, size: int = 5):
  934. session = SqlQuery()
  935. # photos = CRUD(PhotoRecord)
  936. print("准备查询拍摄记录", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  937. statement = (
  938. select(PhotoRecord)
  939. .offset((page - 1) * size)
  940. .limit(size)
  941. .order_by(desc("id"))
  942. .group_by("goods_art_no")
  943. )
  944. list = []
  945. result = session.exec(statement).all()
  946. print("group 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  947. join_conditions = [
  948. {
  949. "model": DeviceConfig,
  950. "on": PhotoRecord.action_id == DeviceConfig.id,
  951. "is_outer": False, # 可选,默认False,设为True则为LEFT JOIN
  952. }
  953. ]
  954. for item in result:
  955. query = (
  956. select(PhotoRecord, DeviceConfig.action_name)
  957. .where(PhotoRecord.goods_art_no == item.goods_art_no)
  958. .join(DeviceConfig, PhotoRecord.action_id == DeviceConfig.id)
  959. )
  960. list_item = session.exec(query).mappings().all()
  961. list.append(
  962. {
  963. "goods_art_no": item.goods_art_no,
  964. "action_time": item.create_time,
  965. "items": list_item,
  966. }
  967. )
  968. # session.close()
  969. print("循环查询 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  970. return {
  971. "code": 0,
  972. "msg": "",
  973. "data": {"list": list, "page": page, "size": size},
  974. }
  975. @app.get("/get_last_photo_record", description="获取最后一条拍照记录")
  976. def get_last_photo_record():
  977. session = SqlQuery()
  978. statement = (
  979. select(PhotoRecord)
  980. .where(PhotoRecord.image_path != None)
  981. .order_by(desc("photo_create_time"))
  982. )
  983. result = session.exec(statement).first()
  984. # session.close()
  985. return {
  986. "code": 0,
  987. "msg": "",
  988. "data": result,
  989. }
  990. @app.get("/get_photo_record_detail", description="通过货号获取拍照记录详情")
  991. def get_photo_record_detail(goods_art_no: str = None):
  992. if goods_art_no == None:
  993. return {"code": 1, "msg": "参数错误", "data": None}
  994. session = SqlQuery()
  995. photos = CRUD(PhotoRecord)
  996. items = photos.read_all(session, conditions={"goods_art_no": goods_art_no})
  997. # session.close()
  998. return {
  999. "code": 0,
  1000. "msg": "",
  1001. "data": {"list": items},
  1002. }
  1003. @app.post("/delect_goods_arts", description="通过货号删除记录")
  1004. def delect_goods_arts(params: PhotoRecordDelete):
  1005. session = SqlQuery()
  1006. photos = CRUD(PhotoRecord)
  1007. for item in params.goods_art_nos:
  1008. photos.deleteConditions(session, conditions={"goods_art_no": item})
  1009. # session.close()
  1010. return {
  1011. "code": 0,
  1012. "msg": "操作成功",
  1013. "data": None,
  1014. }
  1015. def query_config_by_key(key_name):
  1016. """
  1017. 在sys_configs.json格式的数据中查询指定的key,如果匹配则返回整个对象
  1018. Args:
  1019. key_name (str): 要查询的key名称
  1020. Returns:
  1021. dict or None: 匹配的对象或None(如果没有找到)
  1022. """
  1023. try:
  1024. # 获取所有配置数据
  1025. configs = json.load(open("sys_configs.json", encoding="utf-8"))
  1026. # 遍历配置数据查找匹配的key
  1027. for config in configs:
  1028. if config.get("key") == key_name:
  1029. return config
  1030. return None
  1031. except Exception as e:
  1032. print(f"查询配置时出错: {e}")
  1033. return None
  1034. @app.get("/get_sys_config", description="查询系统配置")
  1035. def get_sys_config(key: str = None):
  1036. if key == None:
  1037. return {"code": 1, "msg": "参数错误", "data": None}
  1038. session = SqlQuery()
  1039. photos = CRUD(SysConfigs)
  1040. item = photos.read(session, conditions={"key": key})
  1041. search_data = None if item == None else json.loads(item.value)
  1042. if search_data == None:
  1043. sys_config_json = query_config_by_key(key)
  1044. if sys_config_json != None:
  1045. config = SysConfigs(**sys_config_json)
  1046. session.add(config)
  1047. session.commit() # 合并事务提交
  1048. search_data = json.loads(sys_config_json.get("value"))
  1049. return {
  1050. "code": 0,
  1051. "msg": "",
  1052. "data": search_data,
  1053. }
  1054. @app.post("/update_left_right_config", description="更新左右脚配置")
  1055. def update_left_right_config(params: LeftRightParams):
  1056. session = SqlQuery()
  1057. sysConfig = CRUD(SysConfigs)
  1058. model = sysConfig.read(session, conditions={"key": "action_configs"})
  1059. if model == None:
  1060. return {"code": 1, "msg": "配置不存在", "data": None}
  1061. config_value = json.loads(model.value)
  1062. config_value[params.type] = params.id
  1063. update_value = json.dumps(config_value)
  1064. # 走编辑逻辑
  1065. kwargs = {"key": "action_configs", "value": update_value}
  1066. save_device_config = sysConfig.updateConditions(
  1067. session, conditions={"key": "action_configs"}, **kwargs
  1068. )
  1069. return {"code": 0, "msg": "操作成功", "data": None}
  1070. @app.post("/update_record", description="更新拍照记录")
  1071. def update_record(params: RecordUpdate):
  1072. session = SqlQuery()
  1073. photoRecord = CRUD(PhotoRecord)
  1074. model = photoRecord.read(session, conditions={"id": params.id})
  1075. if model == None:
  1076. return {"code": 1, "msg": "记录不存在", "data": None}
  1077. kwargs = params.__dict__
  1078. save_device_config = photoRecord.update(session, obj_id=params.id, **kwargs)
  1079. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  1080. @app.post("/update_sys_configs", description="创建或修改系统配置")
  1081. def save_sys_configs(params: SysConfigParams):
  1082. session = SqlQuery()
  1083. sysConfig = CRUD(SysConfigs)
  1084. model = sysConfig.read(session, conditions={"key": params.key})
  1085. if model == None:
  1086. return {"code": 1, "msg": "配置不存在", "data": None}
  1087. # 走编辑逻辑
  1088. kwargs = params.__dict__
  1089. save_device_config = sysConfig.updateConditions(
  1090. session, conditions={"key": params.key}, **kwargs
  1091. )
  1092. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  1093. @app.post("/create_main_image", description="创建主图测试")
  1094. def create_main_image(params: MaineImageTest):
  1095. file_path = params.file_path
  1096. onePic = OnePicTest(pic_path=file_path)
  1097. main_out_path = onePic.HandlerMainImage()
  1098. return {"code": 0, "msg": "操作成功", "data": {"main_out_path": main_out_path}}
  1099. def insertEmptyLogoList(session):
  1100. """插入空logo列表"""
  1101. data = {"key": "logo_configs", "value": "[]"}
  1102. config = SysConfigs(**data)
  1103. session.add(config)
  1104. session.commit()
  1105. session.close()
  1106. item = SysConfigs()
  1107. item.key = "logo_configs"
  1108. item.value = "[]"
  1109. return item
  1110. @app.get("/logo_list", description="logo列表")
  1111. def logo_list():
  1112. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  1113. check_path(logo_dir)
  1114. logo_files = os.listdir(logo_dir)
  1115. logo_list = []
  1116. for logoItem in logo_files:
  1117. logo_list.append(f"{logo_dir}{logoItem}")
  1118. return {"code": 0, "msg": "操作成功", "data": logo_list}
  1119. @app.post("/add_logo", description="添加logo")
  1120. def add_logo(params: LogoParams):
  1121. logo_path = params.logo_path.replace("\\", "/")
  1122. session = SqlQuery()
  1123. sysConfig = CRUD(SysConfigs)
  1124. item = sysConfig.read(session, conditions={"key": "logo_configs"})
  1125. if item == None:
  1126. item = insertEmptyLogoList(session)
  1127. if os.path.isfile(logo_path) == False:
  1128. return {"code": 1, "msg": "logo文件不存在", "data": None}
  1129. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  1130. check_path(logo_dir)
  1131. fpath, fname = os.path.split(logo_path)
  1132. logo_path_info = logo_dir + fname
  1133. shutil.copy(logo_path, logo_path_info) # 复制文件
  1134. logo_files = os.listdir(logo_dir)
  1135. logo_list = []
  1136. for logoItem in logo_files:
  1137. logo_list.append(f"{logo_dir}{logoItem}")
  1138. return {
  1139. "code": 0,
  1140. "msg": "",
  1141. "data": {"logo": logo_path_info},
  1142. }
  1143. @app.post("/delete_logo", description="删除logo")
  1144. def delete_logo(params: LogoParamsupdate):
  1145. logo_path = params.path
  1146. if os.path.isfile(logo_path) == False:
  1147. return {"code": 1, "msg": "logo文件不存在", "data": None}
  1148. os.remove(logo_path)
  1149. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  1150. check_path(logo_dir)
  1151. logo_files = os.listdir(logo_dir)
  1152. logo_list = []
  1153. for logoItem in logo_files:
  1154. logo_list.append(f"{logo_dir}{logoItem}")
  1155. return {"code": 0, "msg": "操作成功", "data": logo_list}
  1156. @app.post("/close_other_window", description="关闭窗口")
  1157. def close_other_window():
  1158. hwnd_list = []
  1159. def callback(hwnd, _):
  1160. title = GetWindowText(hwnd)
  1161. if title == "digiCamControl by Duka Istvan":
  1162. hwnd_list.append(hwnd)
  1163. EnumWindows(callback, None)
  1164. if hwnd_list:
  1165. hwnd = hwnd_list[0]
  1166. win32gui.PostMessage(hwnd, win32con.WM_CLOSE, 0, 0)
  1167. return {"code": 0, "msg": "关闭成功", "data": {"status": True}}
  1168. return {"code": 0, "msg": "关闭失败", "data": {"status": False}}
  1169. def syncUserJsonConfigs(token):
  1170. hlm_token = token
  1171. headers = {
  1172. "Authorization": f"Bearer {hlm_token}",
  1173. "content-type": "application/json",
  1174. }
  1175. # 追加配置参数 machine_type 拍照机设备类型;0鞋;1服装
  1176. url = settings.DOMAIN + f"/api/ai_image/camera_machine/get_all_user_configs?machine_type={MACHINE_TYPE}"
  1177. result = requests.get(url=url, headers=headers)
  1178. sys_configs = result.json().get("data", {}).get("configs")
  1179. session = SqlQuery()
  1180. sysConfigs = CRUD(SysConfigs)
  1181. if sys_configs:
  1182. sysConfigs.deleteConditions(session, {})
  1183. configList = []
  1184. for config_keys in sys_configs.keys():
  1185. sys_configs[config_keys]
  1186. configList.append(
  1187. {
  1188. "key": config_keys,
  1189. "value": json.dumps(sys_configs[config_keys], ensure_ascii=False),
  1190. }
  1191. )
  1192. batch_insert_sys_configs(session, configList)
  1193. @app.post("/sync_sys_configs", description="同步线上配置到本地")
  1194. def sync_sys_configs(params: SyncLocalConfigs):
  1195. hlm_token = params.token
  1196. headers = {
  1197. "Authorization": f"Bearer {hlm_token}",
  1198. "content-type": "application/json",
  1199. }
  1200. # 追加配置参数 machine_type 拍照机设备类型;0鞋;1服装
  1201. url = settings.DOMAIN + f"/api/ai_image/camera_machine/get_all_user_configs?machine_type={MACHINE_TYPE}"
  1202. result = requests.get(url=url, headers=headers)
  1203. sys_configs = result.json().get("data", {}).get("configs")
  1204. session = SqlQuery()
  1205. sysConfigs = CRUD(SysConfigs)
  1206. if sys_configs:
  1207. sysConfigs.deleteConditions(session, {})
  1208. configList = []
  1209. for config_keys in sys_configs.keys():
  1210. sys_configs[config_keys]
  1211. configList.append(
  1212. {
  1213. "key": config_keys,
  1214. "value": json.dumps(sys_configs[config_keys], ensure_ascii=False),
  1215. }
  1216. )
  1217. batch_insert_sys_configs(session, configList)
  1218. else:
  1219. all_configs = sysConfigs.read_all(session)
  1220. localConfigData = {}
  1221. for local_config in all_configs:
  1222. localConfigData[local_config.key] = json.loads(local_config.value)
  1223. data_json = json.dumps({"configs": localConfigData,"machine_type":MACHINE_TYPE}, ensure_ascii=False)
  1224. # 同步本地到线上
  1225. url = settings.DOMAIN + "/api/ai_image/camera_machine/update_all_user_configs"
  1226. requests.post(url=url, headers=headers, data=data_json)
  1227. return {"code": 0, "msg": "操作成功", "data": None}
  1228. @app.post("/sync_actions", description="同步左右脚配置到本地")
  1229. def sync_action_configs(params: SyncLocalConfigs):
  1230. hlm_token = params.token
  1231. headers = {
  1232. "Authorization": f"Bearer {hlm_token}",
  1233. "content-type": "application/json",
  1234. }
  1235. url = settings.DOMAIN + f"/api/ai_image/camera_machine/get_all_user_tabs?machine_type={MACHINE_TYPE}"
  1236. result = requests.get(url=url, headers=headers)
  1237. session = SqlQuery()
  1238. deviceConfigs = CRUD(DeviceConfig)
  1239. deviceConfigTabs = CRUD(DeviceConfigTabs)
  1240. tabs = result.json().get("data", {}).get("tabs")
  1241. actions = result.json().get("data", {}).get("actions")
  1242. if tabs:
  1243. # 先删除再创建
  1244. deviceConfigTabs.deleteConditions(session, {})
  1245. deviceConfigs.deleteConditions(session, {})
  1246. batch_insert_device_configsNew(session, tabs, actions)
  1247. else:
  1248. all_actions = deviceConfigs.read_all(session)
  1249. all_tabs = deviceConfigTabs.read_all(session)
  1250. all_tabs_json = [item.model_dump(mode='json') for item in all_tabs]
  1251. all_actions_json = [item.model_dump(mode="json") for item in all_actions]
  1252. data_json = json.dumps(
  1253. {"tabs": all_tabs_json, "actions": all_actions_json,"machine_type":MACHINE_TYPE}, ensure_ascii=False
  1254. )
  1255. sync_url = settings.DOMAIN + "/api/ai_image/camera_machine/sync_actions"
  1256. result = requests.post(url=sync_url, headers=headers, data=data_json)
  1257. tabs = result.json().get("data", {}).get("tabs")
  1258. actions = result.json().get("data", {}).get("actions")
  1259. insert_action_ids = result.json().get("data", {}).get("insert_action_ids")
  1260. if tabs:
  1261. deviceConfigTabs.deleteConditions(session, {})
  1262. deviceConfigs.deleteConditions(session, {})
  1263. batch_insert_device_configsNew(session, tabs, actions)
  1264. if insert_action_ids:
  1265. for action_item in insert_action_ids:
  1266. photos = CRUD(PhotoRecord)
  1267. old_id = action_item.get("old_id")
  1268. new_id = action_item.get("new_id")
  1269. kwargs = {"action_id": new_id}
  1270. photos.updateConditionsAll(
  1271. session, conditions={"action_id": old_id}, **kwargs
  1272. )
  1273. # 因为左右脚线上id可能会发生变化 所以需要重新同步一下本地得配置信息
  1274. # syncUserJsonConfigs(hlm_token)
  1275. return {"code": 0, "msg": "操作成功", "data": None}