from flask import current_app

from flask.ext.script import Manager

from flask.ext.script.commands import Clean, ShowUrls

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 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()