Audiobook Maker Pro v4.2 — production ready

2026-05-22 18:28:47 +06:00
commit 0617a374dd
41 changed files with 15262 additions and 0 deletions
--- a/db.py
+++ b/db.py
@@ -0,0 +1,151 @@
+# db.py - Database Configuration and Operations (v4.2)
+
+import os
+import sqlite3
+from flask import g
+from contextlib import contextmanager
+
+from config import DATABASE
+
+
+def get_db():
+    if 'db' not in g:
+        g.db = sqlite3.connect(DATABASE)
+        g.db.row_factory = sqlite3.Row
+    return g.db
+
+
+def close_db(error=None):
+    db = g.pop('db', None)
+    if db is not None:
+        db.close()
+
+
+@contextmanager
+def get_db_connection():
+    conn = sqlite3.connect(DATABASE)
+    conn.row_factory = sqlite3.Row
+    try:
+        yield conn
+    finally:
+        conn.close()
+
+
+def init_db():
+    """Initialize database tables. Auto-creates parent directory."""
+    # ফোল্ডার না থাকলে নিজে থেকেই তৈরি করবে (Coolify volume mount-এর জন্য জরুরি)
+    db_dir = os.path.dirname(os.path.abspath(DATABASE))
+    if db_dir and not os.path.exists(db_dir):
+        os.makedirs(db_dir, exist_ok=True)
+        print(f"📂 Created data directory: {db_dir}")
+
+    with get_db_connection() as conn:
+        cursor = conn.cursor()
+
+        # Projects table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS projects (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT NOT NULL UNIQUE,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            )
+        ''')
+
+        # Chapters table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS chapters (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                project_id INTEGER NOT NULL,
+                chapter_number INTEGER NOT NULL,
+                title TEXT DEFAULT 'Section',
+                voice TEXT DEFAULT 'af_heart',
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE CASCADE,
+                UNIQUE(project_id, chapter_number)
+            )
+        ''')
+
+        try:
+            cursor.execute('ALTER TABLE chapters ADD COLUMN title TEXT DEFAULT "Section"')
+        except sqlite3.OperationalError:
+            pass
+
+        # Markdown blocks table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS markdown_blocks (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                chapter_id INTEGER NOT NULL,
+                block_order INTEGER NOT NULL,
+                block_type TEXT NOT NULL DEFAULT 'paragraph',
+                content TEXT NOT NULL,
+                tts_text TEXT,
+                audio_data TEXT,
+                audio_format TEXT DEFAULT 'mp3',
+                transcription TEXT,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                FOREIGN KEY (chapter_id) REFERENCES chapters(id) ON DELETE CASCADE
+            )
+        ''')
+
+        # Images table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS block_images (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                block_id INTEGER NOT NULL,
+                image_data TEXT NOT NULL,
+                image_format TEXT DEFAULT 'png',
+                alt_text TEXT,
+                position TEXT DEFAULT 'before',
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                FOREIGN KEY (block_id) REFERENCES markdown_blocks(id) ON DELETE CASCADE
+            )
+        ''')
+
+        # PDF Documents table
+        cursor.execute('''
+            CREATE TABLE IF NOT EXISTS pdf_documents (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                project_id INTEGER,
+                filename TEXT NOT NULL,
+                page_count INTEGER DEFAULT 0,
+                metadata TEXT,
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                FOREIGN KEY (project_id) REFERENCES projects(id) ON DELETE SET NULL
+            )
+        ''')
+
+        # v4.2 publishing migrations
+        migrations = [
+            ('projects', 'is_published', 'INTEGER DEFAULT 0'),
+            ('projects', 'published_at', 'TIMESTAMP'),
+            ('projects', 'thumbnail_data', 'TEXT'),
+            ('projects', 'thumbnail_format', 'TEXT DEFAULT "png"'),
+            ('projects', 'description', 'TEXT DEFAULT ""'),
+            ('projects', 'author', 'TEXT DEFAULT ""'),
+            ('projects', 'category', 'TEXT DEFAULT ""'),
+            ('projects', 'view_count', 'INTEGER DEFAULT 0'),
+        ]
+
+        for table, column, definition in migrations:
+            try:
+                cursor.execute(f'ALTER TABLE {table} ADD COLUMN {column} {definition}')
+                print(f"  ✅ Added {table}.{column}")
+            except sqlite3.OperationalError:
+                pass
+
+        conn.commit()
+        print(f"✅ Database initialized at {DATABASE} (v4.2)")
+
+
+def vacuum_db():
+    with get_db_connection() as conn:
+        conn.execute('VACUUM')
+
+
+def init_app(app):
+    app.teardown_appcontext(close_db)
+    init_db()
+
+    from auth import init_users_table
+    init_users_table()