Add base CLI

This commit is contained in:
Jake Howard 2016-05-18 22:10:19 +01:00
parent b3138ebe3e
commit 2e9e58d8f4
2 changed files with 32 additions and 0 deletions

0
yubi_lock/__init__.py Normal file
View file

32
yubi_lock/cli.py Normal file
View file

@ -0,0 +1,32 @@
import os.path
import click
import logging
FORMAT = "[%(levelname)s]: %(message)s"
logging.basicConfig(format=FORMAT, level=logging.DEBUG)
logging.getLogger("requests").setLevel(logging.WARNING)
class YubiLockCLI(click.MultiCommand):
def list_commands(self, ctx):
return ['status']
def get_command(self, ctx, name):
ns = {}
if name not in self.list_commands(ctx):
return
try:
fn = os.path.join(os.path.dirname(__file__), name + '/cli.py')
with open(fn) as f:
code = compile(f.read(), fn, 'exec')
eval(code, ns, ns)
return ns['cli']
except:
return
cli = YubiLockCLI(help='This tool\'s subcommands are loaded from a plugin folder dynamically.')
if __name__ == '__main__':
cli()