mccutchen · February 2, 2021 21:15 · Feb 2, 2021
diff --git a/ddlogs b/ddlogs
@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+
+import argparse
+import subprocess
+import sys
+import urllib.parse
+
+
+def main(env: str, services: [str]) -> int:
+    filters = [f"env:{env}"]
+    if services:
+        filters.append(" OR ".join(f"service:{svc_name(s)}" for s in services))
+    params = {
+        "query": " ".join(filters),
+    }
+    url = f"https://app.datadoghq.com/logs?{urllib.parse.urlencode(params)}"
+    print(url)
+    return subprocess.call(["open", url])
+
+
+def svc_name(s: str):
+    """
+    Allow service to be specified as any of
+
+        service
+        apps/service
+        apps/service/
+        /path/to/apps/service/
+
+    The form with a trailing slash is likely when using tab completion in the
+    terminal.
+    """
+    return s.strip("/").split("/")[-1]
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Open datadog logs")
+    parser.add_argument("env", metavar="ENV", type=str, help="environment")
+    parser.add_argument(
+        "services",
+        metavar="SERVICE",
+        type=str,
+        nargs="*",
+        help="service (optional, can specify more than one)",
+    )
+    args = parser.parse_args()
+    sys.exit(main(args.env, args.services))
No results found