Add configurable path for Graylog

malcolm/imalcolm.py

@@ -58,9 +58,36 @@ def parse_args():
     return args
 
 
+def graylog_from_yaml(yaml_path):
+    from ruamel.yaml import YAML
+
+    assert yaml_path.endswith(
+        ".yaml"
+    ), f"Expected a/path/to/<yamlname>.yaml, got {yaml_path!r}"
+    with open(yaml_path) as f:
+        text = f.read()
+    # First separate them into their relevant sections
+    yaml = YAML(typ="rt")
+    ds = yaml.load(text)
+    graylog = "graylog2.diamond.ac.uk"
+    for d in ds:
+        assert len(d) == 1, f"Expected section length 1, got {len(d)}"
+        name = list(d)[0]
+        if name == "builtin.defines.string":
+            if d[name]["name"] == "graylog_path":
+                graylog = d[name]["value"]
+                print(graylog)
+    f.close()
+    return graylog
+
+
 def make_logging_config(args):
     from ruamel import yaml
 
+    graylog = "graylog2.diamond.ac.uk"
+    if args.yaml:
+        graylog = graylog_from_yaml(args.yaml)
+    print(graylog)
     log_config = {
         "version": 1,
         "disable_existing_loggers": False,
@@ -99,7 +126,7 @@ def make_logging_config(args):
                 "class": "pygelf.GelfTcpHandler",
                 # Obviously a DLS-specific configuration: the graylog server
                 # address and port
-                "host": "graylog2.diamond.ac.uk",
+                "host": graylog,
                 "port": 12202,
                 "debug": True,
                 "level": "DEBUG",