WIP

Author: z3z1ma

src/dbt_core_interface/project.py

@@ -47,23 +47,23 @@
 logger = logging.getLogger(__name__)
 
 
-def _default_project_dir() -> Path:
+def _get_project_dir() -> Path:
     """Get the default project directory following dbt heuristics."""
     return Path(os.getenv("DBT_PROJECT_DIR", os.getcwd())).expanduser().resolve()
 
 
-def _default_profiles_dir() -> Path:
+def _get_profiles_dir(project_dir: Path | str | None = None) -> Path:
     """Get the default profiles directory following dbt heuristics."""
     if "DBT_PROFILES_DIR" not in os.environ:
-        project_dir = _default_project_dir()
-        if project_dir.is_dir() and project_dir.joinpath("profiles.yml").exists():
-            return project_dir
+        _project_dir = Path(project_dir or _get_project_dir())
+        if _project_dir.is_dir() and _project_dir.joinpath("profiles.yml").exists():
+            return _project_dir
         return Path.home() / ".dbt"
     return Path(os.environ["DBT_PROFILES_DIR"]).expanduser().resolve()
 
 
-DEFAULT_PROFILES_DIR = str(_default_profiles_dir())
-DEFAULT_PROJECT_DIR = str(_default_project_dir())
+DEFAULT_PROFILES_DIR = str(_get_profiles_dir())
+DEFAULT_PROJECT_DIR = str(_get_project_dir())
 
 
 @dataclass
@@ -121,13 +121,16 @@ class DbtProject:
     def __init__(
         self,
         target: str | None = None,
-        profiles_dir: str = DEFAULT_PROFILES_DIR,
         project_dir: str = DEFAULT_PROJECT_DIR,
+        profiles_dir: str = DEFAULT_PROFILES_DIR,
         threads: int = 1,
         vars: dict[str, t.Any] | None = None,
         load: bool = True,
     ) -> None:
         """Initialize the dbt project."""
+        if project_dir is not DEFAULT_PROJECT_DIR and profiles_dir is DEFAULT_PROFILES_DIR:
+            profiles_dir = str(_get_profiles_dir(project_dir).resolve())
+
         self._base_params = DbtConfiguration(
             target=target,
             profiles_dir=profiles_dir,