dbt-postgres 1.8.0b1__tar.gz → 1.8.0b2__tar.gz

{dbt_postgres-1.8.0b1 → dbt_postgres-1.8.0b2}/PKG-INFO

@@ -1,6 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: dbt-postgres
-Version: 1.8.0b1
+Version: 1.8.0b2
+Dynamic: Requires-Dist
 Summary: The set of adapter protocols and base functionality that supports integration with dbt-core
 Project-URL: Homepage, https://github.com/dbt-labs/dbt-postgres
 Project-URL: Documentation, https://docs.getdbt.com
@@ -21,10 +22,11 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8.0
-Requires-Dist: agate<2.0,>=1.0
-Requires-Dist: dbt-adapters<2.0,>=0.1.0a1
-Requires-Dist: dbt-common<2.0,>=0.1.0a1
-Requires-Dist: psycopg2<3.0,>=2.9
+Requires-Dist: dbt-adapters>=0.1.0a1,<2.0
+Requires-Dist: dbt-core>=1.8.0a1
+Requires-Dist: dbt-common>=0.1.0a1,<2.0
+Requires-Dist: agate>=1.0,<2.0
+Requires-Dist: psycopg2-binary>=2.9,<3.0
 Description-Content-Type: text/markdown
 
 <p align="center">

dbt_postgres-1.8.0b2/dbt/adapters/postgres/__version__.py

@@ -0,0 +1 @@
+version = "1.8.0b2"

{dbt_postgres-1.8.0b1 → dbt_postgres-1.8.0b2}/dbt/adapters/postgres/relation.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import FrozenSet, Optional, Set
 
 from dbt.adapters.base.relation import BaseRelation
@@ -20,19 +20,19 @@ from dbt.adapters.postgres.relation_configs import (
 
 @dataclass(frozen=True, eq=False, repr=False)
 class PostgresRelation(BaseRelation):
-    renameable_relations = frozenset(
+    renameable_relations: FrozenSet[RelationType] = field(default_factory=lambda: frozenset(
         {
             RelationType.View,
             RelationType.Table,
             RelationType.MaterializedView,
         }
-    )
-    replaceable_relations = frozenset(
+    ))
+    replaceable_relations: FrozenSet[RelationType] = field(default_factory=lambda: frozenset(
         {
             RelationType.View,
             RelationType.Table,
         }
-    )
+    ))
 
     def __post_init__(self):
         # Check for length of Postgres table/view names.

dbt_postgres-1.8.0b2/hatch_build.py

@@ -0,0 +1,56 @@
+import os
+from typing import Any, Dict
+
+from hatchling.builders.config import BuilderConfig
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+from hatchling.plugin import hookimpl
+
+BASE_DEPS = [
+    # psycopg2 dependency installed in custom hatch_build.py
+    "dbt-adapters>=0.1.0a1,<2.0",
+    # add dbt-core to ensure backwards compatibility of installation, this is not a functional dependency
+    "dbt-core>=1.8.0a1",
+    # installed via dbt-adapters but used directly
+    "dbt-common>=0.1.0a1,<2.0",
+    "agate>=1.0,<2.0",
+]
+
+PSYCOPG2_MESSAGE = """
+No package name override was set.
+Using 'psycopg2-binary' package to satisfy 'psycopg2'
+
+If you experience segmentation faults, silent crashes, or installation errors,
+consider retrying with the 'DBT_PSYCOPG2_NAME' environment variable set to
+'psycopg2'. It may require a compiler toolchain and development libraries!
+""".strip()
+
+
+def _dbt_psycopg2_name():
+    # if the user chose something, use that
+    package_name = os.getenv("DBT_PSYCOPG2_NAME", "")
+    if package_name:
+        return package_name
+
+    # default to psycopg2-binary for all OSes/versions
+    print(PSYCOPG2_MESSAGE)
+    return "psycopg2-binary"
+
+
+class CustomBuildHook(BuildHookInterface[BuilderConfig]):
+    """
+    Custom build hook to install psycopg2 instead of psycopg2-binary based on the presence of `DBT_PSYCOPG2_NAME` env
+    var. This is necessary as psycopg2-binary is better for local development, but psycopg2 is better for production.
+    """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+    def initialize(self, version: str, build_data: Dict) -> None:
+        build_data["dependencies"] = BASE_DEPS
+        psycopg2_pkg_name = _dbt_psycopg2_name()
+        build_data["dependencies"].append(f"{psycopg2_pkg_name}>=2.9,<3.0")
+
+
+@hookimpl
+def hatch_register_build_hook():
+    return CustomBuildHook

{dbt_postgres-1.8.0b1 → dbt_postgres-1.8.0b2}/pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-dynamic = ["version"]
+dynamic = ["version", "dependencies"]
 name = "dbt-postgres"
 description = "The set of adapter protocols and base functionality that supports integration with dbt-core"
 readme = "README.md"
@@ -22,13 +22,7 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
 ]
-dependencies = [
-    "dbt-adapters>=0.1.0a1,<2.0",
-    "psycopg2>=2.9,<3.0",
-    # installed via dbt-adapters but used directly
-    "dbt-common>=0.1.0a1,<2.0",
-    "agate>=1.0,<2.0",
-]
+
 [project.urls]
 Homepage = "https://github.com/dbt-labs/dbt-postgres"
 Documentation = "https://docs.getdbt.com"
@@ -49,6 +43,9 @@ packages = ["dbt"]
 [tool.hatch.version]
 path = "dbt/adapters/postgres/__version__.py"
 
+[tool.hatch.build.hooks.custom]
+path = "./hatch_build.py"
+
 [tool.hatch.envs.default]
 dependencies = [
     "dbt-adapters @ git+https://github.com/dbt-labs/dbt-adapters.git",
@@ -58,7 +55,7 @@ dependencies = [
 [tool.hatch.envs.lint]
 detached = true
 dependencies = [
-    "black",
+    "black>=24.3",
     "flake8",
     "Flake8-pyproject",
 ]
@@ -119,6 +116,8 @@ dependencies = [
     "twine",
     "check-wheel-contents",
 ]
+
+
 [tool.hatch.envs.build.scripts]
 check-all = [
     "- check-wheel",
@@ -166,4 +165,4 @@ env_files = ["test.env"]
 testpaths = [
     "tests/functional",
     "tests/unit",
-]
+]

dbt_postgres-1.8.0b1/dbt/adapters/postgres/__version__.py

@@ -1 +0,0 @@
-version = "1.8.0b1"