66from typing import TYPE_CHECKING
77
88from plain .exceptions import ImproperlyConfigured
9- from plain .postgres .database_url import postgres_cli_args , postgres_cli_env
109
1110if TYPE_CHECKING :
1211 from plain .postgres .connection import DatabaseConnection
@@ -17,10 +16,14 @@ def __init__(self, connection: DatabaseConnection) -> None:
1716 self .connection = connection
1817
1918 def _run (self , cmd : str | list [str ], * , shell : bool = False ) -> None :
19+ from plain .postgres .database_url import postgres_cli_env
20+
2021 env = {** os .environ , ** postgres_cli_env (self .connection .settings_dict )}
2122 subprocess .run (cmd , env = env , check = True , shell = shell )
2223
2324 def create_backup (self , backup_path : Path , * , pg_dump : str = "pg_dump" ) -> None :
25+ from plain .postgres .database_url import postgres_cli_args
26+
2427 settings_dict = self .connection .settings_dict
2528 dbname = settings_dict .get ("DATABASE" )
2629 if not dbname :
@@ -36,6 +39,8 @@ def create_backup(self, backup_path: Path, *, pg_dump: str = "pg_dump") -> None:
3639 def restore_backup (
3740 self , backup_path : Path , * , pg_restore : str = "pg_restore" , psql : str = "psql"
3841 ) -> None :
42+ from plain .postgres .database_url import postgres_cli_args
43+
3944 settings_dict = self .connection .settings_dict
4045 dbname = settings_dict .get ("DATABASE" )
4146 if not dbname :
0 commit comments