aetherforge-platform 1.0.0__py3-none-any.whl

ai-life-assistant-copy/backups/backup_20260404_193836/database.py

@@ -0,0 +1,292 @@
+import sqlite3
+import json
+import os
+from typing import Dict, Any, List, Optional
+from datetime import datetime
+
+
+class AvatarDatabase:
+    def __init__(self, db_path: str = "avatar_system.db"):
+        self.db_path = db_path
+        self._init_database()
+    
+    def _get_connection(self):
+        conn = sqlite3.connect(self.db_path)
+        conn.row_factory = sqlite3.Row
+        return conn
+    
+    def _init_database(self):
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS avatars (
+                avatar_id TEXT PRIMARY KEY,
+                name TEXT NOT NULL,
+                description TEXT,
+                personality_traits TEXT,
+                current_age INTEGER DEFAULT 25,
+                birth_year INTEGER,
+                avatar_image TEXT,
+                is_public INTEGER DEFAULT 0,
+                price REAL,
+                creator_id TEXT,
+                created_at TEXT,
+                updated_at TEXT,
+                settings TEXT
+            )
+        """)
+        
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS memories (
+                memory_id TEXT PRIMARY KEY,
+                avatar_id TEXT NOT NULL,
+                content TEXT NOT NULL,
+                min_age INTEGER DEFAULT 0,
+                max_age INTEGER DEFAULT 150,
+                clarity INTEGER DEFAULT 80,
+                emotional_tone TEXT DEFAULT 'neutral',
+                topic TEXT,
+                parent_id TEXT,
+                created_at TEXT,
+                FOREIGN KEY (avatar_id) REFERENCES avatars(avatar_id)
+            )
+        """)
+        
+        cursor.execute("""
+            CREATE INDEX IF NOT EXISTS idx_memories_avatar ON memories(avatar_id)
+        """)
+        
+        cursor.execute("""
+            CREATE INDEX IF NOT EXISTS idx_memories_age ON memories(avatar_id, min_age, max_age)
+        """)
+        
+        conn.commit()
+        conn.close()
+    
+    def save_avatar(self, avatar_data: Dict[str, Any]):
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            INSERT OR REPLACE INTO avatars 
+            (avatar_id, name, description, personality_traits, current_age, birth_year,
+             avatar_image, is_public, price, creator_id, created_at, updated_at, settings)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        """, (
+            avatar_data["avatar_id"],
+            avatar_data["name"],
+            avatar_data.get("description", ""),
+            json.dumps(avatar_data.get("personality_traits", []), ensure_ascii=False),
+            avatar_data.get("current_age", 25),
+            avatar_data.get("birth_year"),
+            avatar_data.get("avatar_image"),
+            1 if avatar_data.get("is_public") else 0,
+            avatar_data.get("price"),
+            avatar_data.get("creator_id"),
+            avatar_data.get("created_at"),
+            avatar_data.get("updated_at"),
+            json.dumps(avatar_data.get("settings", {}), ensure_ascii=False)
+        ))
+        
+        conn.commit()
+        conn.close()
+    
+    def get_avatar(self, avatar_id: str) -> Optional[Dict[str, Any]]:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("SELECT * FROM avatars WHERE avatar_id = ?", (avatar_id,))
+        row = cursor.fetchone()
+        
+        if row:
+            data = dict(row)
+            data["personality_traits"] = json.loads(data["personality_traits"])
+            data["settings"] = json.loads(data["settings"])
+            data["is_public"] = bool(data["is_public"])
+            conn.close()
+            return data
+        
+        conn.close()
+        return None
+    
+    def list_avatars(self, creator_id: Optional[str] = None) -> List[Dict[str, Any]]:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        if creator_id:
+            cursor.execute("SELECT * FROM avatars WHERE creator_id = ?", (creator_id,))
+        else:
+            cursor.execute("SELECT * FROM avatars")
+        
+        avatars = []
+        for row in cursor.fetchall():
+            data = dict(row)
+            data["personality_traits"] = json.loads(data["personality_traits"])
+            data["settings"] = json.loads(data["settings"])
+            data["is_public"] = bool(data["is_public"])
+            avatars.append(data)
+        
+        conn.close()
+        return avatars
+    
+    def delete_avatar(self, avatar_id: str) -> bool:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("DELETE FROM memories WHERE avatar_id = ?", (avatar_id,))
+        cursor.execute("DELETE FROM avatars WHERE avatar_id = ?", (avatar_id,))
+        
+        deleted = cursor.rowcount > 0
+        conn.commit()
+        conn.close()
+        return deleted
+    
+    def save_memory(self, memory_data: Dict[str, Any]):
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            INSERT OR REPLACE INTO memories 
+            (memory_id, avatar_id, content, min_age, max_age, clarity,
+             emotional_tone, topic, parent_id, created_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        """, (
+            memory_data["memory_id"],
+            memory_data["avatar_id"],
+            memory_data["content"],
+            memory_data.get("min_age", 0),
+            memory_data.get("max_age", 150),
+            memory_data.get("clarity", 80),
+            memory_data.get("emotional_tone", "neutral"),
+            memory_data.get("topic"),
+            memory_data.get("parent_id"),
+            memory_data.get("created_at")
+        ))
+        
+        conn.commit()
+        conn.close()
+    
+    def get_memories_by_age(self, avatar_id: str, target_age: int) -> List[Dict[str, Any]]:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            SELECT * FROM memories 
+            WHERE avatar_id = ? AND min_age <= ? AND max_age >= ?
+            ORDER BY created_at DESC
+        """, (avatar_id, target_age, target_age))
+        
+        memories = [dict(row) for row in cursor.fetchall()]
+        conn.close()
+        return memories
+    
+    def get_all_memories(self, avatar_id: str) -> List[Dict[str, Any]]:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            SELECT * FROM memories WHERE avatar_id = ?
+            ORDER BY created_at DESC
+        """, (avatar_id,))
+        
+        memories = [dict(row) for row in cursor.fetchall()]
+        conn.close()
+        return memories
+    
+    def delete_memory(self, memory_id: str) -> bool:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("DELETE FROM memories WHERE memory_id = ?", (memory_id,))
+        
+        deleted = cursor.rowcount > 0
+        conn.commit()
+        conn.close()
+        return deleted
+    
+    def list_market_avatars(self) -> List[Dict[str, Any]]:
+        conn = self._get_connection()
+        cursor = conn.cursor()
+        
+        cursor.execute("""
+            SELECT * FROM avatars 
+            WHERE is_public = 1 AND price IS NOT NULL
+        """)
+        
+        avatars = []
+        for row in cursor.fetchall():
+            data = dict(row)
+            data["personality_traits"] = json.loads(data["personality_traits"])
+            data["settings"] = json.loads(data["settings"])
+            data["is_public"] = bool(data["is_public"])
+            avatars.append(data)
+        
+        conn.close()
+        return avatars
+
+
+avatar_database = AvatarDatabase()
+
+
+def init_sample_data():
+    sample_avatar = {
+        "avatar_id": "sample_001",
+        "name": "Alex",
+        "description": "一个友好、好奇的虚拟形象",
+        "personality_traits": ["友好", "好奇", "创造力"],
+        "current_age": 25,
+        "birth_year": 2000,
+        "is_public": False,
+        "creator_id": "demo_user",
+        "created_at": datetime.now().isoformat(),
+        "updated_at": datetime.now().isoformat(),
+        "settings": {"developer_mode": False, "auto_organize": True}
+    }
+    
+    avatar_database.save_avatar(sample_avatar)
+    
+    sample_memories = [
+        {
+            "memory_id": "mem_001",
+            "avatar_id": "sample_001",
+            "content": "我第一次学会骑自行车，那天很开心",
+            "min_age": 5,
+            "max_age": 10,
+            "clarity": 90,
+            "emotional_tone": "happy",
+            "topic": "童年",
+            "created_at": datetime.now().isoformat()
+        },
+        {
+            "memory_id": "mem_002",
+            "avatar_id": "sample_001",
+            "content": "高中时参加了数学竞赛，获得了二等奖",
+            "min_age": 15,
+            "max_age": 18,
+            "clarity": 85,
+            "emotional_tone": "proud",
+            "topic": "学业",
+            "created_at": datetime.now().isoformat()
+        },
+        {
+            "memory_id": "mem_003",
+            "avatar_id": "sample_001",
+            "content": "大学毕业后的第一份工作是软件开发工程师",
+            "min_age": 22,
+            "max_age": 30,
+            "clarity": 95,
+            "emotional_tone": "excited",
+            "topic": "职业",
+            "created_at": datetime.now().isoformat()
+        }
+    ]
+    
+    for memory in sample_memories:
+        avatar_database.save_memory(memory)
+    
+    print("Sample data initialized!")
+
+
+if __name__ == "__main__":
+    init_sample_data()