lamindb_setup 0.77.4__py2.py3-none-any.whl → 0.77.6__py2.py3-none-any.whl

lamindb_setup/core/_settings_load.py

@@ -1,105 +1,105 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING, Optional
-from uuid import UUID, uuid4
-
-from lamin_utils import logger
-from pydantic import ValidationError
-
-from ._settings_instance import InstanceSettings
-from ._settings_storage import StorageSettings
-from ._settings_store import (
-    InstanceSettingsStore,
-    UserSettingsStore,
-    current_instance_settings_file,
-    current_user_settings_file,
-)
-from ._settings_user import UserSettings
-
-if TYPE_CHECKING:
-    from pathlib import Path
-
-
-class SettingsEnvFileOutdated(Exception):
-    pass
-
-
-def load_instance_settings(instance_settings_file: Path | None = None):
-    if instance_settings_file is None:
-        instance_settings_file = current_instance_settings_file()
-    if not instance_settings_file.exists():
-        raise SystemExit("No instance is loaded! Call `lamin init` or `lamin load`")
-    try:
-        settings_store = InstanceSettingsStore(_env_file=instance_settings_file)
-    except (ValidationError, TypeError) as error:
-        with open(instance_settings_file) as f:
-            content = f.read()
-        raise SettingsEnvFileOutdated(
-            f"\n\n{error}\n\nYour instance settings file with\n\n{content}\nis invalid"
-            f" (likely outdated), see validation error. Please delete {instance_settings_file} &"
-            " reload (remote) or re-initialize (local) the instance with the same name & storage location."
-        ) from error
-    isettings = setup_instance_from_store(settings_store)
-    return isettings
-
-
-def load_or_create_user_settings() -> UserSettings:
-    """Return current user settings."""
-    current_user_settings = current_user_settings_file()
-    if not current_user_settings.exists():
-        logger.warning("using anonymous user (to identify, call: lamin login)")
-        usettings = UserSettings(handle="anonymous", uid="00000000")
-        from ._settings_save import save_user_settings
-
-        save_user_settings(usettings)
-    else:
-        usettings = load_user_settings(current_user_settings)
-    return usettings
-
-
-def load_user_settings(user_settings_file: Path):
-    try:
-        settings_store = UserSettingsStore(_env_file=user_settings_file)
-    except (ValidationError, TypeError) as error:
-        msg = (
-            "Your user settings file is invalid, please delete"
-            f" {user_settings_file} and log in again."
-        )
-        print(msg)
-        raise SettingsEnvFileOutdated(msg) from error
-    settings = setup_user_from_store(settings_store)
-    return settings
-
-
-def _null_to_value(field, value=None):
-    return field if field != "null" else value
-
-
-def setup_instance_from_store(store: InstanceSettingsStore) -> InstanceSettings:
-    ssettings = StorageSettings(
-        root=store.storage_root,
-        region=_null_to_value(store.storage_region),
-    )
-    return InstanceSettings(
-        id=UUID(store.id),
-        owner=store.owner,
-        name=store.name,
-        storage=ssettings,
-        db=_null_to_value(store.db),
-        schema=_null_to_value(store.schema_str),
-        git_repo=_null_to_value(store.git_repo),
-        keep_artifacts_local=store.keep_artifacts_local,  # type: ignore
-    )
-
-
-def setup_user_from_store(store: UserSettingsStore) -> UserSettings:
-    settings = UserSettings()
-    settings.email = _null_to_value(store.email)
-    settings.password = _null_to_value(store.password)
-    settings.access_token = store.access_token
-    settings.api_key = _null_to_value(store.api_key)
-    settings.uid = store.uid
-    settings.handle = _null_to_value(store.handle, value="anonymous")
-    settings.name = _null_to_value(store.name)
-    settings._uuid = UUID(store.uuid) if store.uuid != "null" else None
-    return settings
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
+from uuid import UUID, uuid4
+
+from lamin_utils import logger
+from pydantic import ValidationError
+
+from ._settings_instance import InstanceSettings
+from ._settings_storage import StorageSettings
+from ._settings_store import (
+    InstanceSettingsStore,
+    UserSettingsStore,
+    current_instance_settings_file,
+    current_user_settings_file,
+)
+from ._settings_user import UserSettings
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+
+class SettingsEnvFileOutdated(Exception):
+    pass
+
+
+def load_instance_settings(instance_settings_file: Path | None = None):
+    if instance_settings_file is None:
+        instance_settings_file = current_instance_settings_file()
+    if not instance_settings_file.exists():
+        raise SystemExit("No instance is loaded! Call `lamin init` or `lamin load`")
+    try:
+        settings_store = InstanceSettingsStore(_env_file=instance_settings_file)
+    except (ValidationError, TypeError) as error:
+        with open(instance_settings_file) as f:
+            content = f.read()
+        raise SettingsEnvFileOutdated(
+            f"\n\n{error}\n\nYour instance settings file with\n\n{content}\nis invalid"
+            f" (likely outdated), see validation error. Please delete {instance_settings_file} &"
+            " reload (remote) or re-initialize (local) the instance with the same name & storage location."
+        ) from error
+    isettings = setup_instance_from_store(settings_store)
+    return isettings
+
+
+def load_or_create_user_settings() -> UserSettings:
+    """Return current user settings."""
+    current_user_settings = current_user_settings_file()
+    if not current_user_settings.exists():
+        logger.warning("using anonymous user (to identify, call: lamin login)")
+        usettings = UserSettings(handle="anonymous", uid="00000000")
+        from ._settings_save import save_user_settings
+
+        save_user_settings(usettings)
+    else:
+        usettings = load_user_settings(current_user_settings)
+    return usettings
+
+
+def load_user_settings(user_settings_file: Path):
+    try:
+        settings_store = UserSettingsStore(_env_file=user_settings_file)
+    except (ValidationError, TypeError) as error:
+        msg = (
+            "Your user settings file is invalid, please delete"
+            f" {user_settings_file} and log in again."
+        )
+        print(msg)
+        raise SettingsEnvFileOutdated(msg) from error
+    settings = setup_user_from_store(settings_store)
+    return settings
+
+
+def _null_to_value(field, value=None):
+    return field if field != "null" else value
+
+
+def setup_instance_from_store(store: InstanceSettingsStore) -> InstanceSettings:
+    ssettings = StorageSettings(
+        root=store.storage_root,
+        region=_null_to_value(store.storage_region),
+    )
+    return InstanceSettings(
+        id=UUID(store.id),
+        owner=store.owner,
+        name=store.name,
+        storage=ssettings,
+        db=_null_to_value(store.db),
+        schema=_null_to_value(store.schema_str),
+        git_repo=_null_to_value(store.git_repo),
+        keep_artifacts_local=store.keep_artifacts_local,  # type: ignore
+    )
+
+
+def setup_user_from_store(store: UserSettingsStore) -> UserSettings:
+    settings = UserSettings()
+    settings.email = _null_to_value(store.email)
+    settings.password = _null_to_value(store.password)
+    settings.access_token = store.access_token
+    settings.api_key = _null_to_value(store.api_key)
+    settings.uid = store.uid
+    settings.handle = _null_to_value(store.handle, value="anonymous")
+    settings.name = _null_to_value(store.name)
+    settings._uuid = UUID(store.uuid) if store.uuid != "null" else None
+    return settings

