config_manager.py

import json
import os
import random
import re
import string
import requests  # 添加 requests 库以支持 HTTP 请求

class ConfigManager:
    CONFIG_FILE = "config.json"

    def __init__(self):
        self.config = self.load_config()

    def load_config(self):
        if os.path.exists(self.CONFIG_FILE):
            with open(self.CONFIG_FILE, 'r') as file:
                return json.load(file)
        return {
            "server_address": "",
            "api_key": "",
            "device_id": self.generate_device_id(),
            "devices": [],
            "time_tables": [],
            "location_id": ""  # 添加location_id默认值
        }

    def save_config(self):
        with open(self.CONFIG_FILE, 'w') as file:
            json.dump(self.config, file, indent=4)

    def generate_device_id(self):
        return ''.join(random.choices(string.ascii_uppercase + string.digits, k=16))

    def set_server_address(self, address):
        self.config["server_address"] = address

    def set_api_key(self, key):
        self.config["api_key"] = key

    def set_remote_video_url(self, url):
        self.config["remote_video_url"] = url

    def set_device_id(self, device_id):
        if re.match(r'^[A-Z0-9]{16}$', device_id):
            self.config["device_id"] = device_id
        else:
            raise ValueError("Device ID must be 16 characters long and contain only uppercase letters and digits.")

    def add_device(self, device_id):
        if re.match(r'^[A-Z0-9]{16}$', device_id) and device_id not in self.config["devices"]:
            self.config["devices"].append({"device_id": device_id})

    def remove_device(self, device_id):
        self.config["devices"] = [device for device in self.config["devices"] if device["device_id"] != device_id]

    def is_valid_device(self, device_id):
        return any(device["device_id"] == device_id for device in self.config["devices"])

    def set_time_tables(self, time_tables):
        self.config["time_tables"] = time_tables

    def get_config(self):
        return self.config

    def fetch_user_data(self):
        server_address = self.config["server_address"]
        api_key = self.config["api_key"]
        url = f"{server_address}api/get_user_data?user_token={api_key}"
        try:
            print(f"Fetching user data from {url}")  # 添加日志记录
            response = requests.get(url)
            response.raise_for_status()
            data = response.json()
            # 提取云端数据中的 devices 和 time_tables
            cloud_data = data.get("data", {})
            devices = cloud_data.get("devices", [])
            time_tables = cloud_data.get("time_tables", [])
            # 更新本地配置
            self.set_devices(devices)
            self.set_time_tables(time_tables)
            self.save_config()
            print("User data fetched and saved successfully")  # 添加日志记录
        except requests.exceptions.RequestException as e:
            print(f"Error fetching user data: {e}")

    def set_devices(self, devices):
        self.config["devices"] = devices

    def set_time_tables(self, time_tables):
        self.config["time_tables"] = time_tables

    def set_location_id(self, location_id):
        self.config["location_id"] = location_id