Do not set SERVER_NAME on case of IP address

LayerManager · Dec 23, 2019 · 5d981f4 · 5d981f4
1 parent 590a578
commit 5d981f4
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/layman/__init__.py b/src/layman/__init__.py
@@ -7,7 +7,18 @@
 
 app = Flask(__name__)
 app.secret_key = os.environ['FLASK_SECRET_KEY']
-if "." in settings.LAYMAN_PROXY_SERVER_NAME:
+
+
+def _is_ip_address(maybe_ip):
+    import socket
+    try:
+        socket.inet_aton(maybe_ip)
+        return True
+    except socket.error:
+        return False
+
+
+if "." in settings.LAYMAN_PROXY_SERVER_NAME and not _is_ip_address(settings.LAYMAN_PROXY_SERVER_NAME):
     app.config['SERVER_NAME'] = settings.LAYMAN_PROXY_SERVER_NAME