Update init_db.py

init_db.py

@@ -4,39 +4,57 @@ from sqlalchemy import text
 
 def init_db():
     with app.app_context():
+        # First, try to create the section table if it doesn't exist
         try:
-            # Try to run migrations first
-            upgrade()
-        except Exception as e:
-            print(f"Migration error (this is normal if tables exist): {e}")
-            
-            # Check if section_id column exists in plant table
-            try:
-                result = db.session.execute(text("SELECT section_id FROM plant LIMIT 1"))
-                print("section_id column exists")
-            except Exception:
-                print("Adding section_id column to plant table")
-                try:
-                    # Add section_id column if it doesn't exist
-                    db.session.execute(text("""
-                        ALTER TABLE plant 
-                        ADD COLUMN IF NOT EXISTS section_id INTEGER 
-                        REFERENCES section(id)
-                    """))
-                    db.session.commit()
-                    print("Added section_id column successfully")
-                except Exception as e:
-                    print(f"Error adding section_id column: {e}")
-                    db.session.rollback()
-        
-        # Create default admin user if it doesn't exist
-        admin = User.query.filter_by(username='admin').first()
-        if not admin:
-            admin = User(username='admin')
-            admin.set_password('admin123')  # You should change this password after first login
-            db.session.add(admin)
+            db.session.execute(text("""
+                CREATE TABLE IF NOT EXISTS section (
+                    id SERIAL PRIMARY KEY,
+                    name VARCHAR(100) NOT NULL,
+                    description TEXT,
+                    icon VARCHAR(200)
+                )
+            """))
             db.session.commit()
-            print("Default admin user created!")
+            print("Section table created or already exists")
+        except Exception as e:
+            print(f"Error creating section table: {e}")
+            db.session.rollback()
+
+        # Then try to add the section_id column if it doesn't exist
+        try:
+            # Check if column exists
+            result = db.session.execute(text("""
+                SELECT column_name 
+                FROM information_schema.columns 
+                WHERE table_name='plant' AND column_name='section_id'
+            """))
+            if not result.fetchone():
+                print("Adding section_id column to plant table")
+                db.session.execute(text("""
+                    ALTER TABLE plant 
+                    ADD COLUMN section_id INTEGER 
+                    REFERENCES section(id)
+                """))
+                db.session.commit()
+                print("Added section_id column successfully")
+            else:
+                print("section_id column already exists")
+        except Exception as e:
+            print(f"Error adding section_id column: {e}")
+            db.session.rollback()
+
+        # Create default admin user if it doesn't exist
+        try:
+            admin = User.query.filter_by(username='admin').first()
+            if not admin:
+                admin = User(username='admin')
+                admin.set_password('admin123')  # You should change this password after first login
+                db.session.add(admin)
+                db.session.commit()
+                print("Default admin user created!")
+        except Exception as e:
+            print(f"Error creating admin user: {e}")
+            db.session.rollback()
 
 if __name__ == '__main__':
     init_db()