lamindb 1.5.2__py3-none-any.whl → 1.6.0__py3-none-any.whl

lamindb/migrations/0100_branch_alter_artifact__branch_code_and_more.py

@@ -0,0 +1,102 @@
+# Generated by Django 5.2 on 2025-05-25 11:59
+
+import django.db.models.deletion
+import django.db.models.functions.datetime
+from django.db import migrations, models
+
+import lamindb.base.fields
+
+
+def update_space_uids_and_create_branches(apps, schema_editor):
+    Space = apps.get_model("lamindb", "Space")
+    Space.objects.filter(uid="00000000").update(uid="A")
+    Branch = apps.get_model("lamindb", "Branch")
+    Branch.objects.get_or_create(
+        id=-1,
+        uid="T",
+        name="Trash",
+        description="The trash.",
+    )
+    Branch.objects.get_or_create(
+        id=0,
+        uid="A",
+        name="Archive",
+        description="The archive.",
+    )
+    Branch.objects.get_or_create(
+        uid="M",
+        name="Main",
+        description="The main & default branch of the instance.",
+    )
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("lamindb", "0099_alter_writelog_seqno"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Branch",
+            fields=[
+                ("id", models.AutoField(primary_key=True, serialize=False)),
+                ("name", models.CharField(db_index=True, max_length=100)),
+                (
+                    "uid",
+                    lamindb.base.fields.CharField(
+                        blank=True,
+                        db_default="M",
+                        db_index=True,
+                        default="M",
+                        editable=False,
+                        max_length=12,
+                        unique=True,
+                    ),
+                ),
+                (
+                    "description",
+                    lamindb.base.fields.CharField(
+                        blank=True, default=None, max_length=255, null=True
+                    ),
+                ),
+                (
+                    "created_at",
+                    lamindb.base.fields.DateTimeField(
+                        blank=True,
+                        db_default=django.db.models.functions.datetime.Now(),
+                        db_index=True,
+                        editable=False,
+                    ),
+                ),
+                (
+                    "created_by",
+                    lamindb.base.fields.ForeignKey(
+                        blank=True,
+                        default=None,
+                        null=True,
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="+",
+                        to="lamindb.user",
+                    ),
+                ),
+            ],
+            options={
+                "abstract": False,
+                "base_manager_name": "objects",
+            },
+        ),
+        migrations.AlterField(
+            model_name="space",
+            name="uid",
+            field=lamindb.base.fields.CharField(
+                blank=True,
+                db_default="A",
+                db_index=True,
+                default="A",
+                editable=False,
+                max_length=12,
+                unique=True,
+            ),
+        ),
+        migrations.RunPython(update_space_uids_and_create_branches),
+    ]

lamindb/migrations/0101_alter_artifact_hash_alter_feature_name_and_more.py

@@ -0,0 +1,444 @@
+# Generated by Django 5.2 on 2025-05-26 09:33
+
+import django.db.models.deletion
+from django.db import migrations, models
+from lamindb_setup.core.hashing import hash_dict, hash_string
+
+import lamindb.base.fields
+
+
+def populate_hash_values(apps, schema_editor):
+    FeatureValue = apps.get_model("lamindb", "FeatureValue")
+
+    # Get records that need updating
+    records_to_update = FeatureValue.objects.filter(hash__isnull=True)
+
+    print(
+        f"\nFound {records_to_update.count()} FeatureValue records with hash=None. Populating..."
+    )
+
+    updated_count = 0
+    skipped_count = 0
+
+    # Iterate efficiently through records
+    for record in records_to_update.iterator():
+        value = record.value
+        new_hash = None
+
+        if not isinstance(value, dict):
+            new_hash = hash_string(str(value))
+        else:
+            new_hash = hash_dict(value)
+
+        try:
+            record.hash = new_hash
+            record.save(update_fields=["hash"])
+            updated_count += 1
+        except Exception as e:
+            # This will likely catch IntegrityError if a duplicate hash exists
+            # for the same feature. You might want to log these or handle them
+            # based on your data cleanup strategy.
+            print(
+                f"  - Could not update record {record.id} (value: {str(value)[:50]}...): {e}"
+            )
+            skipped_count += 1
+
+    print(f"Finished. Updated: {updated_count}, Skipped/Errors: {skipped_count}.")
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("lamindb", "0100_branch_alter_artifact__branch_code_and_more"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="artifact",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="collection",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="feature",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="featurevalue",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="person",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="record",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="recordrecord",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="reference",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="run",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="schema",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="sheet",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="storage",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="transform",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="ulabel",
+            name="_branch_code",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                db_column="_branch_code",
+                db_default=1,
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.RenameField(
+            model_name="artifact",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="collection",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="feature",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="featurevalue",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="person",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="project",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="record",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="recordrecord",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="reference",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="run",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="schema",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="sheet",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="storage",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="transform",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.RenameField(
+            model_name="ulabel",
+            old_name="_branch_code",
+            new_name="branch",
+        ),
+        migrations.AlterField(
+            model_name="artifact",
+            name="hash",
+            field=lamindb.base.fields.CharField(
+                blank=True,
+                db_index=True,
+                default=None,
+                editable=False,
+                max_length=22,
+                null=True,
+            ),
+        ),
+        migrations.AlterField(
+            model_name="feature",
+            name="name",
+            field=lamindb.base.fields.CharField(
+                blank=True, db_index=True, default=None, max_length=150
+            ),
+        ),
+        migrations.AlterField(
+            model_name="feature",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="features",
+                to="lamindb.feature",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="project",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="projects",
+                to="lamindb.project",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="record",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="records",
+                to="lamindb.record",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="reference",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="references",
+                to="lamindb.reference",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="schema",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="schemas",
+                to="lamindb.schema",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="transform",
+            name="hash",
+            field=lamindb.base.fields.CharField(
+                blank=True, db_index=True, default=None, max_length=22, null=True
+            ),
+        ),
+        migrations.AlterField(
+            model_name="ulabel",
+            name="type",
+            field=lamindb.base.fields.ForeignKey(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="ulabels",
+                to="lamindb.ulabel",
+            ),
+        ),
+        migrations.AlterUniqueTogether(
+            name="transform",
+            unique_together={("key", "hash")},
+        ),
+        migrations.AddConstraint(
+            model_name="artifact",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("key__isnull", False)),
+                fields=("storage", "key", "hash"),
+                name="unique_artifact_storage_key_hash",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="feature",
+            name="_expect_many",
+            field=models.BooleanField(db_default=None, default=None, null=True),
+        ),
+        migrations.AlterField(
+            model_name="run",
+            name="uid",
+            field=lamindb.base.fields.CharField(
+                blank=True,
+                db_index=True,
+                default=lamindb.base.uids.base62_16,
+                editable=False,
+                max_length=20,
+                unique=True,
+            ),
+        ),
+        migrations.RemoveConstraint(
+            model_name="featurevalue",
+            name="unique_simple_feature_value",
+        ),
+        migrations.RemoveConstraint(
+            model_name="featurevalue",
+            name="unique_complex_feature_value",
+        ),
+        migrations.AlterUniqueTogether(
+            name="featurevalue",
+            unique_together={("feature", "hash")},
+        ),
+        migrations.RunPython(populate_hash_values),
+    ]

