mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2024-09-19 23:49:07 -04:00
27 lines
928 B
Python
27 lines
928 B
Python
__package__ = 'archivebox.cli'
|
|
|
|
import os
|
|
from importlib import import_module
|
|
|
|
CLI_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
required_attrs = ('__package__', '__command__', '__description__', 'main')
|
|
|
|
|
|
def list_subcommands():
|
|
COMMANDS = {}
|
|
for filename in os.listdir(CLI_DIR):
|
|
if filename.startswith('archivebox_') and filename.endswith('.py'):
|
|
subcommand = filename.replace('archivebox_', '').replace('.py', '')
|
|
module = import_module('.archivebox_{}'.format(subcommand), __package__)
|
|
|
|
assert all(hasattr(module, attr) for attr in required_attrs)
|
|
assert module.__command__.split(' ')[-1] == subcommand
|
|
COMMANDS[subcommand] = module.__description__
|
|
|
|
return COMMANDS
|
|
|
|
|
|
def run_subcommand(subcommand: str, args=None):
|
|
module = import_module('.archivebox_{}'.format(subcommand), __package__)
|
|
return module.main(args) # type: ignore
|