152 lines
4.9 KiB
Python
152 lines
4.9 KiB
Python
# 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()
|