add update() function (#60)

* feat: add update() function to update Julia dependencies

* revert not removing manifest.toml

* update registries when force=True

* more reverts

---------

Co-authored-by: Christopher Doris <github.com/cjdoris>

Author: cjdoris

CHANGELOG.md

@@ -2,7 +2,8 @@
 
 ## Unreleased
 * Support editable dependencies from setuptools (experimental).
-* Better input validation.
+* Add `update()` function.
+* Improved input validation.
 
 ## v0.1.17 (2025-05-13)
 * Respect `JULIAUP_DEPOT_PATH` when searching for Julia using juliaup.

README.md

@@ -59,8 +59,9 @@ Julia v1.*.* and the Example package v0.5.*:
 
 - `juliapkg.executable()` returns a compatible Julia executable.
 - `juliapkg.project()` returns the project into which the packages have been installed.
-- `juliapkg.resolve(force=False)` ensures all the dependencies are installed. You don't
+- `juliapkg.resolve(force=False, dry_run=False)` ensures all the dependencies are installed. You don't
   normally need to do this because the other functions resolve automatically.
+- `juliapkg.update(dry_run=False)` updates the dependencies.
 
 ## Details
 

src/juliapkg/__init__.py

@@ -8,6 +8,7 @@
     resolve,
     rm,
     status,
+    update,
 )
 
 __all__ = [
@@ -20,4 +21,5 @@
     "add",
     "rm",
     "offline",
+    "update",
 ]

src/juliapkg/deps.py

@@ -10,7 +10,7 @@
 
 from .compat import Compat, Version
 from .find_julia import find_julia, julia_version
-from .install_julia import log
+from .install_julia import log, log_script
 from .state import STATE
 
 logger = logging.getLogger("juliapkg")
@@ -363,10 +363,25 @@ def merge_any(dep, kfvs, k):
     return compat, deps
 
 
-def resolve(force=False, dry_run=False):
+def resolve(force=False, dry_run=False, update=False):
+    """
+    Resolve the dependencies.
+
+    Args:
+        force (bool): Force resolution.
+        dry_run (bool): Dry run.
+        update (bool): Update the dependencies.
+
+    Returns:
+        bool: Whether the dependencies are resolved (always True unless dry_run is
+            True).
+    """
+    # update implies force
+    if update:
+        force = True
     # fast check to see if we have already resolved
     if (not force) and STATE["resolved"]:
-        return False
+        return True
     STATE["resolved"] = False
     # use a lock to prevent concurrent resolution
     project = STATE["project"]
@@ -413,9 +428,7 @@ def resolve(force=False, dry_run=False):
             projtoml.extend(
                 f'{pkg.name} = "{pkg.version}"' for pkg in pkgs if pkg.version
             )
-            log("Writing Project.toml:")
-            for line in projtoml:
-                log(" ", line, cont=True)
+            log_script(projtoml, "Writing Project.toml:")
             with open(os.path.join(project, "Project.toml"), "wt") as fp:
                 for line in projtoml:
                     print(line, file=fp)
@@ -437,28 +450,13 @@ def resolve(force=False, dry_run=False):
                 for pkg in add_pkgs:
                     script.append(f"  {pkg.jlstr()},")
                 script.append("])")
-            script.append("Pkg.resolve()")
+            if update:
+                script.append("Pkg.update()")
+            else:
+                script.append("Pkg.resolve()")
             script.append("Pkg.precompile()")
-            log("Installing packages:")
-            for line in script:
-                log(" ", line, cont=True)
-            env = os.environ.copy()
-            if sys.executable:
-                # prefer PythonCall to use the current Python executable
-                # TODO: this is a hack, it would be better for PythonCall to detect that
-                #   Julia is being called from Python
-                env.setdefault("JULIA_PYTHONCALL_EXE", sys.executable)
-            run(
-                [
-                    exe,
-                    "--project=" + project,
-                    "--startup-file=no",
-                    "-e",
-                    "\n".join(script),
-                ],
-                check=True,
-                env=env,
-            )
+            log_script(script, "Installing packages:")
+            run_julia(script, executable=exe, project=project)
         # record that we resolved
         save_meta(
             {
@@ -486,6 +484,39 @@ def resolve(force=False, dry_run=False):
         lock.release()
 
 
+def run_julia(script, executable=None, project=None):
+    """
+    Run a Julia script with the specified executable and project.
+
+    Args:
+        executable (str): Path to the Julia executable.
+        project (str): Path to the Julia project.
+        script (list): List of strings representing the Julia script to run.
+    """
+    if executable is None:
+        executable = STATE["executable"]
+    if project is None:
+        project = STATE["project"]
+
+    env = os.environ.copy()
+    if sys.executable:
+        # prefer PythonCall to use the current Python executable
+        # TODO: this is a hack, it would be better for PythonCall to detect that
+        #   Julia is being called from Python
+        env.setdefault("JULIA_PYTHONCALL_EXE", sys.executable)
+    run(
+        [
+            executable,
+            "--project=" + project,
+            "--startup-file=no",
+            "-e",
+            "\n".join(script),
+        ],
+        check=True,
+        env=env,
+    )
+
+
 def executable():
     resolve()
     return STATE["executable"]
@@ -496,6 +527,20 @@ def project():
     return STATE["project"]
 
 
+def update(dry_run=False):
+    """
+    Resolve and update the dependencies.
+
+    Args:
+        dry_run (bool): Dry run.
+
+    Returns:
+        bool: Whether the dependencies were updated (always True unless dry_run is
+            True).
+    """
+    return resolve(dry_run=dry_run, update=True)
+
+
 def cur_deps_file(target=None):
     if target is None:
         return STATE["deps"]

src/juliapkg/install_julia.py

@@ -24,6 +24,13 @@ def log(*args, cont=False):
     print(prefix, *args)
 
 
+def log_script(script, title=None):
+    if title is not None:
+        log(title)
+    for line in script:
+        log("|", line, cont=True)
+
+
 def all_julia_versions():
     global _all_julia_versions
     if _all_julia_versions is None: