mirror of
https://github.com/ArchiveBox/ArchiveBox
synced 2024-11-10 14:44:18 +00:00
28 lines
928 B
Python
28 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
|