Validating path relative to cli root instead of the absolute one

Author: jhonabreul

lean/click.py

@@ -188,7 +188,7 @@ def __init__(self, exists: bool = False, file_okay: bool = True, dir_okay: bool
     def convert(self, value: str, param: Parameter, ctx: Context) -> Path:
         path = Path(value).expanduser().resolve()
 
-        if not container.path_manager.is_path_valid(path):
+        if not container.path_manager.is_cli_path_valid(path):
             self.fail(f"{self._path_type} '{value}' is not a valid path.", param, ctx)
 
         if self._exists and not path.exists():

lean/commands/cloud/backtest.py

@@ -44,8 +44,7 @@ def backtest(project: str, name: Optional[str], push: bool, open_browser: bool)
     try:
         cloud_project = cloud_project_manager.get_cloud_project(project, push)
     except RuntimeError as e:
-        if cloud_project_manager._project_config_manager.try_get_project_config(Path.cwd() / project,
-                                                                        cloud_project_manager._path_manager):
+        if cloud_project_manager._project_config_manager.try_get_project_config(Path.cwd() / project):
             error_message = f'No project with the given name or id "{project}" found in your cloud projects.'
             error_message += f" Please use `lean cloud backtest --push {project}` to backtest in cloud."
         else:

lean/commands/create_project.py

@@ -292,10 +292,8 @@ def create_project(name: str, language: str) -> None:
             "https://www.lean.io/docs/v2/lean-cli/projects/project-management")
 
     full_path = Path.cwd() / name
-    cli_root_dir = container.lean_config_manager.get_cli_root_directory()
-    relative_path = full_path.relative_to(cli_root_dir)
 
-    if not container.path_manager.is_path_valid(relative_path) or not container.path_manager.is_name_valid(name):
+    if not container.path_manager.is_cli_path_valid(full_path) or not container.path_manager.is_name_valid(name):
         raise MoreInfoError(f"Invalid project name. Can only contain letters, numbers & spaces. Can not start with empty char ' ' or be a reserved name [ {', '.join(reserved_names)} ]",
                          "https://www.lean.io/docs/v2/lean-cli/key-concepts/troubleshooting#02-Common-Errors")
 

lean/components/cloud/cloud_project_manager.py

@@ -66,7 +66,7 @@ def get_cloud_project(self, input: str, push: bool) -> QCProject:
 
         # If the given input is a valid project directory, we try to use that project
         local_path = Path.cwd() / input
-        if self._project_config_manager.try_get_project_config(local_path, self._path_manager):
+        if self._project_config_manager.try_get_project_config(local_path):
             if push:
                 self._push_manager.push_projects([local_path])
 

lean/components/config/project_config_manager.py

@@ -16,7 +16,6 @@
 
 from lean.components.config.storage import Storage
 from lean.components.util.xml_manager import XMLManager
-from lean.components.util.path_manager import PathManager
 from lean.constants import PROJECT_CONFIG_FILE_NAME
 from lean.models.utils import CSharpLibrary
 
@@ -31,15 +30,13 @@ def __init__(self, xml_manager: XMLManager) -> None:
         """
         self._xml_manager = xml_manager
 
-    def try_get_project_config(self, project_directory: Path, path_manager: PathManager) -> Storage:
+    def try_get_project_config(self, project_directory: Path) -> Storage:
         """Returns a Storage instance to get/set the configuration for a project.
 
         :param project_directory: the path to the project to retrieve the configuration of
         :return: the Storage instance containing the project-specific configuration of the given project
         """
-        if path_manager.is_path_valid(project_directory) \
-                and project_directory.is_dir() \
-                and self.get_project_config(project_directory).file.exists():
+        if self.get_project_config(project_directory).file.exists():
             return Storage(str(project_directory / PROJECT_CONFIG_FILE_NAME))
         else:
             return False

lean/components/util/library_manager.py

@@ -54,7 +54,7 @@ def is_lean_library(self, path: Path) -> bool:
         :param path: path to check whether it is a Lean library
         :return: true if the path is a Lean library path
         """
-        if not self._path_manager.is_path_valid(path):
+        if not self._path_manager.is_cli_path_valid(path):
             return False
 
         relative_path = self._path_manager.get_relative_path(path, self._lean_config_manager.get_cli_root_directory())

lean/components/util/path_manager.py

@@ -12,17 +12,21 @@
 # limitations under the License.
 
 from pathlib import Path
+
 from lean.components import reserved_names, forbidden_characters
+from lean.components.config.lean_config_manager import LeanConfigManager
 from lean.components.util.platform_manager import PlatformManager
 
+
 class PathManager:
     """The PathManager class provides utilities for working with paths."""
 
