lamindb_setup 1.18.2__py3-none-any.whl → 1.19.1__py3-none-any.whl

lamindb_setup/_connect_instance.py

@@ -1,438 +1,443 @@
-from __future__ import annotations
-
-import importlib
-import os
-import sys
-import types
-from typing import TYPE_CHECKING, Any
-from uuid import UUID
-
-from lamin_utils import logger
-
-from ._check_setup import _check_instance_setup
-from ._disconnect import disconnect
-from ._init_instance import load_from_isettings
-from ._silence_loggers import silence_loggers
-from .core._hub_core import connect_instance_hub
-from .core._hub_utils import LaminDsnModel
-from .core._settings import settings
-from .core._settings_instance import InstanceSettings
-from .core._settings_load import load_instance_settings
-from .core._settings_storage import StorageSettings
-from .core._settings_store import instance_settings_file
-from .core.cloud_sqlite_locker import unlock_cloud_sqlite_upon_exception
-from .core.django import reset_django
-from .errors import CannotSwitchDefaultInstance, InstanceNotFoundError
-
-if TYPE_CHECKING:
-    from pathlib import Path
-
-    from .core._settings_user import UserSettings
-    from .types import UPathStr
-
-# this is for testing purposes only
-# set to True only to test failed load
-_TEST_FAILED_LOAD = False
-INSTANCE_NOT_FOUND_MESSAGE = (
-    "'{owner}/{name}' not found:"
-    " '{hub_result}'\nCheck your permissions:"
-    " https://lamin.ai/{owner}/{name}"
-)
-
-
-def check_db_dsn_equal_up_to_credentials(db_dsn_hub, db_dsn_local):
-    return (
-        db_dsn_hub.scheme == db_dsn_local.scheme
-        and db_dsn_hub.host == db_dsn_local.host
-        and db_dsn_hub.database == db_dsn_local.database
-        and db_dsn_hub.port == db_dsn_local.port
-    )
-
-
-def update_db_using_local(
-    hub_instance_result: dict[str, str],
-    settings_file: Path,
-    db: str | None = None,
-    raise_permission_error=True,
-) -> str | None:
-    db_updated = None
-    # check if postgres
-    if hub_instance_result["db_scheme"] == "postgresql":
-        if db is not None:
-            # use only the provided db if it is set
-            db_updated = db
-        elif (db_env := os.getenv("LAMINDB_INSTANCE_DB")) is not None:
-            logger.important("loading db URL from env variable LAMINDB_INSTANCE_DB")
-            # read directly from the environment
-            db_updated = db_env
-        else:
-            db_hub = hub_instance_result["db"]
-            db_dsn_hub = LaminDsnModel(db=db_hub)
-            # read from a cached settings file in case the hub result is inexistent
-            if db_dsn_hub.db.user in {None, "none"} and settings_file.exists():
-                isettings = load_instance_settings(settings_file)
-                db_updated = isettings.db
-            else:
-                # just take the default hub result and ensure there is actually a user
-                if (
-                    db_dsn_hub.db.user in {None, "none"}
-                    and db_dsn_hub.db.password in {None, "none"}
-                    and raise_permission_error
-                ):
-                    raise PermissionError(
-                        "No database access, please ask your admin to provide you with"
-                        " a DB URL and pass it via --db <db_url>"
-                    )
-                db_updated = db_hub
-    return db_updated
-
-
-def _connect_instance(
-    owner: str,
-    name: str,
-    *,
-    db: str | None = None,
-    raise_permission_error: bool = True,
-    use_root_db_user: bool = False,
-    use_proxy_db: bool = False,
-    access_token: str | None = None,
-    raise_systemexit: bool = False,
-) -> InstanceSettings:
-    settings_file = instance_settings_file(name, owner)
-    make_hub_request = True
-    if settings_file.exists():
-        isettings = load_instance_settings(settings_file)
-        # skip hub request for a purely local instance
-        if isettings.is_remote:
-            make_hub_request = True
-        else:
-            make_hub_request = False
-            if db is not None and isettings.dialect == "postgresql":
-                isettings._db = db
-    if make_hub_request:
-        # the following will return a string if the instance does not exist on the hub
-        # do not call hub if the user is anonymous
-        if owner != "anonymous":
-            hub_result = connect_instance_hub(
-                owner=owner,
-                name=name,
-                access_token=access_token,
-                use_root_db_user=use_root_db_user,
-                use_proxy_db=use_proxy_db,
-            )
-        else:
-            hub_result = "anonymous-user"
-        # if hub_result is not a string, it means it made a request
-        # that successfully returned metadata
-        if not isinstance(hub_result, str):
-            instance_result, storage_result = hub_result
-            db_updated = update_db_using_local(
-                instance_result,
-                settings_file,
-                db=db,
-                raise_permission_error=raise_permission_error,
-            )
-            ssettings = StorageSettings(
-                root=storage_result["root"],
-                region=storage_result["region"],
-                uid=storage_result["lnid"],
-                uuid=UUID(storage_result["id"]),
-                instance_id=UUID(instance_result["id"]),
-            )
-            isettings = InstanceSettings(
-                id=UUID(instance_result["id"]),
-                owner=owner,
-                name=instance_result["name"],
-                storage=ssettings,
-                db=db_updated,
-                modules=instance_result["schema_str"],
-                git_repo=instance_result["git_repo"],
-                keep_artifacts_local=bool(instance_result["keep_artifacts_local"]),
-                is_on_hub=True,
-                api_url=instance_result["api_url"],
-                schema_id=None
-                if (schema_id := instance_result["schema_id"]) is None
-                else UUID(schema_id),
-                fine_grained_access=bool(instance_result["fine_grained_access"]),
-                db_permissions=instance_result.get("db_permissions", None)
-                if not use_root_db_user
-                else "write",
-            )
-        else:
-            if hub_result != "anonymous-user":
-                message = INSTANCE_NOT_FOUND_MESSAGE.format(
-                    owner=owner, name=name, hub_result=hub_result
-                )
-            else:
-                message = "It is not possible to load an anonymous-owned instance from the hub"
-            exception = (
-                SystemExit(message)
-                if raise_systemexit
-                else InstanceNotFoundError(message)
-            )
-            if settings_file.exists():
-                isettings = load_instance_settings(settings_file)
-                if isettings.is_remote:
-                    raise exception
-            else:
-                raise exception
-    return isettings
-
-
-def reset_django_module_variables():
-    # This function updates all module-level references to Django classes
-    # But it will fail to update function level references
-    # This is not a problem unless for the function that calls ln.connect() itself
-    # So, if a user has
-    #
-    # def my_function():
-    #     import lamindb as ln
-    #     ln.connect(...)
-    #
-    # Then it will **not** work and the `ln` variable becomes stale and hold a reference to the old classes
-    # Other functions that dynamically import are no problem because the variables
-    # are automatically refreshed when the function runs the next time after ln.connect() was called
-    logger.debug("resetting django module variables")
-
-    # django.apps needs to be a local import to refresh variables
-    from django.apps import apps
-
-    app_names = {app.name for app in apps.get_app_configs()}
-    # always copy before iterations over sys.modules
-    # see https://docs.python.org/3/library/sys.html#sys.modules
-    # this whole thing runs about 50ms in a big env
-    for name, module in sys.modules.copy().items():
-        if (
-            module is not None
-            and (not name.startswith("__") or name == "__main__")
-            and name not in sys.builtin_module_names
-        ):
-            try:
-                for k, v in vars(module).items():
-                    if (
-                        isinstance(v, types.ModuleType)
-                        and not k.startswith("_")
-                        and getattr(v, "__name__", None) in app_names
-                    ):
-                        if v.__name__ in sys.modules:
-                            vars(module)[k] = sys.modules[v.__name__]
-                    # Also reset classes from Django apps - but check if the class module starts with any app name
-                    elif hasattr(v, "__module__") and getattr(v, "__module__", None):
-                        class_module = v.__module__
-                        # Check if the class module starts with any of our app names
-                        if any(
-                            class_module.startswith(app_name) for app_name in app_names
-                        ):
-                            if class_module in sys.modules:
-                                fresh_module = sys.modules[class_module]
-                                attr_name = getattr(v, "__name__", k)
-                                if hasattr(fresh_module, attr_name):
-                                    vars(module)[k] = getattr(fresh_module, attr_name)
-            except (AttributeError, TypeError):
-                continue
-
-
-def _connect_cli(
-    instance: str, use_root_db_user: bool = False, use_proxy_db: bool = False
-) -> None:
-    from lamindb_setup import settings as settings_
-
-    owner, name = get_owner_name_from_identifier(instance)
-    isettings = _connect_instance(
-        owner,
-        name,
-        use_root_db_user=use_root_db_user,
-        use_proxy_db=use_proxy_db,
-        raise_systemexit=True,
-    )
-    isettings._persist(write_to_disk=True)
-    if not isettings.is_on_hub or isettings._is_cloud_sqlite:
-        # there are two reasons to call the full-blown connect
-        # (1) if the instance is not on the hub, we need to register
-        # potential users through register_user()
-        # (2) if the instance is cloud sqlite, we need to lock it
-        connect(_write_settings=False, _reload_lamindb=False)
-    else:
-        logger.important(f"connected lamindb: {isettings.slug}")
-    if settings_.dev_dir is None:
-        logger.important_hint(
-            "to map a local dev directory, call: lamin settings set dev-dir ."
-        )
-    return None
-
-
-def validate_connection_state(
-    owner: str, name: str, use_root_db_user: bool = False
-) -> None:
-    from django.db import connection
-
-    if (
-        settings._instance_exists  # exists only for real instances, not for none/none
-        and f"{owner}/{name}" == settings.instance.slug
-        and not use_root_db_user  # always re-connect for root db user
-    ):
-        logger.important(
-            f"doing nothing, already connected lamindb: {settings.instance.slug}"
-        )
-        return None
-    else:
-        if settings._instance_exists:
-            import lamindb as ln
-
-            if ln.context.transform is not None:
-                raise CannotSwitchDefaultInstance(
-                    "Cannot switch default instance while `ln.track()` is live: call `ln.finish()`"
-                )
-        reset_django()
-
-
-@unlock_cloud_sqlite_upon_exception(ignore_prev_locker=True)
-def connect(instance: str | None = None, **kwargs: Any) -> str | tuple | None:
-    """Connect the global default instance.
-
-    If you want to create a read-only database client, use :class:`~lamindb.DB` instead.
-
-    Args:
-        instance: Pass a slug (`account/name`) or URL (`https://lamin.ai/account/name`).
-            If `None`, looks for an environment variable `LAMIN_CURRENT_INSTANCE` to get the instance identifier.
-            If it doesn't find this variable, it connects to the instance that was connected with `lamin connect` through the CLI.
-
-    See Also:
-        Configure an instance for auto-connect via the CLI, see `here <https://docs.lamin.ai/cli#connect>`__.
-    """
-    # validate kwargs
-    valid_kwargs = {
-        "use_root_db_user",
-        "use_proxy_db",
-        "_db",
-        "_write_settings",
-        "_raise_not_found_error",
-        "_reload_lamindb",
-        "_test",
-        "_user",
-    }
-    for kwarg in kwargs:
-        if kwarg not in valid_kwargs:
-            raise TypeError(f"connect() got unexpected keyword argument '{kwarg}'")
-
-    use_root_db_user: bool = kwargs.get("use_root_db_user", False)
-    use_proxy_db = kwargs.get("use_proxy_db", False)
-    # _db is still needed because it is called in init
-    _db: str | None = kwargs.get("_db", None)
-    _write_settings: bool = kwargs.get("_write_settings", False)
-    _raise_not_found_error: bool = kwargs.get("_raise_not_found_error", True)
-    _reload_lamindb: bool = kwargs.get("_reload_lamindb", True)
-    _test: bool = kwargs.get("_test", False)
-
-    isettings: InstanceSettings = None  # type: ignore
-
-    access_token: str | None = None
-    _user: UserSettings | None = kwargs.get("_user", None)
-    if _user is not None:
-        access_token = _user.access_token
-    if instance is None:
-        instance = os.environ.get("LAMIN_CURRENT_INSTANCE")
-
-    try:
-        if instance is None:
-            if settings._instance_exists:
-                isettings = settings.instance
-            else:
-                raise ValueError(
-                    "No instance was connected through the CLI, pass a value to `instance` or connect via the CLI."
-                )
-            if use_root_db_user:
-                reset_django()
-                owner, name = isettings.owner, isettings.name
-            if _db is not None and isettings.dialect == "postgresql":
-                isettings._db = _db
-        else:
-            owner, name = get_owner_name_from_identifier(instance)
-            if _check_instance_setup() and not _test:
-                validate_connection_state(
-                    owner, name, use_root_db_user=use_root_db_user
-                )
-            elif (
-                _write_settings
-                and settings._instance_exists
-                and f"{owner}/{name}" != settings.instance.slug
-            ):
-                disconnect(mute=True)
-
-        if instance is not None or use_root_db_user:
-            try:
-                isettings = _connect_instance(
-                    owner,
-                    name,
-                    db=_db,
-                    access_token=access_token,
-                    use_root_db_user=use_root_db_user,
-                    use_proxy_db=use_proxy_db,
-                )
-            except InstanceNotFoundError as e:
-                if _raise_not_found_error:
-                    raise e
-                else:
-                    return "instance-not-found"
-            if isinstance(isettings, str):
-                return isettings
-        # at this point we have checked already that isettings is not a string
-        # _user is passed to lock cloud sqlite for this user in isettings._load_db()
-        # has no effect if _user is None or if not cloud sqlite instance
-        isettings._locker_user = _user
-        isettings._persist(write_to_disk=_write_settings)
-        if _test:
-            return None
-        silence_loggers()
-        check, msg = isettings._load_db()
-        if not check:
-            local_db = (
-                isettings._is_cloud_sqlite and isettings._sqlite_file_local.exists()
-            )
-            if local_db:
-                logger.warning(
-                    "SQLite file does not exist in the cloud, but exists locally:"
-                    f" {isettings._sqlite_file_local}\nTo push the file to the cloud,"
-                    " call: lamin disconnect"
-                )
-            elif _raise_not_found_error:
-                raise SystemExit(msg)
-            else:
-                logger.warning(
-                    f"instance exists with id {isettings._id.hex}, but database is not"
-                    " loadable: re-initializing"
-                )
-                return "instance-corrupted-or-deleted"
-        # this is for testing purposes only
-        if _TEST_FAILED_LOAD:
-            raise RuntimeError("Technical testing error.")
-
-        load_from_isettings(isettings, user=_user, write_settings=_write_settings)
-        if _reload_lamindb:
-            reset_django_module_variables()
-        if isettings.slug != "none/none":
-            logger.important(f"connected lamindb: {isettings.slug}")
-    except Exception as e:
-        if isettings is not None:
-            if _write_settings:
-                isettings._get_settings_file().unlink(missing_ok=True)  # type: ignore
-            settings._instance_settings = None
-        raise e
-    return None
-
-
-def get_owner_name_from_identifier(identifier: str) -> tuple[str, str]:
-    if "/" in identifier:
-        if identifier.startswith("https://lamin.ai/"):
-            identifier = identifier.replace("https://lamin.ai/", "")
-        split = identifier.split("/")
-        if len(split) > 2:
-            raise ValueError(
-                "The instance identifier needs to be 'owner/name', the instance name"
-                " (owner is current user) or the URL: https://lamin.ai/owner/name."
-            )
-        owner, name = split
-    else:
-        owner = settings.user.handle
-        name = identifier
-    return owner, name
+from __future__ import annotations
+
+import importlib
+import os
+import sys
+import types
+from typing import TYPE_CHECKING, Any
+from uuid import UUID
+
+from lamin_utils import logger
+
+from ._check_setup import _check_instance_setup
+from ._disconnect import disconnect
+from ._init_instance import load_from_isettings
+from ._silence_loggers import silence_loggers
+from .core._settings import settings
+from .core._settings_instance import InstanceSettings
+from .core._settings_load import load_instance_settings
+from .core._settings_storage import StorageSettings
+from .core._settings_store import instance_settings_file
+from .core.cloud_sqlite_locker import unlock_cloud_sqlite_upon_exception
+from .core.django import reset_django
+from .errors import CannotSwitchDefaultInstance, InstanceNotFoundError
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+    from .core._settings_user import UserSettings
+    from .types import UPathStr
+
+# this is for testing purposes only
+# set to True only to test failed load
+_TEST_FAILED_LOAD = False
+INSTANCE_NOT_FOUND_MESSAGE = (
+    "'{owner}/{name}' not found:"
+    " '{hub_result}'\nCheck your permissions:"
+    " https://lamin.ai/{owner}/{name}"
+)
+
+
+def check_db_dsn_equal_up_to_credentials(db_dsn_hub, db_dsn_local):
+    return (
+        db_dsn_hub.scheme == db_dsn_local.scheme
+        and db_dsn_hub.host == db_dsn_local.host
+        and db_dsn_hub.database == db_dsn_local.database
+        and db_dsn_hub.port == db_dsn_local.port
+    )
+
+
+def update_db_using_local(
+    hub_instance_result: dict[str, str],
+    settings_file: Path,
+    db: str | None = None,
+    raise_permission_error=True,
+) -> str | None:
+    db_updated = None
+    # check if postgres
+    if hub_instance_result["db_scheme"] == "postgresql":
+        if db is not None:
+            # use only the provided db if it is set
+            db_updated = db
+        elif (db_env := os.getenv("LAMINDB_INSTANCE_DB")) is not None:
+            logger.important("loading db URL from env variable LAMINDB_INSTANCE_DB")
+            # read directly from the environment
+            db_updated = db_env
+        else:
+            # dynamic import to avoid importing the heavy LaminDsnModel at root
+            from .core._hub_utils import LaminDsnModel
+
+            db_hub = hub_instance_result["db"]
+            db_dsn_hub = LaminDsnModel(db=db_hub)
+            # read from a cached settings file in case the hub result is inexistent
+            if db_dsn_hub.db.user in {None, "none"} and settings_file.exists():
+                isettings = load_instance_settings(settings_file)
+                db_updated = isettings.db
+            else:
+                # just take the default hub result and ensure there is actually a user
+                if (
+                    db_dsn_hub.db.user in {None, "none"}
+                    and db_dsn_hub.db.password in {None, "none"}
+                    and raise_permission_error
+                ):
+                    raise PermissionError(
+                        "No database access, please ask your admin to provide you with"
+                        " a DB URL and pass it via --db <db_url>"
+                    )
+                db_updated = db_hub
+    return db_updated
+
+
+def _connect_instance(
+    owner: str,
+    name: str,
+    *,
+    db: str | None = None,
+    raise_permission_error: bool = True,
+    use_root_db_user: bool = False,
+    use_proxy_db: bool = False,
+    access_token: str | None = None,
+    raise_systemexit: bool = False,
+) -> InstanceSettings:
+    settings_file = instance_settings_file(name, owner)
+    make_hub_request = True
+    if settings_file.exists():
+        isettings = load_instance_settings(settings_file)
+        # skip hub request for a purely local instance
+        if isettings.is_remote:
+            make_hub_request = True
+        else:
+            make_hub_request = False
+            if db is not None and isettings.dialect == "postgresql":
+                isettings._db = db
+    if make_hub_request:
+        # the following will return a string if the instance does not exist on the hub
+        # do not call hub if the user is anonymous
+        if owner != "anonymous":
+            from .core._hub_core import connect_instance_hub
+
+            hub_result = connect_instance_hub(
+                owner=owner,
+                name=name,
+                access_token=access_token,
+                use_root_db_user=use_root_db_user,
+                use_proxy_db=use_proxy_db,
+            )
+        else:
+            hub_result = "anonymous-user"
+        # if hub_result is not a string, it means it made a request
+        # that successfully returned metadata
+        if not isinstance(hub_result, str):
+            instance_result, storage_result = hub_result
+            db_updated = update_db_using_local(
+                instance_result,
+                settings_file,
+                db=db,
+                raise_permission_error=raise_permission_error,
+            )
+            ssettings = StorageSettings(
+                root=storage_result["root"],
+                region=storage_result["region"],
+                uid=storage_result["lnid"],
+                uuid=UUID(storage_result["id"]),
+                instance_id=UUID(instance_result["id"]),
+            )
+            isettings = InstanceSettings(
+                id=UUID(instance_result["id"]),
+                owner=owner,
+                name=instance_result["name"],
+                storage=ssettings,
+                db=db_updated,
+                modules=instance_result["schema_str"],
+                git_repo=instance_result["git_repo"],
+                keep_artifacts_local=bool(instance_result["keep_artifacts_local"]),
+                is_on_hub=True,
+                api_url=instance_result["api_url"],
+                schema_id=None
+                if (schema_id := instance_result["schema_id"]) is None
+                else UUID(schema_id),
+                fine_grained_access=bool(instance_result["fine_grained_access"]),
+                db_permissions=instance_result.get("db_permissions", None)
+                if not use_root_db_user
+                else "write",
+            )
+        else:
+            if hub_result != "anonymous-user":
+                message = INSTANCE_NOT_FOUND_MESSAGE.format(
+                    owner=owner, name=name, hub_result=hub_result
+                )
+            else:
+                message = "It is not possible to load an anonymous-owned instance from the hub"
+            exception = (
+                SystemExit(message)
+                if raise_systemexit
+                else InstanceNotFoundError(message)
+            )
+            if settings_file.exists():
+                isettings = load_instance_settings(settings_file)
+                if isettings.is_remote:
+                    raise exception
+            else:
+                raise exception
+    return isettings
+
+
+def reset_django_module_variables():
+    # This function updates all module-level references to Django classes
+    # But it will fail to update function level references
+    # This is not a problem unless for the function that calls ln.connect() itself
+    # So, if a user has
+    #
+    # def my_function():
+    #     import lamindb as ln
+    #     ln.connect(...)
+    #
+    # Then it will **not** work and the `ln` variable becomes stale and hold a reference to the old classes
+    # Other functions that dynamically import are no problem because the variables
+    # are automatically refreshed when the function runs the next time after ln.connect() was called
+    logger.debug("resetting django module variables")
+
+    # django.apps needs to be a local import to refresh variables
+    from django.apps import apps
+
+    app_names = {app.name for app in apps.get_app_configs()}
+    # always copy before iterations over sys.modules
+    # see https://docs.python.org/3/library/sys.html#sys.modules
+    # this whole thing runs about 50ms in a big env
+    for name, module in sys.modules.copy().items():
+        if (
+            module is not None
+            and (not name.startswith("__") or name == "__main__")
+            and name not in sys.builtin_module_names
+        ):
+            try:
+                module_vars = vars(module)  # references the original
+                # copy to avoid changing size during the loop
+                for k, v in module_vars.copy().items():
+                    if (
+                        isinstance(v, types.ModuleType)
+                        and not k.startswith("_")
+                        and getattr(v, "__name__", None) in app_names
+                    ):
+                        if v.__name__ in sys.modules:
+                            module_vars[k] = sys.modules[v.__name__]
+                    # Also reset classes from Django apps - but check if the class module starts with any app name
+                    elif hasattr(v, "__module__") and getattr(v, "__module__", None):
+                        class_module = v.__module__
+                        # Check if the class module starts with any of our app names
+                        if any(
+                            class_module.startswith(app_name) for app_name in app_names
+                        ):
+                            if class_module in sys.modules:
+                                fresh_module = sys.modules[class_module]
+                                attr_name = getattr(v, "__name__", k)
+                                if hasattr(fresh_module, attr_name):
+                                    module_vars[k] = getattr(fresh_module, attr_name)
+            except (AttributeError, TypeError):
+                continue
+
+
+def _connect_cli(
+    instance: str,
+    use_root_db_user: bool = False,
+) -> None:
+    from lamindb_setup import settings as settings_
+
+    owner, name = get_owner_name_from_identifier(instance)
+    isettings = _connect_instance(
+        owner,
+        name,
+        use_root_db_user=use_root_db_user,
+        raise_systemexit=True,
+    )
+    isettings._persist(write_to_disk=True)
+    if not isettings.is_on_hub or isettings._is_cloud_sqlite:
+        # there are two reasons to call the full-blown connect
+        # (1) if the instance is not on the hub, we need to register
+        # potential users through register_user()
+        # (2) if the instance is cloud sqlite, we need to lock it
+        connect(_write_settings=False, _reload_lamindb=False)
+    else:
+        logger.important(f"connected lamindb: {isettings.slug}")
+    if settings_.dev_dir is None:
+        logger.important_hint(
+            "to map a local dev directory, call: lamin settings set dev-dir ."
+        )
+    return None
+
+
+def validate_connection_state(
+    owner: str, name: str, use_root_db_user: bool = False
+) -> None:
+    from django.db import connection
+
+    if (
+        settings._instance_exists  # exists only for real instances, not for none/none
+        and f"{owner}/{name}" == settings.instance.slug
+        and not use_root_db_user  # always re-connect for root db user
+    ):
+        logger.important(
+            f"doing nothing, already connected lamindb: {settings.instance.slug}"
+        )
+        return None
+    else:
+        if settings._instance_exists:
+            import lamindb as ln
+
+            if ln.context.transform is not None:
+                raise CannotSwitchDefaultInstance(
+                    "Cannot switch default instance while `ln.track()` is live: call `ln.finish()`"
+                )
+        reset_django()
+
+
+@unlock_cloud_sqlite_upon_exception(ignore_prev_locker=True)
+def connect(instance: str | None = None, **kwargs: Any) -> str | tuple | None:
+    """Connect the global default instance.
+
+    If you want to create a read-only database client, use :class:`~lamindb.DB` instead.
+
+    Args:
+        instance: Pass a slug (`account/name`) or URL (`https://lamin.ai/account/name`).
+            If `None`, looks for an environment variable `LAMIN_CURRENT_INSTANCE` to get the instance identifier.
+            If it doesn't find this variable, it connects to the instance that was connected with `lamin connect` through the CLI.
+
+    See Also:
+        Configure an instance for auto-connect via the CLI, see `here <https://docs.lamin.ai/cli#connect>`__.
+    """
+    # validate kwargs
+    valid_kwargs = {
+        "use_root_db_user",
+        "use_proxy_db",
+        "_db",
+        "_write_settings",
+        "_raise_not_found_error",
+        "_reload_lamindb",
+        "_test",
+        "_user",
+    }
+    for kwarg in kwargs:
+        if kwarg not in valid_kwargs:
+            raise TypeError(f"connect() got unexpected keyword argument '{kwarg}'")
+
+    use_root_db_user: bool = kwargs.get("use_root_db_user", False)
+    use_proxy_db = kwargs.get("use_proxy_db", False)
+    # _db is still needed because it is called in init
+    _db: str | None = kwargs.get("_db", None)
+    _write_settings: bool = kwargs.get("_write_settings", False)
+    _raise_not_found_error: bool = kwargs.get("_raise_not_found_error", True)
+    _reload_lamindb: bool = kwargs.get("_reload_lamindb", True)
+    _test: bool = kwargs.get("_test", False)
+
+    isettings: InstanceSettings = None  # type: ignore
+
+    access_token: str | None = None
+    _user: UserSettings | None = kwargs.get("_user", None)
+    if _user is not None:
+        access_token = _user.access_token
+    if instance is None:
+        instance = os.environ.get("LAMIN_CURRENT_INSTANCE")
+
+    try:
+        if instance is None:
+            if settings._instance_exists:
+                isettings = settings.instance
+            else:
+                raise ValueError(
+                    "No instance was connected through the CLI, pass a value to `instance` or connect via the CLI."
+                )
+            if use_root_db_user:
+                reset_django()
+                owner, name = isettings.owner, isettings.name
+            if _db is not None and isettings.dialect == "postgresql":
+                isettings._db = _db
+        else:
+            owner, name = get_owner_name_from_identifier(instance)
+            if _check_instance_setup() and not _test:
+                validate_connection_state(
+                    owner, name, use_root_db_user=use_root_db_user
+                )
+            elif (
+                _write_settings
+                and settings._instance_exists
+                and f"{owner}/{name}" != settings.instance.slug
+            ):
+                disconnect(mute=True)
+
+        if instance is not None or use_root_db_user:
+            try:
+                isettings = _connect_instance(
+                    owner,
+                    name,
+                    db=_db,
+                    access_token=access_token,
+                    use_root_db_user=use_root_db_user,
+                    use_proxy_db=use_proxy_db,
+                )
+            except InstanceNotFoundError as e:
+                if _raise_not_found_error:
+                    raise e
+                else:
+                    return "instance-not-found"
+            if isinstance(isettings, str):
+                return isettings
+        # at this point we have checked already that isettings is not a string
+        # _user is passed to lock cloud sqlite for this user in isettings._load_db()
+        # has no effect if _user is None or if not cloud sqlite instance
+        isettings._locker_user = _user
+        isettings._persist(write_to_disk=_write_settings)
+        if _test:
+            return None
+        silence_loggers()
+        check, msg = isettings._load_db()
+        if not check:
+            local_db = (
+                isettings._is_cloud_sqlite and isettings._sqlite_file_local.exists()
+            )
+            if local_db:
+                logger.warning(
+                    "SQLite file does not exist in the cloud, but exists locally:"
+                    f" {isettings._sqlite_file_local}\nTo push the file to the cloud,"
+                    " call: lamin disconnect"
+                )
+            elif _raise_not_found_error:
+                raise SystemExit(msg)
+            else:
+                logger.warning(
+                    f"instance exists with id {isettings._id.hex}, but database is not"
+                    " loadable: re-initializing"
+                )
+                return "instance-corrupted-or-deleted"
+        # this is for testing purposes only
+        if _TEST_FAILED_LOAD:
+            raise RuntimeError("Technical testing error.")
+
+        load_from_isettings(isettings, user=_user, write_settings=_write_settings)
+        if _reload_lamindb:
+            reset_django_module_variables()
+        if isettings.slug != "none/none":
+            logger.important(f"connected lamindb: {isettings.slug}")
+    except Exception as e:
+        if isettings is not None:
+            if _write_settings:
+                isettings._get_settings_file().unlink(missing_ok=True)  # type: ignore
+            settings._instance_settings = None
+        raise e
+    return None
+
+
+def get_owner_name_from_identifier(identifier: str) -> tuple[str, str]:
+    if "/" in identifier:
+        if identifier.startswith("https://lamin.ai/"):
+            identifier = identifier.replace("https://lamin.ai/", "")
+        split = identifier.split("/")
+        if len(split) > 2:
+            raise ValueError(
+                "The instance identifier needs to be 'owner/name', the instance name"
+                " (owner is current user) or the URL: https://lamin.ai/owner/name."
+            )
+        owner, name = split
+    else:
+        owner = settings.user.handle
+        name = identifier
+    return owner, name