message_handler.py 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943
  1. from .connect_manager import ConnectionManager
  2. from models import WebSocket
  3. import json, asyncio
  4. from mcu.DeviceControl import DeviceControl, checkMcuConnection
  5. from mcu.BlueToothMode import BlueToothMode
  6. from databases import DeviceConfig, SqlQuery, CRUD, PhotoRecord, SysConfigs
  7. from mcu.capture.module_digicam import DigiCam
  8. from mcu.capture.smart_shooter_class import SmartShooter
  9. from models import RecordUpdate
  10. from service.run_main import RunMain
  11. from utils.utils_func import check_path
  12. import time, shutil, os
  13. from service.base import check_move_goods_art_no_folder
  14. from service.deal_image import DealImage
  15. import settings
  16. from middleware import UnicornException
  17. from concurrent.futures import ThreadPoolExecutor
  18. from functools import partial
  19. from logger import logger
  20. import stat
  21. from mcu.one_data_set_control import OneDataSet
  22. # 创建全局线程池
  23. executor = ThreadPoolExecutor(max_workers=4)
  24. async def handlerCutOut(
  25. manager=None, run_main=None, config_data={}, websocket=None, msg_type=""
  26. ):
  27. max_retry_count = 1 # 最多重试1次
  28. retry_count = 0
  29. while retry_count <= max_retry_count:
  30. try:
  31. if retry_count > 0:
  32. logger.info(f"抠图操作重试第{retry_count}次")
  33. # return_data = run_main.check_before_cutout(config_data)
  34. # await run_main.check_for_cutout_image_first_call_back(return_data)
  35. # 将阻塞操作放到线程池中执行
  36. loop = asyncio.get_event_loop()
  37. return_data = await loop.run_in_executor(
  38. executor, partial(run_main.check_before_cutout, config_data)
  39. )
  40. # await run_main.check_for_cutout_image_first_call_back(return_data)
  41. await loop.run_in_executor(
  42. executor,
  43. partial(run_main.check_for_cutout_image_first_call_back, return_data),
  44. )
  45. # 成功则退出循环
  46. break
  47. except UnicornException as e:
  48. logger.error(f"抠图操作发生UnicornException: {e.msg}")
  49. logger.error(f"异常类型: UnicornException")
  50. logger.error(f"当前重试次数: {retry_count}/{max_retry_count}")
  51. # data = manager.jsonMessage(
  52. # code=1,
  53. # msg=e.msg,
  54. # msg_type=msg_type,
  55. # )
  56. # await manager.send_personal_message(data, websocket)
  57. return
  58. except FileNotFoundError as e:
  59. error_msg = str(e)
  60. logger.error(f"抠图操作发生FileNotFoundError: {error_msg}")
  61. logger.error(f"异常类型: FileNotFoundError")
  62. logger.error(f"当前重试次数: {retry_count}/{max_retry_count}")
  63. # 检查是否是原始图缺失错误
  64. if "原始图" in error_msg and retry_count < max_retry_count:
  65. logger.info("检测到原始图缺失,尝试重新拷贝原图并重试...")
  66. try:
  67. # 重新拷贝原图
  68. goods_art_nos = config_data.get("goods_art_nos", [])
  69. image_dir = config_data.get("image_dir", "")
  70. for goods_art_no in goods_art_nos:
  71. dealImage = DealImage(image_dir)
  72. resFlag, path = dealImage.dealMoveImageV2(
  73. goods_art_no=goods_art_no,
  74. )
  75. if not resFlag:
  76. logger.error(f"重新拷贝原图失败: {goods_art_no}")
  77. raise UnicornException(f"重新拷贝原图失败: {goods_art_no}")
  78. logger.info("原图重新拷贝成功,准备重试抠图操作")
  79. retry_count += 1
  80. continue # 重试
  81. except Exception as copy_error:
  82. logger.error(f"重新拷贝原图时发生错误: {str(copy_error)}")
  83. return
  84. else:
  85. logger.error(f"重新拷贝原图时发生错误: 非原始图错误或已达到最大重试次数")
  86. return
  87. except Exception as e:
  88. import traceback
  89. error_msg = str(e)
  90. stack_trace = traceback.format_exc()
  91. logger.error(f"抠图操作发生未知异常: {error_msg}")
  92. logger.error(f"异常类型: {type(e).__name__}")
  93. logger.error(f"当前重试次数: {retry_count}/{max_retry_count}")
  94. logger.error(f"完整堆栈跟踪:\n{stack_trace}")
  95. # 如果是第一次执行且未达到最大重试次数,则重试
  96. if retry_count < max_retry_count:
  97. logger.info("准备重试抠图操作...")
  98. retry_count += 1
  99. continue
  100. return
  101. def handlerFolderDelete(limit_path, goods_art_no_arrays, is_write_txt_log):
  102. check_path(limit_path)
  103. move_folder_array = check_move_goods_art_no_folder(
  104. "output", goods_art_no_arrays, limit_path
  105. )
  106. for goods_art_revice in goods_art_no_arrays:
  107. cutout_goods = f"{limit_path}/{goods_art_revice}"
  108. if os.path.exists(cutout_goods):
  109. for root, dirs, files in os.walk(cutout_goods):
  110. for file in files:
  111. filepath = os.path.join(root, file)
  112. os.chmod(filepath, stat.S_IWRITE)
  113. logger.info(f"解除占用并开始删除目录:{cutout_goods}")
  114. # 尝试多次删除,增加成功率
  115. retry_count = 3
  116. while retry_count > 0:
  117. try:
  118. shutil.rmtree(cutout_goods, onerror=settings.handle_remove_readonly)
  119. del move_folder_array[goods_art_revice]
  120. break
  121. except OSError as e:
  122. logger.info(f"目录已经被删除-OSError:{str(e)}")
  123. break
  124. except (PermissionError) as e:
  125. retry_count -= 1
  126. if retry_count == 0:
  127. logger.info(
  128. f"抠图前目录删除出现问题-PermissionError:{str(e)};{goods_art_revice};{cutout_goods}")
  129. if is_write_txt_log:
  130. error_file_path = f"{cutout_goods}/异常说明-出现目录丢失或缺少图片请点开查看原因.txt"
  131. with open(error_file_path, 'w', encoding='utf-8') as f:
  132. f.write("目录删除失败\n")
  133. f.write(f"原因: 文件被占用或没有删除权限\n")
  134. f.write(f"错误信息: {str(e)}\n")
  135. f.write(f"时间: {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
  136. f.write(f"请关闭可能正在使用此目录的程序后,为当前货号点击重拍后重试\n")
  137. else:
  138. raise UnicornException(f"目录检查出现问题:{str(e)},请关闭错误提示中的被占用文件")
  139. else:
  140. logger.info(f"抠图前目录删除出现问题--PermissionError:{str(e)};{retry_count}")
  141. time.sleep(0.5) # 等待0.5秒后重试
  142. except Exception as e:
  143. retry_count -= 1
  144. if retry_count == 0:
  145. logger.info(f"抠图前目录删除出现问题--Exception:{str(e)};{goods_art_revice};{cutout_goods}")
  146. if is_write_txt_log:
  147. error_file_path = f"{cutout_goods}/异常说明-出现目录丢失或缺少图片请点开查看原因.txt"
  148. with open(error_file_path, 'w', encoding='utf-8') as f:
  149. f.write("目录删除失败\n")
  150. f.write(f"原因: {str(e)}\n")
  151. f.write(f"时间: {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
  152. f.write(f"请检查目录状态后,为当前货号点击重拍后重试\n")
  153. else:
  154. raise UnicornException(f"目录检查出现问题:{str(e)},请关闭错误提示中的被占用文件")
  155. else:
  156. logger.info(f"抠图前目录删除出现问题--Exception:{str(e)};{retry_count}")
  157. time.sleep(0.5) # 等待0.5秒后重试
  158. return move_folder_array
  159. def validate_goods_art_no(goods_art_no):
  160. """
  161. 验证货号输入是否包含特殊字符
  162. """
  163. import re
  164. # 定义不允许的特殊字符,主要是文件系统中可能导致路径问题的字符
  165. invalid_chars = r'[<>:"/\\|?*]'
  166. if re.search(invalid_chars, goods_art_no):
  167. # 找出所有非法字符
  168. invalid_found = re.findall(invalid_chars, goods_art_no)
  169. invalid_str = ', '.join(set(invalid_found))
  170. return False, f"货号包含非法字符: {invalid_str},请修改后再提交"
  171. return True, ""
  172. # socket消息发送逻辑处理方法
  173. async def handlerSend(
  174. manager: ConnectionManager,
  175. receiveData: str,
  176. websocket: WebSocket,
  177. smart_shooter: SmartShooter,
  178. ):
  179. loop = asyncio.get_event_loop()
  180. receiveData = json.loads(receiveData)
  181. # 处理消息发送逻辑
  182. try:
  183. receiveData = json.loads(receiveData.get("text"))
  184. print("receiveData", receiveData)
  185. jsonType = receiveData.get("type")
  186. code = receiveData.get("code")
  187. msg = receiveData.get("msg")
  188. data = receiveData.get("data")
  189. except:
  190. jsonType = "ping"
  191. match jsonType:
  192. case "ping":
  193. """发送心跳"""
  194. data = manager.jsonMessage("pong")
  195. await manager.send_personal_message(data, websocket)
  196. case "pong":
  197. """发送心跳"""
  198. pass
  199. case "forward_message":
  200. data = receiveData.get("data")
  201. dictMsg = {"code": code, "msg": msg, "data": data}
  202. await manager.broadcast(dictMsg)
  203. case "connect_mcu":
  204. device_ctrl = DeviceControl(
  205. websocket_manager=manager, smart_shooter=smart_shooter
  206. )
  207. device_ctrl.mcu_exit = False
  208. # if device_ctrl.serial_ins.check_connect():
  209. # print("未连接")
  210. loop.create_task(checkMcuConnection(device_ctrl), name="mcu")
  211. case "connect_bluetooth":
  212. blue_tooth = BlueToothMode(websocket_manager=manager)
  213. blue_tooth.bluetooth_exit = False
  214. # await blue_tooth.main_func()
  215. print("连接蓝牙信息")
  216. loop.create_task(blue_tooth.main_func(), name="blue_tooth")
  217. # loop.close()
  218. case "init_mcu":
  219. device_ctrl = DeviceControl(
  220. websocket_manager=manager, smart_shooter=smart_shooter
  221. )
  222. # 是否强制初始化
  223. is_force_init = False
  224. loop.create_task(device_ctrl.initDevice(is_force_init), name="init_mcu")
  225. case "get_mcu_info":
  226. device_ctrl = DeviceControl(
  227. websocket_manager=manager, smart_shooter=smart_shooter
  228. )
  229. if not device_ctrl.is_running:
  230. device_ctrl.sendSocketMessage(
  231. code=1, msg="mcu设备未连接,请先连接设备", device_status=0
  232. )
  233. device_ctrl.send_get_all_info_to_mcu()
  234. case "control_mcu":
  235. device_name = data.get("device_name")
  236. value = data.get("value")
  237. if (device_name == "" or device_name == None) or (
  238. value == "" or value == None
  239. ):
  240. data = manager.jsonMessage(code=1, msg="参数错误", msg_type="mcu")
  241. await manager.send_personal_message(data, websocket)
  242. return
  243. device_ctrl = DeviceControl(
  244. websocket_manager=manager, smart_shooter=smart_shooter
  245. )
  246. await device_ctrl.controlDevice(device_name, value)
  247. case "stop_action":
  248. device_ctrl = DeviceControl(
  249. websocket_manager=manager, smart_shooter=smart_shooter
  250. )
  251. if device_ctrl.is_runn_action == True:
  252. print("动作执行中,停止")
  253. device_ctrl.is_stop_action = True
  254. else:
  255. print("动作没有执行,略过")
  256. data = manager.jsonMessage(code=0, msg="执行终止", msg_type="run_mcu_stop")
  257. await manager.send_personal_message(data, websocket)
  258. return
  259. case "run_mcu":
  260. msg_type = "run_mcu"
  261. action_info = data.get("action", "执行左脚程序")
  262. goods_art_no = data.get("goods_art_no", None)
  263. weight = data.get("weight", None)
  264. goods_art_no = goods_art_no.strip()
  265. if goods_art_no == None or goods_art_no == "":
  266. # 判断货号是否存在
  267. data = manager.jsonMessage(
  268. code=1, msg="goods_art_no不能为空", msg_type=msg_type
  269. )
  270. await manager.send_personal_message(data, websocket)
  271. return
  272. is_ok, msg = validate_goods_art_no(goods_art_no)
  273. if not is_ok:
  274. data = manager.jsonMessage(
  275. code=1, msg=msg, msg_type=msg_type
  276. )
  277. await manager.send_personal_message(data, websocket)
  278. return
  279. session = SqlQuery()
  280. sys_configs = CRUD(SysConfigs)
  281. action_configs = sys_configs.read(
  282. session, conditions={"key": "action_configs"}
  283. )
  284. action_configs_json = json.loads(action_configs.value)
  285. action_flag = "left"
  286. if "右" in action_info:
  287. action_flag = "right"
  288. tab_id = action_configs_json.get(action_flag)
  289. photoRecord = CRUD(PhotoRecord)
  290. goods_art_record = photoRecord.read(
  291. session, conditions={"goods_art_no": goods_art_no, "delete_time": None}
  292. )
  293. if goods_art_record != None:
  294. data = manager.jsonMessage(
  295. code=1,
  296. msg=f"货号[{goods_art_no}]已存在,请勿重复拍摄~",
  297. msg_type=msg_type,
  298. )
  299. await manager.send_personal_message(data, websocket)
  300. return
  301. crud = CRUD(DeviceConfig)
  302. condtions = {"tab_id": tab_id}
  303. all_devices = crud.read_all(
  304. session, conditions=condtions, order_by="action_index", ascending=True
  305. )
  306. if len(all_devices) == 0:
  307. # 判断是否有可用配置
  308. data = manager.jsonMessage(code=1, msg="当前没有可用配置")
  309. await manager.send_personal_message(data, websocket, msg_type=msg_type)
  310. return
  311. action_list = [dict(device.__dict__) for device in all_devices]
  312. print("执行拍摄 打印动作列表", condtions)
  313. # logger.info("执行拍摄 打印动作列表", json.dumps(action_list))
  314. device_ctrl = DeviceControl(
  315. websocket_manager=manager, smart_shooter=smart_shooter
  316. )
  317. loop.create_task(
  318. device_ctrl.run_mcu_config(
  319. action_list, goods_art_no, action_info, smart_shooter, weight
  320. ),
  321. name="run_mcu_config",
  322. )
  323. session.close()
  324. case "run_mcu_single":
  325. device_ctrl = DeviceControl(
  326. websocket_manager=manager, smart_shooter=smart_shooter
  327. )
  328. loop.create_task(
  329. device_ctrl.run_mcu_config_single(
  330. data, None, "run_mcu_single", -1, smart_shooter
  331. ),
  332. name="run_mcu_single",
  333. )
  334. case "handler_take_picture":
  335. print("收到单拍指令", 'handler_take_picture')
  336. if data is None:
  337. PointName = "A"
  338. else:
  339. PointName = data.get("point_name", "A")
  340. print("收到单拍指令", PointName)
  341. device_ctrl = DeviceControl(
  342. websocket_manager=manager, smart_shooter=smart_shooter
  343. )
  344. await device_ctrl.controlDevice("laser_position", 0)
  345. blue_tooth = BlueToothMode(
  346. websocket_manager=manager, smart_shooter=smart_shooter
  347. )
  348. session = SqlQuery()
  349. crud = CRUD(PhotoRecord)
  350. record = crud.read(session=session, order_by="id", ascending=False, conditions={"delete_time": None})
  351. if record == None:
  352. # 发送失败消息
  353. data = manager.jsonMessage(
  354. code=1,
  355. msg="单拍失败,请先输入货号或扫码进行组合拍摄",
  356. msg_type="handler_take_picture",
  357. )
  358. await manager.send_personal_message(data, websocket)
  359. return
  360. try:
  361. limit_path = "{}/{}".format(settings.OUTPUT_DIR,
  362. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  363. )
  364. move_folder_array = handlerFolderDelete(limit_path, [record.goods_art_no], False)
  365. except UnicornException as e:
  366. data = manager.jsonMessage(
  367. code=1,
  368. msg=e.msg,
  369. msg_type="handler_take_picture",
  370. )
  371. await manager.send_personal_message(data, websocket)
  372. return
  373. record_action_id = record.action_id
  374. deviceConfig = CRUD(DeviceConfig)
  375. deviceConfigSignle = deviceConfig.read(session, conditions={"id": record_action_id})
  376. deviceConfigAll = deviceConfig.read_all(session, conditions={"tab_id": deviceConfigSignle.tab_id},
  377. order_by="action_index",
  378. ascending=True, )
  379. lastConfig = deviceConfigAll[-1]
  380. if not lastConfig.take_picture:
  381. used_config_id = lastConfig.id
  382. else:
  383. used_config_id = deviceConfigAll[0].id
  384. loop.create_task(
  385. blue_tooth.remote_control_v2.handlerTakePhoto(smart_shooter, session, record, PointName,
  386. used_config_id),
  387. name="run_mcu_config",
  388. )
  389. session.close()
  390. await asyncio.sleep(2.5)
  391. await device_ctrl.controlDevice("laser_position", 1)
  392. case "re_take_picture": # 重拍
  393. msg_type = "re_take_picture"
  394. record_id = data.get("record_id")
  395. session = SqlQuery()
  396. photoRecord = CRUD(PhotoRecord)
  397. goods_art_record = photoRecord.read(session, conditions={"id": record_id, "delete_time": None})
  398. if goods_art_record == None:
  399. data = manager.jsonMessage(
  400. code=1,
  401. msg=f"记录不存在,请核实后重新操作~",
  402. msg_type=msg_type,
  403. )
  404. await manager.send_personal_message(data, websocket)
  405. return
  406. action_id = goods_art_record.action_id
  407. goods_art_no = goods_art_record.goods_art_no
  408. image_index = goods_art_record.image_index
  409. crud = CRUD(DeviceConfig)
  410. condtions = {"id": action_id}
  411. device_action = crud.read(session, conditions=condtions)
  412. result_dict = dict(device_action.__dict__)
  413. print("device_action打印输出====>>>", result_dict)
  414. if device_action == None:
  415. # 判断是否有可用配置
  416. data = manager.jsonMessage(code=1, msg="当前没有可用配置")
  417. await manager.send_personal_message(data, websocket, msg_type=msg_type)
  418. return
  419. # 清除图片记录,执行重拍
  420. reset_data = {"image_path": None}
  421. photoRecord.update(session, record_id, **reset_data)
  422. device_ctrl = DeviceControl(
  423. websocket_manager=manager, smart_shooter=smart_shooter
  424. )
  425. loop.create_task(
  426. device_ctrl.run_mcu_config_single(
  427. result_dict,
  428. goods_art_no,
  429. msg_type=msg_type,
  430. image_index=image_index,
  431. smart_shooter=smart_shooter,
  432. action_id=record_id,
  433. ),
  434. name="run_mcu_config_single",
  435. )
  436. session.close()
  437. case "get_deviation":
  438. device_ctrl = DeviceControl(
  439. websocket_manager=manager, smart_shooter=smart_shooter
  440. )
  441. device_ctrl.getDeviationInfo()
  442. case "set_deviation":
  443. device_ctrl = DeviceControl(
  444. websocket_manager=manager, smart_shooter=smart_shooter
  445. )
  446. value = data.get("value", None)
  447. action_name = data.get("action_name", None)
  448. loop.create_task(
  449. device_ctrl.set_deviation_cmd(value, action_name, "set"),
  450. name="set_deviation",
  451. )
  452. case "move_deviation":
  453. device_ctrl = DeviceControl(
  454. websocket_manager=manager, smart_shooter=smart_shooter
  455. )
  456. value = data.get("value", None)
  457. action_name = data.get("action_name", None)
  458. loop.create_task(
  459. device_ctrl.set_deviation_cmd(value, action_name, "move"),
  460. name="move_deviation",
  461. )
  462. case "get_mcu_other_info":
  463. device_ctrl = DeviceControl(
  464. websocket_manager=manager, smart_shooter=smart_shooter
  465. )
  466. loop.create_task(
  467. device_ctrl.getMcuOtherInfo(),
  468. name="mcu_other_set_get",
  469. )
  470. case "set_mcu_other_info":
  471. device_ctrl = DeviceControl(
  472. websocket_manager=manager, smart_shooter=smart_shooter
  473. )
  474. loop.create_task(
  475. device_ctrl.setMcuOtherInfo(data),
  476. name="setMcuOtherInfo",
  477. )
  478. case "send_command":
  479. device_ctrl = DeviceControl(
  480. websocket_manager=manager, smart_shooter=smart_shooter
  481. )
  482. loop.create_task(
  483. device_ctrl.sendCommand(data.get("command", None)),
  484. name="sendCommand",
  485. )
  486. case "smart_shooter_getinfo":
  487. """
  488. 获取相机信息,是否连接
  489. """
  490. # token
  491. # env
  492. # print("smart_shooter_getinfo",data)
  493. # isMultCameraMode = data.get("isMultiCameraMode",False)
  494. loop.create_task(
  495. smart_shooter.GetCameraInfo(
  496. msg_type="smart_shooter_getinfo",
  497. is_send=True,
  498. # isMultCameraMode=isMultCameraMode,
  499. ),
  500. name="smart_shooter_getinfo",
  501. )
  502. case "smart_shooter_enable_preview":
  503. """
  504. 启动相机或关闭实时预览
  505. """
  506. value = data.get("value", True)
  507. if data is None:
  508. PointName = "A"
  509. else:
  510. PointName = data.get("point_name", "A")
  511. logger.info(f"启动相机或关闭实时预览,PointName:{PointName},value:{value}")
  512. camera_configs = settings.getSysConfigs(
  513. "camera_configs",
  514. "iso_config",
  515. None,
  516. )
  517. msg_type = "smart_shooter_enable_preview_status"
  518. print("传入得msg_type====>>>", msg_type)
  519. temp_A_point = camera_configs.get(PointName, None)
  520. CameraKey = temp_A_point.get("CameraKey", None) if temp_A_point else None
  521. loop.create_task(
  522. smart_shooter.EnableCameraPreview(
  523. enable_status=value,
  524. msg_type=msg_type,
  525. CameraKey=CameraKey,
  526. ),
  527. name=msg_type,
  528. )
  529. case "smart_shooter_auto_focus":
  530. """
  531. 启动相机或关闭实时预览
  532. """
  533. if data is None:
  534. PointName = "A"
  535. else:
  536. PointName = data.get("point_name", "A")
  537. camera_configs = settings.getSysConfigs(
  538. "camera_configs",
  539. "iso_config",
  540. None,
  541. )
  542. msg_type = "smart_shooter_auto_focus"
  543. temp_A_point = camera_configs.get(PointName, None)
  544. CameraKey = temp_A_point.get("CameraKey", None) if temp_A_point else None
  545. status, msg = await smart_shooter.CameraAutofocus(
  546. CameraKey=CameraKey,
  547. )
  548. if not status:
  549. data = manager.jsonMessage(
  550. code=1,
  551. msg=msg,
  552. msg_type=msg_type,
  553. )
  554. await manager.send_personal_message(data, websocket)
  555. case "smart_shooter_get_camera_property":
  556. """
  557. 启动相机或关闭实时预览
  558. """
  559. if data is None:
  560. PointName = "A"
  561. else:
  562. PointName = data.get("point_name", "A")
  563. camera_configs = settings.getSysConfigs(
  564. "camera_configs",
  565. "iso_config",
  566. None,
  567. )
  568. temp_A_point = camera_configs.get(PointName, None)
  569. CameraKey = temp_A_point.get("CameraKey", None) if temp_A_point else None
  570. msg_type = "smart_shooter_get_camera_property"
  571. print("CameraKey", CameraKey)
  572. code = 0
  573. status, info = await smart_shooter.GetCameraProperty(CameraKey=CameraKey)
  574. code = 1 if status == False else 0
  575. msg = info if status == False else "操作成功"
  576. data = info if status == True else {}
  577. data = manager.jsonMessage(
  578. code=code,
  579. msg=msg,
  580. msg_type=msg_type,
  581. data=data,
  582. )
  583. await manager.send_personal_message(data, websocket)
  584. case "smart_shooter_photo_take":
  585. """
  586. 获取相机信息,是否连接
  587. """
  588. if data is None:
  589. PointName = "A"
  590. else:
  591. PointName = data.get("point_name", "A")
  592. camera_configs = settings.getSysConfigs(
  593. "camera_configs",
  594. "iso_config",
  595. None,
  596. )
  597. temp_A_point = camera_configs.get(PointName, None)
  598. CameraKey = temp_A_point.get("CameraKey", None) if temp_A_point else None
  599. device_ctrl = DeviceControl(
  600. websocket_manager=manager, smart_shooter=smart_shooter
  601. )
  602. await device_ctrl.controlDevice("laser_position", 0)
  603. # 兼容主图测试
  604. id = data.get("id", 0)
  605. goods_art_no = data.get("goods_art_no", "")
  606. goods_art_no = goods_art_no.strip()
  607. if goods_art_no:
  608. try:
  609. limit_path = "{}/{}".format(settings.OUTPUT_DIR,
  610. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  611. )
  612. move_folder_array = handlerFolderDelete(limit_path, [goods_art_no], False)
  613. except UnicornException as e:
  614. data = manager.jsonMessage(
  615. code=1,
  616. msg=e.msg,
  617. msg_type="smart_shooter_photo_take",
  618. )
  619. await manager.send_personal_message(data, websocket)
  620. return
  621. if not await device_ctrl.camera_check_mcu_move_is_stop(re_check=True):
  622. logger.warning("拍照前运动检测失败===>,等待0.5秒后再检测一次")
  623. await asyncio.sleep(0.5)
  624. if not await device_ctrl.camera_check_mcu_move_is_stop(re_check=True):
  625. logger.warning("拍照前运动检测二次失败===>,强制拍摄")
  626. await asyncio.sleep(0.5)
  627. is_af = True
  628. loop.create_task(
  629. smart_shooter.CameraShooter(
  630. msg_type="run_mcu",
  631. id=id,
  632. goods_art_no=goods_art_no,
  633. is_af=is_af,
  634. CameraKey=CameraKey,
  635. ),
  636. name="smart_shooter_photo_take",
  637. )
  638. await asyncio.sleep(2.5)
  639. await device_ctrl.controlDevice("laser_position", 1)
  640. case "digicam_take_picture":
  641. msg_type = "re_take_picture"
  642. id = data.get("id", 0)
  643. goods_art_no = data.get("goods_art_no", "")
  644. session = SqlQuery()
  645. photoRecord = CRUD(PhotoRecord)
  646. goods_art_record = photoRecord.read(session, conditions={"id": id, "delete_time": None})
  647. if goods_art_record == None:
  648. data = manager.jsonMessage(
  649. code=1,
  650. msg=f"记录不存在,请核实后重新操作~",
  651. msg_type=msg_type,
  652. )
  653. await manager.send_personal_message(data, websocket)
  654. return
  655. reset_data = {"image_path": None}
  656. photoRecord.update(session, id, **reset_data)
  657. device_ctrl = DeviceControl(websocket_manager=manager) #
  658. loop.create_task(
  659. device_ctrl.only_take_photo(
  660. goods_art_no=goods_art_no,
  661. image_index=goods_art_record.image_index,
  662. record_id=id,
  663. ),
  664. name="sendCommand",
  665. )
  666. session.close()
  667. case "segment_progress":
  668. msg_type = "segment_progress"
  669. obj = None
  670. token = data.get("token", "")
  671. settings.USER_TOKEN = token
  672. token = "Bearer " + token
  673. uuid = data.get("uuid", "")
  674. run_main = RunMain(obj, token, uuid)
  675. goods_art_no_arrays = data.get("goods_art_no", [])
  676. limit_path = "{}/{}".format(settings.OUTPUT_DIR,
  677. time.strftime("%Y-%m-%d", time.localtime(time.time()))
  678. )
  679. try:
  680. move_folder_array = handlerFolderDelete(limit_path, goods_art_no_arrays, True)
  681. except UnicornException as e:
  682. data = manager.jsonMessage(
  683. code=1,
  684. msg=e.msg,
  685. msg_type=msg_type,
  686. )
  687. await manager.send_personal_message(data, websocket)
  688. return
  689. # 该数组表示是否需要后面的移动文件夹操作,减少重复抠图,提升抠图时间和速度
  690. session = SqlQuery()
  691. for goods_art_no in goods_art_no_arrays:
  692. pr = CRUD(PhotoRecord)
  693. images = pr.read_all(session, conditions={"goods_art_no": goods_art_no, "delete_time": None})
  694. if not images:
  695. data = manager.jsonMessage(
  696. code=1,
  697. msg=f"商品货号【{goods_art_no}】在商品档案资料中不存在,请检查货号是否正确",
  698. msg_type=msg_type,
  699. )
  700. await manager.send_personal_message(data, websocket)
  701. return
  702. if move_folder_array.get(goods_art_no) == None:
  703. image_dir = "{}/data/".format(os.getcwd()).replace("\\", "/")
  704. check_path(image_dir)
  705. for idx, itemImg in enumerate(images):
  706. if itemImg.image_path == "" or itemImg.image_path == None:
  707. data = manager.jsonMessage(
  708. code=1,
  709. msg=f"货号【{goods_art_no}】存在没有拍摄完成的图片,请重拍或删除后重试",
  710. msg_type=msg_type,
  711. )
  712. await manager.send_personal_message(data, websocket)
  713. return
  714. # new_file_name = (
  715. # str(idx)+"_"+str(itemImg.goods_art_no) + "_" + str(idx) + ".jpg"
  716. # )
  717. # if not os.path.exists(
  718. # image_dir + "/" + os.path.basename(new_file_name)
  719. # ):
  720. # shutil.copy(itemImg.image_path, image_dir + new_file_name)
  721. dealImage = DealImage(image_dir)
  722. resFlag, path = dealImage.dealMoveImageV2(
  723. goods_art_no=goods_art_no,
  724. )
  725. if not resFlag:
  726. # path
  727. data = manager.jsonMessage(
  728. code=1,
  729. msg=f"抠图操作异常,请检查目录是否存在,或者权限不足",
  730. msg_type=msg_type,
  731. )
  732. await manager.send_personal_message(data, websocket)
  733. return
  734. session.close()
  735. # try:
  736. cutout_mode = settings.getSysConfigs("other_configs", "cutout_mode", "普通抠图")
  737. match cutout_mode:
  738. case "普通抠图":
  739. cutOutMode = "1"
  740. case "精细化抠图":
  741. cutOutMode = "2"
  742. case "旗舰AI抠图":
  743. cutOutMode = "3"
  744. case "旗舰AI抠图-精细":
  745. cutOutMode = "4"
  746. case "旗舰AI抠图-透明物品":
  747. cutOutMode = "5"
  748. case _:
  749. cutOutMode = "1"
  750. config_data = {
  751. "image_dir": limit_path,
  752. "image_order": (
  753. "俯视,侧视,后跟,鞋底,内里,组合,组合2,组合3,组合4,组合5"
  754. ),
  755. "goods_art_no": "",
  756. "goods_art_nos": goods_art_no_arrays,
  757. "is_check_number": False,
  758. "resize_image_view": "后跟",
  759. "cutout_mode": cutOutMode,
  760. "logo_path": "",
  761. "special_goods_art_no_folder_line": "",
  762. "is_use_excel": False, # 是否使用excel
  763. "excel_path": "", # excel路径
  764. "is_check_color_is_all": False,
  765. "cutout_is_pass": True,
  766. "assigned_page_dict": {},
  767. "detail_is_pass": True,
  768. "upload_is_pass": False,
  769. "upload_is_enable": settings.IS_UPLOAD_HLM, # 是否上传到惠利玛商品库,通过config.ini得is_upload开启
  770. "is_filter": False,
  771. "temp_class": {},
  772. "temp_name": "",
  773. "temp_name_list": [],
  774. "target_error_folder": f"{limit_path}/软件-生成详情错误",
  775. "success_handler": [],
  776. }
  777. loop.create_task(
  778. handlerCutOut(
  779. manager=manager,
  780. run_main=run_main,
  781. config_data=config_data,
  782. websocket=websocket,
  783. msg_type=msg_type,
  784. ),
  785. name="handlerCutOut",
  786. )
  787. case "get_dynamic_config":
  788. device_ctrl = DeviceControl(
  789. websocket_manager=manager, smart_shooter=smart_shooter
  790. )
  791. dynamic_configs = device_ctrl.get_dynamic_config()
  792. # print("dynamic_configs", dynamic_configs)
  793. data = manager.jsonMessage(
  794. code=0,
  795. msg="成功",
  796. msg_type="get_dynamic_config",
  797. data=dynamic_configs,
  798. )
  799. await manager.send_personal_message(data, websocket)
  800. case "one_key_down":
  801. device_ctrl = DeviceControl(
  802. websocket_manager=manager, smart_shooter=smart_shooter
  803. )
  804. device_ctrl.one_key_down()
  805. case "get_dynamic_config_signle":
  806. device_ctrl = DeviceControl(
  807. websocket_manager=manager, smart_shooter=smart_shooter
  808. )
  809. name = data.get("name")
  810. value = device_ctrl.get_dynamic_config_signle(name)
  811. # print("dynamic_configs", dynamic_configs)
  812. data = manager.jsonMessage(
  813. code=0,
  814. msg="成功",
  815. msg_type="get_dynamic_config_signle",
  816. data={"value": value},
  817. )
  818. await manager.send_personal_message(data, websocket)
  819. case "set_dynamic_config":
  820. # 设置动态参数
  821. device_ctrl = DeviceControl(
  822. websocket_manager=manager, smart_shooter=smart_shooter
  823. )
  824. readonly = data.get("readonly", False)
  825. if not readonly:
  826. dynamic_configs = device_ctrl.set_dynamic_config(data)
  827. else:
  828. dynamic_configs = False
  829. data = manager.jsonMessage(
  830. code=0,
  831. msg="成功",
  832. msg_type="set_dynamic_config",
  833. data={"status": dynamic_configs},
  834. )
  835. await manager.send_personal_message(data, websocket)
  836. case "get_device_info":
  837. device_ctrl = DeviceControl(
  838. websocket_manager=manager, smart_shooter=smart_shooter
  839. )
  840. device_ctrl.getDeviationInfo()
  841. await asyncio.sleep(0.1)
  842. device_ctrl.get_device_info()
  843. case "to_recheck_focal":
  844. device_ctrl = DeviceControl(
  845. websocket_manager=manager, smart_shooter=smart_shooter
  846. )
  847. device_ctrl.to_recheck_focal("camera_zoom_motor")
  848. case "oneclick_release":
  849. device_ctrl = DeviceControl(
  850. websocket_manager=manager, smart_shooter=smart_shooter
  851. )
  852. device_ctrl.set_stepper_disable("mcu", False)
  853. case "mcu_power_off":
  854. device_ctrl = DeviceControl(
  855. websocket_manager=manager, smart_shooter=smart_shooter
  856. )
  857. device_ctrl.one_key_turnoff("mcu", False)
  858. case "get_scale_data":
  859. '''获取安衡电子秤数据'''
  860. device_ctrl = DeviceControl(
  861. websocket_manager=manager, smart_shooter=smart_shooter
  862. )
  863. if device_ctrl.anheng_scale._running == False:
  864. data = manager.jsonMessage(
  865. code=1,
  866. msg=f"电子秤不存在,请核实后重新操作~",
  867. msg_type="anheng_scale_status",
  868. )
  869. await manager.send_personal_message(data, websocket)
  870. device_ctrl.anheng_scale.is_send_message = True
  871. case "delete_photo_record": # 重拍
  872. msg_type = "delete_photo_record"
  873. record_id = data.get("record_id")
  874. session = SqlQuery()
  875. photoRecord = CRUD(PhotoRecord)
  876. goods_art_record = photoRecord.read(session, conditions={"id": record_id, "delete_time": None})
  877. if goods_art_record == None:
  878. data = manager.jsonMessage(
  879. code=1,
  880. msg=f"记录不存在,请核实后重新操作~",
  881. msg_type=msg_type,
  882. )
  883. await manager.send_personal_message(data, websocket)
  884. return
  885. photoRecord.deleteConditions(session, conditions={"id": record_id, "delete_time": None})
  886. session.close()
  887. data = manager.jsonMessage(
  888. code=0,
  889. msg=f"记录删除成功",
  890. msg_type=msg_type,
  891. )
  892. await manager.send_personal_message(data, websocket)
  893. return
  894. case "get_one_data_set":
  895. onseData = OneDataSet()
  896. result = onseData.create_command_list()
  897. data = manager.jsonMessage(
  898. code=0,
  899. msg=f"获取测试命令",
  900. msg_type='get_one_data_set',
  901. data={"commands": json.loads(result)}
  902. )
  903. await manager.send_personal_message(data, websocket)
  904. case "send_get_one_data_set":
  905. command = data.get("command")
  906. onseData = OneDataSet()
  907. device_ctrl = DeviceControl(
  908. websocket_manager=manager, smart_shooter=smart_shooter
  909. )
  910. if not device_ctrl.is_running:
  911. data = manager.jsonMessage(
  912. code=-1,
  913. msg=f"设备未连接",
  914. msg_type="send_get_one_data_set",
  915. )
  916. await manager.send_personal_message(data, websocket)
  917. return
  918. onseData.mcu = device_ctrl
  919. onseData.deal_send_json_to_mcu(command)
  920. case _:
  921. data = manager.jsonMessage(code=1, msg="未知消息")
  922. await manager.send_personal_message(data, websocket)
  923. return