Initial release: DictIA v0.8.14-alpha (fork de Speakr, AGPL-3.0)
This commit is contained in:
InnovA AI
130
scripts/reset_db.py
Normal file
130
scripts/reset_db.py
Normal file
@@ -0,0 +1,130 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
# Add this near the top if you run this standalone often outside app context
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
# Add project root to path if necessary for 'app' import
|
||||
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
||||
|
||||
# Load environment variables in case DB path relies on them (optional here)
|
||||
# from dotenv import load_dotenv
|
||||
# load_dotenv()
|
||||
|
||||
# Check if running within app context already (e.g., via Flask command)
|
||||
try:
|
||||
from flask import current_app
|
||||
# Ensure app context is pushed if needed for config access
|
||||
app = current_app._get_current_object()
|
||||
# Make sure db is initialized within the app context if needed
|
||||
# (SQLAlchemy initialization in app.py handles this mostly)
|
||||
with app.app_context():
|
||||
db = app.extensions['sqlalchemy'].db # Access db via extensions
|
||||
except (RuntimeError, AttributeError, KeyError):
|
||||
# If not in app context, import directly
|
||||
try:
|
||||
# Ensure this import reflects the updated app.py with the new model
|
||||
from src.app import app, db
|
||||
except ImportError as e:
|
||||
print(f"Error: Could not import 'app' and 'db': {e}")
|
||||
print("Make sure reset_db.py is runnable and PYTHONPATH is set.")
|
||||
sys.exit(1)
|
||||
|
||||
def reset_database(delete_uploads=True):
|
||||
# Determine the database path relative to the instance folder
|
||||
# Use app config if available
|
||||
instance_path = app.instance_path if hasattr(app, 'instance_path') else os.path.join(os.getcwd(), 'instance')
|
||||
try:
|
||||
# Ensure app context for config access if not already present
|
||||
with app.app_context():
|
||||
# Use absolute path from config
|
||||
db_uri = app.config.get('SQLALCHEMY_DATABASE_URI', 'sqlite:///instance/transcriptions.db')
|
||||
# Handle relative vs absolute paths specified in URI
|
||||
if db_uri.startswith('sqlite:///'):
|
||||
# Assume absolute path from URI root if starts with '///'
|
||||
db_path = db_uri.replace('sqlite:///', '/', 1) # Replace only first
|
||||
# Ensure instance path reflects the directory containing the DB
|
||||
instance_path = os.path.dirname(db_path)
|
||||
elif db_uri.startswith('sqlite://'):
|
||||
# Assume relative path from instance folder
|
||||
db_filename = db_uri.split('/')[-1]
|
||||
db_path = os.path.join(instance_path, db_filename)
|
||||
else: # Handle other DB types or formats if needed
|
||||
print(f"Warning: Non-SQLite URI detected: {db_uri}. Deletion logic might need adjustment.")
|
||||
# Attempt to parse or fallback
|
||||
db_filename = db_uri.split('/')[-1] # Best guess
|
||||
db_path = os.path.join(instance_path, db_filename)
|
||||
|
||||
except Exception as config_e:
|
||||
print(f"Error accessing app config for DB path: {config_e}. Using default.")
|
||||
# Fallback if config access fails
|
||||
instance_path = os.path.join(os.getcwd(), 'instance')
|
||||
db_filename = 'transcriptions.db'
|
||||
db_path = os.path.join(instance_path, db_filename)
|
||||
|
||||
# Ensure instance directory exists
|
||||
print(f"Ensuring instance directory exists: {instance_path}")
|
||||
os.makedirs(instance_path, exist_ok=True)
|
||||
print(f"Database path identified as: {db_path}")
|
||||
|
||||
# Remove existing database if it exists
|
||||
if os.path.exists(db_path):
|
||||
print(f"Removing existing database at {db_path}")
|
||||
try:
|
||||
os.remove(db_path)
|
||||
# Also remove journal file if it exists
|
||||
journal_path = db_path + "-journal"
|
||||
if os.path.exists(journal_path):
|
||||
os.remove(journal_path)
|
||||
print(f"Removing existing journal file at {journal_path}")
|
||||
except OSError as e:
|
||||
print(f"Error removing database file: {e}. Check permissions or if it's in use.")
|
||||
# Decide whether to exit or continue
|
||||
# sys.exit(1)
|
||||
|
||||
# Create application context to work with the database
|
||||
try:
|
||||
with app.app_context():
|
||||
print("Creating new database schema (including 'summary' column)...")
|
||||
# Create all tables defined in models (app.py)
|
||||
db.create_all()
|
||||
print("Database schema created successfully!")
|
||||
except Exception as e:
|
||||
print(f"Error creating database schema: {e}")
|
||||
# Attempt rollback if possible (though less relevant for create_all)
|
||||
try:
|
||||
db.session.rollback()
|
||||
except Exception as rb_e:
|
||||
print(f"Rollback attempt failed: {rb_e}")
|
||||
sys.exit(1)
|
||||
|
||||
# Delete all files in the uploads directory if requested
|
||||
if delete_uploads:
|
||||
try:
|
||||
uploads_dir = os.path.join(os.getcwd(), 'uploads')
|
||||
if os.path.exists(uploads_dir):
|
||||
print(f"Deleting all files in uploads directory: {uploads_dir}")
|
||||
for filename in os.listdir(uploads_dir):
|
||||
file_path = os.path.join(uploads_dir, filename)
|
||||
try:
|
||||
if os.path.isfile(file_path):
|
||||
os.remove(file_path)
|
||||
print(f"Deleted file: {file_path}")
|
||||
elif os.path.isdir(file_path):
|
||||
shutil.rmtree(file_path)
|
||||
print(f"Deleted directory: {file_path}")
|
||||
except Exception as e:
|
||||
print(f"Error deleting {file_path}: {e}")
|
||||
print("All files in uploads directory have been deleted.")
|
||||
else:
|
||||
print(f"Uploads directory not found: {uploads_dir}")
|
||||
# Create the directory if it doesn't exist
|
||||
os.makedirs(uploads_dir, exist_ok=True)
|
||||
print(f"Created uploads directory: {uploads_dir}")
|
||||
except Exception as e:
|
||||
print(f"Error cleaning uploads directory: {e}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Attempting to reset the database and clean up all data...")
|
||||
reset_database(delete_uploads=True)
|
||||
print("Database reset process finished.")
|
||||
Reference in New Issue
Block a user