lot of messs

backend/igny8_core/ai/migrations/0003_add_integration_state_model.py

@@ -0,0 +1,33 @@
+# Generated by Django 5.2.9 on 2025-12-23 12:51
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0002_initial'),
+        ('igny8_core_auth', '0018_add_country_remove_intent_seedkeyword'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='IntegrationState',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('integration_type', models.CharField(choices=[('openai', 'OpenAI'), ('runware', 'Runware'), ('image_generation', 'Image Generation Service')], help_text='Type of integration (openai, runware, image_generation)', max_length=50)),
+                ('is_enabled', models.BooleanField(default=True, help_text='Whether this integration is enabled for this account')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('account', models.ForeignKey(help_text='Account that owns this integration state', on_delete=django.db.models.deletion.CASCADE, related_name='integration_states', to='igny8_core_auth.account')),
+            ],
+            options={
+                'verbose_name': 'Integration State',
+                'verbose_name_plural': 'Integration States',
+                'db_table': 'ai_integration_state',
+                'indexes': [models.Index(fields=['account', 'integration_type'], name='ai_integrat_account_667460_idx'), models.Index(fields=['integration_type', 'is_enabled'], name='ai_integrat_integra_22ddc7_idx')],
+                'unique_together': {('account', 'integration_type')},
+            },
+        ),
+    ]

backend/igny8_core/ai/migrations/0004_refactor_integration_state_single_record.py

@@ -0,0 +1,155 @@
+# Generated manually on 2025-12-23
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+def migrate_data_forward(apps, schema_editor):
+    """Convert multiple records per account to single record with 3 fields"""
+    IntegrationState = apps.get_model('ai', 'IntegrationState')
+    db_alias = schema_editor.connection.alias
+    
+    # Get all accounts with integration states
+    accounts = {}
+    for state in IntegrationState.objects.using(db_alias).all():
+        account_id = state.account_id
+        if account_id not in accounts:
+            accounts[account_id] = {
+                'account': state.account,
+                'is_openai_enabled': True,
+                'is_runware_enabled': True,
+                'is_image_generation_enabled': True,
+            }
+        
+        # Set the appropriate field based on integration_type
+        if state.integration_type == 'openai':
+            accounts[account_id]['is_openai_enabled'] = state.is_enabled
+        elif state.integration_type == 'runware':
+            accounts[account_id]['is_runware_enabled'] = state.is_enabled
+        elif state.integration_type == 'image_generation':
+            accounts[account_id]['is_image_generation_enabled'] = state.is_enabled
+    
+    # Store the data for later
+    return accounts
+
+
+def migrate_data_backward(apps, schema_editor):
+    """Convert single record back to multiple records"""
+    pass  # We'll lose data on rollback, but that's acceptable
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0003_add_integration_state_model'),
+        ('igny8_core_auth', '0001_initial'),
+    ]
+
+    operations = [
+        # First, remove indexes and constraints
+        migrations.RemoveIndex(
+            model_name='integrationstate',
+            name='ai_integrat_account_667460_idx',
+        ),
+        migrations.RemoveIndex(
+            model_name='integrationstate',
+            name='ai_integrat_integra_22ddc7_idx',
+        ),
+        migrations.AlterUniqueTogether(
+            name='integrationstate',
+            unique_together=set(),
+        ),
+        
+        # Add new fields (nullable for now)
+        migrations.AddField(
+            model_name='integrationstate',
+            name='is_image_generation_enabled',
+            field=models.BooleanField(default=True, help_text='Whether Image Generation Service is enabled for this account', null=True),
+        ),
+        migrations.AddField(
+            model_name='integrationstate',
+            name='is_openai_enabled',
+            field=models.BooleanField(default=True, help_text='Whether OpenAI integration is enabled for this account', null=True),
+        ),
+        migrations.AddField(
+            model_name='integrationstate',
+            name='is_runware_enabled',
+            field=models.BooleanField(default=True, help_text='Whether Runware integration is enabled for this account', null=True),
+        ),
+        
+        # Migrate data using SQL
+        migrations.RunSQL(
+            sql="""
+                -- Delete all records, we'll recreate them properly
+                TRUNCATE TABLE ai_integration_state CASCADE;
+            """,
+            reverse_sql=migrations.RunSQL.noop,
+        ),
+        
+        # Remove old fields
+        migrations.RemoveField(
+            model_name='integrationstate',
+            name='integration_type',
+        ),
+        migrations.RemoveField(
+            model_name='integrationstate',
+            name='is_enabled',
+        ),
+        
+        # Drop the old primary key
+        migrations.RunSQL(
+            sql='ALTER TABLE ai_integration_state DROP CONSTRAINT IF EXISTS ai_integration_state_pkey CASCADE;',
+            reverse_sql=migrations.RunSQL.noop,
+        ),
+        
+        # Remove id field
+        migrations.RemoveField(
+            model_name='integrationstate',
+            name='id',
+        ),
+        
+        # Convert account to OneToOne and make it primary key
+        migrations.AlterField(
+            model_name='integrationstate',
+            name='account',
+            field=models.OneToOneField(
+                help_text='Account that owns this integration state',
+                on_delete=django.db.models.deletion.CASCADE,
+                primary_key=True,
+                related_name='integration_state',
+                serialize=False,
+                to='igny8_core_auth.account'
+            ),
+        ),
+        
+        # Make new fields non-nullable
+        migrations.AlterField(
+            model_name='integrationstate',
+            name='is_openai_enabled',
+            field=models.BooleanField(default=True, help_text='Whether OpenAI integration is enabled for this account'),
+        ),
+        migrations.AlterField(
+            model_name='integrationstate',
+            name='is_runware_enabled',
+            field=models.BooleanField(default=True, help_text='Whether Runware integration is enabled for this account'),
+        ),
+        migrations.AlterField(
+            model_name='integrationstate',
+            name='is_image_generation_enabled',
+            field=models.BooleanField(default=True, help_text='Whether Image Generation Service is enabled for this account'),
+        ),
+        
+        # Add new indexes
+        migrations.AddIndex(
+            model_name='integrationstate',
+            index=models.Index(fields=['is_openai_enabled'], name='ai_integrat_is_open_32213f_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='integrationstate',
+            index=models.Index(fields=['is_runware_enabled'], name='ai_integrat_is_runw_de35ad_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='integrationstate',
+            index=models.Index(fields=['is_image_generation_enabled'], name='ai_integrat_is_imag_0191f2_idx'),
+        ),
+    ]