|  |  |  | @ -1,7 +1,10 @@ | 
			
		
	
		
			
				
					|  |  |  |  | #!/usr/bin/env python | 
			
		
	
		
			
				
					|  |  |  |  | # -*- coding: utf-8 -*- | 
			
		
	
		
			
				
					|  |  |  |  | """Management script.""" | 
			
		
	
		
			
				
					|  |  |  |  | import os | 
			
		
	
		
			
				
					|  |  |  |  | from flask_script import Manager, Shell, Server | 
			
		
	
		
			
				
					|  |  |  |  | from subprocess import call | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | from flask_script import Manager, Shell, Server, Command | 
			
		
	
		
			
				
					|  |  |  |  | from flask_script.commands import Clean, ShowUrls | 
			
		
	
		
			
				
					|  |  |  |  | from flask_migrate import MigrateCommand | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
	
		
			
				
					|  |  |  | @ -10,7 +13,7 @@ from {{cookiecutter.app_name}}.user.models import User | 
			
		
	
		
			
				
					|  |  |  |  | from {{cookiecutter.app_name}}.settings import DevConfig, ProdConfig | 
			
		
	
		
			
				
					|  |  |  |  | from {{cookiecutter.app_name}}.database import db | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | if os.environ.get("{{cookiecutter.app_name | upper}}_ENV") == 'prod': | 
			
		
	
		
			
				
					|  |  |  |  | if os.environ.get('{{cookiecutter.app_name | upper}}_ENV') == 'prod': | 
			
		
	
		
			
				
					|  |  |  |  |     app = create_app(ProdConfig) | 
			
		
	
		
			
				
					|  |  |  |  | else: | 
			
		
	
		
			
				
					|  |  |  |  |     app = create_app(DevConfig) | 
			
		
	
	
		
			
				
					|  |  |  | @ -22,9 +25,7 @@ manager = Manager(app) | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | def _make_context(): | 
			
		
	
		
			
				
					|  |  |  |  |     """Return context dict for a shell session so you can access | 
			
		
	
		
			
				
					|  |  |  |  |     app, db, and the User model by default. | 
			
		
	
		
			
				
					|  |  |  |  |     """ | 
			
		
	
		
			
				
					|  |  |  |  |     """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} | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
	
		
			
				
					|  |  |  | @ -36,11 +37,29 @@ def test(): | 
			
		
	
		
			
				
					|  |  |  |  |     return exit_code | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | class Lint(Command): | 
			
		
	
		
			
				
					|  |  |  |  |     """Lint and check code style.""" | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  |     def run(self): | 
			
		
	
		
			
				
					|  |  |  |  |         """Run command.""" | 
			
		
	
		
			
				
					|  |  |  |  |         skip = ['requirements'] | 
			
		
	
		
			
				
					|  |  |  |  |         files = [name for name in next(os.walk('.'))[1] if not name.startswith('.') and name not in skip] | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  |         command_line = ['isort', '-rc'] + files | 
			
		
	
		
			
				
					|  |  |  |  |         print('Fixing import order: %s' % ' '.join(command_line)) | 
			
		
	
		
			
				
					|  |  |  |  |         call(command_line) | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  |         command_line = ['flake8'] + files | 
			
		
	
		
			
				
					|  |  |  |  |         print('Checking code style: %s' % ' '.join(command_line)) | 
			
		
	
		
			
				
					|  |  |  |  |         exit(call(command_line)) | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | 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('urls', ShowUrls()) | 
			
		
	
		
			
				
					|  |  |  |  | manager.add_command('clean', Clean()) | 
			
		
	
		
			
				
					|  |  |  |  | manager.add_command('lint', Lint()) | 
			
		
	
		
			
				
					|  |  |  |  | 
 | 
			
		
	
		
			
				
					|  |  |  |  | if __name__ == '__main__': | 
			
		
	
		
			
				
					|  |  |  |  |     manager.run() | 
			
		
	
	
		
			
				
					|  |  |  | 
 |