1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import os
import click
import jinja2
from terminaltables import AsciiTable
from envs.exceptions import EnvsValueException
from . import env
from .util import convert_module, import_mod, list_envs_module, raw_input
SETTINGS_TEMPLATE = jinja2.Environment(loader=jinja2.PackageLoader(
'envs', 'templates')).get_template('settings.jinja2')
@click.group()
def envs():
pass
@envs.command(help='Converts an existing settings file so it uses envs.')
@click.option('--settings-file', prompt=True,
help='Settings Module? ex. settings or yourapp.settings')
def convert_settings(settings_file):
attr_list = convert_module(import_mod(settings_file))
template = SETTINGS_TEMPLATE.render(attr_list=attr_list)
new_settings_filename = raw_input(
'What should we name this new settings file? (DO NOT NAME IT THE SAME AS THE ORIGINAL BECAUSE THERE ARE NO IMPORTS): ')
if not new_settings_filename:
raise ValueError('Settings filename is required.')
with open(new_settings_filename, 'w+') as f:
f.write(template)
click.echo(click.style('Your new settings file {}'.format(new_settings_filename), fg='green'))
@envs.command(help='Shows a list of env instances set in a settings file.')
@click.option('--settings-file', prompt=True,
help='Settings Module? ex. settings or yourapp.settings')
@click.option('--keep-result', prompt=True, help='Keep the result file (.envs_result)?', default=False)
def list_envs(settings_file, keep_result):
envs_result = list_envs_module(settings_file)
table_data = [
['Env Var', 'Var Type', 'Has Default', 'Environment Value'],
]
table_data.extend(
[[row.get('key'), row.get('var_type'), bool(row.get('default')), row.get('value')] for row in envs_result])
click.echo(AsciiTable(table_data).table)
if not keep_result:
os.remove('.envs_result')
@envs.command(help='Make sure that the defined envs with no default value have a value set in the environment.')
@click.option('--settings-file', prompt=True,
help='Settings Module? ex. settings or yourapp.settings')
def check_envs(settings_file):
envs_result = list_envs_module(settings_file)
for row in envs_result:
value = env(row.get('key'), row.get('default'), var_type=row.get('var_type'))
if not value:
raise EnvsValueException('{}: Environment Variable Not Set'.format(row.get('key')))
os.remove('.envs_result')
|
