get_template_part、get_post_format函数使用

本文介绍了WordPress中使用get_template_part和get_post_format函数实现特定文章格式模板加载的方法。当文章格式为image时,会尝试加载content-image.php文件,若该文件不存在,则使用默认模板。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

get_template_part('slug',get_post_format());

get_template_part():根据get_post_format()返回的信息来加载slug开头命名的相应的模板;

get_post_format():获取当前post的分类信息。

例子:

现有模板 content-image.php

那么get_template_part('content',get_post_format());如果当前post分类是image,那么就会调用模板content-image.php;如果返回的post分类模板不存在,那么就会使用默认post模板。

这是app.py代码 import os import logging import traceback import time import threading from flask import Flask, render_template, request, redirect, url_for, send_from_directory, jsonify from werkzeug.utils import secure_filename app = Flask(__name__) app.config['UPLOAD_FOLDER'] = 'uploads' app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024 # 100MB app.config['SECRET_KEY'] = 'your_strong_secret_key_here' app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'] = '/shared/files' # 局域网文件共享目录 app.config['FILE_CLEANUP_INTERVAL'] = 3600 # 文件清理间隔(秒) # 确保上传目录和共享目录存在 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True) os.makedirs(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], exist_ok=True) # 配置详细日志记录 logging.basicConfig( filename='app.log', level=logging.INFO, format='[%(asctime)s] %(levelname)s in %(module)s: %(message)s' ) app.logger = logging.getLogger(__name__) # 文件系统锁确保线程安全 file_lock = threading.Lock() def cleanup_old_files(): """定期清理临时上传目录中的旧文件""" while True: try: with file_lock: now = time.time() for filename in os.listdir(app.config['UPLOAD_FOLDER']): file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename) if os.path.isfile(file_path): file_age = now - os.path.getctime(file_path) if file_age > 24 * 3600: # 删除超过1天的临时文件 os.remove(file_path) app.logger.info(f"Cleaned up old file: {filename}") except Exception as e: app.logger.error(f"File cleanup error: {str(e)}") time.sleep(app.config['FILE_CLEANUP_INTERVAL']) # 启动文件清理线程 cleanup_thread = threading.Thread(target=cleanup_old_files, daemon=True) cleanup_thread.start() @app.errorhandler(500) def internal_error(error): """自定义500错误处理页面""" app.logger.error(f"500 Error: {error}\n{traceback.format_exc()}") return render_template('500.html', error=traceback.format_exc()), 500 @app.errorhandler(404) def not_found_error(error): """自定义404错误处理页面""" return render_template('404.html', error=error.description), 404 @app.route('/') def index(): """安全的主页路由,显示共享文件列表""" try: # 获取共享目录中的文件列表 files = [] with file_lock: for filename in os.listdir(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files']): file_path = os.path.join(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], filename) if os.path.isfile(file_path): files.append({ 'name': filename, 'size': os.path.getsize(file_path), 'modified': time.ctime(os.path.getmtime(file_path)) }) return render_template('index.html', files=files) except Exception as e: app.logger.error(f"Error in index route: {str(e)}") return render_template('500.html', error=str(e)), 500 @app.route('/upload', methods=['POST']) def upload_file(): """安全的文件上传处理""" try: if 'file' not in request.files: app.logger.warning("No file part in request") return jsonify({'status': 'error', 'message': 'No file part'}), 400 file = request.files['file'] if file.filename == '': app.logger.warning("No selected file") return jsonify({'status': 'error', 'message': 'No selected file'}), 400 if file: # 安全处理文件名 filename = secure_filename(file.filename) file_path = os.path.join(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], filename) # 保存文件到共享目录 with file_lock: file.save(file_path) app.logger.info(f"File uploaded: {filename}") return jsonify({'status': 'success', 'filename': filename}) except Exception as e: app.logger.error(f"Upload error: {str(e)}\n{traceback.format_exc()}") return jsonify({'status': 'error', 'message': str(e)}), 500 @app.route('/download/<filename>') def download_file(filename): """文件下载功能""" try: # 验证文件名是否安全 safe_filename = secure_filename(filename) file_path = os.path.join(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], safe_filename) # 检查文件是否存在 if not os.path.exists(file_path): app.logger.warning(f"File not found: {safe_filename}") return render_template('404.html', filename=safe_filename), 404 # 提供文件下载 return send_from_directory( app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], safe_filename, as_attachment=True ) except Exception as e: app.logger.error(f"Download error: {str(e)}\n{traceback.format_exc()}") return render_template('500.html', error=str(e)), 500 @app.route('/delete/<filename>', methods=['POST']) def delete_file(filename): """删除文件功能""" try: # 验证文件名是否安全 safe_filename = secure_filename(filename) file_path = os.path.join(app.config['C:\\Users\21904\PycharmProjects\PythonProject\.venv\your_project\\uploads\shared\files'], safe_filename) # 检查文件是否存在 if not os.path.exists(file_path): app.logger.warning(f"File not found for deletion: {safe_filename}") return jsonify({'status': 'error', 'message': 'File not found'}), 404 # 删除文件 with file_lock: os.remove(file_path) app.logger.info(f"File deleted: {safe_filename}") return jsonify({'status': 'success', 'filename': safe_filename}) except Exception as e: app.logger.error(f"Delete error: {str(e)}\n{traceback.format_exc()}") return jsonify({'status': 'error', 'message': str(e)}), 500 if __name__ == '__main__': app.run(host='0.0.0.0', port=5000, debug=False)
最新发布
08-14
# E:\AI_System\core\config.py import os import json import logging from pathlib import Path from typing import Dict, Any, Optional, Union import hashlib import sys class CoreConfig: """核心配置管理器 - Windows兼容版""" def __init__(self, config_path: Union[str, Path] = None, env_prefix: str = "", default_config: Dict = None): # 确定基础目录 - 修复AttributeError self.BASE_DIR = self._determine_base_dir() # 初始化参数 self.config_path = Path(config_path) if config_path else None self.env_prefix = env_prefix self.default_config = default_config or {} self._config = { 'BASE_DIR': str(self.BASE_DIR), # 确保BASE_DIR在配置中 'LOG_DIR': str(self.BASE_DIR / "logs"), 'CONFIG_DIR': str(self.BASE_DIR / "config"), 'CACHE_DIR': str(self.BASE_DIR / "cache"), 'DATA_DIR': str(self.BASE_DIR / "data"), 'MODEL_DIR': str(self.BASE_DIR / "models"), } self._last_hash = None # 设置安全的日志器 - 避免Unicode错误 self.logger = self._create_safe_logger() # 初始化配置 self._initialize() def _determine_base_dir(self) -> Path: """确定基础目录""" # 1. 尝试从环境变量获取 base_dir_env = os.getenv(f"{self.env_prefix}BASE_DIR") if base_dir_env: return Path(base_dir_env).resolve() # 2. 尝试基于当前文件位置计算 try: return Path(__file__).parent.parent.resolve() except NameError: # 在交互式环境中运行时 return Path.cwd().resolve() # 3. 默认使用当前工作目录 return Path.cwd().resolve() def _create_safe_logger(self) -> logging.Logger: """创建安全的日志器(Windows兼容)""" logger = logging.getLogger('CoreConfig') logger.setLevel(logging.INFO) # 创建安全的控制台处理器 class SafeConsoleHandler(logging.StreamHandler): def emit(self, record): try: msg = self.format(record) # 替换或移除Windows不支持的Unicode字符 safe_msg = msg.encode('utf-8', errors='replace').decode('utf-8') stream = self.stream stream.write(safe_msg + self.terminator) self.flush() except Exception: self.handleError(record) console_handler = SafeConsoleHandler() console_handler.setFormatter(logging.Formatter( '%(asctime)s - %(name)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S' )) logger.addHandler(console_handler) return logger def _initialize(self): """初始化配置""" self.logger.info("初始化配置管理器 | 环境前缀: %s | 基础目录: %s", self.env_prefix, self.BASE_DIR) # 加载环境变量 self._load_environment() # 加载配置文件 if self.config_path and self.config_path.exists(): self._load_config_file() elif self.config_path: self.logger.warning("配置文件不存在: %s", self.config_path) # 合并默认配置 self._merge_defaults() # 创建必要目录 self._create_directories() self.logger.info("配置加载完成 | 条目数: %d", len(self._config)) def _load_environment(self): """加载环境变量""" self.logger.info("加载环境变量...") # 加载系统环境变量 for key, value in os.environ.items(): if key.startswith(self.env_prefix): config_key = key[len(self.env_prefix):].lower() self._config[config_key] = value def _load_config_file(self): """加载配置文件""" file_type = self.config_path.suffix.lower() self.logger.info("加载配置文件: %s (类型: %s)", self.config_path, file_type[1:]) try: if file_type == '.json': with open(self.config_path, 'r', encoding='utf-8') as f: config_data = json.load(f) elif file_type in ('.yaml', '.yml'): import yaml with open(self.config_path, 'r', encoding='utf-8') as f: config_data = yaml.safe_load(f) elif file_type == '.ini': import configparser parser = configparser.ConfigParser() parser.read(self.config_path, encoding='utf-8') config_data = {} for section in parser.sections(): config_data[section] = dict(parser.items(section)) else: raise ValueError(f"不支持的配置文件类型: {file_type}") # 合并配置 self._deep_merge(self._config, config_data) self._last_hash = self._calculate_hash() except Exception as e: self.logger.error("加载配置文件失败: %s", str(e)) def _merge_defaults(self): """合并默认配置""" self.logger.info("合并默认配置...") self._deep_merge(self._config, self.default_config) def _deep_merge(self, target: Dict, source: Dict): """深度合并字典""" for key, value in source.items(): if (key in target and isinstance(target[key], dict) and isinstance(value, dict)): self._deep_merge(target[key], value) else: target[key] = value def _calculate_hash(self) -> str: """计算配置哈希值""" return hashlib.md5(json.dumps(self._config, sort_keys=True).encode()).hexdigest() def _create_directories(self): """创建配置中指定的目录""" dir_keys = ['log_dir', 'cache_dir', 'model_dir', 'data_dir'] for key in dir_keys: path = self.get_path(key) if not path.exists(): path.mkdir(parents=True, exist_ok=True) self.logger.info("创建目录: %s", path) def get(self, key: str, default: Any = None) -> Any: """安全获取配置值""" keys = key.split('.') current = self._config for k in keys: if isinstance(current, dict) and k in current: current = current[k] else: return default return current def get_path(self, key: str, default: Optional[Union[str, Path]] = None) -> Path: """获取路径配置项""" path_str = self.get(key, default) if path_str is None: return self.BASE_DIR / "default" # 安全回退 path = Path(path_str) if not path.is_absolute(): return self.BASE_DIR / path return path def __getattr__(self, name: str) -> Any: """支持点操作符访问配置项""" if name in self._config: return self._config[name] # 支持嵌套属性访问 parts = name.split('.') current = self._config for part in parts: if part in current: current = current[part] else: raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'") return current # E:\AI_System\web_ui\server.py import sys import os import time import logging import json import traceback import threading import platform import psutil import datetime from pathlib import Path from functools import wraps from concurrent.futures import ThreadPoolExecutor from logging.handlers import RotatingFileHandler # ========== 安全工具函数 ========== def safe_path(path_str: str) -> Path: """安全处理路径字符串,确保跨平台兼容性""" try: path = Path(path_str) if not path.is_absolute(): # 如果路径是相对的,基于项目根目录解析 base_dir = Path(__file__).parent.parent.resolve() return (base_dir / path).resolve() return path.resolve() except Exception as e: logging.error(f"路径解析错误: {path_str} - {str(e)}") return Path.cwd().resolve() class SafeConsoleHandler(logging.StreamHandler): """Windows安全的控制台日志处理器""" def emit(self, record): try: msg = self.format(record) # 安全处理Unicode字符 safe_msg = msg.encode('utf-8', errors='replace').decode('utf-8') stream = self.stream stream.write(safe_msg + self.terminator) self.flush() except Exception: self.handleError(record) # ========== 配置系统初始化 ========== # 添加核心模块路径 sys.path.insert(0, str(Path(__file__).parent.parent)) # 导入核心配置 from core.config import CoreConfig as SystemConfig # 全局配置实例 system_config = SystemConfig() # ========== 安全日志系统 ========== def setup_logger() -> logging.Logger: """配置并返回安全的日志系统""" logger = logging.getLogger('WebServer') # 清除所有现有处理器 for handler in logger.handlers[:]: logger.removeHandler(handler) # 安全获取配置项 debug_mode = system_config.get('DEBUG', False) log_level = logging.DEBUG if debug_mode else logging.INFO log_file = system_config.get('logging.file', 'logs/web_server.log') max_log_size = system_config.get('logging.max_size', 10) # MB backup_count = system_config.get('logging.backup_count', 5) # 设置日志级别 logger.setLevel(log_level) # 创建安全的控制台处理器 console_handler = SafeConsoleHandler() console_handler.setLevel(log_level) # 创建安全的文件处理器 try: # 确保日志目录存在 log_path = safe_path(log_file) log_path.parent.mkdir(parents=True, exist_ok=True) file_handler = RotatingFileHandler( filename=str(log_path), maxBytes=max_log_size * 1024 * 1024, backupCount=backup_count, encoding='utf-8' ) file_handler.setLevel(logging.DEBUG) except Exception as e: logging.error(f"无法创建文件日志处理器: {str(e)}") file_handler = logging.NullHandler() # 创建安全的格式化器 formatter = logging.Formatter( '%(asctime)s - %(name)s - %(levelname)s - %(process)d - %(thread)d - %(message)s', datefmt='%Y-%m-%d %H:%M:%S' ) # 应用格式化器 console_handler.setFormatter(formatter) file_handler.setFormatter(formatter) # 添加处理器 logger.addHandler(console_handler) logger.addHandler(file_handler) # 设置根日志器 root_logger = logging.getLogger() root_logger.setLevel(log_level) for handler in root_logger.handlers[:]: root_logger.removeHandler(handler) root_logger.addHandler(console_handler) root_logger.addHandler(file_handler) # 配置第三方库日志器 third_party_loggers = ['werkzeug', 'engineio', 'socketio', 'urllib3'] for log_name in third_party_loggers: lib_logger = logging.getLogger(log_name) lib_logger.setLevel(logging.WARNING) lib_logger.propagate = False lib_logger.addHandler(file_handler) # 安全记录日志系统初始化完成 logger.info("日志系统初始化完成") logger.info("日志级别: %s", 'DEBUG' if debug_mode else 'INFO') logger.info("日志文件: %s", log_file) return logger # 初始化日志系统 logger = setup_logger() # ========== Flask应用初始化 ========== from flask import Flask, render_template, request, jsonify, send_from_directory, current_app # 安全获取模板和静态文件路径 template_dir = safe_path(system_config.get('template_dir', Path(__file__).parent / 'templates')) static_dir = safe_path(system_config.get('static_dir', Path(__file__).parent / 'static')) app = Flask( __name__, template_folder=str(template_dir), static_folder=str(static_dir), static_url_path='/static' ) # 设置Flask调试模式 app.debug = system_config.get('DEBUG', False) app.secret_key = system_config.get('SECRET_KEY', 'default_secret_key') # ========== 环境管理器 ========== class EnvironmentManager: """跨平台兼容的环境管理器""" def __init__(self): self.state = { 'temperature': 22.5, 'humidity': 45.0, 'light_level': 75, 'objects': [], 'last_updated': datetime.datetime.now().isoformat() } self.healthy = True self.lock = threading.Lock() self.running = False self.update_thread = None self.observers = [] logger.info("环境管理器初始化成功") def start(self): """启动环境状态更新线程""" if self.running: return self.running = True self.update_thread = threading.Thread(target=self._update_loop, daemon=True) self.update_thread.start() logger.info("环境管理器已启动") def stop(self): """停止环境管理器""" self.running = False if self.update_thread and self.update_thread.is_alive(): self.update_thread.join(timeout=5) logger.info("环境管理器已停止") def register_observer(self, callback): """注册状态变更观察者""" self.observers.append(callback) logger.debug("注册环境状态观察者: %s", callback.__name__) def _notify_observers(self): """通知所有观察者状态变更""" for observer in self.observers: try: observer(self.state) except Exception as e: logger.error("通知环境观察者失败: %s", str(e)) def _update_loop(self): """环境状态更新循环""" while self.running: try: with self.lock: # 模拟环境变化 self.state['temperature'] = round(20 + 5 * (time.time() % 10) / 10, 1) self.state['humidity'] = round(40 + 10 * (time.time() % 10) / 10, 1) self.state['light_level'] = round(70 + 10 * (time.time() % 10) / 10, 1) self.state['last_updated'] = datetime.datetime.now().isoformat() # 通知观察者 self._notify_observers() except Exception as e: logger.error("环境更新失败: %s", str(e)) time.sleep(1.0) def get_state(self): """获取当前环境状态""" with self.lock: return self.state.copy() def execute_action(self, action, params): """执行环境动作""" logger.info("执行环境动作: %s 参数: %s", action, params) try: if action == "adjust_temperature": return self._adjust_temperature(params) elif action == "adjust_light": return self._adjust_light(params) elif action == "add_object": return self._add_object(params) elif action == "remove_object": return self._remove_object(params) else: logger.warning("未知环境动作: %s", action) return False except Exception as e: logger.error("执行环境动作失败: %s", str(e)) return False def _adjust_temperature(self, params): """调整温度""" value = params.get('value') if value is None: logger.warning("缺少温度值参数") return False try: value = float(value) except ValueError: logger.warning("无效的温度值: %s", value) return False if 10 <= value <= 40: with self.lock: self.state['temperature'] = value return True logger.warning("温度值超出范围 (10-40): %s", value) return False def _adjust_light(self, params): """调整光照强度""" level = params.get('level') if level is None: logger.warning("缺少光照强度参数") return False try: level = float(level) except ValueError: logger.warning("无效的光照强度: %s", level) return False if 0 <= level <= 100: with self.lock: self.state['light_level'] = level return True logger.warning("光照强度超出范围 (0-100): %s", level) return False def _add_object(self, params): """添加环境对象""" obj_name = params.get('name') if not obj_name: logger.warning("缺少对象名称") return False obj_type = params.get('type', 'object') position = params.get('position', 'unknown') # 检查是否已存在 with self.lock: for obj in self.state['objects']: if obj['name'] == obj_name: logger.warning("对象已存在: %s", obj_name) return False # 添加新对象 self.state['objects'].append({ 'name': obj_name, 'type': obj_type, 'position': position, 'added_at': datetime.datetime.now().isoformat() }) return True def _remove_object(self, params): """移除环境对象""" obj_name = params.get('name') if not obj_name: logger.warning("缺少对象名称") return False # 查找并移除对象 with self.lock: for i, obj in enumerate(self.state['objects']): if obj['name'] == obj_name: del self.state['objects'][i] return True logger.warning("未找到对象: %s", obj_name) return False def is_healthy(self): """检查环境健康状态""" # 简单检查:确保所有值在合理范围内 with self.lock: temp = self.state['temperature'] light = self.state['light_level'] if temp < 10 or temp > 40: logger.warning("温度超出健康范围: %s℃", temp) return False if light < 0 or light > 100: logger.warning("光照强度超出健康范围: %s%%", light) return False return self.healthy and self.running # ========== 系统初始化器 ========== class SystemInitializer: """健壮的系统初始化器""" def __init__(self): self.start_time = time.time() self.components = { 'ai_core': None, 'hardware_manager': None, 'life_scheduler': None, 'ai_agent': None, 'environment_manager': None } # 安全获取基础目录 try: self.base_dir = safe_path(system_config.get('BASE_DIR', Path(__file__).parent.parent)) logger.info("系统初始化器创建, 基础目录: %s", self.base_dir) except Exception as e: self.base_dir = Path.cwd().resolve() logger.error("获取基础目录失败, 使用当前目录: %s | 错误: %s", self.base_dir, str(e)) def initialize_system_paths(self): """安全初始化系统路径""" try: # 添加项目根目录到 sys.path if str(self.base_dir) not in sys.path: sys.path.insert(0, str(self.base_dir)) logger.info("项目根目录: %s", self.base_dir) # 添加子目录 sub_dirs = ['agent', 'core', 'utils', 'config', 'cognitive_arch', 'environment'] for sub_dir in sub_dirs: full_path = self.base_dir / sub_dir if full_path.exists() and full_path.is_dir(): if str(full_path) not in sys.path: sys.path.insert(0, str(full_path)) logger.info("添加路径: %s", full_path) else: logger.warning("目录不存在: %s - 已跳过", full_path) except Exception as e: logger.error("系统路径初始化失败: %s", str(e)) def initialize_environment_manager(self): """初始化环境管理器""" try: self.components['environment_manager'] = EnvironmentManager() self.components['environment_manager'].start() logger.info("环境管理器初始化成功") return self.components['environment_manager'] except Exception as e: logger.error("环境管理器初始化失败: %s", str(e)) logger.error(traceback.format_exc()) logger.warning("环境交互功能将不可用") return None def initialize_ai_core(self): """初始化AI核心""" logger.info("AI核心初始化") # 简化实现 self.components['ai_core'] = type('AICore', (), { 'status': 'running', 'get_state': lambda: { "status": "running", "model": system_config.get('DEFAULT_MODEL', 'gpt-3.5-turbo'), "last_update": datetime.datetime.now().isoformat() } })() return self.components['ai_core'] def initialize_hardware_manager(self): """初始化硬件管理器""" logger.info("硬件管理器初始化") # 简化实现 self.components['hardware_manager'] = type('HardwareManager', (), { 'get_status': lambda: { "cpu_usage": psutil.cpu_percent(), "memory_usage": psutil.virtual_memory().percent, "gpu_usage": 0, "disk_usage": psutil.disk_usage('/').percent, "timestamp": datetime.datetime.now().isoformat() } })() return self.components['hardware_manager'] def initialize_life_scheduler(self): """初始化生活调度器""" logger.info("生活调度器初始化") # 简化实现 self.components['life_scheduler'] = type('LifeScheduler', (), { 'get_status': lambda: { "current_activity": "thinking", "next_activity": "learning", "energy": 85, "last_update": datetime.datetime.now().isoformat() } })() return self.components['life_scheduler'] def initialize_ai_agent(self): """初始化AI智能体""" logger.info("开始初始化AI智能体") try: # 动态导入智能体模块 from agent.autonomous_agent import AutonomousAgent # 创建智能体实例 self.components['ai_agent'] = AutonomousAgent() # 将环境管理器传递给智能体(如果存在) if self.components['environment_manager']: # 检查智能体是否有设置环境的方法 if hasattr(self.components['ai_agent'], 'set_environment'): self.components['ai_agent'].set_environment( self.components['environment_manager'] ) logger.info("已将环境管理器连接到智能体") else: logger.warning("智能体没有set_environment方法,无法连接环境管理器") # 启动智能体后台任务 if hasattr(self.components['ai_agent'], 'start'): self.components['ai_agent'].start() logger.info("智能体后台任务已启动") else: logger.warning("智能体没有start方法,无法启动后台任务") logger.info("AI智能体初始化完成") return True except ImportError: logger.warning("无法导入AutonomousAgent,使用模拟智能体") self._create_mock_agent() return False except Exception as e: logger.error("AI智能体初始化失败: %s", str(e)) logger.error(traceback.format_exc()) self._create_mock_agent() return False def _create_mock_agent(self): """创建模拟智能体""" logger.warning("使用模拟智能体作为回退方案") self.components['ai_agent'] = type('MockAgent', (), { 'process_input': lambda self, input, user_id: f"智能体初始化失败,使用模拟模式: 收到消息 '{input}'" })() def start_evolution_monitor(self): """启动进化监视器""" logger.info("进化监视器启动") # 简化实现 def monitor(): while True: logger.debug("进化监视器运行中...") time.sleep(60) threading.Thread(target=monitor, daemon=True).start() def initialize_all(self): """初始化所有系统组件""" logger.info("=" * 50) logger.info("开始初始化AI系统") logger.info("=" * 50) self.initialize_system_paths() self.initialize_ai_core() self.initialize_hardware_manager() self.initialize_life_scheduler() self.initialize_ai_agent() self.initialize_environment_manager() self.start_evolution_monitor() logger.info("所有系统组件初始化完成") return self.components # ========== WebSocket处理 ========== def setup_websocket_handlers(socketio, env_manager): if not socketio or not env_manager: return @socketio.on('connect') def handle_connect(): logger.info('客户端已连接') socketio.emit('system_status', {'status': 'ready'}) # 注册环境状态观察者 env_manager.register_observer(lambda state: socketio.emit('environment_update', state)) @socketio.on('disconnect') def handle_disconnect(): logger.info('客户端已断开连接') @socketio.on('user_message') def handle_user_message(data): user_id = data.get('user_id', 'guest') message = data.get('message', '') logger.info("收到来自 %s 的消息: %s", user_id, message) # 使用线程池处理消息 def process_message(): try: components = current_app.config['SYSTEM_COMPONENTS'] if components['ai_agent']: return components['ai_agent'].process_input(message, user_id) else: return f"已收到您的消息: {message}" except Exception as e: logger.error("消息处理失败: %s", str(e)) return "处理消息时出错" future = ThreadPoolExecutor().submit(process_message) try: response = future.result(timeout=10) socketio.emit('agent_response', { 'user_id': user_id, 'response': response }) except TimeoutError: socketio.emit('agent_response', { 'user_id': user_id, 'response': "处理超时,请重试" }) # ========== 路由注册 ========== def register_routes(app): @app.route('/') def home(): """根路由显示欢迎页面""" current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") return render_template('index.html', current_time=current_time, host=request.host, version="1.0") # 环境路由 @app.route('/environment') def environment_view(): return render_template('environment_view.html') @app.route('/api/environment/state', methods=['GET']) def get_environment_state(): env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager') if not env_manager: return jsonify({"success": False, "error": "环境管理器未初始化"}), 503 try: state = env_manager.get_state() return jsonify(state) except Exception as e: app.logger.error("获取环境状态失败: %s", traceback.format_exc()) return jsonify({"success": False, "error": str(e)}), 500 @app.route('/api/environment/action', methods=['POST']) def execute_environment_action(): env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager') if not env_manager: return jsonify({"success": False, "error": "环境管理器未初始化"}), 503 try: data = request.json action = data.get('action') params = data.get('params', {}) if not action: return jsonify({"success": False, "error": "缺少动作参数"}), 400 success = env_manager.execute_action(action, params) return jsonify({"success": success, "action": action}) except Exception as e: app.logger.error("执行环境动作失败: %s", traceback.format_exc()) return jsonify({"success": False, "error": str(e)}), 500 # 静态文件路由 @app.route('/static/<path:filename>') def static_files(filename): return send_from_directory(app.static_folder, filename) # 健康检查路由 @app.route('/health') def health_check(): components = app.config['SYSTEM_COMPONENTS'] status = { "ai_core": components['ai_core'] is not None, "hardware_manager": components['hardware_manager'] is not None, "life_scheduler": components['life_scheduler'] is not None, "ai_agent": components['ai_agent'] is not None, "environment_manager": components['environment_manager'] and components['environment_manager'].is_healthy(), "timestamp": datetime.datetime.now().isoformat() } return jsonify(status) # 系统状态路由 @app.route('/status') def status(): components = app.config['SYSTEM_COMPONENTS'] system_info = { "uptime": time.time() - app.config['START_TIME'], "ai_core_status": components['ai_core'].get_state() if components['ai_core'] else "uninitialized", "hardware_status": components['hardware_manager'].get_status() if components[ 'hardware_manager'] else "uninitialized", "life_scheduler_status": components['life_scheduler'].get_status() if components[ 'life_scheduler'] else "uninitialized", "environment_status": components['environment_manager'].get_state() if components[ 'environment_manager'] else "uninitialized", "platform": platform.platform(), "python_version": sys.version, "memory_usage": psutil.virtual_memory().percent, "cpu_usage": psutil.cpu_percent(), "thread_count": threading.active_count(), "process_id": os.getpid(), "timestamp": datetime.datetime.now().isoformat() } return jsonify(system_info) # ========== 错误处理 ========== def register_error_handlers(app): @app.errorhandler(404) def page_not_found(e): return render_template('404.html'), 404 @app.errorhandler(500) def internal_server_error(e): return render_template('500.html'), 500 @app.errorhandler(Exception) def handle_general_exception(e): logger.error("未处理异常: %s", str(e)) logger.error(traceback.format_exc()) return render_template('error.html', error=str(e)), 500 # ========== Web应用工厂 ========== def create_app(): # 创建Flask应用 try: template_dir = safe_path(system_config.get('template_dir', Path(__file__).parent / 'templates')) static_dir = safe_path(system_config.get('static_dir', Path(__file__).parent / 'static')) except Exception as e: logger.error("路径配置错误: %s", str(e)) template_dir = Path(__file__).parent / 'templates' static_dir = Path(__file__).parent / 'static' app = Flask( __name__, template_folder=str(template_dir), static_folder=str(static_dir), static_url_path='/static' ) app.secret_key = system_config.get('SECRET_KEY', 'default_secret_key') # 初始化系统组件 try: system_initializer = SystemInitializer() components = system_initializer.initialize_all() app.config['SYSTEM_COMPONENTS'] = components app.config['START_TIME'] = system_initializer.start_time except Exception as e: logger.error("系统初始化失败: %s", str(e)) app.config['SYSTEM_COMPONENTS'] = {} app.config['START_TIME'] = time.time() # 配置SocketIO socketio = None try: from flask_socketio import SocketIO socketio = SocketIO( app, async_mode=system_config.get('ASYNC_MODE', 'threading'), logger=logger.getEffectiveLevel() <= logging.DEBUG, engineio_logger=logger.getEffectiveLevel() <= logging.DEBUG ) env_manager = app.config['SYSTEM_COMPONENTS'].get('environment_manager') if env_manager: setup_websocket_handlers(socketio, env_manager) app.config['SOCKETIO'] = socketio except ImportError: logger.error("未安装flask-socketio,WebSocket功能不可用") except Exception as e: logger.error("SocketIO初始化失败: %s", str(e)) # 注册路由和错误处理 register_routes(app) register_error_handlers(app) return app, socketio # ========== 主程序入口 ========== if __name__ == '__main__': try: app, socketio = create_app() # 服务器配置 host = system_config.get('server.host', '0.0.0.0') port = system_config.get('server.port', 5000) env = os.environ.get('ENV', 'development') # 启动服务器 if env == 'production': # 生产环境使用Waitress服务器 from waitress import serve logger.info("生产服务器启动: http://%s:%s", host, port) serve(app, host=host, port=port, threads=8) else: # 开发环境使用内置服务器 logger.info("开发服务器启动: http://%s:%s", host, port) if socketio: socketio.run( app, host=host, port=port, debug=system_config.get('DEBUG', False), use_reloader=False ) else: app.run( host=host, port=port, debug=system_config.get('DEBUG', False), use_reloader=False ) except Exception as e: logger.critical("服务器启动失败: %s", str(e)) logger.critical(traceback.format_exc()) Microsoft Windows [版本 10.0.22631.2861] (c) Microsoft Corporation。保留所有权利。 C:\Users\Administrator>cd /d E:\AI_System\web_ui E:\AI_System\web_ui>python server.py Traceback (most recent call last): File "E:\AI_System\web_ui\server.py", line 53, in <module> from core.config import CoreConfig as SystemConfig File "E:\AI_System\core\__init__.py", line 12, in <module> system_config = CoreConfig( File "E:\AI_System\core\config.py", line 19, in __init__ self.BASE_DIR = self._determine_base_dir() File "E:\AI_System\core\config.py", line 44, in _determine_base_dir base_dir_env = os.getenv(f"{self.env_prefix}BASE_DIR") File "E:\AI_System\core\config.py", line 199, in __getattr__ if name in self._config: File "E:\AI_System\core\config.py", line 199, in __getattr__ if name in self._config: File "E:\AI_System\core\config.py", line 199, in __getattr__ if name in self._config: [Previous line repeated 982 more times] RecursionError: maximum recursion depth exceeded E:\AI_System\web_ui>
08-13
from flask import Flask, request, jsonify, send_from_directory from flask_cors import CORS import os import numpy as np import pandas as pd import logging import time import json import traceback from werkzeug.utils import secure_filename from new_algorithm import GasSensorDataAnalyzer, AlgorithmSelector, detect_dataset_type, extract_gas_type, extract_concentration, extract_sensor_type # 配置日志 logging.basicConfig( level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', handlers=[ logging.StreamHandler(), # 输出到控制台 logging.FileHandler('app.log', encoding='utf-8') # 输出到文件 ] ) logger = logging.getLogger(__name__) app = Flask(__name__) app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024 # 100MB 文件大小限制 # 配置CORS允许Vue前端访问 CORS(app, resources={ r"/*": { "origins": "http://localhost:5177", # 指定Vue前端地址 "methods": ["GET", "POST", "PUT", "DELETE", "OPTIONS"], "allow_headers": ["Content-Type", "Authorization"], "supports_credentials": True } }) # 配置上传文件夹 UPLOAD_FOLDER = 'uploads' if not os.path.exists(UPLOAD_FOLDER): os.makedirs(UPLOAD_FOLDER) app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER # 允许的文件扩展名 ALLOWED_EXTENSIONS = {'csv', 'xlsx', 'xls'} # 全局数据集存储 data_analyzer = GasSensorDataAnalyzer() X_combined = None y_combined = None gas_types = [] concentrations = [] sensor_types = [] last_activity = time.time() algorithm_selector = AlgorithmSelector(use_chinese=True) def allowed_file(filename): """检查文件扩展名是否合法""" return '.' in filename and \ filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS def save_and_extract_file_info(file): """保存文件并提取气体信息""" try: filename = secure_filename(file.filename) file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename) file.save(file_path) logger.info(f"Saved file: {file_path}") # 从文件名提取信息 sensor_type = extract_sensor_type(filename) gas_type = extract_gas_type(filename) concentration = extract_concentration(filename) return file_path, sensor_type, gas_type, concentration except Exception as e: logger.error(f"Error processing file {file.filename}: {str(e)}\n{traceback.format_exc()}") return None, None, None, None @app.route('/') def index(): """健康检查端点""" global last_activity status = { 'status': 'running', 'version': '1.0.0', 'last_activity': time.ctime(last_activity), 'endpoints': { '/upload': 'POST - Upload data files', '/analyze': 'POST - Analyze data', '/reset': 'POST - Reset data', '/columns': 'GET - Get dataset columns', '/status': 'GET - Service status' } } logger.info(f"Status request: {status}") return jsonify(status) @app.route('/status') def status(): """服务状态检查""" global X_combined, last_activity return jsonify({ 'status': 'active', 'timestamp': time.time(), 'dataset_loaded': X_combined is not None, 'dataset_shape': X_combined.shape if X_combined is not None else None, 'num_classes': len(np.unique(y_combined)) if y_combined is not None else 0 }) @app.route('/upload', methods=['POST']) def upload_files(): """处理文件上传""" global X_combined, y_combined, gas_types, concentrations, sensor_types, last_activity logger.info("Received upload request") # 检查是否有文件 if 'files' not in request.files: logger.error("No file part in request") return jsonify({'error': 'No file part'}), 400 files = request.files.getlist('files') if len(files) == 0 or files[0].filename == '': logger.error("No selected files") return jsonify({'error': 'No selected files'}), 400 # 过滤合法文件 valid_files = [f for f in files if allowed_file(f.filename)] if len(valid_files) < 2: logger.error("Need at least two files for analysis") return jsonify({'error': '需要至少两个文件进行分析'}), 400 # 保存文件并提取信息 file_paths = [] extracted_info = [] for file in valid_files: file_path, sensor_type, gas_type, concentration = save_and_extract_file_info(file) if file_path: file_paths.append(file_path) extracted_info.append({ 'sensor_type': sensor_type, 'gas_type': gas_type, 'concentration': concentration, 'filename': file.filename }) else: logger.error(f"Failed to process file: {file.filename}") if len(file_paths) < 2: logger.error("Failed to process enough files") return jsonify({ 'error': '文件处理失败,请检查文件格式', 'processed_files': [info['filename'] for info in extracted_info], 'details': extracted_info }), 500 # 准备数据加载参数 sensor_types = [info['sensor_type'] for info in extracted_info] gas_types = [info['gas_type'] for info in extracted_info] concentrations = [info['concentration'] for info in extracted_info] # 加载数据 try: X_combined, y_combined = data_analyzer.load_multiple_gas_data( file_paths, gas_types, concentrations, sensor_types ) if X_combined is None or len(X_combined) == 0: logger.error("Failed to load data from files") return jsonify({ 'error': '数据加载失败,请检查文件内容', 'file_details': extracted_info }), 500 logger.info(f"Loaded combined data: {len(X_combined)} samples, {X_combined.shape[1]} features, {len(np.unique(y_combined))} classes") # 获取多维度标签信息 label_info = [] for label in np.unique(y_combined): # 查找对应的标签信息 for key, label_id in data_analyzer.multi_dimension_labels.items(): if label_id == label: parts = key.split('_') if len(parts) >= 3: sensor = parts[0] gas = parts[1] conc = parts[2].replace('ppm', '') try: _, name_dict = data_analyzer.get_or_create_multi_dimension_label(sensor, gas, int(conc)) label_info.append({ 'id': int(label), 'sensor': sensor, 'gas': gas, 'concentration': conc, 'name_cn': name_dict['cn'], 'name_en': name_dict['en'] }) except Exception as e: logger.error(f"Error getting label info for {key}: {str(e)}") label_info.append({ 'id': int(label), 'sensor': sensor, 'gas': gas, 'concentration': conc, 'name_cn': f"未知标签 {label}", 'name_en': f"Unknown Label {label}" }) else: logger.warning(f"Invalid key format: {key}") # 更新最后活动时间 last_activity = time.time() # 返回成功响应 response = { 'message': f'成功上传并合并 {len(file_paths)} 个文件', 'sample_count': len(X_combined), 'feature_count': X_combined.shape[1], 'num_classes': len(np.unique(y_combined)), 'gas_types': gas_types, 'concentrations': concentrations, 'sensor_types': sensor_types, 'label_info': label_info } return jsonify(response), 200 except Exception as e: logger.error(f"Error loading data: {str(e)}\n{traceback.format_exc()}") return jsonify({ 'error': f'数据加载错误: {str(e)}', 'traceback': traceback.format_exc().splitlines() }), 500 @app.route('/analyze', methods=['POST']) def analyze_data(): """执行数据分析""" global X_combined, y_combined, algorithm_selector, last_activity logger.info("Received analyze request") # 检查数据是否已加载 if X_combined is None or y_combined is None: logger.error("No dataset available") return jsonify({'error': '没有可用数据,请先上传文件'}), 400 # 获取前端传递的算法参数 try: data = request.get_json() if not data or 'params' not in data: logger.error("Invalid request parameters") return jsonify({'error': '无效请求参数'}), 400 params = data.get('params', {}) # 设置算法参数 for algo_name, algo_params in params.items(): if algo_name in algorithm_selector.algorithms: logger.info(f"Setting parameters for {algo_name}: {algo_params}") algorithm_selector.set_algorithm_params(algo_name, algo_params) except Exception as e: logger.error(f"Error parsing JSON data: {str(e)}\n{traceback.format_exc()}") return jsonify({'error': 'JSON数据解析错误'}), 400 # 训练模型 try: logger.info(f"Starting model training with {len(X_combined)} samples and {len(np.unique(y_combined))} classes") results = algorithm_selector.train_models(X_combined, y_combined) logger.info("Algorithm training completed") except Exception as e: logger.error(f"Error training models: {str(e)}\n{traceback.format_exc()}") return jsonify({'error': f'模型训练错误: {str(e)}'}), 500 # 提取需要返回的结果(关键修复) response_results = {} for algo_name, result in results.items(): # 如果训练出错,记录错误信息 if 'error' in result: response_results[algo_name] = { 'name': result['name'], 'error': result['error'] } else: # 转换分类报告为字符串(如果存在) if 'classification_report' in result: report = result['classification_report'] if isinstance(report, dict): # 将分类报告字典转换为格式化的字符串 report_str = "\n".join([f"{key}: {value}" for key, value in report.items()]) else: report_str = str(report) else: report_str = "无分类报告" # 提取特征重要性(如果存在) feature_importances = None if 'feature_importances' in result and result['feature_importances'] is not None: # 确保特征重要性是字典格式 if isinstance(result['feature_importances'], dict): feature_importances = result['feature_importances'] elif hasattr(result['model'], 'feature_importances_'): # 尝试从模型中提取特征重要性 try: feature_importances = dict(zip( result['model'].feature_names_in_, result['model'].feature_importances_ )) except Exception as e: logger.warning(f"无法提取特征重要性: {str(e)}") feature_importances = None response_results[algo_name] = { 'name': result['name'], 'train_accuracy': result['train_accuracy'], 'test_accuracy': result['test_accuracy'], 'classification_report': report_str, 'feature_importances': feature_importances } # 更新最后活动时间 last_activity = time.time() return jsonify({ 'message': '分析成功完成', 'results': response_results, 'num_classes': len(np.unique(y_combined)), 'sample_count': len(X_combined) }), 200 @app.route('/reset', methods=['POST']) def reset_data(): """重置数据集""" global X_combined, y_combined, gas_types, concentrations, sensor_types, last_activity X_combined = None y_combined = None gas_types = [] concentrations = [] sensor_types = [] last_activity = time.time() logger.info("Data reset") return jsonify({'message': '数据已重置'}), 200 @app.route('/uploads/<filename>', methods=['GET']) def uploaded_file(filename): """访问上传的文件""" return send_from_directory(app.config['UPLOAD_FOLDER'], filename) if __name__ == '__main__': # 开发环境 - 允许Vue前端访问 app.run(host='0.0.0.0', port=5000, debug=True) 这是你上次修改的代码,上次的代码还能够上传数据,这个功能也要保留的同时,能够分析数据
06-22
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值