-    def __init__(self, platform_manager: PlatformManager) -> None:
+    def __init__(self, lean_config_manager: LeanConfigManager, platform_manager: PlatformManager) -> None:
         """Creates a new PathManager instance.
 
         :param platform_manager: the PlatformManager used when checking which operating system is in use
         """
+        self._lean_config_manager = lean_config_manager
         self._platform_manager = platform_manager
 
     def get_relative_path(self, destination: Path, source: Path = Path.cwd()) -> Path:
@@ -78,3 +82,14 @@ def is_path_valid(self, path: Path) -> bool:
                 if forbidden_character in component:
                     return False
         return True
+
+    def is_cli_path_valid(self, path: Path) -> bool:
+        """Returns whether the given path is a valid project path in the current operating system.
+
+        :param path: the path to validate
+        :return: True if the path is valid on the current operating system, False if not
+        """
+        cli_root_dir = self._lean_config_manager.get_cli_root_directory()
+        relative_path = path.relative_to(cli_root_dir)
+
+        return relative_path == Path(".") or self.is_path_valid(relative_path)

lean/container.py

@@ -14,17 +14,18 @@
 from typing import Union, Any
 
 from lean.components.api.api_client import APIClient
+from lean.components.config.lean_config_manager import LeanConfigManager
+from lean.components.config.project_config_manager import ProjectConfigManager
+from lean.components.util.path_manager import PathManager
 from lean.components.cloud.cloud_project_manager import CloudProjectManager
 from lean.components.cloud.cloud_runner import CloudRunner
 from lean.components.cloud.data_downloader import DataDownloader
 from lean.components.cloud.module_manager import ModuleManager
 from lean.components.cloud.pull_manager import PullManager
 from lean.components.cloud.push_manager import PushManager
 from lean.components.config.cli_config_manager import CLIConfigManager
-from lean.components.config.lean_config_manager import LeanConfigManager
 from lean.components.config.optimizer_config_manager import OptimizerConfigManager
 from lean.components.config.output_config_manager import OutputConfigManager
-from lean.components.config.project_config_manager import ProjectConfigManager
 from lean.components.config.storage import Storage
 from lean.components.docker.docker_manager import DockerManager
 from lean.components.docker.lean_runner import LeanRunner
@@ -34,7 +35,6 @@
 from lean.components.util.market_hours_database import MarketHoursDatabase
 from lean.components.util.name_generator import NameGenerator
 from lean.components.util.organization_manager import OrganizationManager
-from lean.components.util.path_manager import PathManager
 from lean.components.util.platform_manager import PlatformManager
 from lean.components.util.project_manager import ProjectManager
 from lean.components.util.task_manager import TaskManager
@@ -63,7 +63,6 @@ def initialize(self,
         self.platform_manager = PlatformManager()
         self.task_manager = TaskManager(self.logger)
         self.name_generator = NameGenerator()
-        self.path_manager = PathManager(self.platform_manager)
         self.temp_manager = TempManager()
         self.xml_manager = XMLManager()
         self.http_client = HTTPClient(self.logger)
@@ -92,6 +91,7 @@ def initialize(self,
                                                      self.project_config_manager,
                                                      self.module_manager,
                                                      self.cache_storage)
+        self.path_manager = PathManager(self.lean_config_manager, self.platform_manager)
         self.output_config_manager = OutputConfigManager(self.lean_config_manager)
         self.optimizer_config_manager = OptimizerConfigManager(self.logger)
 

tests/commands/test_build.py

@@ -22,12 +22,18 @@
 from lean.models.docker import DockerImage
 from lean.constants import CUSTOM_FOUNDATION, CUSTOM_RESEARCH, CUSTOM_ENGINE
 from tests.conftest import initialize_container
+from tests.test_helpers import create_fake_lean_cli_directory
 
 CUSTOM_FOUNDATION_IMAGE = DockerImage(name=CUSTOM_FOUNDATION, tag="latest")
 CUSTOM_ENGINE_IMAGE = DockerImage(name=CUSTOM_ENGINE, tag="latest")
 CUSTOM_RESEARCH_IMAGE = DockerImage(name=CUSTOM_RESEARCH, tag="latest")
 
 
+@pytest.fixture(autouse=True)
+def create_fake_cli_directory() -> None:
+    create_fake_lean_cli_directory()
+
+
 def create_fake_repositories() -> None:
     lean_dir = Path.cwd() / "Lean"
 

tests/commands/test_create_project.py

@@ -153,7 +153,7 @@ def test_create_project_aborts_when_path_invalid() -> None:
     create_fake_lean_cli_directory()
 
     path_manager = mock.Mock()
-    path_manager.is_path_valid.return_value = False
+    path_manager.is_cli_path_valid.return_value = False
     container.path_manager= path_manager
 
     result = CliRunner().invoke(lean, ["create-project", "--language", "python", "My First Project"])