138 lines
3.8 KiB
Python
Executable File
138 lines
3.8 KiB
Python
Executable File
#!/usr/bin/python3
|
|
## A simple tool to configure services and create sql users/databases
|
|
|
|
import os
|
|
import subprocess
|
|
import secrets
|
|
import argparse
|
|
|
|
|
|
# only allow aA-zZ, 0-9 and below "special" characters in passwords
|
|
ALLOWED_CHARS = [ord('!'), ord('@'), ord('#'), ord('%'), ord('&')]\
|
|
+ list(range(48,58))\
|
|
+ list(range(64,90))\
|
|
+ list(range(97,123))
|
|
|
|
# convert ints back to characters
|
|
ALLOWED_CHARS = [chr(c) for c in ALLOWED_CHARS]
|
|
|
|
|
|
def gen_pass(l):
|
|
"""gen_pass returns a password of length l from ALLOWED_CHARS"""
|
|
|
|
password = ""
|
|
for i in range(int(l)):
|
|
password += secrets.choice(ALLOWED_CHARS)
|
|
|
|
return password
|
|
|
|
def get_var(key):
|
|
"""get_var() gets user input to confirm/replace empty keys"""
|
|
|
|
while True:
|
|
user_input = input(
|
|
f"{key} is empty, enter value (leave blank for password):\n")
|
|
|
|
if user_input == "":
|
|
prompt = "provided value is empty, choose an option below\n"\
|
|
+ "\t1) generate password\n"\
|
|
+ "\t2) leave blank\n"\
|
|
+ "\t3) back\n"
|
|
choice = input(prompt)
|
|
while choice not in ["1","2","3"]:
|
|
print(f"error: {choice} unrecognized, please enter 1,2 or 3")
|
|
choice = input(prompt)
|
|
|
|
if choice == "1":
|
|
return gen_pass(32)
|
|
if choice == "2":
|
|
return user_input
|
|
if choice == "3":
|
|
continue
|
|
else:
|
|
# run until user confirms input
|
|
confirm = input(f"set {key} to {user_input} (y/N)?")
|
|
if confirm in ["y", "Y"]:
|
|
return user_input
|
|
|
|
def parse_env(env_file):
|
|
"""parse_env returns a dictionary of env vars parsed from the base file
|
|
|
|
:service: is a string of the service name to parse, must match folder name
|
|
"""
|
|
f = open(env_file)
|
|
c = dict()
|
|
for line in f:
|
|
# skip comments
|
|
if line[0] == "#":
|
|
continue
|
|
if line.isspace():
|
|
continue
|
|
|
|
kv = line.strip('\n').split("=")
|
|
|
|
key = kv[0]
|
|
val = kv[1].strip('"')
|
|
|
|
if val == "":
|
|
val = get_var(key)
|
|
|
|
c[key] = val
|
|
|
|
return c
|
|
|
|
def gen_env(kv):
|
|
"""gen_env takes in a dictionary and returns the formatted env file text as a string
|
|
|
|
:kv: is a dictionary of strings mapping env var to value
|
|
ex) "HOME": "/home/user/"
|
|
"""
|
|
env = ""
|
|
for key, value in kv.items():
|
|
env += f"{key}=\"{value}\"\n"
|
|
|
|
return env
|
|
|
|
def config_service(service, force=False):
|
|
"""config_service processes service env vars to generate private .env file
|
|
|
|
:service: is a string of the name of the folder to write the .env to
|
|
also requires file service/env to source base config from
|
|
:force: is an optional parameter to overwrite existing file
|
|
default is False
|
|
"""
|
|
# setup directory
|
|
subprocess.run(["mkdir", "-p", f"{service}/priv"])
|
|
priv_file = f"{service}/priv/env"
|
|
|
|
# prevent overwrite
|
|
if os.path.isfile(priv_file) and not force:
|
|
print(f"{priv_file} already exists... skipping")
|
|
return parse_env(priv_file)
|
|
|
|
c = parse_env(f"{service}/env")
|
|
|
|
f = open(priv_file, 'w')
|
|
f.write(gen_env(c))
|
|
f.close()
|
|
|
|
return c
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description="takes in a service to configure a private .env file and generate sql users/databases")
|
|
|
|
parser.add_argument('service', nargs="+", help="service to configure env for")
|
|
parser.add_argument('-f', '--force', action="store_true", help="overwrite existing files")
|
|
|
|
args = parser.parse_args()
|
|
|
|
sql = ""
|
|
for service in args.service:
|
|
print(f"\nsetting up {service}...")
|
|
c = config_service(service, args.force)
|
|
|
|
print("success")
|
|
|
|
main()
|