api.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840
  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. # 可用模板列表-需要导包
  101. temp_name_list = []
  102. for tempItem in params.temp_list:
  103. temp_class[tempItem.template_id] = tempItem.template_local_classes
  104. temp_name_list.append(tempItem.template_id)
  105. # 抠图时用到的货号列表,与生成详情图有所区别
  106. goods_art_no_arrays = []
  107. # # 详情图生成需要对同款商品进行分组,保证详情图可以生成多个色
  108. goods_art_no_group_arrays = []
  109. for _, goods_row in excel_df.iterrows():
  110. goods_art_no = str(goods_row["商品货号"])
  111. goods_art_no_arrays.append(goods_art_no)
  112. goods_no = str(goods_row["款号"])
  113. a001_df = goods_art_dirs.get_group(goods_no)
  114. goods_art_groups = a001_df["商品货号"].tolist()
  115. if goods_art_groups in goods_art_no_group_arrays:
  116. continue
  117. goods_art_no_group_arrays.append(goods_art_groups)
  118. limit_path = "output/{}".format(
  119. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  120. )
  121. # 该数组表示是否需要后面的移动文件夹操作,减少重复抠图,提升抠图时间和速度
  122. move_folder_array = check_move_goods_art_no_folder(
  123. "output", goods_art_no_arrays, limit_path
  124. )
  125. for index, row in excel_df.iterrows():
  126. goods_art_no_image_dir = str(row["文件夹名称"])
  127. goods_art_no = str(row["商品货号"])
  128. print("货号数据", goods_art_no)
  129. if not goods_art_no:
  130. raise UnicornException("货号不能为空")
  131. session = SqlQuery()
  132. pr = CRUD(PhotoRecord)
  133. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  134. if not images:
  135. raise UnicornException("没有可用货号数据")
  136. # 货号目录不存在再去进行移动和创建操作
  137. if move_folder_array.get(goods_art_no) == None:
  138. image_dir = "{}/data/{}".format(os.getcwd(), goods_art_no).replace(
  139. "\\", "/"
  140. )
  141. check_path(image_dir)
  142. for idx, itemImg in enumerate(images):
  143. if itemImg.image_path == "" or itemImg.image_path == None:
  144. raise UnicornException(
  145. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  146. )
  147. new_file_name = str(itemImg.goods_art_no) + "_" + str(idx) + ".jpg"
  148. if not os.path.exists(
  149. image_dir + "/" + os.path.basename(new_file_name)
  150. ):
  151. shutil.copy(itemImg.image_path, image_dir + new_file_name)
  152. dealImage = DealImage(image_dir)
  153. resFlag, path = dealImage.dealMoveImage(
  154. image_dir=image_dir,
  155. callback_func=None,
  156. goods_art_no=goods_art_no_image_dir,
  157. )
  158. if not resFlag:
  159. raise UnicornException(path)
  160. shutil.rmtree(image_dir)
  161. # path = os.path.dirname(path)
  162. config_data = {
  163. "image_dir": limit_path, # 这个目录作为本次生成的图片目录非常重要 例:./output/当前日期
  164. "image_order": params.template_image_order,
  165. "goods_art_no": goods_art_no,
  166. "goods_art_nos": goods_art_no_arrays,
  167. "is_check_number": False,
  168. "resize_image_view": "后跟",
  169. "cutout_mode": settings.CUTOUT_MODE,
  170. "logo_path": params.logo_path,
  171. "special_goods_art_no_folder_line": "",
  172. "is_use_excel": (False if params.excel_path == "" else True), # 是否使用excel
  173. "excel_path": params.excel_path, # excel路径
  174. "is_check_color_is_all": False,
  175. "cutout_is_pass": True,
  176. "assigned_page_dict": {},
  177. "detail_is_pass": True,
  178. "upload_is_pass": False,
  179. "upload_is_enable": False,
  180. "is_filter": False,
  181. "temp_class": temp_class,
  182. "temp_name": params.temp_name,
  183. "temp_name_list": temp_name_list,
  184. "target_error_folder": f"{limit_path}/软件-生成详情错误",
  185. "success_handler": [],
  186. }
  187. # 动态导入类
  188. temp_class_dict = {}
  189. for key, class_path in config_data["temp_class"].items():
  190. module_path, class_name = class_path.rsplit(".", 1)
  191. module = importlib.import_module(module_path)
  192. cls = getattr(module, class_name)
  193. temp_class_dict[key] = cls
  194. config_data["temp_class"] = temp_class_dict
  195. # 此处对抠图进行批量处理,保证所有的图片在生成详情图之前已经完成抠图,以保证详情图生成的效率
  196. return_data = run_main.check_before_cutout(config_data)
  197. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  198. check_for_detail_first_res = None
  199. try:
  200. if cutout_res:
  201. return_data_check_before_detail = run_main.check_before_detail(config_data)
  202. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  203. return_data_check_before_detail
  204. )
  205. if isinstance(check_for_detail_first_res, partial):
  206. result = check_for_detail_first_res()
  207. try:
  208. config_data = result["config_data"]
  209. except:
  210. config_data = result
  211. if config_data["sign_text"] == "已结束详情处理":
  212. print("config_data", config_data)
  213. if config_data["upload_is_enable"]:
  214. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  215. check_path(to_deal_dir)
  216. print("to_deal_dir", to_deal_dir)
  217. if os.path.exists(to_deal_dir):
  218. upload_pic = UploadPic(
  219. windows=None,
  220. to_deal_dir=to_deal_dir,
  221. config_data=config_data,
  222. token=token,
  223. )
  224. upload_pic.run()
  225. out_put_dir = config_data["out_put_dir"]
  226. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  227. "\\", "/"
  228. )
  229. handler_result_folder = os.path.dirname(out_put_dir_path)
  230. if len(config_data["success_handler"])==0:
  231. for good_art in goods_art_no_arrays:
  232. handler_result.append(
  233. {"goods_art_no": good_art, "success": False, "info": "处理失败"}
  234. )
  235. else:
  236. handler_result = config_data["success_handler"]
  237. else:
  238. for good_art in goods_art_no_arrays:
  239. handler_result.append(
  240. {"goods_art_no": good_art, "success": False, "info": "处理失败"}
  241. )
  242. except Exception as e:
  243. for good_art in goods_art_no_arrays:
  244. handler_result.append(
  245. {"goods_art_no": good_art, "success": False, "info": str(e)}
  246. )
  247. handler_result_folder = "/".join(handler_result_folder.split("/")[:-1])
  248. return {
  249. "code": 0,
  250. "msg": "",
  251. "data": {"output_folder": handler_result_folder, "list": handler_result},
  252. }
  253. def group_by_style_number(data):
  254. result = {}
  255. for goods_id, info in data.items():
  256. style_number = info["款号"]
  257. if style_number not in result:
  258. result[style_number] = []
  259. result[style_number].append(goods_id)
  260. return result
  261. @app.post("/handle_detail")
  262. async def handle_detail(request: Request, params: HandlerDetail):
  263. obj = None
  264. token = "Bearer " + params.token
  265. run_main = RunMain(obj, token)
  266. goods_art_no_arrays = params.goods_art_no
  267. is_only_cutout = params.is_only_cutout
  268. handler_result = []
  269. handler_result_folder = ""
  270. if params.excel_path != "" and params.excel_path != None:
  271. return fromExcelHandler(params)
  272. # goods_art_list = run_main.data_mode_generate_detail.get_goods_art_no_info(
  273. # goods_art_list=goods_art_no_arrays
  274. # )
  275. # goods_art_nos = []
  276. # goods_art_list_dict = group_by_style_number(goods_art_list)
  277. path = ""
  278. limit_path = "output/{}".format(
  279. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  280. )
  281. # 该数组表示是否需要后面的移动文件夹操作,减少重复抠图,提升抠图时间和速度
  282. move_folder_array = check_move_goods_art_no_folder(
  283. "output", goods_art_no_arrays, limit_path
  284. )
  285. for goods_art_no in goods_art_no_arrays:
  286. if not goods_art_no:
  287. raise UnicornException("货号不能为空")
  288. session = SqlQuery()
  289. pr = CRUD(PhotoRecord)
  290. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  291. if not images:
  292. raise UnicornException("没有可用货号数据")
  293. if is_only_cutout != 1:
  294. detail_counts = len(params.template_image_order.split(","))
  295. image_counts = len(images)
  296. if image_counts < detail_counts:
  297. raise UnicornException(
  298. f"货号:[{goods_art_no}],实际照片数量:{image_counts}张,小于详情图要求数量:{detail_counts}张"
  299. )
  300. if move_folder_array.get(goods_art_no) == None:
  301. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  302. check_path(image_dir)
  303. for idx, itemImg in enumerate(images):
  304. if itemImg.image_path == "" or itemImg.image_path == None:
  305. raise UnicornException(
  306. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  307. )
  308. new_file_name = str(itemImg.goods_art_no) + "_" + str(idx) + ".jpg"
  309. if not os.path.exists(
  310. image_dir + "/" + os.path.basename(new_file_name)
  311. ):
  312. shutil.copy(itemImg.image_path, image_dir + new_file_name)
  313. dealImage = DealImage(image_dir)
  314. resFlag, path = dealImage.dealMoveImage(
  315. image_dir=image_dir, callback_func=None, goods_art_no=goods_art_no
  316. )
  317. if not resFlag:
  318. raise UnicornException(path)
  319. # try:
  320. temp_class = {}
  321. temp_name_list = []
  322. for tempItem in params.temp_list:
  323. temp_class[tempItem.template_id] = tempItem.template_local_classes
  324. temp_name_list.append(tempItem.template_id)
  325. config_data = {
  326. "image_dir": limit_path,
  327. "image_order": (
  328. "俯视,侧视,后跟,鞋底,内里"
  329. if params.template_image_order == None
  330. or params.template_image_order == ""
  331. else params.template_image_order
  332. ),
  333. "goods_art_no": "",
  334. "goods_art_nos": goods_art_no_arrays,
  335. "is_check_number": False,
  336. "resize_image_view": "后跟",
  337. "cutout_mode": settings.CUTOUT_MODE,
  338. "logo_path": params.logo_path,
  339. "special_goods_art_no_folder_line": "",
  340. "is_use_excel": False, # 是否使用excel
  341. "excel_path": "", # excel路径
  342. "is_check_color_is_all": False,
  343. "cutout_is_pass": True,
  344. "assigned_page_dict": {},
  345. "detail_is_pass": True,
  346. "upload_is_pass": False,
  347. "upload_is_enable": settings.IS_UPLOAD_HLM, # 是否上传到惠利玛商品库,通过config.ini得is_upload开启
  348. "is_filter": False,
  349. "temp_class": temp_class,
  350. "temp_name": params.temp_name,
  351. "temp_name_list": temp_name_list,
  352. "target_error_folder": f"{limit_path}/软件-生成详情错误",
  353. "success_handler": [],
  354. }
  355. print("image_dir=====>>>>>", config_data["image_dir"])
  356. # 动态导入类
  357. temp_class_dict = {}
  358. for key, class_path in config_data["temp_class"].items():
  359. module_path, class_name = class_path.rsplit(".", 1)
  360. module = importlib.import_module(module_path)
  361. cls = getattr(module, class_name)
  362. temp_class_dict[key] = cls
  363. config_data["temp_class"] = temp_class_dict
  364. return_data = run_main.check_before_cutout(config_data)
  365. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  366. check_for_detail_first_res = None
  367. if cutout_res == True:
  368. sys_path = format(os.getcwd()).replace("\\", "/")
  369. handler_result_folder = f"{sys_path}/{limit_path}"
  370. for goods_art_item in goods_art_no_arrays:
  371. handler_result.append(
  372. {
  373. "goods_art_no": goods_art_item,
  374. "success": True,
  375. "info": "处理成功",
  376. }
  377. )
  378. if is_only_cutout == 1:
  379. return {
  380. "code": 0,
  381. "msg": "",
  382. "data": {"output_folder": handler_result_folder, "list": handler_result},
  383. }
  384. handler_result = []
  385. try:
  386. return_data_check_before_detail = run_main.check_before_detail(config_data)
  387. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  388. return_data_check_before_detail
  389. )
  390. if isinstance(check_for_detail_first_res, partial):
  391. result = check_for_detail_first_res()
  392. try:
  393. config_data = result["config_data"]
  394. except:
  395. config_data = result
  396. if config_data["sign_text"] == "已结束详情处理":
  397. # at_pic = AutoDealPics()
  398. print("config_data", config_data)
  399. # if config_data["upload_is_enable"]:
  400. # to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  401. # check_path(to_deal_dir)
  402. # print("to_deal_dir", to_deal_dir)
  403. # if os.path.exists(to_deal_dir):
  404. # upload_pic = UploadPic(
  405. # windows=None,
  406. # to_deal_dir=to_deal_dir,
  407. # config_data=config_data,
  408. # token=token,
  409. # )
  410. # upload_pic.run()
  411. out_put_dir = config_data.get("out_put_dir")
  412. if out_put_dir:
  413. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace(
  414. "\\", "/"
  415. )
  416. handler_result_folder = os.path.dirname(out_put_dir_path)
  417. else:
  418. handler_result_folder = ""
  419. if len(config_data["success_handler"]) == 0:
  420. for good_art in goods_art_no_arrays:
  421. handler_result.append(
  422. {"goods_art_no": good_art, "success": False, "info": "处理失败"}
  423. )
  424. else:
  425. handler_result = config_data["success_handler"]
  426. else:
  427. handler_result.append(
  428. {"goods_art_no": "", "success": False, "info": "处理失败"}
  429. )
  430. except Exception as e:
  431. print(e.__traceback__.tb_frame.f_globals["__file__"]) # 发生异常所在的文件
  432. print(e.__traceback__.tb_lineno) # 发生异常所在的行数
  433. handler_result.append({"goods_art_no": "", "success": False, "info": str(e)})
  434. return {
  435. "code": 0,
  436. "msg": "",
  437. "data": {"output_folder": handler_result_folder, "list": handler_result},
  438. }
  439. @app.get("/get_device_tabs", description="获取可执行程序命令列表")
  440. def get_device_tabs(type: int):
  441. session = SqlQuery()
  442. statement = (
  443. select(DeviceConfigTabs)
  444. .where(DeviceConfigTabs.mode_type == type)
  445. .order_by(asc("id"))
  446. )
  447. result = session.exec(statement).all()
  448. session.close()
  449. sys = CRUD(SysConfigs)
  450. action_configs = sys.read(session, conditions={"key": "action_configs"})
  451. session.close()
  452. return {
  453. "code": 0,
  454. "msg": "",
  455. "data": {"tabs": result, "select_configs": json.loads(action_configs.value)},
  456. }
  457. @app.post("/update_tab_name", description="更改tab名称")
  458. def update_tab_name(params: DeviceConfigTabsReq):
  459. if params.mode_name == "":
  460. return {"code": 1, "msg": "名称不能为空", "data": {}}
  461. session = SqlQuery()
  462. tabModel = CRUD(DeviceConfigTabs)
  463. kwargs = {"mode_name": params.mode_name}
  464. tabModel.updateConditions(session, conditions={"id": params.id}, **kwargs)
  465. session.close()
  466. return {
  467. "code": 0,
  468. "msg": "",
  469. "data": None,
  470. }
  471. @app.post("/get_device_configs", description="获取可执行程序命令列表")
  472. def get_device_configs(params: ModelGetDeviceConfig):
  473. tab_id = params.tab_id
  474. session = SqlQuery()
  475. configModel = CRUD(DeviceConfig)
  476. configList = configModel.read_all(
  477. session,
  478. conditions={"tab_id": tab_id},
  479. order_by="action_index",
  480. ascending=True,
  481. )
  482. return {
  483. "code": 0,
  484. "msg": "",
  485. "data": {"list": configList},
  486. }
  487. @app.post("/device_config_detail", description="获取可执行程序详情")
  488. def device_config_detail(params: ModelGetDeviceConfigDetail):
  489. action_id = params.id
  490. session = SqlQuery()
  491. configModel = CRUD(DeviceConfig)
  492. model = configModel.read(session, conditions={"id": action_id})
  493. if model == None:
  494. return {"code": 1, "msg": "数据不存在", "data": None}
  495. return {"code": 0, "msg": "", "data": model}
  496. @app.post("/device_config_detail_query", description="通过条件获取可执行程序详情")
  497. def device_config_detail_query():
  498. # tab_id = params.tab_id
  499. # action_name = params.action_name
  500. session = SqlQuery()
  501. sys = CRUD(SysConfigs)
  502. action_configs = sys.read(session, conditions={"key": "action_configs"})
  503. action_configs_value = json.loads(action_configs.value)
  504. left_config = action_configs_value.get("left")
  505. configModel = CRUD(DeviceConfig)
  506. model = configModel.read(
  507. session, conditions={"tab_id": left_config, "action_name": "侧视"}
  508. )
  509. if model == None:
  510. model = configModel.read(session, conditions={"tab_id": left_config})
  511. return {"code": 0, "msg": "", "data": model}
  512. @app.post("/remove_config", description="删除一条可执行命令")
  513. def get_device_configs(params: ModelGetDeviceConfigDetail):
  514. action_id = params.id
  515. session = SqlQuery()
  516. configModel = CRUD(DeviceConfig)
  517. model = configModel.read(session, conditions={"id": action_id})
  518. if model == None:
  519. return {"code": 1, "msg": "数据不存在", "data": None}
  520. if model.is_system == True:
  521. return {"code": 1, "msg": "系统配置不允许删除", "data": None}
  522. configArray = configModel.read_all(session, conditions={"tab_id": model.tab_id})
  523. if len(configArray) == 1:
  524. return {"code": 1, "msg": "请至少保留一个配置", "data": None}
  525. configModel.delete(session, obj_id=action_id)
  526. return {"code": 0, "msg": "删除成功", "data": None}
  527. @app.post("/save_device_config", description="创建或修改一条可执行命令")
  528. def save_device_config(params: SaveDeviceConfig):
  529. action_id = params.id
  530. session = SqlQuery()
  531. deviceConfig = CRUD(DeviceConfig)
  532. if action_id == None or action_id == 0:
  533. # 走新增逻辑
  534. params.id = None
  535. save_device_config = deviceConfig.create(session, obj_in=params)
  536. else:
  537. model = deviceConfig.read(session, conditions={"id": action_id})
  538. if model == None:
  539. return {"code": 1, "msg": "数据不存在", "data": None}
  540. # 走编辑逻辑
  541. kwargs = params.__dict__
  542. save_device_config = deviceConfig.update(session, obj_id=action_id, **kwargs)
  543. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  544. @app.post("/reset_config", description="创建或修改一条可执行命令")
  545. def reset_config(params: ModelGetDeviceConfig):
  546. tab_id = params.tab_id
  547. if tab_id == None or tab_id == "":
  548. return {"code": 1, "msg": "参数错误", "data": None}
  549. session = SqlQuery()
  550. deviceConfig = CRUD(DeviceConfig)
  551. first_config = deviceConfig.read(session, conditions={"tab_id": tab_id})
  552. res = deviceConfig.deleteConditions(session, conditions={"tab_id": tab_id})
  553. if res is False:
  554. return {"code": 1, "msg": "操作失败", "data": None}
  555. actions = json.load(open("action.json", encoding="utf-8"))
  556. for data in actions:
  557. data["tab_id"] = tab_id
  558. data["is_system"] = first_config.is_system
  559. device_config = DeviceConfig(**data)
  560. session.add(device_config)
  561. session.commit()
  562. # session.close()
  563. return {"code": 0, "msg": "操作成功", "data": None}
  564. @app.get("/get_photo_records", description="获取拍照记录")
  565. def get_photo_records(page: int = 1, size: int = 5):
  566. session = SqlQuery()
  567. # photos = CRUD(PhotoRecord)
  568. print("准备查询拍摄记录", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  569. statement = (
  570. select(PhotoRecord)
  571. .offset((page - 1) * size)
  572. .limit(size)
  573. .order_by(desc("id"))
  574. .group_by("goods_art_no")
  575. )
  576. list = []
  577. result = session.exec(statement).all()
  578. print("group 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  579. join_conditions = [
  580. {
  581. "model": DeviceConfig,
  582. "on": PhotoRecord.action_id == DeviceConfig.id,
  583. "is_outer": False, # 可选,默认False,设为True则为LEFT JOIN
  584. }
  585. ]
  586. for item in result:
  587. query = (
  588. select(PhotoRecord, DeviceConfig.action_name)
  589. .where(PhotoRecord.goods_art_no == item.goods_art_no)
  590. .join(DeviceConfig, PhotoRecord.action_id == DeviceConfig.id)
  591. )
  592. list_item = session.exec(query).mappings().all()
  593. list.append(
  594. {
  595. "goods_art_no": item.goods_art_no,
  596. "action_time": item.create_time,
  597. "items": list_item,
  598. }
  599. )
  600. # session.close()
  601. print("循环查询 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  602. return {
  603. "code": 0,
  604. "msg": "",
  605. "data": {"list": list, "page": page, "size": size},
  606. }
  607. @app.get("/get_last_photo_record", description="获取最后一条拍照记录")
  608. def get_last_photo_record():
  609. session = SqlQuery()
  610. statement = (
  611. select(PhotoRecord)
  612. .where(PhotoRecord.image_path != None)
  613. .order_by(desc("photo_create_time"))
  614. )
  615. result = session.exec(statement).first()
  616. # session.close()
  617. return {
  618. "code": 0,
  619. "msg": "",
  620. "data": result,
  621. }
  622. @app.get("/get_photo_record_detail", description="通过货号获取拍照记录详情")
  623. def get_photo_record_detail(goods_art_no: str = None):
  624. if goods_art_no == None:
  625. return {"code": 1, "msg": "参数错误", "data": None}
  626. session = SqlQuery()
  627. photos = CRUD(PhotoRecord)
  628. items = photos.read_all(session, conditions={"goods_art_no": goods_art_no})
  629. # session.close()
  630. return {
  631. "code": 0,
  632. "msg": "",
  633. "data": {"list": items},
  634. }
  635. @app.post("/delect_goods_arts", description="通过货号删除记录")
  636. def delect_goods_arts(params: PhotoRecordDelete):
  637. session = SqlQuery()
  638. photos = CRUD(PhotoRecord)
  639. for item in params.goods_art_nos:
  640. photos.deleteConditions(session, conditions={"goods_art_no": item})
  641. # session.close()
  642. return {
  643. "code": 0,
  644. "msg": "操作成功",
  645. "data": None,
  646. }
  647. @app.get("/get_sys_config", description="查询系统配置")
  648. def get_sys_config(key: str = None):
  649. if key == None:
  650. return {"code": 1, "msg": "参数错误", "data": None}
  651. session = SqlQuery()
  652. photos = CRUD(SysConfigs)
  653. item = photos.read(session, conditions={"key": key})
  654. # session.close()
  655. return {
  656. "code": 0,
  657. "msg": "",
  658. "data": json.loads(item.value),
  659. }
  660. @app.post("/update_left_right_config", description="更新左右脚配置")
  661. def update_left_right_config(params: LeftRightParams):
  662. session = SqlQuery()
  663. sysConfig = CRUD(SysConfigs)
  664. model = sysConfig.read(session, conditions={"key": "action_configs"})
  665. if model == None:
  666. return {"code": 1, "msg": "配置不存在", "data": None}
  667. config_value = json.loads(model.value)
  668. config_value[params.type] = params.id
  669. update_value = json.dumps(config_value)
  670. # 走编辑逻辑
  671. kwargs = {"key": "action_configs", "value": update_value}
  672. save_device_config = sysConfig.updateConditions(
  673. session, conditions={"key": "action_configs"}, **kwargs
  674. )
  675. return {"code": 0, "msg": "操作成功", "data": None}
  676. @app.post("/update_record", description="更新拍照记录")
  677. def update_record(params: RecordUpdate):
  678. session = SqlQuery()
  679. photoRecord = CRUD(PhotoRecord)
  680. model = photoRecord.read(session, conditions={"id": params.id})
  681. if model == None:
  682. return {"code": 1, "msg": "记录不存在", "data": None}
  683. kwargs = params.__dict__
  684. save_device_config = photoRecord.update(session, obj_id=params.id, **kwargs)
  685. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  686. @app.post("/update_sys_configs", description="创建或修改系统配置")
  687. def save_sys_configs(params: SysConfigParams):
  688. session = SqlQuery()
  689. sysConfig = CRUD(SysConfigs)
  690. model = sysConfig.read(session, conditions={"key": params.key})
  691. if model == None:
  692. return {"code": 1, "msg": "配置不存在", "data": None}
  693. # 走编辑逻辑
  694. kwargs = params.__dict__
  695. save_device_config = sysConfig.updateConditions(
  696. session, conditions={"key": params.key}, **kwargs
  697. )
  698. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  699. @app.post("/create_main_image", description="创建主图测试")
  700. def create_main_image(params: MaineImageTest):
  701. file_path = params.file_path
  702. onePic = OnePicTest(pic_path=file_path)
  703. main_out_path = onePic.HandlerMainImage()
  704. return {"code": 0, "msg": "操作成功", "data": {"main_out_path": main_out_path}}
  705. def insertEmptyLogoList(session):
  706. """插入空logo列表"""
  707. data = {"key": "logo_configs", "value": "[]"}
  708. config = SysConfigs(**data)
  709. session.add(config)
  710. session.commit()
  711. session.close()
  712. item = SysConfigs()
  713. item.key = "logo_configs"
  714. item.value = "[]"
  715. return item
  716. @app.get("/logo_list", description="logo列表")
  717. def logo_list():
  718. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  719. check_path(logo_dir)
  720. logo_files = os.listdir(logo_dir)
  721. logo_list = []
  722. for logoItem in logo_files:
  723. logo_list.append(f"{logo_dir}{logoItem}")
  724. return {"code": 0, "msg": "操作成功", "data": logo_list}
  725. @app.post("/add_logo", description="添加logo")
  726. def add_logo(params: LogoParams):
  727. logo_path = params.logo_path.replace("\\", "/")
  728. session = SqlQuery()
  729. sysConfig = CRUD(SysConfigs)
  730. item = sysConfig.read(session, conditions={"key": "logo_configs"})
  731. if item == None:
  732. item = insertEmptyLogoList(session)
  733. if os.path.isfile(logo_path) == False:
  734. return {"code": 1, "msg": "logo文件不存在", "data": None}
  735. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  736. check_path(logo_dir)
  737. fpath, fname = os.path.split(logo_path)
  738. logo_path_info = logo_dir + fname
  739. shutil.copy(logo_path, logo_path_info) # 复制文件
  740. logo_files = os.listdir(logo_dir)
  741. logo_list = []
  742. for logoItem in logo_files:
  743. logo_list.append(f"{logo_dir}{logoItem}")
  744. return {
  745. "code": 0,
  746. "msg": "",
  747. "data": {"logo": logo_path_info},
  748. }
  749. @app.post("/delete_logo", description="删除logo")
  750. def delete_logo(params: LogoParamsupdate):
  751. logo_path = params.path
  752. if os.path.isfile(logo_path) == False:
  753. return {"code": 1, "msg": "logo文件不存在", "data": None}
  754. os.remove(logo_path)
  755. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  756. check_path(logo_dir)
  757. logo_files = os.listdir(logo_dir)
  758. logo_list = []
  759. for logoItem in logo_files:
  760. logo_list.append(f"{logo_dir}{logoItem}")
  761. return {"code": 0, "msg": "操作成功", "data": logo_list}
  762. @app.post("/close_other_window", description="关闭窗口")
  763. def close_other_window():
  764. hwnd_list = []
  765. def callback(hwnd, _):
  766. title = GetWindowText(hwnd)
  767. if title == "digiCamControl by Duka Istvan":
  768. hwnd_list.append(hwnd)
  769. EnumWindows(callback, None)
  770. if hwnd_list:
  771. hwnd = hwnd_list[0]
  772. win32gui.PostMessage(hwnd, win32con.WM_CLOSE, 0, 0)
  773. return {"code": 0, "msg": "关闭成功", "data": {"status": True}}
  774. return {"code": 0, "msg": "关闭失败", "data": {"status": False}}