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

lamindb_setup/core/_hub_crud.py

@@ -1,227 +1,227 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-from uuid import UUID, uuid4
-
-from lamin_utils import logger
-from supabase.client import Client  # noqa
-
-
-def select_instance_by_owner_name(
-    owner: str,
-    name: str,
-    client: Client,
-) -> dict | None:
-    # this won't find an instance without the default storage
-    data = (
-        client.table("instance")
-        .select(
-            "*, account!inner!instance_account_id_28936e8f_fk_account_id(*),"
-            " storage!inner!storage_instance_id_359fca71_fk_instance_id(*)"
-        )
-        .eq("name", name)
-        .eq("account.handle", owner)
-        .eq("storage.is_default", True)
-        .execute()
-        .data
-    )
-    if len(data) == 0:
-        return None
-    result = data[0]
-    # this is a list
-    # assume only one default storage
-    result["storage"] = result["storage"][0]
-    return result
-
-
-# --------------- ACCOUNT ----------------------
-def select_account_by_handle(
-    handle: str,
-    client: Client,
-):
-    data = client.table("account").select("*").eq("handle", handle).execute().data
-    if len(data) == 0:
-        return None
-    return data[0]
-
-
-def select_account_handle_name_by_lnid(lnid: str, client: Client):
-    data = (
-        client.table("account").select("handle, name").eq("lnid", lnid).execute().data
-    )
-    if not data:
-        return None
-    return data[0]
-
-
-# --------------- INSTANCE ----------------------
-def select_instance_by_name(
-    account_id: str,
-    name: str,
-    client: Client,
-):
-    data = (
-        client.table("instance")
-        .select("*")
-        .eq("account_id", account_id)
-        .eq("name", name)
-        .execute()
-        .data
-    )
-    if len(data) != 0:
-        return data[0]
-
-    data = (
-        client.table("instance_previous_name")
-        .select(
-            "instance!instance_previous_name_instance_id_17ac5d61_fk_instance_id(*)"
-        )
-        .eq("instance.account_id", account_id)
-        .eq("previous_name", name)
-        .execute()
-        .data
-    )
-    if len(data) != 0:
-        return data[0]["instance"]
-
-    return None
-
-
-def select_instance_by_id(
-    instance_id: str,
-    client: Client,
-):
-    response = client.table("instance").select("*").eq("id", instance_id).execute()
-    if len(response.data) == 0:
-        return None
-    return response.data[0]
-
-
-def select_instance_by_id_with_storage(
-    instance_id: str,
-    client: Client,
-):
-    # this won't find an instance without the default storage
-    data = (
-        client.table("instance")
-        .select("*, storage!inner!storage_instance_id_359fca71_fk_instance_id(*)")
-        .eq("id", instance_id)
-        .eq("storage.is_default", True)
-        .execute()
-        .data
-    )
-    if len(data) == 0:
-        return None
-    result = data[0]
-    # this is a list
-    # assume only one default storage
-    result["storage"] = result["storage"][0]
-    return result
-
-
-def update_instance(instance_id: str, instance_fields: dict, client: Client):
-    response = (
-        client.table("instance").update(instance_fields).eq("id", instance_id).execute()
-    )
-    if len(response.data) == 0:
-        raise PermissionError(
-            f"Update of instance with {instance_id} was not successful. Probably, you"
-            " don't have sufficient permissions."
-        )
-    return response.data[0]
-
-
-# --------------- COLLABORATOR ----------------------
-
-
-def select_collaborator(
-    instance_id: str,
-    account_id: str,
-    client: Client,
-):
-    data = (
-        client.table("account_instance")
-        .select("*")
-        .eq("instance_id", instance_id)
-        .eq("account_id", account_id)
-        .execute()
-        .data
-    )
-    if len(data) == 0:
-        return None
-    return data[0]
-
-
-# --------------- STORAGE ----------------------
-
-
-def select_default_storage_by_instance_id(
-    instance_id: str, client: Client
-) -> dict | None:
-    data = (
-        client.table("storage")
-        .select("*")
-        .eq("instance_id", instance_id)
-        .eq("is_default", True)
-        .execute()
-        .data
-    )
-    if len(data) == 0:
-        return None
-    return data[0]
-
-
-# --------------- DBUser ----------------------
-
-
-def insert_db_user(
-    *,
-    name: str,
-    db_user_name: str,
-    db_user_password: str,
-    instance_id: UUID,
-    client: Client,
-) -> None:
-    fields = (
-        {
-            "id": uuid4().hex,
-            "instance_id": instance_id.hex,
-            "name": name,
-            "db_user_name": db_user_name,
-            "db_user_password": db_user_password,
-        },
-    )
-    data = client.table("db_user").insert(fields).execute().data
-    return data[0]
-
-
-def select_db_user_by_instance(instance_id: str, client: Client):
-    """Get db_user for which client has permission."""
-    data = (
-        client.table("db_user")
-        .select("*")
-        .eq("instance_id", instance_id)
-        .execute()
-        .data
-    )
-    if len(data) == 0:
-        return None
-    elif len(data) > 1:
-        for item in data:
-            if item["name"] == "write":
-                return item
-        logger.warning("found multiple db credentials, using the first one")
-    return data[0]
-
-
-def _delete_instance_record(instance_id: UUID, client: Client) -> None:
-    if not isinstance(instance_id, UUID):
-        instance_id = UUID(instance_id)
-    response = client.table("instance").delete().eq("id", instance_id.hex).execute()
-    if response.data:
-        logger.important(f"deleted instance record on hub {instance_id.hex}")
-    else:
-        raise PermissionError(
-            f"Deleting of instance with {instance_id.hex} was not successful. Probably, you"
-            " don't have sufficient permissions."
-        )
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+from uuid import UUID, uuid4
+
+from lamin_utils import logger
+from supabase.client import Client  # noqa
+
+
+def select_instance_by_owner_name(
+    owner: str,
+    name: str,
+    client: Client,
+) -> dict | None:
+    # this won't find an instance without the default storage
+    data = (
+        client.table("instance")
+        .select(
+            "*, account!inner!instance_account_id_28936e8f_fk_account_id(*),"
+            " storage!inner!storage_instance_id_359fca71_fk_instance_id(*)"
+        )
+        .eq("name", name)
+        .eq("account.handle", owner)
+        .eq("storage.is_default", True)
+        .execute()
+        .data
+    )
+    if len(data) == 0:
+        return None
+    result = data[0]
+    # this is a list
+    # assume only one default storage
+    result["storage"] = result["storage"][0]
+    return result
+
+
+# --------------- ACCOUNT ----------------------
+def select_account_by_handle(
+    handle: str,
+    client: Client,
+):
+    data = client.table("account").select("*").eq("handle", handle).execute().data
+    if len(data) == 0:
+        return None
+    return data[0]
+
+
+def select_account_handle_name_by_lnid(lnid: str, client: Client):
+    data = (
+        client.table("account").select("handle, name").eq("lnid", lnid).execute().data
+    )
+    if not data:
+        return None
+    return data[0]
+
+
+# --------------- INSTANCE ----------------------
+def select_instance_by_name(
+    account_id: str,
+    name: str,
+    client: Client,
+):
+    data = (
+        client.table("instance")
+        .select("*")
+        .eq("account_id", account_id)
+        .eq("name", name)
+        .execute()
+        .data
+    )
+    if len(data) != 0:
+        return data[0]
+
+    data = (
+        client.table("instance_previous_name")
+        .select(
+            "instance!instance_previous_name_instance_id_17ac5d61_fk_instance_id(*)"
+        )
+        .eq("instance.account_id", account_id)
+        .eq("previous_name", name)
+        .execute()
+        .data
+    )
+    if len(data) != 0:
+        return data[0]["instance"]
+
+    return None
+
+
+def select_instance_by_id(
+    instance_id: str,
+    client: Client,
+):
+    response = client.table("instance").select("*").eq("id", instance_id).execute()
+    if len(response.data) == 0:
+        return None
+    return response.data[0]
+
+
+def select_instance_by_id_with_storage(
+    instance_id: str,
+    client: Client,
+):
+    # this won't find an instance without the default storage
+    data = (
+        client.table("instance")
+        .select("*, storage!inner!storage_instance_id_359fca71_fk_instance_id(*)")
+        .eq("id", instance_id)
+        .eq("storage.is_default", True)
+        .execute()
+        .data
+    )
+    if len(data) == 0:
+        return None
+    result = data[0]
+    # this is a list
+    # assume only one default storage
+    result["storage"] = result["storage"][0]
+    return result
+
+
+def update_instance(instance_id: str, instance_fields: dict, client: Client):
+    response = (
+        client.table("instance").update(instance_fields).eq("id", instance_id).execute()
+    )
+    if len(response.data) == 0:
+        raise PermissionError(
+            f"Update of instance with {instance_id} was not successful. Probably, you"
+            " don't have sufficient permissions."
+        )
+    return response.data[0]
+
+
+# --------------- COLLABORATOR ----------------------
+
+
+def select_collaborator(
+    instance_id: str,
+    account_id: str,
+    client: Client,
+):
+    data = (
+        client.table("account_instance")
+        .select("*")
+        .eq("instance_id", instance_id)
+        .eq("account_id", account_id)
+        .execute()
+        .data
+    )
+    if len(data) == 0:
+        return None
+    return data[0]
+
+
+# --------------- STORAGE ----------------------
+
+
+def select_default_storage_by_instance_id(
+    instance_id: str, client: Client
+) -> dict | None:
+    data = (
+        client.table("storage")
+        .select("*")
+        .eq("instance_id", instance_id)
+        .eq("is_default", True)
+        .execute()
+        .data
+    )
+    if len(data) == 0:
+        return None
+    return data[0]
+
+
+# --------------- DBUser ----------------------
+
+
+def insert_db_user(
+    *,
+    name: str,
+    db_user_name: str,
+    db_user_password: str,
+    instance_id: UUID,
+    client: Client,
+) -> None:
+    fields = (
+        {
+            "id": uuid4().hex,
+            "instance_id": instance_id.hex,
+            "name": name,
+            "db_user_name": db_user_name,
+            "db_user_password": db_user_password,
+        },
+    )
+    data = client.table("db_user").insert(fields).execute().data
+    return data[0]
+
+
+def select_db_user_by_instance(instance_id: str, client: Client):
+    """Get db_user for which client has permission."""
+    data = (
+        client.table("db_user")
+        .select("*")
+        .eq("instance_id", instance_id)
+        .execute()
+        .data
+    )
+    if len(data) == 0:
+        return None
+    elif len(data) > 1:
+        for item in data:
+            if item["name"] == "write":
+                return item
+        logger.warning("found multiple db credentials, using the first one")
+    return data[0]
+
+
+def _delete_instance_record(instance_id: UUID, client: Client) -> None:
+    if not isinstance(instance_id, UUID):
+        instance_id = UUID(instance_id)
+    response = client.table("instance").delete().eq("id", instance_id.hex).execute()
+    if response.data:
+        logger.important(f"deleted instance record on hub {instance_id.hex}")
+    else:
+        raise PermissionError(
+            f"Deleting of instance with {instance_id.hex} was not successful. Probably, you"
+            " don't have sufficient permissions."
+        )

lamindb_setup/core/_private_django_api.py

@@ -1,83 +1,83 @@
-from __future__ import annotations
-
-import os
-from pathlib import Path
-
-
-def find_vscode_stubs_folder() -> Path | None:
-    # Possible locations of VSCode extensions
-    possible_locations = [
-        Path.home() / ".vscode" / "extensions",  # Linux and macOS
-        Path.home() / ".vscode-server" / "extensions",  # Remote development
-        Path(os.environ.get("APPDATA", "")) / "Code" / "User" / "extensions",  # Windows
-        Path("/usr/share/code/resources/app/extensions"),  # Some Linux distributions
-    ]
-    for location in possible_locations:
-        if location.exists():
-            # Look for Pylance extension folder
-            pylance_folders = list(location.glob("ms-python.vscode-pylance-*"))
-            if pylance_folders:
-                # Sort to get the latest version
-                latest_pylance = sorted(pylance_folders)[-1]
-                stubs_folder = (
-                    latest_pylance / "dist" / "bundled" / "stubs" / "django-stubs"
-                )
-                if stubs_folder.exists():
-                    return stubs_folder
-
-    return None
-
-
-def private_django_api(reverse=False):
-    from django import db
-
-    # the order here matters
-    # changing it might break the tests
-    attributes = [
-        "MultipleObjectsReturned",
-        "add_to_class",
-        "adelete",
-        "refresh_from_db",
-        "asave",
-        "clean",
-        "clean_fields",
-        "date_error_message",
-        "get_constraints",
-        "get_deferred_fields",
-        "prepare_database_save",
-        "save_base",
-        "serializable_value",
-        "unique_error_message",
-        "validate_constraints",
-        "validate_unique",
-    ]
-    if reverse:
-        attributes.append("arefresh_from_db")
-        attributes.append("full_clean")
-    else:
-        attributes.append("a_refresh_from_db")
-        attributes.append("full__clean")
-
-    django_path = Path(db.__file__).parent.parent
-
-    encoding = "utf8" if os.name == "nt" else None
-
-    def prune_file(file_path):
-        content = file_path.read_text(encoding=encoding)
-        original_content = content
-
-        for attr in attributes:
-            old_name = f"_{attr}" if reverse else attr
-            new_name = attr if reverse else f"_{attr}"
-            content = content.replace(old_name, new_name)
-
-        if content != original_content:
-            file_path.write_text(content, encoding=encoding)
-
-    for file_path in django_path.rglob("*.py"):
-        prune_file(file_path)
-
-    pylance_path = find_vscode_stubs_folder()
-    if pylance_path is not None:
-        for file_path in pylance_path.rglob("*.pyi"):
-            prune_file(file_path)
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+
+def find_vscode_stubs_folder() -> Path | None:
+    # Possible locations of VSCode extensions
+    possible_locations = [
+        Path.home() / ".vscode" / "extensions",  # Linux and macOS
+        Path.home() / ".vscode-server" / "extensions",  # Remote development
+        Path(os.environ.get("APPDATA", "")) / "Code" / "User" / "extensions",  # Windows
+        Path("/usr/share/code/resources/app/extensions"),  # Some Linux distributions
+    ]
+    for location in possible_locations:
+        if location.exists():
+            # Look for Pylance extension folder
+            pylance_folders = list(location.glob("ms-python.vscode-pylance-*"))
+            if pylance_folders:
+                # Sort to get the latest version
+                latest_pylance = sorted(pylance_folders)[-1]
+                stubs_folder = (
+                    latest_pylance / "dist" / "bundled" / "stubs" / "django-stubs"
+                )
+                if stubs_folder.exists():
+                    return stubs_folder
+
+    return None
+
+
+def private_django_api(reverse=False):
+    from django import db
+
+    # the order here matters
+    # changing it might break the tests
+    attributes = [
+        "MultipleObjectsReturned",
+        "add_to_class",
+        "adelete",
+        "refresh_from_db",
+        "asave",
+        "clean",
+        "clean_fields",
+        "date_error_message",
+        "get_constraints",
+        "get_deferred_fields",
+        "prepare_database_save",
+        "save_base",
+        "serializable_value",
+        "unique_error_message",
+        "validate_constraints",
+        "validate_unique",
+    ]
+    if reverse:
+        attributes.append("arefresh_from_db")
+        attributes.append("full_clean")
+    else:
+        attributes.append("a_refresh_from_db")
+        attributes.append("full__clean")
+
+    django_path = Path(db.__file__).parent.parent
+
+    encoding = "utf8" if os.name == "nt" else None
+
+    def prune_file(file_path):
+        content = file_path.read_text(encoding=encoding)
+        original_content = content
+
+        for attr in attributes:
+            old_name = f"_{attr}" if reverse else attr
+            new_name = attr if reverse else f"_{attr}"
+            content = content.replace(old_name, new_name)
+
+        if content != original_content:
+            file_path.write_text(content, encoding=encoding)
+
+    for file_path in django_path.rglob("*.py"):
+        prune_file(file_path)
+
+    pylance_path = find_vscode_stubs_folder()
+    if pylance_path is not None:
+        for file_path in pylance_path.rglob("*.pyi"):
+            prune_file(file_path)