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

lamindb_setup/core/_hub_utils.py

@@ -1,100 +1,100 @@
-from __future__ import annotations
-
-from typing import Any, ClassVar
-from urllib.parse import urlparse, urlunparse
-
-from pydantic import BaseModel, Field, GetCoreSchemaHandler
-from pydantic_core import CoreSchema, core_schema
-
-
-def validate_db_arg(db: str | None) -> None:
-    if db is not None:
-        LaminDsnModel(db=db)
-
-
-class LaminDsn(str):
-    allowed_schemes: ClassVar[set[str]] = {
-        "postgresql",
-        # future enabled schemes
-        # "snowflake",
-        # "bigquery"
-    }
-
-    @classmethod
-    def __get_pydantic_core_schema__(
-        cls, source_type: Any, handler: GetCoreSchemaHandler
-    ) -> CoreSchema:
-        return core_schema.no_info_after_validator_function(
-            cls.validate,
-            core_schema.str_schema(),
-            serialization=core_schema.plain_serializer_function_ser_schema(str),
-        )
-
-    @classmethod
-    def validate(cls, v: Any) -> LaminDsn:
-        if isinstance(v, str):
-            parsed = urlparse(v)
-            if parsed.scheme not in cls.allowed_schemes:
-                raise ValueError(f"Invalid scheme: {parsed.scheme}")
-            return cls(v)
-        elif isinstance(v, cls):
-            return v
-        else:
-            raise ValueError(f"Invalid value for LaminDsn: {v}")
-
-    @property
-    def user(self) -> str | None:
-        return urlparse(self).username
-
-    @property
-    def password(self) -> str | None:
-        return urlparse(self).password
-
-    @property
-    def host(self) -> str | None:
-        return urlparse(self).hostname
-
-    @property
-    def port(self) -> int | None:
-        return urlparse(self).port
-
-    @property
-    def database(self) -> str:
-        return urlparse(self).path.lstrip("/")
-
-    @property
-    def scheme(self) -> str:
-        return urlparse(self).scheme
-
-    @classmethod
-    def build(
-        cls,
-        *,
-        scheme: str,
-        user: str | None = None,
-        password: str | None = None,
-        host: str,
-        port: int | None = None,
-        database: str | None = None,
-        query: str | None = None,
-        fragment: str | None = None,
-    ) -> LaminDsn:
-        netloc = host
-        if port is not None:
-            netloc = f"{netloc}:{port}"
-        if user is not None:
-            auth = user
-            if password is not None:
-                auth = f"{auth}:{password}"
-            netloc = f"{auth}@{netloc}"
-
-        path = f"/{database}" if database else ""
-
-        url = urlunparse((scheme, netloc, path, "", query or "", fragment or ""))
-        return cls(url)
-
-
-class LaminDsnModel(BaseModel):
-    db: LaminDsn = Field(..., description="The database DSN")
-
-    model_config = {"arbitrary_types_allowed": True}
+from __future__ import annotations
+
+from typing import Any, ClassVar
+from urllib.parse import urlparse, urlunparse
+
+from pydantic import BaseModel, Field, GetCoreSchemaHandler
+from pydantic_core import CoreSchema, core_schema
+
+
+def validate_db_arg(db: str | None) -> None:
+    if db is not None:
+        LaminDsnModel(db=db)
+
+
+class LaminDsn(str):
+    allowed_schemes: ClassVar[set[str]] = {
+        "postgresql",
+        # future enabled schemes
+        # "snowflake",
+        # "bigquery"
+    }
+
+    @classmethod
+    def __get_pydantic_core_schema__(
+        cls, source_type: Any, handler: GetCoreSchemaHandler
+    ) -> CoreSchema:
+        return core_schema.no_info_after_validator_function(
+            cls.validate,
+            core_schema.str_schema(),
+            serialization=core_schema.plain_serializer_function_ser_schema(str),
+        )
+
+    @classmethod
+    def validate(cls, v: Any) -> LaminDsn:
+        if isinstance(v, str):
+            parsed = urlparse(v)
+            if parsed.scheme not in cls.allowed_schemes:
+                raise ValueError(f"Invalid scheme: {parsed.scheme}")
+            return cls(v)
+        elif isinstance(v, cls):
+            return v
+        else:
+            raise ValueError(f"Invalid value for LaminDsn: {v}")
+
+    @property
+    def user(self) -> str | None:
+        return urlparse(self).username
+
+    @property
+    def password(self) -> str | None:
+        return urlparse(self).password
+
+    @property
+    def host(self) -> str | None:
+        return urlparse(self).hostname
+
+    @property
+    def port(self) -> int | None:
+        return urlparse(self).port
+
+    @property
+    def database(self) -> str:
+        return urlparse(self).path.lstrip("/")
+
+    @property
+    def scheme(self) -> str:
+        return urlparse(self).scheme
+
+    @classmethod
+    def build(
+        cls,
+        *,
+        scheme: str,
+        user: str | None = None,
+        password: str | None = None,
+        host: str,
+        port: int | None = None,
+        database: str | None = None,
+        query: str | None = None,
+        fragment: str | None = None,
+    ) -> LaminDsn:
+        netloc = host
+        if port is not None:
+            netloc = f"{netloc}:{port}"
+        if user is not None:
+            auth = user
+            if password is not None:
+                auth = f"{auth}:{password}"
+            netloc = f"{auth}@{netloc}"
+
+        path = f"/{database}" if database else ""
+
+        url = urlunparse((scheme, netloc, path, "", query or "", fragment or ""))
+        return cls(url)
+
+
+class LaminDsnModel(BaseModel):
+    db: LaminDsn = Field(..., description="The database DSN")
+
+    model_config = {"arbitrary_types_allowed": True}

lamindb_setup/core/_private_django_api.py

@@ -1,80 +1,80 @@
-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 = [
-        "add_to_class",
-        "arefresh_from_db",
-        "adelete",
-        "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("full_clean")
-    else:
-        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 = [
+        "add_to_class",
+        "arefresh_from_db",
+        "adelete",
+        "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("full_clean")
+    else:
+        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)