Adds configurable hosts for hamilton ui

Settings mini is used by default by the Hamilton UI
command. This changes it to enable changing
ALLOWED_HOSTS by passing in
some environment variables.

It then automatically adds a few of them, like
the current IP, etc.

Author: skrawcz

ui/backend/server/server/settings_mini.py

@@ -1,6 +1,10 @@
+import logging
 import os
+import socket
 from pathlib import Path
 
+logger = logging.getLogger(__name__)
+
 
 def get_from_env(
     var_name: str, valid_values: list = None, allow_missing: bool = False, default_value: str = None
@@ -34,7 +38,19 @@ def get_from_env(
 
 HAMILTON_AUTH_MODE = "permissive"
 
-ALLOWED_HOSTS = ["localhost", "backend", "0.0.0.0", "127.0.0.1"]
+# set up allowed hosts
+try:
+    hostname = socket.gethostname()
+    local_ip = socket.gethostbyname(hostname)
+    hosts_to_add = [local_ip, hostname]
+except Exception:
+    logger.warning("Could not get hostname or local IP")
+    hosts_to_add = []
+ALLOWED_HOSTS = ["localhost", "backend", "0.0.0.0", "127.0.0.1", "colab.research.google.com"]
+ALLOWED_HOSTS = (
+    ALLOWED_HOSTS + hosts_to_add + os.environ.get("HAMILTON_ALLOWED_HOSTS", "").split(",")
+)
+logger.debug(f"ALLOWED_HOSTS: {ALLOWED_HOSTS}")
 
 HAMILTON_BLOB_STORE = "local"