mirrors/ArchiveBox
1
0
Fork 0
mirror of https://github.com/ArchiveBox/ArchiveBox.git synced 2025-05-27 13:14:24 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

colorized and better command sorting in help msg

Browse source
This commit is contained in:
Nick Sweeting 2019-04-11 06:59:14 -04:00
parent 718e25c973
commit 4ca9a0beac
3 changed files with 20 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

9
archivebox/cli/__init__.py
View file

@ -8,8 +8,11 @@ CLI_DIR = os.path.dirname(os.path.abspath(__file__))
required_attrs = ('__package__', '__command__', '__description__', 'main')
order = ('help', 'version', 'init', 'list', 'update', 'add', 'remove')
def list_subcommands():
COMMANDS = {}
COMMANDS = []
for filename in os.listdir(CLI_DIR):
if filename.startswith('archivebox_') and filename.endswith('.py'):
subcommand = filename.replace('archivebox_', '').replace('.py', '')
@ -17,9 +20,9 @@ def list_subcommands():
assert all(hasattr(module, attr) for attr in required_attrs)
assert module.__command__.split(' ')[-1] == subcommand
COMMANDS[subcommand] = module.__description__
COMMANDS.append((subcommand, module.__description__))
return COMMANDS
return dict(sorted(COMMANDS, key=lambda cmd: order.index(cmd[0]) if cmd[0] in order else 10 + len(cmd[0])))
def run_subcommand(subcommand: str, args=None):