niku/infra-as-code
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added default values for taking input

Browse Source
This commit is contained in:
strNophix 2022-04-08 14:20:40 +02:00
parent cd736a2606
commit acda51a6b4
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
service.py
View File

@ -18,9 +18,11 @@ RE_TEXT = re.compile(r"^\w+$")
RE_NUM = re.compile(r"^\d+$") RE_NUM = re.compile(r"^\d+$")
def _take_input(prompt: str, pattern: re.Pattern) -> str: def _take_input(prompt: str, pattern: re.Pattern, default: str="") -> str:
while True: while True:
i = input(prompt) i = input(prompt)
if i == "":
i = default
if pattern.match(i): if pattern.match(i):
return i return i
@ -36,22 +38,22 @@ def main() -> int:
break break
if c == "1": if c == "1":
customer_name = _take_input("Customer name (example=opc): ", customer_name = _take_input("Customer name (example=opg): ",
RE_TEXT) RE_TEXT)
env_name = _take_input("Environment name (example=prod): ", env_name = _take_input("Environment name (example=prod): ",
RE_TEXT) RE_TEXT)
amnt_nginx_web = _take_input( amnt_nginx_web = _take_input(
"Number of nginx webservers (example=1): ", RE_NUM) "Number of nginx webservers (default=1): ", RE_NUM, default="1")
amnt_nginx_lb = _take_input( amnt_nginx_lb = _take_input(
"Number of nginx loadbalancers (example=1): ", RE_NUM) "Number of nginx loadbalancers (default=1): ", RE_NUM, default="1")
amnt_psql = _take_input( amnt_psql = _take_input(
"Number of postgres instances (example=1): ", RE_NUM) "Number of postgres instances (default=1): ", RE_NUM, default="1")
ip_format = _take_input( ip_format = _take_input(
"Format of ip (example=192.168.56.{}, `{}` will be replaced with the correct octet at runtime): ", "Format of ip (default=192.168.56.{}, `{}` will be replaced with the correct octet at runtime): ",
re.compile(r".+")) re.compile(r".+"), default="192.168.56.{}")
ip_int = _take_input( ip_int = _take_input(
"Number to start formatting the IP from (example=10): ", "Number to start formatting the IP from (default=10): ",
RE_NUM) RE_NUM, default="10")
sub.call([ sub.call([
"python3", "cli.py", "create", customer_name, env_name, "python3", "cli.py", "create", customer_name, env_name,
@ -61,21 +63,21 @@ def main() -> int:
]) ])
if c == "2": if c == "2":
customer_name = _take_input("Customer name (example=opc): ", customer_name = _take_input("Customer name (example=opg): ",
RE_TEXT) RE_TEXT)
env_name = _take_input("Environment name (example=prod): ", env_name = _take_input("Environment name (example=prod): ",
RE_TEXT) RE_TEXT)
sub.call(["python3", "cli.py", "delete", customer_name, env_name]) sub.call(["python3", "cli.py", "delete", customer_name, env_name])
if c == "3": if c == "3":
customer_name = _take_input("Customer name (example=opc): ", customer_name = _take_input("Customer name (example=opg): ",
RE_TEXT) RE_TEXT)
env_name = _take_input("Environment name (example=prod): ", env_name = _take_input("Environment name (example=prod): ",
RE_TEXT) RE_TEXT)
sub.call(["python3", "cli.py", "recover", customer_name, env_name]) sub.call(["python3", "cli.py", "recover", customer_name, env_name])
if c == "4": if c == "4":
customer_name = _take_input("Customer name (example=opc): ", customer_name = _take_input("Customer name (example=opg): ",
RE_TEXT) RE_TEXT)
sub.call(["python3", "cli.py", "list", customer_name]) sub.call(["python3", "cli.py", "list", customer_name])