diff options
Diffstat (limited to 'docker/common')
| -rwxr-xr-x | docker/common/hautoproxy/start.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/docker/common/hautoproxy/start.py b/docker/common/hautoproxy/start.py index 877b22157..bbdd876b5 100755 --- a/docker/common/hautoproxy/start.py +++ b/docker/common/hautoproxy/start.py @@ -6,7 +6,6 @@ docker/kubernetes environment variables.''' import argparse import os -import sys from jinja2 import Environment, FileSystemLoader import re import urlparse @@ -14,6 +13,7 @@ import urlparse re_url = re.compile( '^(?P<name>.*)_PORT_(?P<port>\d+)_(?P<proto>(UDP|TCP))$') + def parse_args(): p = argparse.ArgumentParser() p.add_argument('--output', '-o', @@ -24,6 +24,7 @@ def parse_args(): default='/etc/haproxy/templates') return p.parse_args() + def discover_services(): services = [] for k in os.environ: @@ -31,7 +32,7 @@ def discover_services(): if mo: parts = urlparse.urlparse(os.environ[k]) - remote_host,remote_port = parts.netloc.split(':') + remote_host, remote_port = parts.netloc.split(':') service_name = '%(name)s-%(port)s' % mo.groupdict() services.append({ @@ -45,6 +46,7 @@ def discover_services(): return services + def main(): args = parse_args() services = discover_services() @@ -62,4 +64,3 @@ def main(): if __name__ == '__main__': main() - |