Setup graylog logging outside of Zocalo configuration

This should remove the need for the ZOCALO_CONFIG environment variable to be set when starting a Murfey server

Author: d-j-hatton

src/murfey/server/__init__.py

@@ -13,11 +13,11 @@
 from threading import Thread
 from typing import Any, Dict, List, NamedTuple, Tuple
 
+import graypy
 import mrcfile
 import numpy as np
 import uvicorn
 import workflows
-import zocalo.configuration
 from backports.entry_points_selectable import entry_points
 from fastapi import Request
 from fastapi.templating import Jinja2Templates
@@ -273,18 +273,18 @@ def run():
         default=0,
     )
 
+    security_config = get_security_config()
     # setup logging
-    zc = zocalo.configuration.from_file()
-    zc.activate()
+    if security_config.graylog_host:
+        handler = graypy.GELFUDPHandler(
+            security_config.graylog_host, security_config.graylog_port, level_names=True
+        )
+        root_logger = logging.getLogger()
+        root_logger.addHandler(handler)
 
     # Install a log filter to all existing handlers.
-    # At this stage this will exclude console loggers, but will cover
-    # any Graylog logging set up by the environment activation
     LogFilter.install()
 
-    zc.add_command_line_options(parser)
-
-    security_config = get_security_config()
     workflows.transport.load_configuration_file(security_config.rabbitmq_credentials)
 
     args = parser.parse_args()

src/murfey/util/config.py

@@ -8,7 +8,7 @@
 
 import yaml
 from backports.entry_points_selectable import entry_points
-from pydantic import BaseModel, BaseSettings, Extra
+from pydantic import BaseModel, BaseSettings, Extra, validator
 
 
 class MachineConfig(BaseModel, extra=Extra.allow):  # type: ignore
@@ -94,6 +94,16 @@ class Security(BaseModel):
     auth_type: Literal["password", "cookie"] = "password"
     cookie_key: str = ""
     feedback_queue: str = "murfey_feedback"
+    graylog_host: str = ""
+    graylog_port: Optional[int] = None
+
+    @validator("graylog_port")
+    def check_port_present_if_host_is(
+        cls, v: Optional[int], values: dict, **kwargs
+    ) -> Optional[int]:
+        if values["graylog_host"] and v is None:
+            raise ValueError("The Graylog port must be set if the Graylog host is")
+        return v
 
 
 def security_from_file(config_file_path: Path) -> Security: