mirror of
https://github.com/spiral-project/ihatemoney.git
synced 2025-05-06 05:01:48 +02:00
91 lines
2.5 KiB
Python
Executable file
91 lines
2.5 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import os
|
|
import random
|
|
import sys
|
|
import getpass
|
|
|
|
from flask_script import Manager, Command, Option
|
|
from flask_migrate import Migrate, MigrateCommand
|
|
from werkzeug.security import generate_password_hash
|
|
|
|
from ihatemoney.run import create_app
|
|
from ihatemoney.models import db, Project
|
|
from ihatemoney.utils import create_jinja_env
|
|
|
|
|
|
class GeneratePasswordHash(Command):
|
|
|
|
"""Get password from user and hash it without printing it in clear text."""
|
|
|
|
def run(self):
|
|
password = getpass.getpass(prompt='Password: ')
|
|
print(generate_password_hash(password))
|
|
|
|
|
|
class GenerateConfig(Command):
|
|
def get_options(self):
|
|
return [
|
|
Option('config_file', choices=[
|
|
'ihatemoney.cfg',
|
|
'apache-vhost.conf',
|
|
'gunicorn.conf.py',
|
|
'supervisord.conf',
|
|
'nginx.conf',
|
|
]),
|
|
]
|
|
|
|
@staticmethod
|
|
def gen_secret_key():
|
|
return ''.join([
|
|
random.SystemRandom().choice(
|
|
'abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)')
|
|
for i in range(50)])
|
|
|
|
def run(self, config_file):
|
|
env = create_jinja_env('conf-templates', strict_rendering=True)
|
|
template = env.get_template('%s.j2' % config_file)
|
|
|
|
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=self.gen_secret_key(),
|
|
))
|
|
|
|
|
|
class DeleteProject(Command):
|
|
def run(self, project_name):
|
|
demo_project = Project.query.get(project_name)
|
|
db.session.delete(demo_project)
|
|
db.session.commit()
|
|
|
|
|
|
def main():
|
|
QUIET_COMMANDS = ('generate_password_hash', 'generate-config')
|
|
|
|
backup_stderr = sys.stderr
|
|
# Hack to divert stderr for commands generating content to stdout
|
|
# to avoid confusing the user
|
|
if len(sys.argv) > 1 and sys.argv[1] in QUIET_COMMANDS:
|
|
sys.stderr = open(os.devnull, 'w')
|
|
|
|
app = create_app()
|
|
Migrate(app, db)
|
|
|
|
# Restore stderr (among other: to be able to display help)
|
|
sys.stderr = backup_stderr
|
|
|
|
manager = Manager(app)
|
|
manager.add_command('db', MigrateCommand)
|
|
manager.add_command('generate_password_hash', GeneratePasswordHash)
|
|
manager.add_command('generate-config', GenerateConfig)
|
|
manager.add_command('delete-project', DeleteProject)
|
|
manager.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|