From 4e7a384e32a825b2a04ea4ed2a88c3e56414c497 Mon Sep 17 00:00:00 2001 From: ShenYuhan Date: Tue, 30 Jun 2020 14:56:54 +0800 Subject: [PATCH] get url when start vdl. (#690) --- visualdl/server/args.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/visualdl/server/args.py b/visualdl/server/args.py index cf43fc30..124dd20b 100644 --- a/visualdl/server/args.py +++ b/visualdl/server/args.py @@ -14,11 +14,12 @@ # ======================================================================= import sys +import socket from argparse import ArgumentParser from visualdl.server.log import logger -default_host = '127.0.0.1' +default_host = None default_port = 8040 default_cache_timeout = 20 @@ -50,6 +51,16 @@ def validate_args(args): sys.exit(-1) +def get_host(host=None, port=default_port): + if not host: + host = socket.getfqdn() + try: + socket.create_connection((host, port), timeout=1) + except socket.error: + host = 'localhost' + return host + + def format_args(args): validate_args(args) @@ -63,6 +74,9 @@ def format_args(args): if args.api_only: args.open_browser = False + if not args.host: + args.host = get_host(args.host, args.port) + return args -- GitLab