wu_xinjun
2022-02-28 9411c4e47500e271d8fb88143c86e6dcb2d89684
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(