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