kinda working safe point
This commit is contained in:
parent
b9a82af12f
commit
6dda02141e
31 changed files with 4302 additions and 2937 deletions
|
@ -1,52 +0,0 @@
|
|||
"""
|
||||
Script to add storage_name column to the file table.
|
||||
Run this once from the command line:
|
||||
python migrations/add_storage_name.py
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import sqlite3
|
||||
|
||||
# Add the parent directory to the path so we can import the app
|
||||
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
||||
|
||||
from config import Config
|
||||
from app import create_app
|
||||
|
||||
def add_storage_name_column():
|
||||
"""Add storage_name column to file table"""
|
||||
# Get database path from config
|
||||
app = create_app()
|
||||
with app.app_context():
|
||||
db_path = app.config['SQLALCHEMY_DATABASE_URI'].replace('sqlite:///', '')
|
||||
|
||||
print(f"Database path: {db_path}")
|
||||
|
||||
# Connect to the database
|
||||
conn = sqlite3.connect(db_path)
|
||||
cursor = conn.cursor()
|
||||
|
||||
try:
|
||||
# Check if column already exists
|
||||
cursor.execute("PRAGMA table_info(file)")
|
||||
columns = [column[1] for column in cursor.fetchall()]
|
||||
|
||||
if 'storage_name' not in columns:
|
||||
print("Adding storage_name column to file table...")
|
||||
cursor.execute("ALTER TABLE file ADD COLUMN storage_name TEXT")
|
||||
|
||||
# Update existing files to use name as storage_name
|
||||
cursor.execute("UPDATE file SET storage_name = name WHERE is_folder = 0")
|
||||
|
||||
conn.commit()
|
||||
print("Column added successfully!")
|
||||
else:
|
||||
print("Column already exists.")
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
if __name__ == "__main__":
|
||||
add_storage_name_column()
|
Loading…
Add table
Add a link
Reference in a new issue