api.py 58 KB

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