lamindb_setup/core/_settings_save.py

@@ -1,81 +1,81 @@
-from __future__ import annotations
-
-from pathlib import Path
-from typing import TYPE_CHECKING, Any, Optional, get_type_hints
-from uuid import UUID
-
-from ._settings_store import (
-    InstanceSettingsStore,
-    UserSettingsStore,
-    current_user_settings_file,
-    user_settings_file_email,
-    user_settings_file_handle,
-)
-
-if TYPE_CHECKING:
-    from ._settings_user import UserSettings
-    from .upath import UPath
-
-
-def save_user_settings(settings: UserSettings):
-    type_hints = get_type_hints(UserSettingsStore)
-    prefix = "lamin_user_"
-    save_settings(settings, current_user_settings_file(), type_hints, prefix)
-    if settings.email is not None:
-        save_settings(
-            settings, user_settings_file_email(settings.email), type_hints, prefix
-        )
-    if settings.handle is not None and settings.handle != "anonymous":
-        save_settings(
-            settings, user_settings_file_handle(settings.handle), type_hints, prefix
-        )
-
-
-def save_settings(
-    settings: Any,
-    settings_file: Path,
-    type_hints: dict[str, Any],
-    prefix: str,
-):
-    with open(settings_file, "w") as f:
-        for store_key, type in type_hints.items():
-            if type == Optional[str]:
-                type = str
-            if type == Optional[bool]:
-                type = bool
-            if "__" not in store_key:
-                if store_key == "model_config":
-                    continue
-                if store_key == "storage_root":
-                    value = settings.storage.root_as_str
-                elif store_key == "storage_region":
-                    value = settings.storage.region
-                else:
-                    if store_key in {"db", "schema_str", "name_", "uuid", "id"}:
-                        settings_key = f"_{store_key.rstrip('_')}"
-                    else:
-                        settings_key = store_key
-                    value = getattr(settings, settings_key, None)
-                if value is None:
-                    value = "null"
-                elif isinstance(value, UUID):
-                    value = value.hex
-                else:
-                    value = type(value)
-                f.write(f"{prefix}{store_key}={value}\n")
-
-
-def save_instance_settings(settings: Any, settings_file: Path):
-    type_hints = get_type_hints(InstanceSettingsStore)
-    prefix = "lamindb_instance_"
-    save_settings(settings, settings_file, type_hints, prefix)
-
-
-def save_system_storage_settings(
-    cache_path: str | Path | UPath | None, settings_file: Path
-):
-    cache_path = "null" if cache_path is None else cache_path
-    if isinstance(cache_path, Path):  # also True for UPath
-        cache_path = cache_path.as_posix()
-    with open(settings_file, "w") as f:
-        f.write(f"lamindb_cache_path={cache_path}")
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING, Any, Optional, get_type_hints
+from uuid import UUID
+
+from ._settings_store import (
+    InstanceSettingsStore,
+    UserSettingsStore,
+    current_user_settings_file,
+    user_settings_file_email,
+    user_settings_file_handle,
+)
+
+if TYPE_CHECKING:
+    from ._settings_user import UserSettings
+    from .upath import UPath
+
+
+def save_user_settings(settings: UserSettings):
+    type_hints = get_type_hints(UserSettingsStore)
+    prefix = "lamin_user_"
+    save_settings(settings, current_user_settings_file(), type_hints, prefix)
+    if settings.email is not None:
+        save_settings(
+            settings, user_settings_file_email(settings.email), type_hints, prefix
+        )
+    if settings.handle is not None and settings.handle != "anonymous":
+        save_settings(
+            settings, user_settings_file_handle(settings.handle), type_hints, prefix
+        )
+
+
+def save_settings(
+    settings: Any,
+    settings_file: Path,
+    type_hints: dict[str, Any],
+    prefix: str,
+):
+    with open(settings_file, "w") as f:
+        for store_key, type in type_hints.items():
+            if type == Optional[str]:
+                type = str
+            if type == Optional[bool]:
+                type = bool
+            if "__" not in store_key:
+                if store_key == "model_config":
+                    continue
+                if store_key == "storage_root":
+                    value = settings.storage.root_as_str
+                elif store_key == "storage_region":
+                    value = settings.storage.region
+                else:
+                    if store_key in {"db", "schema_str", "name_", "uuid", "id"}:
+                        settings_key = f"_{store_key.rstrip('_')}"
+                    else:
+                        settings_key = store_key
+                    value = getattr(settings, settings_key, None)
+                if value is None:
+                    value = "null"
+                elif isinstance(value, UUID):
+                    value = value.hex
+                else:
+                    value = type(value)
+                f.write(f"{prefix}{store_key}={value}\n")
+
+
+def save_instance_settings(settings: Any, settings_file: Path):
+    type_hints = get_type_hints(InstanceSettingsStore)
+    prefix = "lamindb_instance_"
+    save_settings(settings, settings_file, type_hints, prefix)
+
+
+def save_system_storage_settings(
+    cache_path: str | Path | UPath | None, settings_file: Path
+):
+    cache_path = "null" if cache_path is None else cache_path
+    if isinstance(cache_path, Path):  # also True for UPath
+        cache_path = cache_path.as_posix()
+    with open(settings_file, "w") as f:
+        f.write(f"lamindb_cache_path={cache_path}")