Initial solution

Author: JosueNina

lean/commands/cloud/push.py

@@ -34,7 +34,10 @@
 @option("--key",
               type=PathParameter(exists=True, file_okay=True, dir_okay=False),
               help="Path to the encryption key to use")
-def push(project: Optional[Path], encrypt: Optional[bool], decrypt: Optional[bool], key: Optional[Path]) -> None:
+@option("--force",
+        is_flag=True, default=False,
+        help="Force push even if there's a lock conflict")
+def push(project: Optional[Path], encrypt: Optional[bool], decrypt: Optional[bool], key: Optional[Path], force: Optional[bool]) -> None:
     """Push local projects to QuantConnect.
 
     This command overrides the content of cloud files with the content of their respective local counterparts.
@@ -61,11 +64,11 @@ def push(project: Optional[Path], encrypt: Optional[bool], decrypt: Optional[boo
 
         if encrypt and key is not None:
             from lean.components.util.encryption_helper import validate_encryption_key_registered_with_cloud
-            validate_encryption_key_registered_with_cloud(key, container.organization_manager, container.api_client) 
+            validate_encryption_key_registered_with_cloud(key, container.organization_manager, container.api_client)
 
-        push_manager.push_project(project, encryption_action, key)
+        push_manager.push_project(project, encryption_action, key, force)
     else:
         if key is not None:
             raise RuntimeError(f"Encryption key can only be specified when pushing a single project.")
         projects_to_push = [p.parent for p in Path.cwd().rglob(PROJECT_CONFIG_FILE_NAME)]
-        push_manager.push_projects(projects_to_push, [], encryption_action, key)
+        push_manager.push_projects(projects_to_push, [], encryption_action, key, force)

lean/components/api/project_client.py

@@ -83,7 +83,8 @@ def update(self,
                python_venv: Optional[int] = None,
                files: Optional[List[Dict[str, str]]] = None,
                libraries: Optional[List[int]] = None,
-               encryption_key: Optional[str] = None) -> None:
+               encryption_key: Optional[str] = None,
+               code_source_id: Optional[str] = "cli") -> None:
         """Updates an existing project.
 
         :param project_id: the id of the project to update
