mirror of
https://github.com/ArchiveBox/ArchiveBox
synced 2024-11-30 08:00:22 +00:00
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
__package__ = 'archivebox.cli'
|
|
|
|
import os
|
|
|
|
from typing import Dict
|
|
from importlib import import_module
|
|
|
|
CLI_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# these common commands will appear sorted before any others for ease-of-use
|
|
display_first = ('help', 'version', 'init', 'info', 'list', 'update', 'add', 'remove')
|
|
|
|
# every imported command module must have these properties in order to be valid
|
|
required_attrs = ('__package__', '__command__', 'main')
|
|
|
|
# basic checks to make sure imported files are valid subcommands
|
|
is_cli_module = lambda fname: fname.startswith('archivebox_') and fname.endswith('.py')
|
|
is_valid_cli_module = lambda module, subcommand: (
|
|
all(hasattr(module, attr) for attr in required_attrs)
|
|
and module.__command__.split(' ')[-1] == subcommand
|
|
)
|
|
|
|
def list_subcommands() -> Dict[str, str]:
|
|
"""find and import all valid archivebox_<subcommand>.py files in CLI_DIR"""
|
|
|
|
COMMANDS = []
|
|
for filename in os.listdir(CLI_DIR):
|
|
if is_cli_module(filename):
|
|
subcommand = filename.replace('archivebox_', '').replace('.py', '')
|
|
module = import_module('.archivebox_{}'.format(subcommand), __package__)
|
|
assert is_valid_cli_module(module, subcommand)
|
|
COMMANDS.append((subcommand, module.__description__)) # type: ignore
|
|
globals()[subcommand] = module.main
|
|
module.main.__doc__ = module.__description__
|
|
|
|
display_order = lambda cmd: (
|
|
display_first.index(cmd[0])
|
|
if cmd[0] in display_first else
|
|
100 + len(cmd[0])
|
|
)
|
|
|
|
return dict(sorted(COMMANDS, key=display_order))
|
|
|
|
|
|
def run_subcommand(subcommand: str, args=None) -> None:
|
|
"""run a given ArchiveBox subcommand with the given list of args"""
|
|
|
|
module = import_module('.archivebox_{}'.format(subcommand), __package__)
|
|
module.main(args) # type: ignore
|
|
|
|
|
|
SUBCOMMANDS = list_subcommands()
|
|
|
|
__all__ = (
|
|
'SUBCOMMANDS',
|
|
'list_subcommands',
|
|
'run_subcommand',
|
|
*SUBCOMMANDS.keys(),
|
|
)
|