api.py 40 KB

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