From 8c2322adaed3d4374a7524bdb9ae8fb27f74fe7f Mon Sep 17 00:00:00 2001
From: wu_xinjun <bent20140204@outlook.com>
Date: 星期一, 21 二月 2022 17:00:05 +0800
Subject: [PATCH] optimize the udp2ply

---
 src/engine/DeviceManager.py |    8 ++++----
 1 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/engine/DeviceManager.py b/src/engine/DeviceManager.py
index 7909d32..715cc21 100644
--- a/src/engine/DeviceManager.py
+++ b/src/engine/DeviceManager.py
@@ -19,13 +19,13 @@
         channel.queue_declare(queue=device_id + "_callback")
         return channel
 
-    def get_device(self, device_id: str):
+    def get_device(self, device_id: str, save_folder: str):
         device = self._devices.get(device_id, None)
         if device is None:
             start = datetime.datetime.now()
-            if not os.path.exists('./Images'):
-                os.mkdir('./Images')
-            path = f"./Images/{device_id}-{start.strftime('%Y-%m-%d-%H-%M-%S')}"
+            if not os.path.exists(save_folder):
+                os.mkdir(save_folder)
+            path = os,path.join(save_folder,f"{device_id}-{start.strftime('%Y-%m-%d-%H-%M-%S')}")
             if not os.path.exists(path):
                 os.mkdir(path)
             device = dict(

--
Gitblit v1.9.1