lamindb/migrations/0102_remove_writelog_branch_code_and_more.py

@@ -0,0 +1,72 @@
+# Generated by Django 5.2 on 2025-05-27 11:29
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("lamindb", "0101_alter_artifact_hash_alter_feature_name_and_more"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="writelog",
+            name="branch_code",
+        ),
+        migrations.RemoveField(
+            model_name="writelog",
+            name="space_uid",
+        ),
+        migrations.AddField(
+            model_name="writelog",
+            name="branch",
+            field=models.ForeignKey(
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.branch",
+            ),
+        ),
+        migrations.AddField(
+            model_name="writelog",
+            name="space",
+            field=models.ForeignKey(
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                to="lamindb.space",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="writelog",
+            name="run_uid",
+            field=models.CharField(default="0000000000000000", max_length=20),
+        ),
+        migrations.AlterField(
+            model_name="writelog",
+            name="record_uid",
+            field=models.JSONField(db_index=True, default=0),
+            preserve_default=False,
+        ),
+        migrations.AlterModelOptions(
+            name="migrationstate",
+            options={"base_manager_name": "objects"},
+        ),
+        migrations.AlterModelOptions(
+            name="tablestate",
+            options={"base_manager_name": "objects"},
+        ),
+        migrations.AlterField(
+            model_name="writelog",
+            name="migration_state",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.PROTECT, to="lamindb.migrationstate"
+            ),
+        ),
+        migrations.AlterField(
+            model_name="writelog",
+            name="table",
+            field=models.ForeignKey(
+                on_delete=django.db.models.deletion.PROTECT, to="lamindb.tablestate"
+            ),
+        ),
+    ]

lamindb/migrations/0103_remove_writelog_migration_state_and_more.py

@@ -0,0 +1,46 @@
+# Generated by Django 5.2 on 2025-05-29 12:02
+
+from django.db import migrations
+
+
+def fix_artifact_kind(apps, schema_editor):
+    Artifact = apps.get_model("lamindb", "Artifact")
+    Artifact.objects.filter(kind="__lamindb__").update(kind="__lamindb__run__")
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("lamindb", "0102_remove_writelog_branch_code_and_more"),
+    ]
+
+    operations = [
+        migrations.RunPython(fix_artifact_kind),
+        migrations.RemoveField(
+            model_name="writelog",
+            name="migration_state",
+        ),
+        migrations.RemoveField(
+            model_name="writelog",
+            name="table",
+        ),
+        migrations.RemoveField(
+            model_name="writelog",
+            name="branch",
+        ),
+        migrations.RemoveField(
+            model_name="writelog",
+            name="space",
+        ),
+        migrations.DeleteModel(
+            name="WriteLogLock",
+        ),
+        migrations.DeleteModel(
+            name="MigrationState",
+        ),
+        migrations.DeleteModel(
+            name="TableState",
+        ),
+        migrations.DeleteModel(
+            name="WriteLog",
+        ),
+    ]