api.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872
  1. from natsort.natsort import order_by_index
  2. from sqlalchemy import func
  3. from models import *
  4. import requests
  5. import json
  6. from logger import logger
  7. from serial.tools import list_ports
  8. from model import PhotoRecord
  9. import settings, datetime
  10. import pandas as pd
  11. from utils.hlm_http_request import forward_request
  12. from utils.utils_func import check_path
  13. from sockets.socket_client import socket_manager
  14. from mcu.DeviceControl import DeviceControl
  15. import time, shutil, os
  16. from sqlalchemy import and_, asc, desc
  17. from functools import partial
  18. from service.deal_image import DealImage
  19. from databases import DeviceConfig, SysConfigs, SqlQuery, CRUD, select, DeviceConfigTabs
  20. from service.run_main import RunMain
  21. import importlib
  22. from service.auto_deal_pics.upload_pic import UploadPic
  23. from service.OnePicTest import OnePicTest
  24. from service.base import check_move_goods_art_no_folder
  25. import hashlib
  26. import win32api, win32gui, win32con
  27. from win32gui import EnumWindows, GetWindowText
  28. # from service.AutoDealPics import AutoDealPics
  29. # for plugin in settings.plugins:
  30. # module_path, class_name = plugin.rsplit(".", 1)
  31. # print("module_path", module_path, class_name)
  32. # module = importlib.import_module(module_path)
  33. # getattr(module, class_name)
  34. def calculate_md5(filepath):
  35. # 打开文件,以二进制只读模式打开
  36. with open(filepath, "rb") as f:
  37. # 创建MD5哈希对象
  38. md5hash = hashlib.md5()
  39. # 循环读取文件的内容并更新哈希对象
  40. for chunk in iter(lambda: f.read(4096), b""):
  41. md5hash.update(chunk)
  42. # 返回MD5哈希的十六进制表示
  43. return md5hash.hexdigest()
  44. @app.get("/")
  45. async def index():
  46. # await socket_manager.send_message(msg="测试")
  47. return {"message": "Hello World"}
  48. @app.get("/scan_serials", description="扫描可用的设备端口")
  49. async def scanSerials():
  50. """扫描串口"""
  51. ports = list_ports.comports()
  52. print("Scanning", ports)
  53. return {"message": "Hello World"}
  54. @app.api_route(
  55. "/forward_request", methods=["GET", "POST"], description="代理转发hlm项目得请求"
  56. )
  57. async def forwardRequest(request: HlmForwardRequest):
  58. """
  59. 转发HTTP请求到目标URL
  60. :param request: FastAPI Request对象
  61. :return: 目标接口的响应
  62. """
  63. try:
  64. if request.method == "GET":
  65. params = request.query_params
  66. elif request.method == "POST":
  67. params = json.dump(request.query_params)
  68. else:
  69. raise UnicornException("仅支持GET和POST方法")
  70. target_url = request.target_url
  71. method = request.method.upper()
  72. headers = request.headers
  73. if not target_url:
  74. raise UnicornException("目标url地址不能为空")
  75. # 调用 hlm_http_request 中的 forward_request 函数
  76. response = forward_request(
  77. target_url, params=params, method=method, headers=headers
  78. )
  79. return response
  80. except requests.RequestException as e:
  81. raise UnicornException(e)
  82. except Exception as e:
  83. raise UnicornException(e)
  84. def fromExcelHandler(params: HandlerDetail):
  85. excel_path = params.excel_path
  86. token = "Bearer " + params.token
  87. excel_df = pd.read_excel(excel_path, sheet_name=0, header=0)
  88. handler_result = []
  89. handler_result_folder = ""
  90. if "文件夹名称" not in excel_df.columns:
  91. raise UnicornException("缺失 [文件夹名称] 列")
  92. if "商品货号" not in excel_df.columns:
  93. raise UnicornException("缺失 [商品货号] 列")
  94. if "款号" not in excel_df.columns:
  95. raise UnicornException("缺失 [款号] 列")
  96. goods_art_dirs = excel_df.groupby(excel_df["款号"])
  97. obj = None
  98. run_main = RunMain(obj, token)
  99. temp_class = {}
  100. temp_name_list = []
  101. for tempItem in params.temp_list:
  102. temp_class[tempItem.template_id] = tempItem.template_local_classes
  103. temp_name_list.append(tempItem.template_id)
  104. goods_art_no_arrays = []
  105. for _, goods_row in excel_df.iterrows():
  106. goods_art_no = str(goods_row["商品货号"])
  107. goods_art_no_arrays.append(goods_art_no)
  108. limit_path = "output/{}".format(
  109. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  110. )
  111. move_folder_array = check_move_goods_art_no_folder(
  112. "output", goods_art_no_arrays, limit_path
  113. )
  114. for index, row in excel_df.iterrows():
  115. goods_art_no_image_dir = str(row["文件夹名称"])
  116. goods_art_no = str(row["商品货号"])
  117. print("货号数据", goods_art_no)
  118. goods_no = str(row["款号"])
  119. a001_df = goods_art_dirs.get_group(goods_no)
  120. goods_art_nos = a001_df["商品货号"].tolist()
  121. print("goods_art_nos", goods_art_nos)
  122. if not goods_art_no:
  123. raise UnicornException("货号不能为空")
  124. session = SqlQuery()
  125. pr = CRUD(PhotoRecord)
  126. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  127. if not images:
  128. raise UnicornException("没有可用货号数据")
  129. # 货号目录不存在再去进行移动和创建操作
  130. if move_folder_array.get(goods_art_no) == None:
  131. image_dir = "{}/data/{}".format(os.getcwd(), goods_no).replace("\\", "/")
  132. check_path(image_dir)
  133. for itemImg in images:
  134. if itemImg.image_path == "" or itemImg.image_path == None:
  135. raise UnicornException(
  136. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  137. )
  138. new_file_name = (
  139. str(itemImg.goods_art_no) + "_" + str(itemImg.id) + ".jpg"
  140. )
  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. try:
  156. config_data = {
  157. "image_dir": limit_path, # 这个目录作为本次生成的图片目录非常重要 例:./output/当前日期
  158. "image_order": params.template_image_order,
  159. "goods_art_no": goods_art_no,
  160. "goods_art_nos": goods_art_no_arrays,
  161. "is_check_number": False,
  162. "resize_image_view": "后跟",
  163. "cutout_mode": settings.CUTOUT_MODE,
  164. "logo_path": params.logo_path,
  165. "special_goods_art_no_folder_line": "",
  166. "is_use_excel": (False if params.excel_path == "" else True), # 是否使用excel
  167. "excel_path": params.excel_path, # excel路径
  168. "is_check_color_is_all": False,
  169. "cutout_is_pass": True,
  170. "assigned_page_dict": {},
  171. "detail_is_pass": True,
  172. "upload_is_pass": False,
  173. "upload_is_enable": False,
  174. "is_filter": False,
  175. "temp_class": temp_class,
  176. "temp_name": params.temp_name,
  177. "temp_name_list": temp_name_list,
  178. "target_error_folder": f"{limit_path}/软件-生成详情错误",
  179. }
  180. # 动态导入类
  181. temp_class_dict = {}
  182. for key, class_path in config_data["temp_class"].items():
  183. module_path, class_name = class_path.rsplit(".", 1)
  184. module = importlib.import_module(module_path)
  185. cls = getattr(module, class_name)
  186. temp_class_dict[key] = cls
  187. config_data["temp_class"] = temp_class_dict
  188. # 此处对抠图进行批量处理,保证所有的图片在生成详情图之前已经完成抠图,以保证详情图生成的效率
  189. return_data = run_main.check_before_cutout(config_data)
  190. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  191. check_for_detail_first_res = None
  192. if cutout_res:
  193. return_data_check_before_detail = run_main.check_before_detail(config_data)
  194. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  195. return_data_check_before_detail
  196. )
  197. if isinstance(check_for_detail_first_res, partial):
  198. result = check_for_detail_first_res()
  199. try:
  200. config_data = result["config_data"]
  201. except:
  202. config_data = result
  203. if config_data["sign_text"] == "已结束详情处理":
  204. # at_pic = AutoDealPics()
  205. print("config_data", config_data)
  206. if config_data["upload_is_enable"]:
  207. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  208. check_path(to_deal_dir)
  209. print("to_deal_dir", to_deal_dir)
  210. if os.path.exists(to_deal_dir):
  211. upload_pic = UploadPic(
  212. windows=None,
  213. to_deal_dir=to_deal_dir,
  214. config_data=config_data,
  215. token=token,
  216. )
  217. upload_pic.run()
  218. out_put_dir = config_data["out_put_dir"]
  219. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace("\\", "/")
  220. handler_result_folder = os.path.dirname(out_put_dir_path)
  221. handler_result.append(
  222. {"goods_art_no": goods_art_no, "success": True, "info": "处理成功"}
  223. )
  224. else:
  225. handler_result.append(
  226. {"goods_art_no": goods_art_no, "success": False, "info": "处理失败"}
  227. )
  228. except Exception as e:
  229. handler_result.append(
  230. {"goods_art_no": goods_art_no, "success": False, "info": str(e)}
  231. )
  232. handler_result_folder = "/".join(handler_result_folder.split("/")[:-1])
  233. return {
  234. "code": 0,
  235. "msg": "",
  236. "data": {"output_folder": handler_result_folder, "list": handler_result},
  237. }
  238. def group_by_style_number(data):
  239. result = {}
  240. for goods_id, info in data.items():
  241. style_number = info["款号"]
  242. if style_number not in result:
  243. result[style_number] = []
  244. result[style_number].append(goods_id)
  245. return result
  246. @app.post("/handle_detail")
  247. async def handle_detail(request: Request, params: HandlerDetail):
  248. obj = None
  249. token = "Bearer " + params.token
  250. run_main = RunMain(obj, token)
  251. goods_art_no_array = params.goods_art_no
  252. is_only_cutout = params.is_only_cutout
  253. handler_result = []
  254. handler_result_folder = ""
  255. if params.excel_path != "" and params.excel_path != None:
  256. return fromExcelHandler(params)
  257. goods_art_list = run_main.data_mode_generate_detail.get_goods_art_no_info(
  258. goods_art_list=goods_art_no_array
  259. )
  260. goods_art_nos = []
  261. # for goods_no_values in goods_art_list_dict.values():
  262. # if goods_art_no in goods_no_values:
  263. # goods_art_nos = goods_no_values
  264. # break
  265. goods_art_list_dict = group_by_style_number(goods_art_list)
  266. path = ""
  267. for goods_art_no in goods_art_no_array:
  268. try:
  269. if not goods_art_no:
  270. raise UnicornException("货号不能为空")
  271. session = SqlQuery()
  272. pr = CRUD(PhotoRecord)
  273. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  274. if not images:
  275. raise UnicornException("没有可用货号数据")
  276. if is_only_cutout != 1:
  277. detail_counts = len(params.template_image_order.split(","))
  278. image_counts = len(images)
  279. if image_counts < detail_counts:
  280. raise UnicornException(
  281. f"货号:[{goods_art_no}],实际照片数量:{image_counts}张,小于详情图要求数量:{detail_counts}张"
  282. )
  283. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  284. check_path(image_dir)
  285. for itemImg in images:
  286. if itemImg.image_path == "" or itemImg.image_path == None:
  287. raise UnicornException(
  288. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  289. )
  290. new_file_name = (
  291. str(itemImg.goods_art_no) + "_" + str(itemImg.id) + ".jpg"
  292. )
  293. if not os.path.exists(
  294. image_dir + "/" + os.path.basename(new_file_name)
  295. ):
  296. shutil.copy(itemImg.image_path, image_dir + "/" + new_file_name)
  297. dealImage = DealImage(image_dir)
  298. resFlag, path = dealImage.dealMoveImage(
  299. image_dir=image_dir, callback_func=None, goods_art_no=goods_art_no
  300. )
  301. if not resFlag:
  302. raise UnicornException(path)
  303. temp_class = {}
  304. temp_name_list = []
  305. for tempItem in params.temp_list:
  306. temp_class[tempItem.template_id] = tempItem.template_local_classes
  307. temp_name_list.append(tempItem.template_id)
  308. config_data = {
  309. "image_dir": path,
  310. "image_order": (
  311. "俯视,侧视,后跟,鞋底,内里"
  312. if params.template_image_order == None
  313. or params.template_image_order == ""
  314. else params.template_image_order
  315. ),
  316. "goods_art_no": goods_art_no,
  317. "goods_art_nos": (
  318. goods_art_nos if len(goods_art_nos) > 0 else [goods_art_no]
  319. ),
  320. "is_check_number": False,
  321. "resize_image_view": "后跟",
  322. "cutout_mode": settings.CUTOUT_MODE,
  323. "logo_path": params.logo_path,
  324. "special_goods_art_no_folder_line": "",
  325. "is_use_excel": (
  326. False if params.excel_path == "" else True
  327. ), # 是否使用excel
  328. "excel_path": params.excel_path, # excel路径
  329. "is_check_color_is_all": False,
  330. "cutout_is_pass": True,
  331. "assigned_page_dict": {},
  332. "detail_is_pass": True,
  333. "upload_is_pass": False,
  334. "upload_is_enable": True,
  335. "is_filter": False,
  336. "temp_class": temp_class,
  337. "temp_name": params.temp_name,
  338. "temp_name_list": temp_name_list,
  339. "target_error_folder": f"{path}/软件-生成详情错误",
  340. }
  341. print("image_dir=====>>>>>", config_data["image_dir"])
  342. # 动态导入类
  343. temp_class_dict = {}
  344. for key, class_path in config_data["temp_class"].items():
  345. module_path, class_name = class_path.rsplit(".", 1)
  346. module = importlib.import_module(module_path)
  347. cls = getattr(module, class_name)
  348. temp_class_dict[key] = cls
  349. config_data["temp_class"] = temp_class_dict
  350. return_data = run_main.check_before_cutout(config_data)
  351. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  352. check_for_detail_first_res = None
  353. if cutout_res == True:
  354. out_put_dir = return_data["data"]["image_dir"] + "/" + goods_art_no
  355. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  356. "\\", "/"
  357. )
  358. # print("out_put_dir_path", out_put_dir_path)
  359. handler_result_folder = os.path.dirname(out_put_dir_path)
  360. handler_result.append(
  361. {"goods_art_no": goods_art_no, "success": True, "info": "处理成功"}
  362. )
  363. # print("检测到需要裁剪图片,开始裁剪图片", return_data)
  364. except Exception as e:
  365. handler_result.append(
  366. {"goods_art_no": goods_art_no, "success": False, "info": str(e)}
  367. )
  368. if is_only_cutout == 1:
  369. return {
  370. "code": 0,
  371. "msg": "",
  372. "data": {"output_folder": handler_result_folder, "list": handler_result},
  373. }
  374. handler_result = []
  375. try:
  376. for goods_art_no_arr in goods_art_list_dict.values():
  377. temp_class = {}
  378. temp_name_list = []
  379. for tempItem in params.temp_list:
  380. temp_class[tempItem.template_id] = tempItem.template_local_classes
  381. temp_name_list.append(tempItem.template_id)
  382. config_data = {
  383. "image_dir": path,
  384. "image_order": (
  385. "俯视,侧视,后跟,鞋底,内里"
  386. if params.template_image_order is None
  387. or params.template_image_order == ""
  388. else params.template_image_order
  389. ),
  390. "goods_art_no": "",
  391. "goods_art_nos": goods_art_no_arr,
  392. "is_check_number": False,
  393. "resize_image_view": "后跟",
  394. "cutout_mode": settings.CUTOUT_MODE,
  395. "logo_path": params.logo_path,
  396. "special_goods_art_no_folder_line": "",
  397. "is_use_excel": (
  398. False if params.excel_path == "" else True
  399. ), # 是否使用excel
  400. "excel_path": params.excel_path, # excel路径
  401. "is_check_color_is_all": False,
  402. "cutout_is_pass": True,
  403. "assigned_page_dict": {},
  404. "detail_is_pass": True,
  405. "upload_is_pass": False,
  406. "upload_is_enable": True,
  407. "is_filter": False,
  408. "temp_class": temp_class,
  409. "temp_name": params.temp_name,
  410. "temp_name_list": temp_name_list,
  411. "target_error_folder": f"{path}/软件-生成详情错误",
  412. }
  413. # 动态导入类
  414. temp_class_dict = {}
  415. for key, class_path in config_data["temp_class"].items():
  416. module_path, class_name = class_path.rsplit(".", 1)
  417. module = importlib.import_module(module_path)
  418. cls = getattr(module, class_name)
  419. temp_class_dict[key] = cls
  420. config_data["temp_class"] = temp_class_dict
  421. return_data_check_before_detail = run_main.check_before_detail(config_data)
  422. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  423. return_data_check_before_detail
  424. )
  425. if isinstance(check_for_detail_first_res, partial):
  426. result = check_for_detail_first_res()
  427. try:
  428. config_data = result["config_data"]
  429. except:
  430. config_data = result
  431. if config_data["sign_text"] == "已结束详情处理":
  432. # at_pic = AutoDealPics()
  433. print("config_data", config_data)
  434. if config_data["upload_is_enable"]:
  435. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  436. check_path(to_deal_dir)
  437. print("to_deal_dir", to_deal_dir)
  438. if os.path.exists(to_deal_dir):
  439. upload_pic = UploadPic(
  440. windows=None,
  441. to_deal_dir=to_deal_dir,
  442. config_data=config_data,
  443. token=token,
  444. )
  445. upload_pic.run()
  446. out_put_dir = config_data["out_put_dir"]
  447. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  448. "\\", "/"
  449. )
  450. handler_result_folder = os.path.dirname(out_put_dir_path)
  451. handler_result.append(
  452. {"goods_art_no": "", "success": True, "info": "处理成功"}
  453. )
  454. else:
  455. handler_result.append(
  456. {"goods_art_no": "", "success": False, "info": "处理失败"}
  457. )
  458. except Exception as e:
  459. handler_result.append({"goods_art_no": "", "success": False, "info": str(e)})
  460. return {
  461. "code": 0,
  462. "msg": "",
  463. "data": {"output_folder": handler_result_folder, "list": handler_result},
  464. }
  465. @app.get("/get_device_tabs", description="获取可执行程序命令列表")
  466. def get_device_tabs(type: int):
  467. session = SqlQuery()
  468. statement = (
  469. select(DeviceConfigTabs)
  470. .where(DeviceConfigTabs.mode_type == type)
  471. .order_by(asc("id"))
  472. )
  473. result = session.exec(statement).all()
  474. session.close()
  475. sys = CRUD(SysConfigs)
  476. action_configs = sys.read(session, conditions={"key": "action_configs"})
  477. session.close()
  478. return {
  479. "code": 0,
  480. "msg": "",
  481. "data": {"tabs": result, "select_configs": json.loads(action_configs.value)},
  482. }
  483. @app.post("/update_tab_name", description="更改tab名称")
  484. def update_tab_name(params: DeviceConfigTabsReq):
  485. if params.mode_name == "":
  486. return {"code": 1, "msg": "名称不能为空", "data": {}}
  487. session = SqlQuery()
  488. tabModel = CRUD(DeviceConfigTabs)
  489. kwargs = {"mode_name": params.mode_name}
  490. tabModel.updateConditions(session, conditions={"id": params.id}, **kwargs)
  491. session.close()
  492. return {
  493. "code": 0,
  494. "msg": "",
  495. "data": None,
  496. }
  497. @app.post("/get_device_configs", description="获取可执行程序命令列表")
  498. def get_device_configs(params: ModelGetDeviceConfig):
  499. tab_id = params.tab_id
  500. session = SqlQuery()
  501. configModel = CRUD(DeviceConfig)
  502. configList = configModel.read_all(
  503. session,
  504. conditions={"tab_id": tab_id},
  505. order_by="action_index",
  506. ascending=True,
  507. )
  508. return {
  509. "code": 0,
  510. "msg": "",
  511. "data": {"list": configList},
  512. }
  513. @app.post("/device_config_detail", description="获取可执行程序详情")
  514. def get_device_configs(params: ModelGetDeviceConfigDetail):
  515. action_id = params.id
  516. session = SqlQuery()
  517. configModel = CRUD(DeviceConfig)
  518. model = configModel.read(session, conditions={"id": action_id})
  519. if model == None:
  520. return {"code": 1, "msg": "数据不存在", "data": None}
  521. return {"code": 0, "msg": "", "data": model}
  522. @app.post("/device_config_detail_query", description="通过条件获取可执行程序详情")
  523. def device_config_detail_query():
  524. # tab_id = params.tab_id
  525. # action_name = params.action_name
  526. session = SqlQuery()
  527. sys = CRUD(SysConfigs)
  528. action_configs = sys.read(session, conditions={"key": "action_configs"})
  529. action_configs_value = json.loads(action_configs.value)
  530. left_config = action_configs_value.get("left")
  531. configModel = CRUD(DeviceConfig)
  532. model = configModel.read(
  533. session, conditions={"tab_id": left_config, "action_name": "侧视"}
  534. )
  535. if model == None:
  536. model = configModel.read(session, conditions={"tab_id": left_config})
  537. return {"code": 0, "msg": "", "data": model}
  538. @app.post("/remove_config", description="删除一条可执行命令")
  539. def get_device_configs(params: ModelGetDeviceConfigDetail):
  540. action_id = params.id
  541. session = SqlQuery()
  542. configModel = CRUD(DeviceConfig)
  543. model = configModel.read(session, conditions={"id": action_id})
  544. if model == None:
  545. return {"code": 1, "msg": "数据不存在", "data": None}
  546. if model.is_system == True:
  547. return {"code": 1, "msg": "系统配置不允许删除", "data": None}
  548. configArray = configModel.read_all(session, conditions={"tab_id": model.tab_id})
  549. if len(configArray) == 1:
  550. return {"code": 1, "msg": "请至少保留一个配置", "data": None}
  551. configModel.delete(session, obj_id=action_id)
  552. return {"code": 0, "msg": "删除成功", "data": None}
  553. @app.post("/save_device_config", description="创建或修改一条可执行命令")
  554. def save_device_config(params: SaveDeviceConfig):
  555. action_id = params.id
  556. session = SqlQuery()
  557. deviceConfig = CRUD(DeviceConfig)
  558. if action_id == None or action_id == 0:
  559. # 走新增逻辑
  560. params.id = None
  561. save_device_config = deviceConfig.create(session, obj_in=params)
  562. else:
  563. model = deviceConfig.read(session, conditions={"id": action_id})
  564. if model == None:
  565. return {"code": 1, "msg": "数据不存在", "data": None}
  566. # 走编辑逻辑
  567. kwargs = params.__dict__
  568. save_device_config = deviceConfig.update(session, obj_id=action_id, **kwargs)
  569. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  570. @app.post("/reset_config", description="创建或修改一条可执行命令")
  571. def reset_config(params: ModelGetDeviceConfig):
  572. tab_id = params.tab_id
  573. if tab_id == None or tab_id == "":
  574. return {"code": 1, "msg": "参数错误", "data": None}
  575. session = SqlQuery()
  576. deviceConfig = CRUD(DeviceConfig)
  577. first_config = deviceConfig.read(session, conditions={"tab_id": tab_id})
  578. res = deviceConfig.deleteConditions(session, conditions={"tab_id": tab_id})
  579. if res is False:
  580. return {"code": 1, "msg": "操作失败", "data": None}
  581. actions = json.load(open("action.json", encoding="utf-8"))
  582. for data in actions:
  583. data["tab_id"] = tab_id
  584. data["is_system"] = first_config.is_system
  585. device_config = DeviceConfig(**data)
  586. session.add(device_config)
  587. session.commit()
  588. session.close()
  589. return {"code": 0, "msg": "操作成功", "data": None}
  590. @app.get("/get_photo_records", description="获取拍照记录")
  591. def get_photo_records(page: int = 1, size: int = 5):
  592. session = SqlQuery()
  593. photos = CRUD(PhotoRecord)
  594. print("准备查询拍摄记录", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  595. statement = (
  596. select(PhotoRecord)
  597. .offset((page - 1) * size)
  598. .limit(size)
  599. .order_by(desc("id"))
  600. .group_by("goods_art_no")
  601. )
  602. list = []
  603. result = session.exec(statement).all()
  604. print("group 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  605. join_conditions = [
  606. {
  607. "model": DeviceConfig,
  608. "on": PhotoRecord.action_id == DeviceConfig.id,
  609. "is_outer": False, # 可选,默认False,设为True则为LEFT JOIN
  610. }
  611. ]
  612. for item in result:
  613. print(item)
  614. # list_item = photos.read_all(
  615. # session,
  616. # conditions={"goods_art_no": item.goods_art_no},
  617. # join_conditions=join_conditions,
  618. # )
  619. query = (
  620. select(PhotoRecord, DeviceConfig.action_name)
  621. .where(PhotoRecord.goods_art_no == item.goods_art_no)
  622. .join(DeviceConfig, PhotoRecord.action_id == DeviceConfig.id)
  623. )
  624. list_item = session.exec(query).mappings().all()
  625. session.close()
  626. list.append(
  627. {
  628. "goods_art_no": item.goods_art_no,
  629. "action_time": item.create_time,
  630. "items": list_item,
  631. }
  632. )
  633. session.close()
  634. print("循环查询 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  635. return {
  636. "code": 0,
  637. "msg": "",
  638. "data": {"list": list, "page": page, "size": size},
  639. }
  640. @app.get("/get_last_photo_record", description="获取最后一条拍照记录")
  641. def get_last_photo_record():
  642. session = SqlQuery()
  643. statement = (
  644. select(PhotoRecord)
  645. .where(PhotoRecord.image_path != None)
  646. .order_by(desc("photo_create_time"))
  647. )
  648. result = session.exec(statement).first()
  649. session.close()
  650. return {
  651. "code": 0,
  652. "msg": "",
  653. "data": result,
  654. }
  655. @app.get("/get_photo_record_detail", description="通过货号获取拍照记录详情")
  656. def get_photo_records(goods_art_no: str = None):
  657. if goods_art_no == None:
  658. return {"code": 1, "msg": "参数错误", "data": None}
  659. session = SqlQuery()
  660. photos = CRUD(PhotoRecord)
  661. items = photos.read_all(session, conditions={"goods_art_no": goods_art_no})
  662. session.close()
  663. return {
  664. "code": 0,
  665. "msg": "",
  666. "data": {"list": items},
  667. }
  668. @app.post("/delect_goods_arts", description="通过货号删除记录")
  669. def delect_goods_arts(params: PhotoRecordDelete):
  670. session = SqlQuery()
  671. photos = CRUD(PhotoRecord)
  672. for item in params.goods_art_nos:
  673. photos.deleteConditions(session, conditions={"goods_art_no": item})
  674. session.close()
  675. return {
  676. "code": 0,
  677. "msg": "操作成功",
  678. "data": None,
  679. }
  680. @app.get("/get_sys_config", description="查询系统配置")
  681. def get_sys_config(key: str = None):
  682. if key == None:
  683. return {"code": 1, "msg": "参数错误", "data": None}
  684. session = SqlQuery()
  685. photos = CRUD(SysConfigs)
  686. item = photos.read(session, conditions={"key": key})
  687. session.close()
  688. return {
  689. "code": 0,
  690. "msg": "",
  691. "data": json.loads(item.value),
  692. }
  693. @app.post("/update_left_right_config", description="更新左右脚配置")
  694. def update_left_right_config(params: LeftRightParams):
  695. session = SqlQuery()
  696. sysConfig = CRUD(SysConfigs)
  697. model = sysConfig.read(session, conditions={"key": "action_configs"})
  698. if model == None:
  699. return {"code": 1, "msg": "配置不存在", "data": None}
  700. config_value = json.loads(model.value)
  701. config_value[params.type] = params.id
  702. update_value = json.dumps(config_value)
  703. # 走编辑逻辑
  704. kwargs = {"key": "action_configs", "value": update_value}
  705. save_device_config = sysConfig.updateConditions(
  706. session, conditions={"key": "action_configs"}, **kwargs
  707. )
  708. # return {"code": 0, "msg": "操作成功", "data": save_device_config}
  709. return {"code": 0, "msg": "操作成功", "data": None}
  710. @app.post("/update_sys_configs", description="创建或修改系统配置")
  711. def save_sys_configs(params: SysConfigParams):
  712. session = SqlQuery()
  713. sysConfig = CRUD(SysConfigs)
  714. model = sysConfig.read(session, conditions={"key": params.key})
  715. if model == None:
  716. return {"code": 1, "msg": "配置不存在", "data": None}
  717. # 走编辑逻辑
  718. kwargs = params.__dict__
  719. save_device_config = sysConfig.updateConditions(
  720. session, conditions={"key": params.key}, **kwargs
  721. )
  722. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  723. @app.post("/create_main_image", description="创建主图测试")
  724. def create_main_image(params: MaineImageTest):
  725. file_path = params.file_path
  726. onePic = OnePicTest(pic_path=file_path)
  727. # session = SqlQuery()
  728. # sysConfig = CRUD(SysConfigs)
  729. # model = sysConfig.read(session, conditions={"key": params.key})
  730. # if model == None:
  731. # return {"code": 1, "msg": "配置不存在", "data": None}
  732. # # 走编辑逻辑
  733. # kwargs = params.__dict__
  734. # save_device_config = sysConfig.updateConditions(
  735. # session, conditions={"key": params.key}, **kwargs
  736. # )
  737. main_out_path = onePic.HandlerMainImage()
  738. return {"code": 0, "msg": "操作成功", "data": {"main_out_path": main_out_path}}
  739. def insertEmptyLogoList(session):
  740. """插入空logo列表"""
  741. data = {"key": "logo_configs", "value": "[]"}
  742. config = SysConfigs(**data)
  743. session.add(config)
  744. session.commit()
  745. session.close()
  746. item = SysConfigs()
  747. item.key = "logo_configs"
  748. item.value = "[]"
  749. return item
  750. @app.get("/logo_list", description="logo列表")
  751. def logo_list():
  752. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  753. check_path(logo_dir)
  754. logo_files = os.listdir(logo_dir)
  755. logo_list = []
  756. for logoItem in logo_files:
  757. logo_list.append(f"{logo_dir}{logoItem}")
  758. return {"code": 0, "msg": "操作成功", "data": logo_list}
  759. @app.post("/add_logo", description="添加logo")
  760. def add_logo(params: LogoParams):
  761. logo_path = params.logo_path.replace("\\", "/")
  762. session = SqlQuery()
  763. sysConfig = CRUD(SysConfigs)
  764. item = sysConfig.read(session, conditions={"key": "logo_configs"})
  765. if item == None:
  766. item = insertEmptyLogoList(session)
  767. if os.path.isfile(logo_path) == False:
  768. return {"code": 1, "msg": "logo文件不存在", "data": None}
  769. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  770. check_path(logo_dir)
  771. fpath, fname = os.path.split(logo_path)
  772. logo_path_info = logo_dir + fname
  773. shutil.copy(logo_path, logo_path_info) # 复制文件
  774. logo_files = os.listdir(logo_dir)
  775. logo_list = []
  776. for logoItem in logo_files:
  777. logo_list.append(f"{logo_dir}{logoItem}")
  778. return {
  779. "code": 0,
  780. "msg": "",
  781. "data": {"logo": logo_path_info},
  782. }
  783. @app.post("/delete_logo", description="删除logo")
  784. def delete_logo(params: LogoParamsupdate):
  785. logo_path = params.path
  786. if os.path.isfile(logo_path) == False:
  787. return {"code": 1, "msg": "logo文件不存在", "data": None}
  788. os.remove(logo_path)
  789. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  790. check_path(logo_dir)
  791. logo_files = os.listdir(logo_dir)
  792. logo_list = []
  793. for logoItem in logo_files:
  794. logo_list.append(f"{logo_dir}{logoItem}")
  795. return {"code": 0, "msg": "操作成功", "data": logo_list}
  796. @app.post("/close_other_window", description="关闭窗口")
  797. def close_other_window():
  798. hwnd_list = []
  799. def callback(hwnd, _):
  800. title = GetWindowText(hwnd)
  801. if title == "digiCamControl by Duka Istvan":
  802. hwnd_list.append(hwnd)
  803. EnumWindows(callback, None)
  804. if hwnd_list:
  805. hwnd = hwnd_list[0]
  806. win32gui.PostMessage(hwnd, win32con.WM_CLOSE, 0, 0)
  807. return {"code": 0, "msg": "关闭成功", "data": {"status": True}}
  808. return {"code": 0, "msg": "关闭失败", "data": {"status": False}}