api.py 30 KB

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