You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.9 KiB
80 lines
2.9 KiB
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""Management script."""
|
|
import os
|
|
from glob import glob
|
|
from subprocess import call
|
|
|
|
from flask_migrate import MigrateCommand
|
|
from flask_script import Command, Manager, Option, Server, Shell
|
|
from flask_script.commands import Clean, ShowUrls
|
|
|
|
from {{cookiecutter.app_name}}.app import create_app
|
|
from {{cookiecutter.app_name}}.database import db
|
|
from {{cookiecutter.app_name}}.settings import DevConfig, ProdConfig
|
|
from {{cookiecutter.app_name}}.user.models import User
|
|
|
|
CONFIG = ProdConfig if os.environ.get('{{cookiecutter.app_name | upper}}_ENV') == 'prod' else DevConfig
|
|
HERE = os.path.abspath(os.path.dirname(__file__))
|
|
TEST_PATH = os.path.join(HERE, 'tests')
|
|
|
|
app = create_app(CONFIG) # pylint: disable=invalid-name
|
|
manager = Manager(app) # pylint: disable=invalid-name
|
|
|
|
|
|
def _make_context():
|
|
"""Return context dict for a shell session so you can access app, db, and the User model by default."""
|
|
return {'app': app, 'db': db, 'User': User}
|
|
|
|
|
|
@manager.command
|
|
def test():
|
|
"""Run the tests."""
|
|
import pytest
|
|
exit_code = pytest.main([TEST_PATH, '--verbose'])
|
|
return exit_code
|
|
|
|
|
|
class Lint(Command):
|
|
"""Lint and check code style with flake8, isort and, optionally, pylint."""
|
|
|
|
def get_options(self):
|
|
"""Command line options."""
|
|
return (
|
|
Option('-f', '--fix-imports', action='store_true', dest='fix_imports', default=False,
|
|
help='Fix imports using isort, before linting'),
|
|
Option('-p', '--pylint', action='store_true', dest='use_pylint', default=False,
|
|
help='Use pylint after flake8, for an extended strict check'),
|
|
)
|
|
|
|
def run(self, fix_imports, use_pylint): # pylint: disable=arguments-differ,method-hidden
|
|
"""Run command."""
|
|
skip = ['requirements']
|
|
root_files = glob('*.py')
|
|
root_directories = [name for name in next(os.walk('.'))[1] if not name.startswith('.')]
|
|
files_and_directories = [arg for arg in root_files + root_directories if arg not in skip]
|
|
|
|
def execute_tool(description, *args):
|
|
"""Execute a checking tool with its arguments."""
|
|
command_line = list(args) + files_and_directories
|
|
print('%s: %s' % (description, ' '.join(command_line)))
|
|
rv = call(command_line)
|
|
if rv is not 0:
|
|
exit(rv)
|
|
|
|
if fix_imports:
|
|
execute_tool('Fixing import order', 'isort', '-rc')
|
|
execute_tool('Checking code style', 'flake8')
|
|
if use_pylint:
|
|
execute_tool('Checking code style', 'pylint', '--rcfile=.pylintrc')
|
|
|
|
|
|
manager.add_command('server', Server())
|
|
manager.add_command('shell', Shell(make_context=_make_context))
|
|
manager.add_command('db', MigrateCommand)
|
|
manager.add_command('urls', ShowUrls())
|
|
manager.add_command('clean', Clean())
|
|
manager.add_command('lint', Lint())
|
|
|
|
if __name__ == '__main__':
|
|
manager.run()
|
|
|