Files
spring-cloud-dataflow-samples/dataflow-website/recipes/polyglot/polyglot-python-processor/util/task_args.py
2019-05-16 12:37:34 +02:00

34 lines
934 B
Python

import os
import sys
from collections import defaultdict
def get_cmd_arg(name):
d = defaultdict(list)
for cmd_args in sys.argv[1:]:
cmd_arg = cmd_args.split('=')
if len(cmd_arg) == 2:
d[cmd_arg[0].lstrip('-')].append(cmd_arg[1])
if name in d:
return d[name][0]
else:
print('Unknown command line arg requested: {}'.format(name))
def get_env_var(name):
if name in os.environ:
return os.environ[name]
else:
print('Unknown environment variable requested: {}'.format(name))
def get_kafka_binder_brokers():
return get_env_var('SPRING_CLOUD_STREAM_KAFKA_BINDER_BROKERS')
def get_input_channel():
return get_cmd_arg("spring.cloud.stream.bindings.input.destination")
def get_output_channel():
return get_cmd_arg("spring.cloud.stream.bindings.output.destination")
def get_reverse_string():
return get_cmd_arg("reverestring")