20 lines
563 B
Python
20 lines
563 B
Python
import os
|
|
|
|
import click
|
|
|
|
|
|
class KiwiCLI(click.MultiCommand):
|
|
def list_commands(self, ctx):
|
|
result = []
|
|
for filename in os.listdir(os.path.abspath(os.path.dirname(__file__))):
|
|
if filename.startswith("cmd_") and filename.endswith(".py"):
|
|
result.append(filename[4:-3])
|
|
result.sort()
|
|
return result
|
|
|
|
def get_command(self, ctx, name):
|
|
try:
|
|
mod = __import__(f"kiwi_scp.commands.cmd_{name}", None, None, ["cmd"])
|
|
except ImportError:
|
|
return
|
|
return mod.cmd
|