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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#!/usr/bin/env python
import getpass
import os
import random
import sys
import click
from flask.cli import FlaskGroup
from werkzeug.security import generate_password_hash
from ihatemoney.models import Project, db
from ihatemoney.run import create_app
from ihatemoney.utils import create_jinja_env
@click.group(cls=FlaskGroup, create_app=create_app)
def cli():
"""IHateMoney Management script"""
@cli.command(
context_settings={"ignore_unknown_options": True, "allow_extra_args": True}
)
@click.pass_context
def runserver(ctx):
"""Deprecated, use the "run" command instead"""
click.secho(
'"runserver" is deprecated, please use the standard "run" flask command',
fg="red",
)
run = cli.get_command(ctx, "run")
ctx.forward(run)
@click.command(name="generate_password_hash")
def password_hash():
"""Get password from user and hash it without printing it in clear text."""
password = getpass.getpass(prompt="Password: ")
print(generate_password_hash(password))
@click.command()
@click.argument(
"config_file",
type=click.Choice(
[
"ihatemoney.cfg",
"apache-vhost.conf",
"gunicorn.conf.py",
"supervisord.conf",
"nginx.conf",
]
),
)
def generate_config(config_file):
"""Generate front-end server configuration"""
def gen_secret_key():
return "".join(
[
random.SystemRandom().choice(
"abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)"
)
for i in range(50)
]
)
env = create_jinja_env("conf-templates", strict_rendering=True)
template = env.get_template(f"{config_file}.j2")
bin_path = os.path.dirname(sys.executable)
pkg_path = os.path.abspath(os.path.dirname(__file__))
print(
template.render(
pkg_path=pkg_path,
bin_path=bin_path,
sys_prefix=sys.prefix,
secret_key=gen_secret_key(),
)
)
@cli.command()
@click.argument("project_name")
def delete_project(project_name):
"""Delete a project"""
project = Project.query.get(project_name)
if project is None:
click.secho(f'Project "{project_name}" not found', fg="red")
else:
db.session.delete(project)
db.session.commit()
if __name__ == "__main__":
cli()
|
