feat: add attribute to store identity overrides storage type (#3109)

api/projects/migrations/0021_add_identity_overrides_storage_type.py

@@ -0,0 +1,47 @@
+# Generated by Django 3.2.23 on 2023-12-07 12:33
+from django.apps.registry import Apps
+from django.db import migrations, models
+from django.db.backends.base.schema import BaseDatabaseSchemaEditor
+
+from projects.models import IdentityOverridesStorageType
+
+
+def apply_defaults(apps: Apps, schema_editor: BaseDatabaseSchemaEditor) -> None:
+    apps.get_model("projects", "Project").objects.all().update(
+        identity_overrides_storage_type=IdentityOverridesStorageType.IDENTITY_RECORD
+    )
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("projects", "0020_add_environment_feature_state_version_logic"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="project",
+            name="identity_overrides_storage_type",
+            field=models.CharField(
+                choices=[
+                    ("IDENTITY_RECORD", "Identity Record"),
+                    ("FLAGSMITH_OVERRIDES", "Flagsmith Overrides"),
+                ],
+                default=None,
+                max_length=50,
+                null=True,
+            ),
+        ),
+        migrations.RunPython(apply_defaults, reverse_code=migrations.RunPython.noop),
+        migrations.AlterField(
+            model_name="project",
+            name="identity_overrides_storage_type",
+            field=models.CharField(
+                choices=[
+                    ("IDENTITY_RECORD", "Identity Record"),
+                    ("FLAGSMITH_OVERRIDES", "Flagsmith Overrides"),
+                ],
+                default="IDENTITY_RECORD",
+                max_length=50,
+            ),
+        ),
+    ]

api/projects/models.py

@@ -30,6 +30,11 @@
 environment_cache = caches[settings.ENVIRONMENT_CACHE_NAME]
 
 
+class IdentityOverridesStorageType(models.TextChoices):
+    IDENTITY_RECORD = "IDENTITY_RECORD", "Identity Record"
+    FLAGSMITH_OVERRIDES = "FLAGSMITH_OVERRIDES", "Flagsmith Overrides"
+
+
 class Project(LifecycleModelMixin, SoftDeleteExportableModel):
     name = models.CharField(max_length=2000)
     created_date = models.DateTimeField("DateCreated", auto_now_add=True)
@@ -71,6 +76,11 @@ class Project(LifecycleModelMixin, SoftDeleteExportableModel):
         default=100,
         help_text="Max segments overrides allowed for any (one) environment within this project",
     )
+    identity_overrides_storage_type = models.CharField(
+        max_length=50,
+        choices=IdentityOverridesStorageType.choices,
+        default=IdentityOverridesStorageType.IDENTITY_RECORD,
+    )
 
     objects = ProjectManager()