api.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876
  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. import hashlib
  25. import win32api, win32gui, win32con
  26. from win32gui import EnumWindows, GetWindowText
  27. # from service.AutoDealPics import AutoDealPics
  28. # for plugin in settings.plugins:
  29. # module_path, class_name = plugin.rsplit(".", 1)
  30. # print("module_path", module_path, class_name)
  31. # module = importlib.import_module(module_path)
  32. # getattr(module, class_name)
  33. def calculate_md5(filepath):
  34. # 打开文件,以二进制只读模式打开
  35. with open(filepath, "rb") as f:
  36. # 创建MD5哈希对象
  37. md5hash = hashlib.md5()
  38. # 循环读取文件的内容并更新哈希对象
  39. for chunk in iter(lambda: f.read(4096), b""):
  40. md5hash.update(chunk)
  41. # 返回MD5哈希的十六进制表示
  42. return md5hash.hexdigest()
  43. @app.get("/")
  44. async def index():
  45. # await socket_manager.send_message(msg="测试")
  46. return {"message": "Hello World"}
  47. @app.get("/scan_serials", description="扫描可用的设备端口")
  48. async def scanSerials():
  49. """扫描串口"""
  50. ports = list_ports.comports()
  51. print("Scanning", ports)
  52. return {"message": "Hello World"}
  53. @app.api_route(
  54. "/forward_request", methods=["GET", "POST"], description="代理转发hlm项目得请求"
  55. )
  56. async def forwardRequest(request: HlmForwardRequest):
  57. """
  58. 转发HTTP请求到目标URL
  59. :param request: FastAPI Request对象
  60. :return: 目标接口的响应
  61. """
  62. try:
  63. if request.method == "GET":
  64. params = request.query_params
  65. elif request.method == "POST":
  66. params = json.dump(request.query_params)
  67. else:
  68. raise UnicornException("仅支持GET和POST方法")
  69. target_url = request.target_url
  70. method = request.method.upper()
  71. headers = request.headers
  72. if not target_url:
  73. raise UnicornException("目标url地址不能为空")
  74. # 调用 hlm_http_request 中的 forward_request 函数
  75. response = forward_request(
  76. target_url, params=params, method=method, headers=headers
  77. )
  78. return response
  79. except requests.RequestException as e:
  80. raise UnicornException(e)
  81. except Exception as e:
  82. raise UnicornException(e)
  83. def fromExcelHandler(params: HandlerDetail):
  84. excel_path = params.excel_path
  85. token = "Bearer " + params.token
  86. excel_df = pd.read_excel(excel_path, sheet_name=0, header=0)
  87. handler_result = []
  88. handler_result_folder = ""
  89. if "文件夹名称" not in excel_df.columns:
  90. raise UnicornException("缺失 [文件夹名称] 列")
  91. if "商品货号" not in excel_df.columns:
  92. raise UnicornException("缺失 [商品货号] 列")
  93. if "款号" not in excel_df.columns:
  94. raise UnicornException("缺失 [款号] 列")
  95. goods_art_dirs = excel_df.groupby(excel_df["款号"])
  96. obj = None
  97. run_main = RunMain(obj, token)
  98. temp_class = {}
  99. temp_name_list = []
  100. for tempItem in params.temp_list:
  101. temp_class[tempItem.template_id] = tempItem.template_local_classes
  102. temp_name_list.append(tempItem.template_id)
  103. goods_art_no_arrays = []
  104. for index, row in excel_df.iterrows():
  105. goods_art_no_image_dir = str(row["文件夹名称"])
  106. goods_art_no = str(row["商品货号"])
  107. goods_art_no_arrays.append(goods_art_no)
  108. print("货号数据", goods_art_no)
  109. goods_no = str(row["款号"])
  110. a001_df = goods_art_dirs.get_group(goods_no)
  111. goods_art_nos = a001_df["商品货号"].tolist()
  112. print("goods_art_nos", goods_art_nos)
  113. if not goods_art_no:
  114. raise UnicornException("货号不能为空")
  115. session = SqlQuery()
  116. pr = CRUD(PhotoRecord)
  117. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  118. if not images:
  119. raise UnicornException("没有可用货号数据")
  120. image_dir = "{}/data/{}".format(os.getcwd(), goods_no).replace("\\", "/")
  121. check_path(image_dir)
  122. for itemImg in images:
  123. if itemImg.image_path == "" or itemImg.image_path == None:
  124. raise UnicornException(
  125. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  126. )
  127. new_file_name = (
  128. str(itemImg.goods_art_no) + "_" + str(itemImg.id) + ".jpg"
  129. )
  130. if not os.path.exists(
  131. image_dir + "/" + os.path.basename(new_file_name)
  132. ):
  133. shutil.copy(itemImg.image_path, image_dir + "/" + new_file_name)
  134. dealImage = DealImage(image_dir)
  135. resFlag, path = dealImage.dealMoveImage(
  136. image_dir=image_dir,
  137. callback_func=None,
  138. goods_art_no=goods_art_no_image_dir,
  139. )
  140. if not resFlag:
  141. raise UnicornException(path)
  142. # if goods_art_no_image_dir not in path:
  143. # path = path + "/" + goods_art_no_image_dir
  144. path = os.path.dirname(path)
  145. try:
  146. print("path", path)
  147. config_data = {
  148. "image_dir": path,
  149. "image_order": params.template_image_order,
  150. "goods_art_no": goods_art_no,
  151. "goods_art_nos": goods_art_no_arrays,
  152. "is_check_number": False,
  153. "resize_image_view": "后跟",
  154. "cutout_mode": settings.CUTOUT_MODE,
  155. "logo_path": params.logo_path,
  156. "special_goods_art_no_folder_line": "",
  157. "is_use_excel": (
  158. False if params.excel_path == "" else True
  159. ), # 是否使用excel
  160. "excel_path": params.excel_path, # excel路径
  161. "is_check_color_is_all": False,
  162. "cutout_is_pass": True,
  163. "assigned_page_dict": {},
  164. "detail_is_pass": True,
  165. "upload_is_pass": False,
  166. "upload_is_enable": False,
  167. "is_filter": False,
  168. "temp_class": temp_class,
  169. "temp_name": params.temp_name,
  170. "temp_name_list": temp_name_list,
  171. "target_error_folder": f"{path}/软件-生成详情错误",
  172. }
  173. # 动态导入类
  174. temp_class_dict = {}
  175. for key, class_path in config_data["temp_class"].items():
  176. module_path, class_name = class_path.rsplit(".", 1)
  177. module = importlib.import_module(module_path)
  178. cls = getattr(module, class_name)
  179. temp_class_dict[key] = cls
  180. config_data["temp_class"] = temp_class_dict
  181. return_data = run_main.check_before_cutout(config_data)
  182. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  183. check_for_detail_first_res = None
  184. if cutout_res:
  185. return_data_check_before_detail = run_main.check_before_detail(
  186. config_data
  187. )
  188. print(
  189. "return_data_check_before_detail======> 测试 ==>",
  190. return_data_check_before_detail,
  191. )
  192. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  193. return_data_check_before_detail
  194. )
  195. if isinstance(check_for_detail_first_res, partial):
  196. result = check_for_detail_first_res()
  197. try:
  198. config_data = result["config_data"]
  199. except:
  200. config_data = result
  201. if config_data["sign_text"] == "已结束详情处理":
  202. # at_pic = AutoDealPics()
  203. print("config_data", config_data)
  204. if config_data["upload_is_enable"]:
  205. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  206. check_path(to_deal_dir)
  207. print("to_deal_dir", to_deal_dir)
  208. if os.path.exists(to_deal_dir):
  209. upload_pic = UploadPic(
  210. windows=None,
  211. to_deal_dir=to_deal_dir,
  212. config_data=config_data,
  213. token=token,
  214. )
  215. upload_pic.run()
  216. out_put_dir = config_data["out_put_dir"]
  217. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  218. "\\", "/"
  219. )
  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(
  422. config_data
  423. )
  424. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  425. return_data_check_before_detail
  426. )
  427. if isinstance(check_for_detail_first_res, partial):
  428. result = check_for_detail_first_res()
  429. try:
  430. config_data = result["config_data"]
  431. except:
  432. config_data = result
  433. if config_data["sign_text"] == "已结束详情处理":
  434. # at_pic = AutoDealPics()
  435. print("config_data", config_data)
  436. if config_data["upload_is_enable"]:
  437. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  438. check_path(to_deal_dir)
  439. print("to_deal_dir", to_deal_dir)
  440. if os.path.exists(to_deal_dir):
  441. upload_pic = UploadPic(
  442. windows=None,
  443. to_deal_dir=to_deal_dir,
  444. config_data=config_data,
  445. token=token,
  446. )
  447. upload_pic.run()
  448. out_put_dir = config_data["out_put_dir"]
  449. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  450. "\\", "/"
  451. )
  452. handler_result_folder = os.path.dirname(out_put_dir_path)
  453. handler_result.append(
  454. {"goods_art_no": "", "success": True, "info": "处理成功"}
  455. )
  456. else:
  457. handler_result.append(
  458. {"goods_art_no": "", "success": False, "info": "处理失败"}
  459. )
  460. except Exception as e:
  461. handler_result.append(
  462. {"goods_art_no": "", "success": False, "info": str(e)}
  463. )
  464. return {
  465. "code": 0,
  466. "msg": "",
  467. "data": {"output_folder": handler_result_folder, "list": handler_result},
  468. }
  469. @app.get("/get_device_tabs", description="获取可执行程序命令列表")
  470. def get_device_tabs(type: int):
  471. session = SqlQuery()
  472. statement = (
  473. select(DeviceConfigTabs)
  474. .where(DeviceConfigTabs.mode_type == type)
  475. .order_by(asc("id"))
  476. )
  477. result = session.exec(statement).all()
  478. session.close()
  479. sys = CRUD(SysConfigs)
  480. action_configs = sys.read(session, conditions={"key": "action_configs"})
  481. session.close()
  482. return {
  483. "code": 0,
  484. "msg": "",
  485. "data": {"tabs": result, "select_configs": json.loads(action_configs.value)},
  486. }
  487. @app.post("/update_tab_name", description="更改tab名称")
  488. def update_tab_name(params: DeviceConfigTabsReq):
  489. if params.mode_name == "":
  490. return {"code": 1, "msg": "名称不能为空", "data": {}}
  491. session = SqlQuery()
  492. tabModel = CRUD(DeviceConfigTabs)
  493. kwargs = {"mode_name": params.mode_name}
  494. tabModel.updateConditions(session, conditions={"id": params.id}, **kwargs)
  495. session.close()
  496. return {
  497. "code": 0,
  498. "msg": "",
  499. "data": None,
  500. }
  501. @app.post("/get_device_configs", description="获取可执行程序命令列表")
  502. def get_device_configs(params: ModelGetDeviceConfig):
  503. tab_id = params.tab_id
  504. session = SqlQuery()
  505. configModel = CRUD(DeviceConfig)
  506. configList = configModel.read_all(
  507. session,
  508. conditions={"tab_id": tab_id},
  509. order_by="action_index",
  510. ascending=True,
  511. )
  512. return {
  513. "code": 0,
  514. "msg": "",
  515. "data": {"list": configList},
  516. }
  517. @app.post("/device_config_detail", description="获取可执行程序详情")
  518. def get_device_configs(params: ModelGetDeviceConfigDetail):
  519. action_id = params.id
  520. session = SqlQuery()
  521. configModel = CRUD(DeviceConfig)
  522. model = configModel.read(session, conditions={"id": action_id})
  523. if model == None:
  524. return {"code": 1, "msg": "数据不存在", "data": None}
  525. return {"code": 0, "msg": "", "data": model}
  526. @app.post("/device_config_detail_query", description="通过条件获取可执行程序详情")
  527. def device_config_detail_query():
  528. # tab_id = params.tab_id
  529. # action_name = params.action_name
  530. session = SqlQuery()
  531. sys = CRUD(SysConfigs)
  532. action_configs = sys.read(session, conditions={"key": "action_configs"})
  533. action_configs_value = json.loads(action_configs.value)
  534. left_config = action_configs_value.get("left")
  535. configModel = CRUD(DeviceConfig)
  536. model = configModel.read(
  537. session, conditions={"tab_id": left_config, "action_name": "侧视"}
  538. )
  539. if model == None:
  540. model = configModel.read(session, conditions={"tab_id": left_config})
  541. return {"code": 0, "msg": "", "data": model}
  542. @app.post("/remove_config", description="删除一条可执行命令")
  543. def get_device_configs(params: ModelGetDeviceConfigDetail):
  544. action_id = params.id
  545. session = SqlQuery()
  546. configModel = CRUD(DeviceConfig)
  547. model = configModel.read(session, conditions={"id": action_id})
  548. if model == None:
  549. return {"code": 1, "msg": "数据不存在", "data": None}
  550. if model.is_system == True:
  551. return {"code": 1, "msg": "系统配置不允许删除", "data": None}
  552. configArray = configModel.read_all(session, conditions={"tab_id": model.tab_id})
  553. if len(configArray) == 1:
  554. return {"code": 1, "msg": "请至少保留一个配置", "data": None}
  555. configModel.delete(session, obj_id=action_id)
  556. return {"code": 0, "msg": "删除成功", "data": None}
  557. @app.post("/save_device_config", description="创建或修改一条可执行命令")
  558. def save_device_config(params: SaveDeviceConfig):
  559. action_id = params.id
  560. session = SqlQuery()
  561. deviceConfig = CRUD(DeviceConfig)
  562. if action_id == None or action_id == 0:
  563. # 走新增逻辑
  564. params.id = None
  565. save_device_config = deviceConfig.create(session, obj_in=params)
  566. else:
  567. model = deviceConfig.read(session, conditions={"id": action_id})
  568. if model == None:
  569. return {"code": 1, "msg": "数据不存在", "data": None}
  570. # 走编辑逻辑
  571. kwargs = params.__dict__
  572. save_device_config = deviceConfig.update(session, obj_id=action_id, **kwargs)
  573. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  574. @app.post("/reset_config", description="创建或修改一条可执行命令")
  575. def reset_config(params: ModelGetDeviceConfig):
  576. tab_id = params.tab_id
  577. if tab_id == None or tab_id == "":
  578. return {"code": 1, "msg": "参数错误", "data": None}
  579. session = SqlQuery()
  580. deviceConfig = CRUD(DeviceConfig)
  581. first_config = deviceConfig.read(session, conditions={"tab_id": tab_id})
  582. res = deviceConfig.deleteConditions(session, conditions={"tab_id": tab_id})
  583. if res is False:
  584. return {"code": 1, "msg": "操作失败", "data": None}
  585. actions = json.load(open("action.json", encoding="utf-8"))
  586. for data in actions:
  587. data["tab_id"] = tab_id
  588. data["is_system"] = first_config.is_system
  589. device_config = DeviceConfig(**data)
  590. session.add(device_config)
  591. session.commit()
  592. session.close()
  593. return {"code": 0, "msg": "操作成功", "data": None}
  594. @app.get("/get_photo_records", description="获取拍照记录")
  595. def get_photo_records(page: int = 1, size: int = 5):
  596. session = SqlQuery()
  597. photos = CRUD(PhotoRecord)
  598. print("准备查询拍摄记录", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  599. statement = (
  600. select(PhotoRecord)
  601. .offset((page - 1) * size)
  602. .limit(size)
  603. .order_by(desc("id"))
  604. .group_by("goods_art_no")
  605. )
  606. list = []
  607. result = session.exec(statement).all()
  608. print("group 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  609. join_conditions = [
  610. {
  611. "model": DeviceConfig,
  612. "on": PhotoRecord.action_id == DeviceConfig.id,
  613. "is_outer": False, # 可选,默认False,设为True则为LEFT JOIN
  614. }
  615. ]
  616. for item in result:
  617. print(item)
  618. # list_item = photos.read_all(
  619. # session,
  620. # conditions={"goods_art_no": item.goods_art_no},
  621. # join_conditions=join_conditions,
  622. # )
  623. query = (
  624. select(PhotoRecord, DeviceConfig.action_name)
  625. .where(PhotoRecord.goods_art_no == item.goods_art_no)
  626. .join(DeviceConfig, PhotoRecord.action_id == DeviceConfig.id)
  627. )
  628. list_item = session.exec(query).mappings().all()
  629. session.close()
  630. list.append(
  631. {
  632. "goods_art_no": item.goods_art_no,
  633. "action_time": item.create_time,
  634. "items": list_item,
  635. }
  636. )
  637. session.close()
  638. print("循环查询 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  639. return {
  640. "code": 0,
  641. "msg": "",
  642. "data": {"list": list, "page": page, "size": size},
  643. }
  644. @app.get("/get_last_photo_record", description="获取最后一条拍照记录")
  645. def get_last_photo_record():
  646. session = SqlQuery()
  647. statement = (
  648. select(PhotoRecord)
  649. .where(PhotoRecord.image_path != None)
  650. .order_by(desc("photo_create_time"))
  651. )
  652. result = session.exec(statement).first()
  653. session.close()
  654. return {
  655. "code": 0,
  656. "msg": "",
  657. "data": result,
  658. }
  659. @app.get("/get_photo_record_detail", description="通过货号获取拍照记录详情")
  660. def get_photo_records(goods_art_no: str = None):
  661. if goods_art_no == None:
  662. return {"code": 1, "msg": "参数错误", "data": None}
  663. session = SqlQuery()
  664. photos = CRUD(PhotoRecord)
  665. items = photos.read_all(session, conditions={"goods_art_no": goods_art_no})
  666. session.close()
  667. return {
  668. "code": 0,
  669. "msg": "",
  670. "data": {"list": items},
  671. }
  672. @app.post("/delect_goods_arts", description="通过货号删除记录")
  673. def delect_goods_arts(params: PhotoRecordDelete):
  674. session = SqlQuery()
  675. photos = CRUD(PhotoRecord)
  676. for item in params.goods_art_nos:
  677. photos.deleteConditions(session, conditions={"goods_art_no": item})
  678. session.close()
  679. return {
  680. "code": 0,
  681. "msg": "操作成功",
  682. "data": None,
  683. }
  684. @app.get("/get_sys_config", description="查询系统配置")
  685. def get_sys_config(key: str = None):
  686. if key == None:
  687. return {"code": 1, "msg": "参数错误", "data": None}
  688. session = SqlQuery()
  689. photos = CRUD(SysConfigs)
  690. item = photos.read(session, conditions={"key": key})
  691. session.close()
  692. return {
  693. "code": 0,
  694. "msg": "",
  695. "data": json.loads(item.value),
  696. }
  697. @app.post("/update_left_right_config", description="更新左右脚配置")
  698. def update_left_right_config(params: LeftRightParams):
  699. session = SqlQuery()
  700. sysConfig = CRUD(SysConfigs)
  701. model = sysConfig.read(session, conditions={"key": "action_configs"})
  702. if model == None:
  703. return {"code": 1, "msg": "配置不存在", "data": None}
  704. config_value = json.loads(model.value)
  705. config_value[params.type] = params.id
  706. update_value = json.dumps(config_value)
  707. # 走编辑逻辑
  708. kwargs = {"key": "action_configs", "value": update_value}
  709. save_device_config = sysConfig.updateConditions(
  710. session, conditions={"key": "action_configs"}, **kwargs
  711. )
  712. # return {"code": 0, "msg": "操作成功", "data": save_device_config}
  713. return {"code": 0, "msg": "操作成功", "data": None}
  714. @app.post("/update_sys_configs", description="创建或修改系统配置")
  715. def save_sys_configs(params: SysConfigParams):
  716. session = SqlQuery()
  717. sysConfig = CRUD(SysConfigs)
  718. model = sysConfig.read(session, conditions={"key": params.key})
  719. if model == None:
  720. return {"code": 1, "msg": "配置不存在", "data": None}
  721. # 走编辑逻辑
  722. kwargs = params.__dict__
  723. save_device_config = sysConfig.updateConditions(
  724. session, conditions={"key": params.key}, **kwargs
  725. )
  726. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  727. @app.post("/create_main_image", description="创建主图测试")
  728. def create_main_image(params: MaineImageTest):
  729. file_path = params.file_path
  730. onePic = OnePicTest(pic_path=file_path)
  731. # session = SqlQuery()
  732. # sysConfig = CRUD(SysConfigs)
  733. # model = sysConfig.read(session, conditions={"key": params.key})
  734. # if model == None:
  735. # return {"code": 1, "msg": "配置不存在", "data": None}
  736. # # 走编辑逻辑
  737. # kwargs = params.__dict__
  738. # save_device_config = sysConfig.updateConditions(
  739. # session, conditions={"key": params.key}, **kwargs
  740. # )
  741. main_out_path = onePic.HandlerMainImage()
  742. return {"code": 0, "msg": "操作成功", "data": {"main_out_path": main_out_path}}
  743. def insertEmptyLogoList(session):
  744. """插入空logo列表"""
  745. data = {"key": "logo_configs", "value": "[]"}
  746. config = SysConfigs(**data)
  747. session.add(config)
  748. session.commit()
  749. session.close()
  750. item = SysConfigs()
  751. item.key = "logo_configs"
  752. item.value = "[]"
  753. return item
  754. @app.get("/logo_list", description="logo列表")
  755. def logo_list():
  756. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  757. check_path(logo_dir)
  758. logo_files = os.listdir(logo_dir)
  759. logo_list = []
  760. for logoItem in logo_files:
  761. logo_list.append(f"{logo_dir}{logoItem}")
  762. return {"code": 0, "msg": "操作成功", "data": logo_list}
  763. @app.post("/add_logo", description="添加logo")
  764. def add_logo(params: LogoParams):
  765. logo_path = params.logo_path.replace("\\", "/")
  766. session = SqlQuery()
  767. sysConfig = CRUD(SysConfigs)
  768. item = sysConfig.read(session, conditions={"key": "logo_configs"})
  769. if item == None:
  770. item = insertEmptyLogoList(session)
  771. if os.path.isfile(logo_path) == False:
  772. return {"code": 1, "msg": "logo文件不存在", "data": None}
  773. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  774. check_path(logo_dir)
  775. fpath, fname = os.path.split(logo_path)
  776. logo_path_info = logo_dir + fname
  777. shutil.copy(logo_path, logo_path_info) # 复制文件
  778. logo_files = os.listdir(logo_dir)
  779. logo_list = []
  780. for logoItem in logo_files:
  781. logo_list.append(f"{logo_dir}{logoItem}")
  782. return {
  783. "code": 0,
  784. "msg": "",
  785. "data": {"logo": logo_path_info},
  786. }
  787. @app.post("/delete_logo", description="删除logo")
  788. def delete_logo(params: LogoParamsupdate):
  789. logo_path = params.path
  790. if os.path.isfile(logo_path) == False:
  791. return {"code": 1, "msg": "logo文件不存在", "data": None}
  792. os.remove(logo_path)
  793. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  794. check_path(logo_dir)
  795. logo_files = os.listdir(logo_dir)
  796. logo_list = []
  797. for logoItem in logo_files:
  798. logo_list.append(f"{logo_dir}{logoItem}")
  799. return {"code": 0, "msg": "操作成功", "data": logo_list}
  800. @app.post("/close_other_window", description="关闭窗口")
  801. def close_other_window():
  802. hwnd_list = []
  803. def callback(hwnd, _):
  804. title = GetWindowText(hwnd)
  805. if title == "digiCamControl by Duka Istvan":
  806. hwnd_list.append(hwnd)
  807. EnumWindows(callback, None)
  808. if hwnd_list:
  809. hwnd = hwnd_list[0]
  810. win32gui.PostMessage(hwnd, win32con.WM_CLOSE, 0, 0)
  811. return {"code": 0, "msg": "关闭成功", "data": {"status": True}}
  812. return {"code": 0, "msg": "关闭失败", "data": {"status": False}}