api.py 46 KB

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