import datetime
import os
import pika

class DeviceManager:
    def __init__(self, host, user, password):
        self.host = host
        self.user = user
        self.password = password
        self._devices = {}

    def _create_controll_channel(self, device_id: str):
        credentials = pika.PlainCredentials(self.user, self.password)
        parameters = pika.ConnectionParameters(
            credentials=credentials, host=self.host)
        # 连接队列服务器
        connection = pika.BlockingConnection(parameters)
        channel = connection.channel()
        channel.queue_declare(queue=device_id + "_callback")
        return channel

    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(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(
                power_on=True, 
                srart=start, 
                path=path, 
                images=0, 
                micro_wave=0,
                controll_channel=self._create_controll_channel(device_id)
                )
            self._devices[device_id] = device
        return device