From 1f55f65fa0e1891b0a7b41af605bea1259cedfd6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B5=B5=E6=9D=B0?= Date: Mon, 22 Sep 2025 14:40:25 +0100 Subject: [PATCH] =?UTF-8?q?feat:=20=E8=87=AA=E5=8A=A8=E6=8F=90=E4=BA=A4=20?= =?UTF-8?q?-=20=E5=91=A8=E4=B8=80=202025/09/22=2014:40:25.43?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- init_database.py | 174 +++++++++++++++++++++++------------------------ 1 file changed, 87 insertions(+), 87 deletions(-) diff --git a/init_database.py b/init_database.py index 56fdea3..38fd940 100644 --- a/init_database.py +++ b/init_database.py @@ -63,15 +63,15 @@ class DatabaseInitializer: print(f"🔗 连接地址: {self.db_url}") print(f"⏰ 初始化时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print("=" * 80) - - try: - # 设置日志 - setup_logging(Config.LOG_LEVEL, Config.LOG_FILE) - # 测试数据库连接 + try: + # 设置日志 + setup_logging(Config.LOG_LEVEL, Config.LOG_FILE) + + # 测试数据库连接 if not self._test_connection(): - return False - + return False + # 检查是否需要重置数据库 if force_reset: if not self._reset_database(): @@ -84,8 +84,8 @@ class DatabaseInitializer: # 执行数据库迁移 if not self._run_migrations(): return False - - # 插入初始数据 + + # 插入初始数据 if not self._insert_initial_data(): return False @@ -99,9 +99,9 @@ class DatabaseInitializer: print("\n" + "=" * 80) print("🎉 数据库初始化完成！") print("=" * 80) - return True - - except Exception as e: + return True + + except Exception as e: print(f"\n❌ 数据库初始化失败: {e}") self.logger.error(f"数据库初始化失败: {e}", exc_info=True) return False @@ -154,7 +154,7 @@ class DatabaseInitializer: if created_tables: print(f"✅ 新创建表: {', '.join(created_tables)}") - else: + else: print("✅ 所有表已存在") return True @@ -220,17 +220,17 @@ class DatabaseInitializer: # 检查表是否存在 inspector = inspect(db_manager.engine) if 'vehicle_data' not in inspector.get_table_names(): - print(" ➕ 创建vehicle_data表...") - VehicleData.__table__.create(session.bind, checkfirst=True) - print(" ✅ vehicle_data表创建成功") - else: - print(" ✅ vehicle_data表已存在") - - session.commit() + print(" ➕ 创建vehicle_data表...") + VehicleData.__table__.create(session.bind, checkfirst=True) + print(" ✅ vehicle_data表创建成功") + else: + print(" ✅ vehicle_data表已存在") + + session.commit() return True - except Exception as e: + except Exception as e: print(f" ❌ 车辆数据表迁移失败: {e}") - return False + return False def _migrate_conversation_enhancements(self) -> bool: """迁移对话增强字段""" @@ -357,9 +357,9 @@ class DatabaseInitializer: else: print(f" 📊 所有字段都已存在，跳过 {skipped_count} 个字段") - return True - - except Exception as e: + return True + + except Exception as e: print(f" ❌ 添加字段过程失败: {e}") return False @@ -391,15 +391,15 @@ class DatabaseInitializer: return result[0] > 0 except Exception: - return False + return False def _insert_initial_data(self) -> bool: - """插入初始数据""" + """插入初始数据""" print("\n📊 插入初始数据...") - try: - with db_manager.get_session() as session: - # 检查是否已有数据 + try: + with db_manager.get_session() as session: + # 检查是否已有数据 existing_count = session.query(KnowledgeEntry).count() if existing_count > 0: print(f" ✅ 数据库中已有 {existing_count} 条知识库条目，跳过初始数据插入") @@ -541,53 +541,53 @@ class DatabaseInitializer: ] def _add_sample_vehicle_data(self) -> bool: - """添加示例车辆数据""" - try: - from src.vehicle.vehicle_data_manager import VehicleDataManager - - vehicle_manager = VehicleDataManager() - success = vehicle_manager.add_sample_vehicle_data() - - if success: - print(" ✅ 示例车辆数据添加成功") - else: - print(" ❌ 示例车辆数据添加失败") + """添加示例车辆数据""" + try: + from src.vehicle.vehicle_data_manager import VehicleDataManager + vehicle_manager = VehicleDataManager() + success = vehicle_manager.add_sample_vehicle_data() + + if success: + print(" ✅ 示例车辆数据添加成功") + else: + print(" ❌ 示例车辆数据添加失败") + return success - except Exception as e: - print(f" ❌ 添加示例车辆数据失败: {e}") + except Exception as e: + print(f" ❌ 添加示例车辆数据失败: {e}") return False def _verify_existing_knowledge(self) -> bool: - """验证现有的知识库条目""" - try: - with db_manager.get_session() as session: - # 获取所有未验证的知识库条目 - unverified_entries = session.query(KnowledgeEntry).filter( - KnowledgeEntry.is_verified == False - ).all() - - if unverified_entries: - print(f" 📝 发现 {len(unverified_entries)} 条未验证的知识库条目") + """验证现有的知识库条目""" + try: + with db_manager.get_session() as session: + # 获取所有未验证的知识库条目 + unverified_entries = session.query(KnowledgeEntry).filter( + KnowledgeEntry.is_verified == False + ).all() - # 将现有的知识库条目标记为已验证 - for entry in unverified_entries: - entry.is_verified = True - entry.verified_by = "system_init" - entry.verified_at = datetime.now() + if unverified_entries: + print(f" 📝 发现 {len(unverified_entries)} 条未验证的知识库条目") + + # 将现有的知识库条目标记为已验证 + for entry in unverified_entries: + entry.is_verified = True + entry.verified_by = "system_init" + entry.verified_at = datetime.now() if not hasattr(entry, 'search_frequency'): entry.search_frequency = 0 if not hasattr(entry, 'relevance_score'): entry.relevance_score = 0.7 - - session.commit() - print(f" ✅ 成功验证 {len(unverified_entries)} 条知识库条目") - else: - print(" ✅ 所有知识库条目已验证") - + + session.commit() + print(f" ✅ 成功验证 {len(unverified_entries)} 条知识库条目") + else: + print(" ✅ 所有知识库条目已验证") + return True - except Exception as e: - print(f" ❌ 验证知识库条目失败: {e}") + except Exception as e: + print(f" ❌ 验证知识库条目失败: {e}") return False def _verify_database_integrity(self) -> bool: @@ -676,14 +676,14 @@ class DatabaseInitializer: return 0 def check_database_status(self) -> Dict[str, Any]: - """检查数据库状态""" + """检查数据库状态""" print("\n" + "=" * 80) - print("📊 数据库状态检查") + print("📊 数据库状态检查") print("=" * 80) - - try: - with db_manager.get_session() as session: - # 检查各表的记录数 + + try: + with db_manager.get_session() as session: + # 检查各表的记录数 tables_info = { 'work_orders': WorkOrder, 'conversations': Conversation, @@ -710,21 +710,21 @@ class DatabaseInitializer: except Exception as e: status["tables"][table_name] = f"错误: {e}" print(f"⚠️ {table_name}: 检查失败 - {e}") - - # 检查车辆数据详情 + + # 检查车辆数据详情 if 'vehicle_data' in status["tables"] and isinstance(status["tables"]['vehicle_data'], int): vehicle_count = status["tables"]['vehicle_data'] if vehicle_count > 0: - vehicle_ids = session.query(VehicleData.vehicle_id).distinct().all() - print(f" - 车辆数量: {len(vehicle_ids)}") + vehicle_ids = session.query(VehicleData.vehicle_id).distinct().all() + print(f" - 车辆数量: {len(vehicle_ids)}") status["vehicle_count"] = len(vehicle_ids) - for vehicle_id in vehicle_ids[:3]: # 显示前3个车辆 - vehicle_data_types = session.query(VehicleData.data_type).filter( - VehicleData.vehicle_id == vehicle_id[0] - ).distinct().all() - print(f" - 车辆 {vehicle_id[0]}: {len(vehicle_data_types)} 种数据类型") - + for vehicle_id in vehicle_ids[:3]: # 显示前3个车辆 + vehicle_data_types = session.query(VehicleData.data_type).filter( + VehicleData.vehicle_id == vehicle_id[0] + ).distinct().all() + print(f" - 车辆 {vehicle_id[0]}: {len(vehicle_data_types)} 种数据类型") + # 检查知识库验证状态 if 'knowledge_entries' in status["tables"] and isinstance(status["tables"]['knowledge_entries'], int): verified_count = session.query(KnowledgeEntry).filter(KnowledgeEntry.is_verified == True).count() @@ -737,12 +737,12 @@ class DatabaseInitializer: } print(f"\n📊 总记录数: {status['total_records']}") - print("\n✅ 数据库状态检查完成") - + print("\n✅ 数据库状态检查完成") + return status - - except Exception as e: - print(f"❌ 数据库状态检查失败: {e}") + + except Exception as e: + print(f"❌ 数据库状态检查失败: {e}") return {"error": str(e)} def main():