made corrections based on feedback

migrations/add_feedback_fixes_2025_01_07.py

@@ -0,0 +1,110 @@
+"""
+Migration: Add fields from feedback fixes
+Date: 2025-01-07
+
+Adds the following fields:
+- projects.is_archived (INTEGER, default 0)
+- companies.product_service (TEXT, nullable)
+- companies.clients (TEXT, nullable - stored as JSON string)
+- investor_members.linkedin (VARCHAR, nullable)
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# Add parent directory to path to import app modules
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from sqlalchemy import create_engine, text
+from app.db.db import DATABASE_URL, engine
+
+
+def check_column_exists(conn, table_name, column_name):
+    """Check if a column exists in a table"""
+    result = conn.execute(text(f"PRAGMA table_info({table_name})"))
+    columns = [row[1] for row in result]
+    return column_name in columns
+
+
+def upgrade():
+    """Add new columns to tables"""
+    print("Running migration: Add feedback fixes fields")
+    print("=" * 60)
+    
+    with engine.begin() as conn:  # Use begin() for transaction management
+        # 1. Add is_archived to projects table
+        print("\n1. Adding 'is_archived' column to projects table...")
+        if check_column_exists(conn, "projects", "is_archived"):
+            print("   ✓ Column 'is_archived' already exists. Skipping.")
+        else:
+            conn.execute(text("ALTER TABLE projects ADD COLUMN is_archived INTEGER DEFAULT 0 NOT NULL"))
+            # Set default value for existing rows
+            conn.execute(text("UPDATE projects SET is_archived = 0 WHERE is_archived IS NULL"))
+            print("   ✓ Successfully added 'is_archived' column to projects table")
+        
+        # 2. Add product_service to companies table
+        print("\n2. Adding 'product_service' column to companies table...")
+        if check_column_exists(conn, "companies", "product_service"):
+            print("   ✓ Column 'product_service' already exists. Skipping.")
+        else:
+            conn.execute(text("ALTER TABLE companies ADD COLUMN product_service TEXT"))
+            print("   ✓ Successfully added 'product_service' column to companies table")
+        
+        # 3. Add clients to companies table
+        print("\n3. Adding 'clients' column to companies table...")
+        if check_column_exists(conn, "companies", "clients"):
+            print("   ✓ Column 'clients' already exists. Skipping.")
+        else:
+            conn.execute(text("ALTER TABLE companies ADD COLUMN clients TEXT"))
+            print("   ✓ Successfully added 'clients' column to companies table")
+        
+        # 4. Add linkedin to investor_members table
+        print("\n4. Adding 'linkedin' column to investor_members table...")
+        if check_column_exists(conn, "investor_members", "linkedin"):
+            print("   ✓ Column 'linkedin' already exists. Skipping.")
+        else:
+            conn.execute(text("ALTER TABLE investor_members ADD COLUMN linkedin VARCHAR"))
+            print("   ✓ Successfully added 'linkedin' column to investor_members table")
+    
+    print("\n" + "=" * 60)
+    print("Migration completed successfully!")
+
+
+def downgrade():
+    """Remove added columns from tables"""
+    print("Running downgrade: Remove feedback fixes fields")
+    print("=" * 60)
+    
+    # Note: SQLite doesn't support DROP COLUMN directly
+    print("\nWarning: SQLite doesn't support DROP COLUMN directly.")
+    print("To remove these columns, you would need to:")
+    print("1. Create new tables without the columns")
+    print("2. Copy data from old tables to new tables")
+    print("3. Drop old tables and rename new tables")
+    print("\nColumns to remove:")
+    print("  - projects.is_archived")
+    print("  - companies.product_service")
+    print("  - companies.clients")
+    print("  - investor_members.linkedin")
+
+
+if __name__ == "__main__":
+    import argparse
+    
+    parser = argparse.ArgumentParser(description="Run database migration")
+    parser.add_argument(
+        "direction",
+        choices=["upgrade", "downgrade"],
+        default="upgrade",
+        nargs="?",
+        help="Migration direction (default: upgrade)"
+    )
+    
+    args = parser.parse_args()
+    
+    if args.direction == "upgrade":
+        upgrade()
+    else:
+        downgrade()
+