lamindb_setup 1.8.3__py3-none-any.whl → 1.9.1__py3-none-any.whl

lamindb_setup/_migrate.py

@@ -1,259 +1,266 @@
-from __future__ import annotations
-
-from django.db import connection
-from django.db.migrations.loader import MigrationLoader
-from lamin_utils import logger
-from packaging import version
-
-from ._check_setup import _check_instance_setup, disable_auto_connect
-from .core._settings import settings
-from .core.django import setup_django
-
-
-# for the django-based synching code, see laminhub_rest
-def check_whether_migrations_in_sync(db_version_str: str):
-    from importlib import metadata
-
-    try:
-        installed_version_str = metadata.version("lamindb")
-    except metadata.PackageNotFoundError:
-        return None
-    if db_version_str is None:
-        logger.warning("no lamindb version stored to compare with installed version")
-        return None
-    installed_version = version.parse(installed_version_str)
-    db_version = version.parse(db_version_str)
-    if installed_version.major < db_version.major:
-        logger.warning(
-            f"the database ({db_version_str}) is far ahead of your installed lamindb package ({installed_version_str})"
-        )
-        logger.important(
-            f"please update lamindb: pip install lamindb>={db_version.major}"
-        )
-    elif (
-        installed_version.major == db_version.major
-        and installed_version.minor < db_version.minor
-    ):
-        db_version_lower = f"{db_version.major}.{db_version.minor}"
-        logger.important(
-            f"the database ({db_version_str}) is ahead of your installed lamindb"
-            f" package ({installed_version_str})"
-        )
-        logger.important(
-            f"consider updating lamindb: pip install lamindb>={db_version_lower}"
-        )
-    elif installed_version.major > db_version.major:
-        logger.warning(
-            f"the database ({db_version_str}) is far behind your installed lamindb package"
-            f" ({installed_version_str})"
-        )
-        logger.important(
-            "if you are an admin, migrate your database: lamin migrate deploy"
-        )
-    elif (
-        installed_version.major == db_version.major
-        and installed_version.minor > db_version.minor
-    ):
-        pass
-        # if the database is behind by a minor version, we don't want to spam the user
-        # logger.important(
-        #     f"the database ({db_version_str}) is behind your installed lamindb package"
-        #     f" ({installed_version_str})"
-        # )
-        # logger.important("consider migrating your database: lamin migrate deploy")
-
-
-# for tests, see lamin-cli
-class migrate:
-    """Manage migrations.
-
-    Examples:
-
-    >>> import lamindb as ln
-    >>> ln.setup.migrate.create()
-    >>> ln.setup.migrate.deploy()
-    >>> ln.setup.migrate.check()
-
-    """
-
-    @classmethod
-    @disable_auto_connect
-    def create(cls) -> None:
-        """Create a migration."""
-        if _check_instance_setup():
-            raise RuntimeError("Restart Python session to create migration or use CLI!")
-        setup_django(settings.instance, create_migrations=True)
-
-    @classmethod
-    @disable_auto_connect
-    def deploy(cls) -> None:
-        """Deploy a migration."""
-        from ._schema_metadata import update_schema_in_hub
-
-        if _check_instance_setup():
-            raise RuntimeError("Restart Python session to migrate or use CLI!")
-        from lamindb_setup.core._hub_client import call_with_fallback_auth
-        from lamindb_setup.core._hub_crud import (
-            select_collaborator,
-            update_instance,
-        )
-
-        if settings.instance.is_on_hub:
-            # double check that user is an admin, otherwise will fail below
-            # due to insufficient SQL permissions with cryptic error
-            collaborator = call_with_fallback_auth(
-                select_collaborator,
-                instance_id=settings.instance._id,
-                account_id=settings.user._uuid,
-            )
-            if collaborator is None or collaborator["role"] != "admin":
-                raise SystemExit(
-                    "❌ Only admins can deploy migrations, please ensure that you're an"
-                    f" admin: https://lamin.ai/{settings.instance.slug}/settings"
-                )
-            # we need lamindb to be installed, otherwise we can't populate the version
-            # information in the hub
-            import lamindb
-
-        # this sets up django and deploys the migrations
-        setup_django(settings.instance, deploy_migrations=True)
-        # this populates the hub
-        if settings.instance.is_on_hub:
-            logger.important(f"updating lamindb version in hub: {lamindb.__version__}")
-            # TODO: integrate update of instance table within update_schema_in_hub & below
-            if settings.instance.dialect != "sqlite":
-                update_schema_in_hub()
-            call_with_fallback_auth(
-                update_instance,
-                instance_id=settings.instance._id.hex,
-                instance_fields={"lamindb_version": lamindb.__version__},
-            )
-
-    @classmethod
-    @disable_auto_connect
-    def check(cls) -> bool:
-        """Check whether Registry definitions are in sync with migrations."""
-        from django.core.management import call_command
-
-        setup_django(settings.instance)
-        try:
-            call_command("makemigrations", check_changes=True)
-        except SystemExit:
-            logger.error(
-                "migrations are not in sync with ORMs, please create a migration: lamin"
-                " migrate create"
-            )
-            return False
-        return True
-
-    @classmethod
-    @disable_auto_connect
-    def squash(
-        cls, package_name, migration_nr, start_migration_nr: str | None = None
-    ) -> None:
-        """Squash migrations."""
-        from django.core.management import call_command
-
-        setup_django(settings.instance)
-        if start_migration_nr is not None:
-            call_command(
-                "squashmigrations", package_name, start_migration_nr, migration_nr
-            )
-        else:
-            call_command("squashmigrations", package_name, migration_nr)
-
-    @classmethod
-    @disable_auto_connect
-    def show(cls) -> None:
-        """Show migrations."""
-        from django.core.management import call_command
-
-        setup_django(settings.instance)
-        call_command("showmigrations")
-
-    @classmethod
-    def defined_migrations(cls, latest: bool = False):
-        from io import StringIO
-
-        from django.core.management import call_command
-
-        def parse_migration_output(output):
-            """Parse the output of the showmigrations command to get migration names."""
-            lines = output.splitlines()
-
-            # Initialize an empty dict to store migration names of each module
-            migration_names = {}
-
-            # Process each line
-            for line in lines:
-                if " " not in line:
-                    # CLI displays the module name in bold
-                    name = line.strip().replace("\x1b[1m", "")
-                    migration_names[name] = []
-                    continue
-                # Strip whitespace and split the line into status and migration name
-                migration_name = line.strip().split("] ")[-1].split(" ")[0]
-                # The second part is the migration name
-                migration_names[name].append(migration_name)
-
-            return migration_names
-
-        out = StringIO()
-        call_command("showmigrations", stdout=out)
-        out.seek(0)
-        output = out.getvalue()
-        if latest:
-            return {k: v[-1] for k, v in parse_migration_output(output).items()}
-        else:
-            return parse_migration_output(output)
-
-    @classmethod
-    def deployed_migrations(cls, latest: bool = False):
-        """Get the list of deployed migrations from Migration table in DB."""
-        if latest:
-            latest_migrations = {}
-            with connection.cursor() as cursor:
-                # query to get the latest migration for each app that is not squashed
-                cursor.execute(
-                    """
-                    SELECT app, name
-                    FROM django_migrations
-                    WHERE id IN (
-                        SELECT MAX(id)
-                        FROM django_migrations
-                        WHERE name NOT LIKE '%%_squashed_%%'
-                        GROUP BY app
-                    )
-                """
-                )
-                # fetch all the results
-                for app, name in cursor.fetchall():
-                    latest_migrations[app] = name
-
-            return latest_migrations
-        else:
-            # Load all migrations using Django's migration loader
-            loader = MigrationLoader(connection)
-            squashed_replacements = set()
-            for _key, migration in loader.disk_migrations.items():
-                if hasattr(migration, "replaces"):
-                    squashed_replacements.update(migration.replaces)
-
-            deployed_migrations: dict = {}
-            with connection.cursor() as cursor:
-                cursor.execute(
-                    """
-                    SELECT app, name, deployed
-                    FROM django_migrations
-                    ORDER BY app, deployed DESC
-                """
-                )
-                for app, name, _deployed in cursor.fetchall():
-                    # skip migrations that are part of a squashed migration
-                    if (app, name) in squashed_replacements:
-                        continue
-
-                    if app not in deployed_migrations:
-                        deployed_migrations[app] = []
-                    deployed_migrations[app].append(name)
-            return deployed_migrations
+from __future__ import annotations
+
+from django.db import connection
+from django.db.migrations.loader import MigrationLoader
+from lamin_utils import logger
+from packaging import version
+
+from ._check_setup import _check_instance_setup, disable_auto_connect
+from .core._settings import settings
+from .core.django import setup_django
+
+
+# for the django-based synching code, see laminhub_rest
+def check_whether_migrations_in_sync(db_version_str: str):
+    from importlib import metadata
+
+    try:
+        installed_version_str = metadata.version("lamindb")
+    except metadata.PackageNotFoundError:
+        return None
+    if db_version_str is None:
+        logger.warning("no lamindb version stored to compare with installed version")
+        return None
+    installed_version = version.parse(installed_version_str)
+    db_version = version.parse(db_version_str)
+    if installed_version.major < db_version.major:
+        logger.warning(
+            f"the database ({db_version_str}) is far ahead of your installed lamindb package ({installed_version_str})"
+        )
+        logger.important(
+            f"please update lamindb: pip install lamindb>={db_version.major}"
+        )
+    elif (
+        installed_version.major == db_version.major
+        and installed_version.minor < db_version.minor
+    ):
+        db_version_lower = f"{db_version.major}.{db_version.minor}"
+        logger.important(
+            f"the database ({db_version_str}) is ahead of your installed lamindb"
+            f" package ({installed_version_str})"
+        )
+        logger.important(
+            f"consider updating lamindb: pip install lamindb>={db_version_lower}"
+        )
+    elif installed_version.major > db_version.major:
+        logger.warning(
+            f"the database ({db_version_str}) is far behind your installed lamindb package"
+            f" ({installed_version_str})"
+        )
+        logger.important(
+            "if you are an admin, migrate your database: lamin migrate deploy"
+        )
+    elif (
+        installed_version.major == db_version.major
+        and installed_version.minor > db_version.minor
+    ):
+        pass
+        # if the database is behind by a minor version, we don't want to spam the user
+        # logger.important(
+        #     f"the database ({db_version_str}) is behind your installed lamindb package"
+        #     f" ({installed_version_str})"
+        # )
+        # logger.important("consider migrating your database: lamin migrate deploy")
+
+
+# for tests, see lamin-cli
+class migrate:
+    """Manage migrations.
+
+    Examples:
+
+    >>> import lamindb as ln
+    >>> ln.setup.migrate.create()
+    >>> ln.setup.migrate.deploy()
+    >>> ln.setup.migrate.check()
+
+    """
+
+    @classmethod
+    @disable_auto_connect
+    def create(cls) -> None:
+        """Create a migration."""
+        if _check_instance_setup():
+            raise RuntimeError("Restart Python session to create migration or use CLI!")
+        setup_django(settings.instance, create_migrations=True)
+
+    @classmethod
+    @disable_auto_connect
+    def deploy(cls, package_name: str | None = None, number: int | None = None) -> None:
+        """Deploy a migration."""
+        from ._schema_metadata import update_schema_in_hub
+
+        if _check_instance_setup():
+            raise RuntimeError("Restart Python session to migrate or use CLI!")
+        from lamindb_setup.core._hub_client import call_with_fallback_auth
+        from lamindb_setup.core._hub_crud import (
+            select_collaborator,
+            update_instance,
+        )
+
+        if settings.instance.is_on_hub:
+            # double check that user is an admin, otherwise will fail below
+            # due to insufficient SQL permissions with cryptic error
+            collaborator = call_with_fallback_auth(
+                select_collaborator,
+                instance_id=settings.instance._id,
+                account_id=settings.user._uuid,
+            )
+            if collaborator is None or collaborator["role"] != "admin":
+                raise SystemExit(
+                    "❌ Only admins can deploy migrations, please ensure that you're an"
+                    f" admin: https://lamin.ai/{settings.instance.slug}/settings"
+                )
+            # we need lamindb to be installed, otherwise we can't populate the version
+            # information in the hub
+            import lamindb
+
+        # this sets up django and deploys the migrations
+        if package_name is not None and number is not None:
+            setup_django(
+                settings.instance,
+                deploy_migrations=True,
+                appname_number=(package_name, number),
+            )
+        else:
+            setup_django(settings.instance, deploy_migrations=True)
+        # this populates the hub
+        if settings.instance.is_on_hub:
+            logger.important(f"updating lamindb version in hub: {lamindb.__version__}")
+            # TODO: integrate update of instance table within update_schema_in_hub & below
+            if settings.instance.dialect != "sqlite":
+                update_schema_in_hub()
+            call_with_fallback_auth(
+                update_instance,
+                instance_id=settings.instance._id.hex,
+                instance_fields={"lamindb_version": lamindb.__version__},
+            )
+
+    @classmethod
+    @disable_auto_connect
+    def check(cls) -> bool:
+        """Check whether Registry definitions are in sync with migrations."""
+        from django.core.management import call_command
+
+        setup_django(settings.instance)
+        try:
+            call_command("makemigrations", check_changes=True)
+        except SystemExit:
+            logger.error(
+                "migrations are not in sync with ORMs, please create a migration: lamin"
+                " migrate create"
+            )
+            return False
+        return True
+
+    @classmethod
+    @disable_auto_connect
+    def squash(
+        cls, package_name, migration_nr, start_migration_nr: str | None = None
+    ) -> None:
+        """Squash migrations."""
+        from django.core.management import call_command
+
+        setup_django(settings.instance)
+        if start_migration_nr is not None:
+            call_command(
+                "squashmigrations", package_name, start_migration_nr, migration_nr
+            )
+        else:
+            call_command("squashmigrations", package_name, migration_nr)
+
+    @classmethod
+    @disable_auto_connect
+    def show(cls) -> None:
+        """Show migrations."""
+        from django.core.management import call_command
+
+        setup_django(settings.instance)
+        call_command("showmigrations")
+
+    @classmethod
+    def defined_migrations(cls, latest: bool = False):
+        from io import StringIO
+
+        from django.core.management import call_command
+
+        def parse_migration_output(output):
+            """Parse the output of the showmigrations command to get migration names."""
+            lines = output.splitlines()
+
+            # Initialize an empty dict to store migration names of each module
+            migration_names = {}
+
+            # Process each line
+            for line in lines:
+                if " " not in line:
+                    # CLI displays the module name in bold
+                    name = line.strip().replace("\x1b[1m", "")
+                    migration_names[name] = []
+                    continue
+                # Strip whitespace and split the line into status and migration name
+                migration_name = line.strip().split("] ")[-1].split(" ")[0]
+                # The second part is the migration name
+                migration_names[name].append(migration_name)
+
+            return migration_names
+
+        out = StringIO()
+        call_command("showmigrations", stdout=out)
+        out.seek(0)
+        output = out.getvalue()
+        if latest:
+            return {k: v[-1] for k, v in parse_migration_output(output).items()}
+        else:
+            return parse_migration_output(output)
+
+    @classmethod
+    def deployed_migrations(cls, latest: bool = False):
+        """Get the list of deployed migrations from Migration table in DB."""
+        if latest:
+            latest_migrations = {}
+            with connection.cursor() as cursor:
+                # query to get the latest migration for each app that is not squashed
+                cursor.execute(
+                    """
+                    SELECT app, name
+                    FROM django_migrations
+                    WHERE id IN (
+                        SELECT MAX(id)
+                        FROM django_migrations
+                        WHERE name NOT LIKE '%%_squashed_%%'
+                        GROUP BY app
+                    )
+                """
+                )
+                # fetch all the results
+                for app, name in cursor.fetchall():
+                    latest_migrations[app] = name
+
+            return latest_migrations
+        else:
+            # Load all migrations using Django's migration loader
+            loader = MigrationLoader(connection)
+            squashed_replacements = set()
+            for _key, migration in loader.disk_migrations.items():
+                if hasattr(migration, "replaces"):
+                    squashed_replacements.update(migration.replaces)
+
+            deployed_migrations: dict = {}
+            with connection.cursor() as cursor:
+                cursor.execute(
+                    """
+                    SELECT app, name, deployed
+                    FROM django_migrations
+                    ORDER BY app, deployed DESC
+                """
+                )
+                for app, name, _deployed in cursor.fetchall():
+                    # skip migrations that are part of a squashed migration
+                    if (app, name) in squashed_replacements:
+                        continue
+
+                    if app not in deployed_migrations:
+                        deployed_migrations[app] = []
+                    deployed_migrations[app].append(name)
+            return deployed_migrations

