Add base CLI
This commit is contained in:
parent
b3138ebe3e
commit
2e9e58d8f4
2 changed files with 32 additions and 0 deletions
0
yubi_lock/__init__.py
Normal file
0
yubi_lock/__init__.py
Normal file
32
yubi_lock/cli.py
Normal file
32
yubi_lock/cli.py
Normal 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()
|
Reference in a new issue