Add Team Roles feature.

2025-06-28 22:39:28 +02:00
parent e7593dc840
commit d8ec7d636e
11 changed files with 672 additions and 37 deletions
--- a/migrate_roles_teams.py
+++ b/migrate_roles_teams.py
@@ -0,0 +1,89 @@
+from app import app, db
+from models import User, Team, Role, SystemSettings
+from sqlalchemy import text
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def migrate_roles_teams():
+    with app.app_context():
+        logger.info("Starting migration for roles and teams...")
+        
+        # Check if the team table exists
+        try:
+            # Create the team table if it doesn't exist
+            db.engine.execute(text("""
+                CREATE TABLE IF NOT EXISTS team (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    name VARCHAR(100) NOT NULL UNIQUE,
+                    description VARCHAR(255),
+                    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+                )
+            """))
+            logger.info("Team table created or already exists")
+        except Exception as e:
+            logger.error(f"Error creating team table: {e}")
+            return
+        
+        # Check if the user table has the role and team_id columns
+        try:
+            # Check if role column exists
+            result = db.engine.execute(text("PRAGMA table_info(user)"))
+            columns = [row[1] for row in result]
+            
+            if 'role' not in columns:
+                # Use the enum name instead of the value
+                db.engine.execute(text("ALTER TABLE user ADD COLUMN role VARCHAR(20) DEFAULT 'TEAM_MEMBER'"))
+                logger.info("Added role column to user table")
+            
+            if 'team_id' not in columns:
+                db.engine.execute(text("ALTER TABLE user ADD COLUMN team_id INTEGER REFERENCES team(id)"))
+                logger.info("Added team_id column to user table")
+            
+            # Create a default team for existing users
+            default_team = Team.query.filter_by(name="Default Team").first()
+            if not default_team:
+                default_team = Team(name="Default Team", description="Default team for existing users")
+                db.session.add(default_team)
+                db.session.commit()
+                logger.info("Created default team")
+            
+            # Map string role values to enum values
+            role_mapping = {
+                'Team Member': Role.TEAM_MEMBER,
+                'TEAM_MEMBER': Role.TEAM_MEMBER,
+                'Team Leader': Role.TEAM_LEADER,
+                'TEAM_LEADER': Role.TEAM_LEADER,
+                'Supervisor': Role.SUPERVISOR,
+                'SUPERVISOR': Role.SUPERVISOR,
+                'Administrator': Role.ADMIN,
+                'admin': Role.ADMIN,
+                'ADMIN': Role.ADMIN
+            }
+            
+            # Assign all existing users to the default team and set role based on admin status
+            users = User.query.all()
+            for user in users:
+                if user.team_id is None:
+                    user.team_id = default_team.id
+                
+                # Handle role conversion properly
+                if isinstance(user.role, str):
+                    # Try to map the string to an enum value
+                    user.role = role_mapping.get(user.role, Role.TEAM_MEMBER)
+                elif user.role is None:
+                    # Set default role based on admin status
+                    user.role = Role.ADMIN if user.is_admin else Role.TEAM_MEMBER
+            
+            db.session.commit()
+            logger.info(f"Assigned {len(users)} existing users to default team and updated roles")
+            
+        except Exception as e:
+            logger.error(f"Error updating user table: {e}")
+            return
+        
+        logger.info("Migration completed successfully")
+
+if __name__ == "__main__":
+    migrate_roles_teams()