message_handler.py 34 KB

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