mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-22 21:03:00 +00:00
47 lines
1.1 KiB
Python
Executable file
47 lines
1.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import logging
|
|
|
|
import kiwi
|
|
from kiwi.subcommands import *
|
|
|
|
|
|
def set_verbosity(logger, handler, verbosity):
|
|
if verbosity >= 2:
|
|
log_level = logging.DEBUG
|
|
log_format = "[%(asctime)s] %(levelname)s @ %(filename)s:%(funcName)s:%(lineno)d: %(message)s"
|
|
elif verbosity >= 1:
|
|
log_level = logging.INFO
|
|
log_format = "[%(asctime)s] %(levelname)s: %(message)s"
|
|
else:
|
|
log_level = logging.WARNING
|
|
log_format = "%(levelname)s: %(message)s"
|
|
|
|
logger.setLevel(log_level)
|
|
handler.setFormatter(logging.Formatter(log_format))
|
|
|
|
|
|
def main():
|
|
commands = [
|
|
InitCommand,
|
|
ShowCommand,
|
|
LogsCommand
|
|
]
|
|
|
|
for cmd in commands:
|
|
cmd.setup()
|
|
|
|
args = kiwi.Parser().get_args()
|
|
|
|
log_handler = logging.StreamHandler()
|
|
logging.getLogger().addHandler(log_handler)
|
|
|
|
set_verbosity(logging.getLogger(), log_handler, args.verbosity)
|
|
|
|
for cmd in commands:
|
|
if cmd.command == args.command:
|
|
cmd.run()
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|