dev flag for coiled script

Author: gjoseph92

scripts/coiled-envs.py

@@ -1,6 +1,8 @@
 """
 Make Coiled software environments for stackstac in multiple regions,
 and upload all notebook files into a Coiled notebook (aka Job).
+
+Pass --dev to build off the latest commit on `origin/main` instead.
 """
 
 import asyncio
@@ -9,6 +11,7 @@
 from pathlib import Path
 import sys
 from typing import List
+import warnings
 
 import aiotools
 import coiled
@@ -79,23 +82,35 @@ async def make_coiled_stuff(
         print(f"Created notebook {name}")
 
 
+def run(cmd: str, **kwargs) -> str:
+    return subprocess.run(
+        cmd, shell=True, check=True, capture_output=True, text=True, **kwargs
+    ).stdout.strip()
+
+
 if __name__ == "__main__":
-    proc = subprocess.run(
-        "poetry export --without-hashes -E binder -E viz",
-        shell=True,
-        check=True,
-        capture_output=True,
-        text=True,
-    )
-    deps = proc.stdout.splitlines()
+    # TODO use click or something to make an actual CLI.
+    dev = len(sys.argv) == 2 and sys.argv[-1] == "--dev"
 
-    # TODO single-source the version! this is annoying
-    version = subprocess.run(
-        "poetry version -s", shell=True, check=True, capture_output=True, text=True
-    ).stdout.strip()
-    print(f"Version: {version}")
-    deps += [f"stackstac[binder,viz]=={version}"]
+    deps = run("poetry export --without-hashes -E binder -E viz").splitlines()
+
+    if dev:
+        name = "stackstac-dev"
+        subprocess.run("git fetch", shell=True, check=True)
+        main = run("git rev-parse main")
+        origin_main = run("git rev-parse origin/main")
+        if main != origin_main:
+            warnings.warn("Your local main branch is not up to date with origin/main")
+        print(f"Commit (origin/main): {origin_main}")
+        stackstac_dep = f"git+https://github.com/gjoseph92/stackstac.git@{origin_main}#egg=stackstac[binder,viz]"
+    else:
+        name = "stackstac"
+        # TODO single-source the version! this is annoying
+        version = run("poetry version -s")
+        print(f"Version: {version}")
+        stackstac_dep = f"stackstac[binder,viz]=={version}"
+    deps += [stackstac_dep]
 
     asyncio.run(
-        make_coiled_stuff(deps, regions=["us-west-2", "eu-central-1"], name="stackstac")
+        make_coiled_stuff(deps, regions=["us-west-2", "eu-central-1"], name=name)
     )