cli

Author: snopoke

taskbadger/__init__.py

@@ -1,2 +1,2 @@
 from .integrations import Action, EmailIntegration
-from .sdk import Task, init, update_task
+from .sdk import Task, init, init_from_env, update_task

taskbadger/cli.py

@@ -0,0 +1,24 @@
+import subprocess
+from typing import List
+
+import typer
+
+import taskbadger as tb
+
+app = typer.Typer()
+
+tb.init_from_env()
+
+
+@app.command()
+def monitor(name: str, command: List[str] = typer.Argument(None)):
+    task = tb.Task.create(name)
+    result = subprocess.run(command, env={"TASKBADGER_TASK_ID": task.id})
+    if result.returncode != 0:
+        task.success()
+    else:
+        task.error(data={"return_code": result.returncode})
+
+
+if __name__ == "__main__":
+    app()

taskbadger/sdk.py

@@ -1,4 +1,5 @@
 import dataclasses
+import os
 from typing import List
 
 from _contextvars import ContextVar
@@ -14,6 +15,14 @@
 _local = ContextVar("taskbadger_client")
 
 
+def init_from_env():
+    _init(
+        os.environ["TASKBADGER_ORG"],
+        os.environ["TASKBADGER_PROJECT"],
+        os.environ["TASKBADGER_TOKEN"],
+    )
+
+
 def init(organization_slug: str, project_slug: str, token: str):
     _init("https://taskbadger.net", organization_slug, project_slug, token)