From 9a7606a740419fffffdc38750ddb929b98e4ad88 Mon Sep 17 00:00:00 2001 From: Daniel Date: Mon, 4 Dec 2023 15:31:06 +0100 Subject: [PATCH] move copy_env_files to env_manager --- src/coordinator.py | 10 +--------- src/env_manager.py | 10 ++++++++++ 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/src/coordinator.py b/src/coordinator.py index 99d1f66..e306bba 100644 --- a/src/coordinator.py +++ b/src/coordinator.py @@ -1,4 +1,3 @@ -import shutil from pathlib import Path from loguru import logger @@ -24,14 +23,7 @@ class Coordinator: def setup_test(self) -> None: logger.info("calling setup_test()") self.DIR.create_all_dirs() - self._copy_env_files() - - def _copy_env_files(self) -> None: - """Copies all env files to STATES/env_files. Files will be renamed to their own TYPE value.""" - env_files_dir = self.DIR.STATES / "env_files" - env_files_dir.mkdir(exist_ok=True) - for env_file in self.ENV.env_files: - shutil.copy(env_file.env_path, env_files_dir / env_file.env_type) + self.ENV.copy_env_files(self.DIR) def run_test(self) -> None: logger.info("calling run_test()") diff --git a/src/env_manager.py b/src/env_manager.py index 5bca1f6..239f9a9 100644 --- a/src/env_manager.py +++ b/src/env_manager.py @@ -1,9 +1,12 @@ +import shutil from pathlib import Path from typing import NamedTuple from dotenv import dotenv_values from loguru import logger +from src.dirmanager import DirManager + class EnvFile(NamedTuple): env_path: Path @@ -87,3 +90,10 @@ class EnvManager: dependency_rule = DependencyRule(child=child_runner_class.name, dependency=dependency.name) dependency_rules.append(dependency_rule) return dependency_rules + + def copy_env_files(self, DIR: DirManager) -> None: + """Copies all env files to STATES/env_files. Files will be renamed to their own TYPE value.""" + env_files_dir = DIR.STATES / "env_files" + env_files_dir.mkdir(exist_ok=True) + for env_file in self.env_files: + shutil.copy(env_file.env_path, env_files_dir / env_file.env_type)