Git add files that are changed

Author: robbievanleeuwen

README.md

@@ -7,6 +7,7 @@ Creates PRs for dependency updates in python projects.
 - Currently only supports a single specifier.
 - File structure is fixed
 - GH actions must only use major version
+- Recommend to have a clean git before running (or at least no changes to pyproject etc.)
 
 ## TODO
 

src/upgrade_dependencies/main.py

@@ -13,7 +13,7 @@
 
 from upgrade_dependencies.dependency import GitHubDependency, PyPIDependency
 from upgrade_dependencies.project import Project
-from upgrade_dependencies.utils import run_shell_command
+from upgrade_dependencies.utils import get_git_status, run_shell_command
 
 if TYPE_CHECKING:
     from upgrade_dependencies.dependency import Dependency
@@ -330,22 +330,24 @@ def update(
         version: _description_
         project_path: _description_
     """
-    # TODO: print status
+    # TODO: print status of update
     project = Project(
         project_path=project_path,
         gh_pat=GH_PAT,
     )
 
+    # search for dependency and save old version
     try:
         dep = project.get_dependency(name=dependency)
         old_ver = str(sorted(dep.specifier, key=str)[0].version)
     except RuntimeError as e:
         rprint(f"Cannot find {dependency} in {project.name}.")
         raise typer.Exit(code=1) from e
 
+    # fetch data from pypi/github
     asyncio.run(dep.save_data())
 
-    # get version
+    # get latest/desired version
     if version is None:
         version = str(dep.get_latest_version())
 
@@ -358,11 +360,19 @@ def update(
 
     run_shell_command(["git", "checkout", "-b", branch_name])
 
+    # get status of files before changes
+    files_before = get_git_status()
+
     # update dependency
     project.update_dependency(dependency=dep, version=version)
 
-    # stage changes - TODO: get files that are changed
-    run_shell_command(["git", "add", ".github/"])
+    # get status of files after changes
+    files_after = get_git_status()
+
+    # get only the files that were changed
+    changed_files = [f for f in files_after if f not in files_before]
+
+    run_shell_command(["git", "add", *changed_files])
 
     # commit the changes
     if isinstance(dep, GitHubDependency) and dep.action:

src/upgrade_dependencies/utils.py

@@ -250,15 +250,40 @@ def update_pre_commit(
         yaml.dump(data, temp_f)  # pyright: ignore
 
 
-def run_shell_command(shell_args: list[str]) -> None:
+def run_shell_command(shell_args: list[str]) -> Any:
     """_summary_.
 
     Args:
         shell_args: _description_
+
+    Returns:
+        _description_
     """
     try:
-        subprocess.run(shell_args, check=True, capture_output=True, text=True)  # noqa: S603
+        res = subprocess.run(shell_args, check=True, capture_output=True, text=True)  # noqa: S603
     except subprocess.CalledProcessError as e:
         msg = f"Command failed with return code {e.returncode}.\n"
         msg += f"Error output: {e.stderr}"
         raise RuntimeError(msg) from e
+
+    return res
+
+
+def get_git_status() -> list[str]:
+    """Get the list of modified or untracked files from git status.
+
+    Returns:
+        _description_
+    """
+    result = run_shell_command(["git", "status", "-s"])
+
+    # parse the result to get the list of files
+    changed_files: list[str] = []
+
+    for line in result.stdout.strip().split("\n"):
+        status, file_path = line.split(maxsplit=1)  # status and file name
+
+        if status in ["M", "A", "D"]:  # Modified, Added, or Deleted
+            changed_files.append(file_path)
+
+    return changed_files