import redis


class RedisClient:

    def __init__(self, config_path="config.yaml"):
        self.redis = redis.Redis(
            "10.10.12.101",
            6379,
            0,
            decode_responses=True,
        )

    def set_online(self, device_id: str):
        key = f"device:online:{device_id}"
        self.redis.set(key, 1)

    def set_offline(self, device_id: str):
        key = f"device:online:{device_id}"
        self.redis.delete(key)

    def is_device_online(self, device_id: str) -> bool:
        key = f"device:online:{device_id}"
        return self.redis.exists(key) == 1