entrouvert
/
publik-common
17
0
Fork 0
Code Issues Pull Requests Releases Activity

misc: apply double-quote-string-fixer (#79788)

This commit is contained in:
Valentin Deniaud 2023-08-16 11:53:24 +02:00
parent 1118551c85
commit af691f64f7
1 changed files with 31 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
publik-create-databases
View File

@ -8,23 +8,23 @@ import sys
publik = { publik = {
"authentic2-multitenant": { 'authentic2-multitenant': {
"database": "authentic2_multitenant", 'database': 'authentic2_multitenant',
"user": "authentic-multitenant", 'user': 'authentic-multitenant',
"extensions": ["unaccent", "pg_trgm"], 'extensions': ['unaccent', 'pg_trgm'],
}, },
"bijoe": {}, 'bijoe': {},
"chrono": {}, 'chrono': {},
"combo": {"extensions": ["unaccent"]}, 'combo': {'extensions': ['unaccent']},
"fargo": {}, 'fargo': {},
"hobo": {}, 'hobo': {},
"passerelle": {}, 'passerelle': {},
"wcs": {}, 'wcs': {},
"welco": {}, 'welco': {},
} }
def run(command, database="postgres", fake=False): def run(command, database='postgres', fake=False):
cmd = 'sudo -u postgres psql -c "%s" %s' % (command.replace('"', '\\"'), database) cmd = 'sudo -u postgres psql -c "%s" %s' % (command.replace('"', '\\"'), database)
if fake: if fake:
print(cmd) print(cmd)
@ -33,14 +33,14 @@ def run(command, database="postgres", fake=False):
def write_setting(brique, host, port, password): def write_setting(brique, host, port, password):
if brique == "wcs": if brique == 'wcs':
print("warning: setting w.c.s. connection password is not implemented") print('warning: setting w.c.s. connection password is not implemented')
return return
settings_d = "/etc/%s/settings.d" % brique settings_d = '/etc/%s/settings.d' % brique
settings = "%s/database.py" % settings_d settings = '%s/database.py' % settings_d
if not os.path.isdir(settings_d): if not os.path.isdir(settings_d):
os.system('mkdir -p %s' % settings_d) os.system('mkdir -p %s' % settings_d)
with open(settings, "w") as fh: with open(settings, 'w') as fh:
fh.write( fh.write(
"DATABASES['default']['HOST'] = '{host}'\n" "DATABASES['default']['HOST'] = '{host}'\n"
"DATABASES['default']['PORT'] = {port}\n" "DATABASES['default']['PORT'] = {port}\n"
@ -50,14 +50,14 @@ def write_setting(brique, host, port, password):
def main(args): def main(args):
for brique, data in publik.items(): for brique, data in publik.items():
database = data.get("database", brique) database = data.get('database', brique)
user = data.get("user", brique) user = data.get('user', brique)
extensions = data.get("extensions") extensions = data.get('extensions')
run('CREATE USER "%s";' % user, fake=args.fake) run('CREATE USER "%s";' % user, fake=args.fake)
if user == "wcs": if user == 'wcs':
run("ALTER USER wcs CREATEDB;", fake=args.fake) run('ALTER USER wcs CREATEDB;', fake=args.fake)
if args.password: if args.password:
password = "".join(random.choice(string.ascii_letters + string.digits) for _ in range(16)) password = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(16))
run("ALTER USER \"%s\" with password '%s';" % (user, password), fake=args.fake) run("ALTER USER \"%s\" with password '%s';" % (user, password), fake=args.fake)
run( run(
"CREATE DATABASE {} WITH OWNER = \"{}\" TEMPLATE = template0 " "CREATE DATABASE {} WITH OWNER = \"{}\" TEMPLATE = template0 "
@ -66,20 +66,20 @@ def main(args):
) )
if extensions: if extensions:
for e in extensions: for e in extensions:
run("CREATE EXTENSION %s;" % e, database=database, fake=args.fake) run('CREATE EXTENSION %s;' % e, database=database, fake=args.fake)
if not args.fake and args.password: if not args.fake and args.password:
write_setting(brique, args.host, args.port, password) write_setting(brique, args.host, args.port, password)
if __name__ == "__main__": if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--password", action="store_true", help="generate password") parser.add_argument('--password', action='store_true', help='generate password')
parser.add_argument("--fake", action="store_true", help="dry-run") parser.add_argument('--fake', action='store_true', help='dry-run')
parser.add_argument("--host", default="localhost") parser.add_argument('--host', default='localhost')
parser.add_argument("--port", default="5432") parser.add_argument('--port', default='5432')
args = parser.parse_args() args = parser.parse_args()
if not args.fake and os.geteuid() != 0: if not args.fake and os.geteuid() != 0:
sys.exit("You need to have privileges to run this script, please try again with sudo.") sys.exit('You need to have privileges to run this script, please try again with sudo.')
main(args) main(args)