mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-25 05:53:00 +00:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
import attr
|
|
import click
|
|
|
|
from ..config import Config
|
|
|
|
|
|
class KiwiCLI(click.MultiCommand):
|
|
"""Command Line Interface spread over multiple files in this directory"""
|
|
|
|
def list_commands(self, ctx):
|
|
"""list all the commands defined by cmd_*.py files in this directory"""
|
|
|
|
return (
|
|
filename[4:-3]
|
|
for filename in os.listdir(os.path.abspath(os.path.dirname(__file__)))
|
|
if filename.startswith("cmd_") and filename.endswith(".py")
|
|
)
|
|
|
|
def get_command(self, ctx, name):
|
|
"""import and return a specific command"""
|
|
|
|
try:
|
|
mod = __import__(f"kiwi_scp.commands.cmd_{name}", None, None, ["cmd"])
|
|
except ImportError:
|
|
return
|
|
return mod.cmd
|
|
|
|
|
|
@attr.s
|
|
class KiwiCTX:
|
|
"""this class is used as the commands' shared context"""
|
|
|
|
instance: Path = attr.ib(factory=lambda: Path('.'))
|
|
|
|
@property
|
|
def config(self) -> Config:
|
|
"""shorthand: get the current configuration"""
|
|
|
|
return Config.from_instance(self.instance)
|
|
|
|
|
|
pass_kiwi_ctx = click.make_pass_decorator(KiwiCTX, ensure=True)
|