| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- from dataclasses import dataclass
- import yaml
- @dataclass()
- class Command:
- name: str
- docstring: str | None = None
- signature: str | None = None
- def parse_command_file(filepath: str) -> str:
- content = open(filepath, 'r').read()
- lines = content.split('\n')
- commands: list[Command] = []
- idx = 0
- docs: list[str] = []
- while idx < len(lines):
- line = lines[idx]
- idx += 1
- if line.startswith('# '):
- docs.append(line[2:])
- elif line.strip().endswith('() {'):
- name = line.split()[0][:-2]
- while lines[idx].strip() != '}':
- idx += 1
- docstring, signature = None, name
- docs_dict = yaml.safe_load('\n'.join(docs).replace('@yaml', ''))
- if docs_dict is not None:
- docstring = docs_dict.get('docstring')
- arguments = docs_dict.get('arguments', None)
- if 'signature' in docs_dict:
- signature = docs_dict['signature']
- else:
- if arguments is not None:
- for param, settings in arguments.items():
- if 'required' in settings:
- signature += f' <{param}>'
- else:
- signature += f' [<{param}>]'
- command = Command(name, docstring, signature)
- commands.append(command)
- docs = []
- function_docs = ''
- for cmd in commands:
- if cmd.docstring is not None:
- function_docs += f'{cmd.signature or cmd.name} - {cmd.docstring}\n'
- return function_docs
- if __name__ == '__main__':
- import sys
- if len(sys.argv) < 2:
- print('Usage: python parse_commands.py <file>')
- sys.exit(1)
- filepath = sys.argv[1]
- filepaths = filepath.split(',')
- for filepath in filepaths:
- docs = parse_command_file(filepath)
- print(docs)
|
