api.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  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("/send_test")
  48. async def index():
  49. data = {"data1": 1, "data2": 2, "data3": 3, "data4": 4}
  50. await socket_manager.send_message(msg="测试", data=data)
  51. return {"message": "Hello World"}
  52. @app.get("/scan_serials", description="扫描可用的设备端口")
  53. async def scanSerials():
  54. """扫描串口"""
  55. ports = list_ports.comports()
  56. print("Scanning", ports)
  57. return {"message": "Hello World"}
  58. @app.get("/test_conndevice")
  59. def test_conndevice():
  60. device_control = DeviceControl()
  61. p_list = []
  62. temp_ports_dict = {}
  63. # while 1:
  64. time.sleep(1)
  65. ports_dict = device_control.scan_serial_port()
  66. temp_ports_dict = ports_dict
  67. if not ports_dict:
  68. # 全部清空 移除所有串口
  69. if p_list:
  70. _p = p_list.pop()
  71. device_control.remove_port(_p)
  72. # continue
  73. if ports_dict:
  74. # print(plist)
  75. for index, _i in enumerate(p_list):
  76. if _i not in ports_dict:
  77. _p = p_list.pop(index)
  78. device_control.remove_port(_p)
  79. for _port_name, _port_value in ports_dict.items():
  80. if _port_name not in p_list:
  81. try:
  82. p_list.append(_port_name)
  83. device_control.add_port_by_linkage(_port_name)
  84. except BaseException as e:
  85. print(e.__traceback__.tb_frame.f_globals["__file__"]) # 发生异常所在的文件
  86. print(e.__traceback__.tb_lineno) # 发生异常所在的行数
  87. print("串口不存在{} {}".format(_port_name, e))
  88. # threading.Thread(target=self.add_port, args=(_port_name, _port_value)).start()
  89. # self.add_port(_p)
  90. @app.api_route("/forward_request", methods=["GET", "POST"], description="代理转发hlm项目得请求")
  91. async def forwardRequest(request: HlmForwardRequest):
  92. """
  93. 转发HTTP请求到目标URL
  94. :param request: FastAPI Request对象
  95. :return: 目标接口的响应
  96. """
  97. try:
  98. if request.method == "GET":
  99. params = request.query_params
  100. elif request.method == "POST":
  101. params = json.dump(request.query_params)
  102. else:
  103. raise UnicornException("仅支持GET和POST方法")
  104. target_url = request.target_url
  105. method = request.method.upper()
  106. headers = request.headers
  107. if not target_url:
  108. raise UnicornException("目标url地址不能为空")
  109. # 调用 hlm_http_request 中的 forward_request 函数
  110. response = forward_request(target_url, params=params, method=method, headers=headers)
  111. return response
  112. except requests.RequestException as e:
  113. raise UnicornException(e)
  114. except Exception as e:
  115. raise UnicornException(e)
  116. def fromExcelHandler(params: HandlerDetail):
  117. excel_path = params.excel_path
  118. excel_df = pd.read_excel(excel_path, sheet_name=0, header=0)
  119. handler_result = []
  120. handler_result_folder = ""
  121. if "文件夹名称" not in excel_df.columns:
  122. raise UnicornException("缺失 [文件夹名称] 列")
  123. if "商品货号" not in excel_df.columns:
  124. raise UnicornException("缺失 [商品货号] 列")
  125. for index, row in excel_df.iterrows():
  126. goods_art_no_image_dir = str(row["文件夹名称"])
  127. goods_art_no = str(row["商品货号"])
  128. try:
  129. if not goods_art_no:
  130. raise UnicornException("货号不能为空")
  131. token = "Bearer " + params.token
  132. session = SqlQuery()
  133. pr = CRUD(PhotoRecord)
  134. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  135. if not images:
  136. raise UnicornException("没有可用货号数据")
  137. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  138. check_path(image_dir)
  139. for itemImg in images:
  140. if itemImg.image_path == "" or itemImg.image_path == None:
  141. raise UnicornException(
  142. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  143. )
  144. if not os.path.exists(image_dir + "/" + os.path.basename(itemImg.image_path)):
  145. shutil.copy(itemImg.image_path, image_dir)
  146. dealImage = DealImage(image_dir)
  147. resFlag, path = dealImage.dealMoveImage(
  148. image_dir=image_dir,
  149. callback_func=None,
  150. goods_art_no=goods_art_no_image_dir,
  151. )
  152. if not resFlag:
  153. raise UnicornException(path)
  154. temp_class = {}
  155. temp_name_list = []
  156. for tempItem in params.temp_list:
  157. temp_class[tempItem.template_id] = tempItem.template_local_classes
  158. temp_name_list.append(tempItem.template_id)
  159. # if goods_art_no_image_dir not in path:
  160. # path = path + "/" + goods_art_no_image_dir
  161. config_data = {
  162. "image_dir": path,
  163. "image_order": params.template_image_order,
  164. "goods_art_no": goods_art_no,
  165. "is_check_number": False,
  166. "resize_image_view": "后跟",
  167. "cutout_mode": settings.CUTOUT_MODE,
  168. "logo_path": params.logo_path,
  169. "special_goods_art_no_folder_line": "",
  170. "is_use_excel": (
  171. False if params.excel_path == "" else True
  172. ), # 是否使用excel
  173. "excel_path": params.excel_path, # excel路径
  174. "is_check_color_is_all": False,
  175. "cutout_is_pass": False,
  176. "assigned_page_dict": {},
  177. "detail_is_pass": False,
  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"{path}/软件-生成详情错误",
  185. }
  186. # 动态导入类
  187. temp_class_dict = {}
  188. for key, class_path in config_data["temp_class"].items():
  189. module_path, class_name = class_path.rsplit(".", 1)
  190. module = importlib.import_module(module_path)
  191. cls = getattr(module, class_name)
  192. temp_class_dict[key] = cls
  193. config_data["temp_class"] = temp_class_dict
  194. obj = None
  195. run_main = RunMain(obj, token)
  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. if cutout_res == True:
  200. return_data_check_before_detail = run_main.check_before_detail(config_data)
  201. print(
  202. "return_data_check_before_detail======> 测试 ==>",
  203. return_data_check_before_detail,
  204. )
  205. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  206. return_data_check_before_detail
  207. )
  208. if isinstance(check_for_detail_first_res, partial):
  209. result = check_for_detail_first_res()
  210. try:
  211. config_data = result["config_data"]
  212. except:
  213. config_data = result
  214. if config_data["sign_text"] == "已结束详情处理":
  215. # at_pic = AutoDealPics()
  216. print("config_data", config_data)
  217. if config_data["upload_is_enable"]:
  218. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  219. check_path(to_deal_dir)
  220. print("to_deal_dir", to_deal_dir)
  221. if os.path.exists(to_deal_dir):
  222. upload_pic = UploadPic(
  223. windows=None,
  224. to_deal_dir=to_deal_dir,
  225. config_data=config_data,
  226. token=token,
  227. )
  228. upload_pic.run()
  229. out_put_dir = config_data["out_put_dir"]
  230. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace("\\", "/")
  231. handler_result_folder = os.path.dirname(out_put_dir_path)
  232. handler_result.append({"goods_art_no": goods_art_no, "success": True, "info": "处理成功"})
  233. else:
  234. handler_result.append({"goods_art_no": goods_art_no, "success": False, "info": "处理失败"})
  235. except Exception as e:
  236. handler_result.append({"goods_art_no": goods_art_no, "success": False, "info": str(e)})
  237. handler_result_folder = "/".join(handler_result_folder.split("/")[:-1])
  238. return {
  239. "code": 0,
  240. "msg": "",
  241. "data": {"output_folder": handler_result_folder, "list": handler_result},
  242. }
  243. @app.post("/handle_detail")
  244. async def handle_detail(request: Request, params: HandlerDetail):
  245. goods_art_no_array = params.goods_art_no
  246. handler_result = []
  247. handler_result_folder = ""
  248. if params.excel_path != "" and params.excel_path != None:
  249. return fromExcelHandler(params)
  250. for goods_art_no in goods_art_no_array:
  251. try:
  252. if not goods_art_no:
  253. raise UnicornException("货号不能为空")
  254. token = "Bearer " + params.token
  255. session = SqlQuery()
  256. pr = CRUD(PhotoRecord)
  257. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no})
  258. if not images:
  259. raise UnicornException("没有可用货号数据")
  260. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  261. check_path(image_dir)
  262. for itemImg in images:
  263. if itemImg.image_path == "" or itemImg.image_path == None:
  264. raise UnicornException(
  265. f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试"
  266. )
  267. if not os.path.exists(image_dir + "/" + os.path.basename(itemImg.image_path)):
  268. shutil.copy(itemImg.image_path, image_dir)
  269. dealImage = DealImage(image_dir)
  270. resFlag, path = dealImage.dealMoveImage(
  271. image_dir=image_dir, callback_func=None, goods_art_no=goods_art_no
  272. )
  273. if not resFlag:
  274. raise UnicornException(path)
  275. temp_class = {}
  276. temp_name_list = []
  277. for tempItem in params.temp_list:
  278. temp_class[tempItem.template_id] = tempItem.template_local_classes
  279. temp_name_list.append(tempItem.template_id)
  280. config_data = {
  281. "image_dir": path,
  282. "image_order": params.template_image_order,
  283. "goods_art_no": goods_art_no,
  284. "is_check_number": False,
  285. "resize_image_view": "后跟",
  286. "cutout_mode": settings.CUTOUT_MODE,
  287. "logo_path": params.logo_path,
  288. "special_goods_art_no_folder_line": "",
  289. "is_use_excel": (
  290. False if params.excel_path == "" else True
  291. ), # 是否使用excel
  292. "excel_path": params.excel_path, # excel路径
  293. "is_check_color_is_all": False,
  294. "cutout_is_pass": False,
  295. "assigned_page_dict": {},
  296. "detail_is_pass": False,
  297. "upload_is_pass": False,
  298. "upload_is_enable": False,
  299. "is_filter": False,
  300. "temp_class": temp_class,
  301. "temp_name": params.temp_name,
  302. "temp_name_list": temp_name_list,
  303. "target_error_folder": f"{path}/软件-生成详情错误",
  304. }
  305. # 动态导入类
  306. temp_class_dict = {}
  307. for key, class_path in config_data["temp_class"].items():
  308. module_path, class_name = class_path.rsplit(".", 1)
  309. module = importlib.import_module(module_path)
  310. cls = getattr(module, class_name)
  311. temp_class_dict[key] = cls
  312. config_data["temp_class"] = temp_class_dict
  313. obj = None
  314. run_main = RunMain(obj, token)
  315. return_data = run_main.check_before_cutout(config_data)
  316. cutout_res = run_main.check_for_cutout_image_first_call_back(return_data)
  317. check_for_detail_first_res = None
  318. if cutout_res == True:
  319. return_data_check_before_detail = run_main.check_before_detail(config_data)
  320. print(
  321. "return_data_check_before_detail======> 测试 ==>",
  322. return_data_check_before_detail,
  323. )
  324. check_for_detail_first_res = run_main.check_for_detail_first_call_back(
  325. return_data_check_before_detail
  326. )
  327. if isinstance(check_for_detail_first_res, partial):
  328. result = check_for_detail_first_res()
  329. try:
  330. config_data = result["config_data"]
  331. except:
  332. config_data = result
  333. if config_data["sign_text"] == "已结束详情处理":
  334. # at_pic = AutoDealPics()
  335. print("config_data", config_data)
  336. if config_data["upload_is_enable"]:
  337. to_deal_dir = "{}/软件-详情图生成".format(config_data["image_dir"])
  338. check_path(to_deal_dir)
  339. print("to_deal_dir", to_deal_dir)
  340. if os.path.exists(to_deal_dir):
  341. upload_pic = UploadPic(
  342. windows=None,
  343. to_deal_dir=to_deal_dir,
  344. config_data=config_data,
  345. token=token,
  346. )
  347. upload_pic.run()
  348. out_put_dir = config_data["out_put_dir"]
  349. out_put_dir_path = "{}/{}".format(os.getcwd(), out_put_dir).replace("\\", "/")
  350. handler_result_folder = os.path.dirname(out_put_dir_path)
  351. handler_result.append({"goods_art_no": goods_art_no, "success": True, "info": "处理成功"})
  352. else:
  353. handler_result.append({"goods_art_no": goods_art_no, "success": False, "info": "处理失败"})
  354. except Exception as e:
  355. handler_result.append({"goods_art_no": goods_art_no, "success": False, "info": str(e)})
  356. return {
  357. "code": 0,
  358. "msg": "",
  359. "data": {"output_folder": handler_result_folder, "list": handler_result},
  360. }
  361. @app.get("/get_device_tabs", description="获取可执行程序命令列表")
  362. def get_device_tabs(type:int):
  363. session = SqlQuery()
  364. statement = (
  365. select(DeviceConfigTabs)
  366. .where(DeviceConfigTabs.mode_type == type).order_by(asc("id"))
  367. )
  368. result = session.exec(statement).all()
  369. session.close()
  370. sys = CRUD(SysConfigs)
  371. action_configs = sys.read(session, conditions={"key": "action_configs"})
  372. session.close()
  373. return {
  374. "code": 0,
  375. "msg": "",
  376. "data": {"tabs": result, "select_configs": json.loads(action_configs.value)},
  377. }
  378. @app.post("/update_tab_name", description="更改tab名称")
  379. def update_tab_name(params: DeviceConfigTabsReq):
  380. if params.mode_name == "":
  381. return {"code": 1, "msg": "名称不能为空", "data": {}}
  382. session = SqlQuery()
  383. tabModel = CRUD(DeviceConfigTabs)
  384. kwargs = {"mode_name": params.mode_name}
  385. tabModel.updateConditions(session, conditions={"id": params.id}, **kwargs)
  386. session.close()
  387. sys = CRUD(SysConfigs)
  388. action_configs = sys.read(session, conditions={"key": "action_configs"})
  389. session.close()
  390. return {
  391. "code": 0,
  392. "msg": "",
  393. "data": None,
  394. }
  395. @app.post("/get_device_configs", description="获取可执行程序命令列表")
  396. def get_device_configs(params: ModelGetDeviceConfig):
  397. tab_id = params.tab_id
  398. session = SqlQuery()
  399. configModel = CRUD(DeviceConfig)
  400. configList = configModel.read_all(
  401. session,
  402. conditions={"tab_id": tab_id},
  403. order_by="action_index",
  404. ascending=True,
  405. )
  406. return {
  407. "code": 0,
  408. "msg": "",
  409. "data": {"list": configList},
  410. }
  411. @app.post("/device_config_detail", description="获取可执行程序详情")
  412. def get_device_configs(params: ModelGetDeviceConfigDetail):
  413. action_id = params.id
  414. session = SqlQuery()
  415. configModel = CRUD(DeviceConfig)
  416. model = configModel.read(session, conditions={"id": action_id})
  417. if model == None:
  418. return {"code": 1, "msg": "数据不存在", "data": None}
  419. return {"code": 0, "msg": "", "data": model}
  420. @app.post("/device_config_detail_query", description="通过条件获取可执行程序详情")
  421. def get_device_configs(params: ModelGetDeviceConfigDetailQuery):
  422. tab_id = params.tab_id
  423. action_name = params.action_name
  424. session = SqlQuery()
  425. configModel = CRUD(DeviceConfig)
  426. model = configModel.read(
  427. session, conditions={"tab_id": tab_id, "action_name": action_name}
  428. )
  429. if model == None:
  430. return {"code": 1, "msg": "数据不存在", "data": None}
  431. return {"code": 0, "msg": "", "data": model}
  432. @app.post("/remove_config", description="删除一条可执行命令")
  433. def get_device_configs(params: ModelGetDeviceConfigDetail):
  434. action_id = params.id
  435. session = SqlQuery()
  436. configModel = CRUD(DeviceConfig)
  437. model = configModel.read(session, conditions={"id": action_id})
  438. if model == None:
  439. return {"code": 1, "msg": "数据不存在", "data": None}
  440. if model.is_system == True:
  441. return {"code": 1, "msg": "系统配置不允许删除", "data": None}
  442. configModel.delete(session, obj_id=action_id)
  443. return {"code": 0, "msg": "删除成功", "data": None}
  444. @app.post("/save_device_config", description="创建或修改一条可执行命令")
  445. def save_device_config(params: SaveDeviceConfig):
  446. action_id = params.id
  447. session = SqlQuery()
  448. deviceConfig = CRUD(DeviceConfig)
  449. if action_id == None or action_id == 0:
  450. # 走新增逻辑
  451. params.id = None
  452. save_device_config = deviceConfig.create(session, obj_in=params)
  453. else:
  454. model = deviceConfig.read(session, conditions={"id": action_id})
  455. if model == None:
  456. return {"code": 1, "msg": "数据不存在", "data": None}
  457. # 走编辑逻辑
  458. kwargs = params.__dict__
  459. save_device_config = deviceConfig.update(session, obj_id=action_id, **kwargs)
  460. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  461. @app.post("/reset_config", description="创建或修改一条可执行命令")
  462. def reset_config(params: ModelGetDeviceConfig):
  463. tab_id = params.tab_id
  464. if tab_id == None or tab_id == "":
  465. return {"code": 1, "msg": "参数错误", "data": None}
  466. session = SqlQuery()
  467. deviceConfig = CRUD(DeviceConfig)
  468. res = deviceConfig.deleteConditions(session, conditions={"tab_id": tab_id})
  469. if res is False:
  470. return {"code": 1, "msg": "操作失败", "data": None}
  471. actions = json.load(open("action.json", encoding="utf-8"))
  472. act = []
  473. for item in actions:
  474. if item.get("tab_id") == tab_id:
  475. act.append(item)
  476. batch_insert_device_configs(session, act)
  477. return {"code": 0, "msg": "操作成功", "data": None}
  478. @app.get("/get_photo_records", description="获取拍照记录")
  479. def get_photo_records(page: int = 1, size: int = 5):
  480. session = SqlQuery()
  481. photos = CRUD(PhotoRecord)
  482. print("准备查询拍摄记录", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  483. statement = (
  484. select(PhotoRecord)
  485. .offset((page - 1) * size)
  486. .limit(size)
  487. .order_by(desc("id"))
  488. .group_by("goods_art_no")
  489. )
  490. list = []
  491. result = session.exec(statement).all()
  492. print("group 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  493. for item in result:
  494. list_item = photos.read_all(session, conditions={"goods_art_no": item.goods_art_no})
  495. list.append(
  496. {
  497. "goods_art_no": item.goods_art_no,
  498. "action_time": item.create_time,
  499. "items": list_item,
  500. }
  501. )
  502. session.close()
  503. print("循环查询 完成 ", datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
  504. return {
  505. "code": 0,
  506. "msg": "",
  507. "data": {"list": list, "page": page, "size": size},
  508. }
  509. @app.get("/get_last_photo_record", description="获取最后一条拍照记录")
  510. def get_last_photo_record():
  511. session = SqlQuery()
  512. statement = select(PhotoRecord).where(PhotoRecord.image_path != None).order_by(desc("photo_create_time"))
  513. result = session.exec(statement).first()
  514. session.close()
  515. return {
  516. "code": 0,
  517. "msg": "",
  518. "data": result,
  519. }
  520. @app.get("/get_photo_record_detail", description="通过货号获取拍照记录详情")
  521. def get_photo_records(goods_art_no: str = None):
  522. if goods_art_no == None:
  523. return {"code": 1, "msg": "参数错误", "data": None}
  524. session = SqlQuery()
  525. photos = CRUD(PhotoRecord)
  526. items = photos.read_all(session, conditions={"goods_art_no": goods_art_no})
  527. session.close()
  528. return {
  529. "code": 0,
  530. "msg": "",
  531. "data": {"list": items},
  532. }
  533. @app.post("/delect_goods_arts", description="通过货号删除记录")
  534. def delect_goods_arts(params: PhotoRecordDelete):
  535. session = SqlQuery()
  536. photos = CRUD(PhotoRecord)
  537. for item in params.goods_art_nos:
  538. photos.deleteConditions(session, conditions={"goods_art_no": item})
  539. session.close()
  540. return {
  541. "code": 0,
  542. "msg": "操作成功",
  543. "data": None,
  544. }
  545. @app.get("/get_sys_config", description="查询系统配置")
  546. def get_sys_config(key: str = None):
  547. if key == None:
  548. return {"code": 1, "msg": "参数错误", "data": None}
  549. session = SqlQuery()
  550. photos = CRUD(SysConfigs)
  551. item = photos.read(session, conditions={"key": key})
  552. session.close()
  553. return {
  554. "code": 0,
  555. "msg": "",
  556. "data": json.loads(item.value),
  557. }
  558. @app.post("/update_sys_configs", description="创建或修改系统配置")
  559. def save_sys_configs(params: SysConfigParams):
  560. session = SqlQuery()
  561. sysConfig = CRUD(SysConfigs)
  562. model = sysConfig.read(session, conditions={"key": params.key})
  563. if model == None:
  564. return {"code": 1, "msg": "配置不存在", "data": None}
  565. # 走编辑逻辑
  566. kwargs = params.__dict__
  567. save_device_config = sysConfig.updateConditions(session, conditions={"key": params.key}, **kwargs)
  568. return {"code": 0, "msg": "操作成功", "data": save_device_config}
  569. @app.post("/create_main_image", description="创建主图测试")
  570. def create_main_image(params: MaineImageTest):
  571. file_path = params.file_path
  572. onePic = OnePicTest(pic_path=file_path)
  573. # session = SqlQuery()
  574. # sysConfig = CRUD(SysConfigs)
  575. # model = sysConfig.read(session, conditions={"key": params.key})
  576. # if model == None:
  577. # return {"code": 1, "msg": "配置不存在", "data": None}
  578. # # 走编辑逻辑
  579. # kwargs = params.__dict__
  580. # save_device_config = sysConfig.updateConditions(
  581. # session, conditions={"key": params.key}, **kwargs
  582. # )
  583. main_out_path = onePic.HandlerMainImage()
  584. return {"code": 0, "msg": "操作成功", "data": {"main_out_path": main_out_path}}
  585. def insertEmptyLogoList(session):
  586. """插入空logo列表"""
  587. data = {"key": "logo_configs", "value": "[]"}
  588. config = SysConfigs(**data)
  589. session.add(config)
  590. session.commit()
  591. session.close()
  592. item = SysConfigs()
  593. item.key = "logo_configs"
  594. item.value = "[]"
  595. return item
  596. @app.get("/logo_list", description="logo列表")
  597. def logo_list():
  598. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  599. check_path(logo_dir)
  600. logo_files = os.listdir(logo_dir)
  601. logo_list = []
  602. for logoItem in logo_files:
  603. logo_list.append(f"{logo_dir}{logoItem}")
  604. return {"code": 0, "msg": "操作成功", "data": logo_list}
  605. @app.post("/add_logo", description="添加logo")
  606. def add_logo(params: LogoParams):
  607. logo_path = params.logo_path.replace("\\", "/")
  608. session = SqlQuery()
  609. sysConfig = CRUD(SysConfigs)
  610. item = sysConfig.read(session, conditions={"key": "logo_configs"})
  611. if item == None:
  612. item = insertEmptyLogoList(session)
  613. if os.path.isfile(logo_path) == False:
  614. return {"code": 1, "msg": "logo文件不存在", "data": None}
  615. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  616. check_path(logo_dir)
  617. fpath, fname = os.path.split(logo_path)
  618. logo_path_info = logo_dir + fname
  619. shutil.copy(logo_path, logo_path_info) # 复制文件
  620. logo_files = os.listdir(logo_dir)
  621. logo_list = []
  622. for logoItem in logo_files:
  623. logo_list.append(f"{logo_dir}{logoItem}")
  624. return {
  625. "code": 0,
  626. "msg": "",
  627. "data": {"logo": logo_path_info},
  628. }
  629. @app.post("/delete_logo", description="删除logo")
  630. def delete_logo(params: LogoParamsupdate):
  631. logo_path = params.path
  632. if os.path.isfile(logo_path) == False:
  633. return {"code": 1, "msg": "logo文件不存在", "data": None}
  634. os.remove(logo_path)
  635. logo_dir = "{}/data/logo/".format(os.getcwd()).replace("\\", "/")
  636. check_path(logo_dir)
  637. logo_files = os.listdir(logo_dir)
  638. logo_list = []
  639. for logoItem in logo_files:
  640. logo_list.append(f"{logo_dir}{logoItem}")
  641. return {"code": 0, "msg": "操作成功", "data": logo_list}
  642. @app.post("/close_other_window", description="关闭窗口")
  643. def close_other_window():
  644. hwnd_list = []
  645. def callback(hwnd, _):
  646. title = GetWindowText(hwnd)
  647. if title == "digiCamControl by Duka Istvan":
  648. hwnd_list.append(hwnd)
  649. EnumWindows(callback, None)
  650. if hwnd_list:
  651. hwnd = hwnd_list[0]
  652. win32gui.PostMessage(hwnd, win32con.WM_CLOSE, 0, 0)
  653. return {"code": 0, "msg": "关闭成功", "data": {"status":True}}
  654. return {"code": 0, "msg": "关闭失败", "data": {"status": False}}