import pprint
import subprocess
import os
import os.path
from flask import current_app
from flask.ext.script import Manager
from flask.ext.script.commands import Clean, ShowUrls
os.environ['ENNSTATUS_SETTINGS'] = '../test/config.py'
from ennstatus import create_app
manager = Manager(create_app())
@manager.command
def dumpconfig():
"""Show config."""
pprint.pprint(current_app.config)
def checkers(checker):
def get_files(root, files):
for f in files:
if os.path.splitext(f)[1] == '.py':
yield os.path.join(root, f)
for root, dirs, files in os.walk('./'):
for f in get_files(root, files):
subprocess.call([checker, f])
@manager.command
def pep8():
"""Check all py files for pep8 compat."""
checkers('pep8')
@manager.command
def pep257():
"""Check all py files for pep257 compat."""
checkers('pep257')
manager.add_command('clean', Clean())
manager.add_command('urls', ShowUrls())
if __name__ == "__main__":
manager.run()