lamindb_setup/_register_instance.py

@@ -1,35 +1,35 @@
-from __future__ import annotations
-
-from lamin_utils import logger
-
-from .core._settings import settings
-from .core._settings_storage import base62
-from .core.django import setup_django
-
-
-def register(_test: bool = False):
-    """Register an instance on the hub."""
-    from ._check_setup import _check_instance_setup
-    from .core._hub_core import init_instance_hub, init_storage_hub
-
-    logger.warning("note that register() is only for testing purposes")
-
-    isettings = settings.instance
-    if not _check_instance_setup() and not _test:
-        setup_django(isettings)
-
-    ssettings = settings.instance.storage
-    if ssettings._uid is None and _test:
-        # because django isn't up, we can't get it from the database
-        ssettings._uid = base62(8)
-    # cannot yet populate the instance id here
-    ssettings._instance_id = None
-    # flag auto_populate_instance can be removed once FK migration is over
-    init_storage_hub(ssettings, auto_populate_instance=False)
-    init_instance_hub(isettings)
-    isettings._is_on_hub = True
-    isettings._persist()
-    if isettings.dialect != "sqlite" and not _test:
-        from ._schema_metadata import update_schema_in_hub
-
-        update_schema_in_hub()
+from __future__ import annotations
+
+from lamin_utils import logger
+
+from .core._settings import settings
+from .core._settings_storage import base62
+from .core.django import setup_django
+
+
+def register(_test: bool = False):
+    """Register an instance on the hub."""
+    from ._check_setup import _check_instance_setup
+    from .core._hub_core import init_instance_hub, init_storage_hub
+
+    logger.warning("note that register() is only for testing purposes")
+
+    isettings = settings.instance
+    if not _check_instance_setup() and not _test:
+        setup_django(isettings)
+
+    ssettings = settings.instance.storage
+    if ssettings._uid is None and _test:
+        # because django isn't up, we can't get it from the database
+        ssettings._uid = base62(8)
+    # cannot yet populate the instance id here
+    ssettings._instance_id = None
+    # flag auto_populate_instance can be removed once FK migration is over
+    init_storage_hub(ssettings, auto_populate_instance=False)
+    init_instance_hub(isettings)
+    isettings._is_on_hub = True
+    isettings._persist()
+    if isettings.dialect != "sqlite" and not _test:
+        from ._schema_metadata import update_schema_in_hub
+
+        update_schema_in_hub()