@@ -94,6 +95,7 @@ def update(self,
         :param python_venv: the python venv id for the project, or None if the python venv shouldn't be changed
         :param files: the list of files for the project
         :param libraries: the list of libraries referenced by the project
+        :param code_source_id: the source of the code changes (e.g., "cli")
         """
         request_parameters = {
             "projectId": project_id
@@ -136,7 +138,10 @@ def update(self,
 
         if encryption_key is not None:
             request_parameters["encryptionKey"] = encryption_key
-            
+
+        if code_source_id is not None:
+            request_parameters["codeSourceId"] = code_source_id
+
         self._api.post("projects/update", request_parameters, data_as_json=False)
 
     def delete(self, project_id: int) -> None:

lean/components/cloud/push_manager.py

@@ -46,17 +46,17 @@ def __init__(self,
         self._organization_manager = organization_manager
         self._cloud_projects = []
 
-    def push_project(self, project: Path, encryption_action: Optional[ActionType]=None, encryption_key: Optional[Path]=None) -> None:
+    def push_project(self, project: Path, encryption_action: Optional[ActionType]=None, encryption_key: Optional[Path]=None, force: Optional[bool]=False) -> None:
         """Pushes the given project from the local drive to the cloud.
 
         It will also push every library referenced by the project and add or remove references.
 
         :param project: path to the directory containing the local project that needs to be pushed
         """
         libraries = self._project_manager.get_project_libraries(project)
-        self.push_projects([project], libraries, encryption_action, encryption_key)
+        self.push_projects([project], libraries, encryption_action, encryption_key, force)
 
-    def push_projects(self, projects_to_push: List[Path], associated_libraries_to_push: Optional[List[Path]]=[], encryption_action: Optional[ActionType]=None, encryption_key: Optional[Path]=None) -> None:
+    def push_projects(self, projects_to_push: List[Path], associated_libraries_to_push: Optional[List[Path]]=[], encryption_action: Optional[ActionType]=None, encryption_key: Optional[Path]=None, force: Optional[bool]=False) -> None:
         """Pushes the given projects from the local drive to the cloud.
 
         It will also push every library referenced by each project and add or remove references.
@@ -78,7 +78,7 @@ def push_projects(self, projects_to_push: List[Path], associated_libraries_to_pu
             relative_path = path.relative_to(Path.cwd())
             try:
                 self._logger.info(f"[{index}/{len(all_projects_to_push)}] Pushing '{relative_path}'")
-                self._push_project(path, organization_id, encryption_action_value, encryption_key_value)
+                self._push_project(path, organization_id, encryption_action_value, encryption_key_value, force=force)
             except Exception as ex:
                 from traceback import format_exc
                 self._logger.debug(format_exc().strip())
@@ -95,7 +95,7 @@ def _get_local_libraries_cloud_ids(self, project_dir: Path) -> List[int]:
 
         return local_libraries_cloud_ids
 
-    def _push_project(self, project_path: Path, organization_id: str, encryption_action: Optional[ActionType], encryption_key: Optional[Path], suggested_rename_path: Path = None) -> None:
+    def _push_project(self, project_path: Path, organization_id: str, encryption_action: Optional[ActionType], encryption_key: Optional[Path], force: Optional[bool], suggested_rename_path: Path = None) -> None:
         """Pushes a single local project to the cloud.
 
         Raises an error with a descriptive message if the project cannot be pushed.
@@ -111,7 +111,6 @@ def _push_project(self, project_path: Path, organization_id: str, encryption_act
         if suggested_rename_path and suggested_rename_path != project_path:
             potential_new_name = suggested_rename_path.relative_to(Path.cwd()).as_posix()
 
-
         project_config = self._project_config_manager.get_project_config(project_path)
         cloud_id = project_config.get("cloud-id")
         local_encryption_state = project_config.get("encrypted", False)
@@ -163,7 +162,7 @@ def _push_project(self, project_path: Path, organization_id: str, encryption_act
             encryption_key = local_encryption_key
             encryption_action = ActionType.ENCRYPT if local_encryption_state else ActionType.DECRYPT
         # Finalize pushing by updating locally modified metadata, files and libraries
-        self._push_metadata(project_path, cloud_project, encryption_action, encryption_key)
+        self._push_metadata(project_path, cloud_project, encryption_action, encryption_key, force)
 
     def _get_files(self, project: Path, encryption_action: Optional[ActionType], encryption_key: Optional[Path]) -> List[Dict[str, str]]:
         """Pushes the files of a local project to the cloud.
@@ -193,7 +192,7 @@ def _get_files(self, project: Path, encryption_action: Optional[ActionType], enc
 
         return files
 
-    def _push_metadata(self, project: Path, cloud_project: QCProject, encryption_action: Optional[ActionType], encryption_key: Optional[Path]) -> None:
+    def _push_metadata(self, project: Path, cloud_project: QCProject, encryption_action: Optional[ActionType], encryption_key: Optional[Path], force: Optional[bool]) -> None:
         """Pushes local project description and parameters to the cloud.
 
         Does nothing if the cloud is already up-to-date.
@@ -260,13 +259,16 @@ def _push_metadata(self, project: Path, cloud_project: QCProject, encryption_act
 
             if "encryption_key" in update_args:
                 del update_args["encryption_key"]
+            if not force:
+                update_args["code_source_id"] = "cli"
             updated_keys = list(update_args)
             if len(updated_keys) == 1:
                 updated_keys_str = updated_keys[0]
             elif len(updated_keys) == 2:
                 updated_keys_str = " and ".join(updated_keys)
             else:
                 updated_keys_str = ", ".join(updated_keys[:-1]) + f", and {updated_keys[-1]}"
+
             self._logger.info(f"Successfully updated {updated_keys_str} for '{cloud_project.name}'")
 
     def _get_cloud_project(self, project_id: int, organization_id: str) -> QCProject:

tests/commands/cloud/test_push.py

@@ -82,7 +82,7 @@ def test_cloud_push_pushes_single_project_when_project_option_given() -> None:
 
     assert result.exit_code == 0
 
-    push_manager.push_project.assert_called_once_with(Path.cwd() / "Python Project", None, None)
+    push_manager.push_project.assert_called_once_with(Path.cwd() / "Python Project", None, None, False)
 
 
 def test_cloud_push_aborts_when_given_directory_is_not_lean_project() -> None:
@@ -469,3 +469,29 @@ def _get_expected_encrypted_files_content() -> dict:
 UGw0ehtO8qY5FmPGcUlkBGuqmd7r6aLE4mosoZrc/UyZb+clWNYJITRLFJbQpWm3EU/Xrt5UM8uWwEdV
 bFWAAkX56MyDHwJefC1nkA=="""
 }
+
+def test_cloud_push_sets_code_source_id_to_cli() -> None:
+    create_fake_lean_cli_directory()
+    project_path = Path.cwd() / "Python Project"
+
+    api_client = mock.Mock()
+    cloud_project = create_api_project(1, "Python Project")
+    api_client.projects.create = mock.MagicMock(return_value=cloud_project)
+    api_client.files.get_all = mock.MagicMock(return_value=[
+        QCFullFile(name="file.py", content="print(123)", modified=datetime.now(), isLibrary=False)
+    ])
+
+    init_container(api_client_to_use=api_client)
+
+    result = CliRunner().invoke(lean, ["cloud", "push", "--project", project_path])
+
+    assert result.exit_code == 0
+    expected_arguments = {
+        "name": "Python Project",
+        "description": "",
+        "files": mock.ANY,
+        "libraries": [],
+        "encryption_key": "",
+        "codeSourceId": "cli"
+    }
+    api_client.projects.update.assert_called_once_with(1, **expected_arguments)