init commit

APP/db.py

@@ -0,0 +1,91 @@
+import sqlite3
+import configparser
+import os
+import logging
+
+class DatabaseManager:
+    def __init__(self, config_file='config.ini'):
+        self.config_file = config_file
+        self.config = configparser.ConfigParser()
+        self.logger = logging.getLogger(__name__)
+        self.load_config()
+
+    def load_config(self):
+        if os.path.exists(self.config_file):
+            self.config.read(self.config_file)
+        self.db_path = self.config.get('openvpn_monitor', 'db_path', fallback='openvpn_monitor.db')
+
+    def get_connection(self):
+        """Get a database connection"""
+        return sqlite3.connect(self.db_path)
+
+    def init_database(self):
+        """Initialize the database schema"""
+        # Create directory if needed
+        db_dir = os.path.dirname(self.db_path)
+        if db_dir and not os.path.exists(db_dir):
+            try:
+                os.makedirs(db_dir)
+            except OSError:
+                pass
+
+        self.logger.info(f"Using database: {self.db_path}")
+        
+        conn = self.get_connection()
+        cursor = conn.cursor()
+        
+        try:
+            # 1. Clients Table
+            cursor.execute('''
+            CREATE TABLE IF NOT EXISTS clients (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                common_name TEXT UNIQUE NOT NULL,
+                real_address TEXT,
+                status TEXT DEFAULT 'Active',
+                total_bytes_received INTEGER DEFAULT 0,
+                total_bytes_sent INTEGER DEFAULT 0,
+                last_bytes_received INTEGER DEFAULT 0,
+                last_bytes_sent INTEGER DEFAULT 0,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                last_activity TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            )
+            ''')
+            
+            # 2. Raw Usage History
+            cursor.execute('''
+            CREATE TABLE IF NOT EXISTS usage_history (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                client_id INTEGER,
+                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                bytes_received INTEGER,
+                bytes_sent INTEGER,
+                bytes_received_rate_mbps REAL,
+                bytes_sent_rate_mbps REAL,
+                FOREIGN KEY (client_id) REFERENCES clients (id)
+            )
+            ''')
+            cursor.execute('CREATE INDEX IF NOT EXISTS idx_usage_ts ON usage_history(timestamp)')
+
+            # 3. Aggregated Stats Tables
+            tables = ['stats_5min', 'stats_15min', 'stats_hourly', 'stats_6h', 'stats_daily']
+            
+            for table in tables:
+                cursor.execute(f'''
+                CREATE TABLE IF NOT EXISTS {table} (
+                    timestamp TEXT NOT NULL,
+                    client_id INTEGER NOT NULL,
+                    bytes_received INTEGER DEFAULT 0,
+                    bytes_sent INTEGER DEFAULT 0,
+                    PRIMARY KEY (timestamp, client_id),
+                    FOREIGN KEY (client_id) REFERENCES clients (id)
+                )
+                ''')
+                cursor.execute(f'CREATE INDEX IF NOT EXISTS idx_{table}_ts ON {table}(timestamp)')
+            
+            conn.commit()
+            self.logger.info("Database initialized with full schema")
+        except Exception as e:
+            self.logger.error(f"Database initialization error: {e}")
+        finally:
+            conn.close()