Rooba
/
pokecrystal
mirror of https://github.com/pret/pokecrystal.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Don't bother supporting numbered lists

This commit is contained in:
Remy Oukaour 2017-12-13 16:57:29 -05:00
parent 00f28bb8f4
commit 3e5b6322e1
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tools/toc.py
View File

@ -5,7 +5,6 @@
Usage: python3 toc.py [-n] files.md... Usage: python3 toc.py [-n] files.md...
Replace a "## TOC" heading in a Markdown file with a table of contents, Replace a "## TOC" heading in a Markdown file with a table of contents,
generated from the other headings in the file. Supports multiple files. generated from the other headings in the file. Supports multiple files.
Use "-n" for numbered list items.
Headings must start with "##" signs to be detected. Headings must start with "##" signs to be detected.
""" """
@ -46,7 +45,7 @@ def get_toc_items(lines, toc_index):
anchor = name_to_anchor(name) anchor = name_to_anchor(name)
yield TocItem(name, anchor, level) yield TocItem(name, anchor, level)
def toc_string(toc_items, numeric): def toc_string(toc_items):
lines = ['## %s' % toc_name, ''] lines = ['## %s' % toc_name, '']
for name, anchor, level in toc_items: for name, anchor, level in toc_items:
padding = ' ' * level padding = ' ' * level
@ -54,7 +53,7 @@ def toc_string(toc_items, numeric):
lines.append(line) lines.append(line)
return '\n'.join(lines) + '\n' return '\n'.join(lines) + '\n'
def add_toc(filename, numeric): def add_toc(filename):
with open(filename, 'r', encoding='utf-8') as f: with open(filename, 'r', encoding='utf-8') as f:
lines = f.readlines() lines = f.readlines()
toc_index = get_toc_index(lines) toc_index = get_toc_index(lines)
@ -66,27 +65,19 @@ def add_toc(filename, numeric):
with open(filename, 'w', encoding='utf-8') as f: with open(filename, 'w', encoding='utf-8') as f:
for i, line in enumerate(lines): for i, line in enumerate(lines):
if i == toc_index: if i == toc_index:
f.write(toc_string(toc_items, numeric)) f.write(toc_string(toc_items))
else: else:
f.write(line) f.write(line)
return True # OK return True # OK
def main(): def main():
if len(sys.argv) < 2: if len(sys.argv) < 2:
print('*** ERROR: Not enough arguments') print('*** ERROR: No filenames specified')
print(__doc__) print(__doc__)
exit(1) exit(1)
del sys.argv[0] for filename in sys.argv[1:]:
numeric = False
if sys.argv[0] == '-n':
numeric = True
del sys.argv[0]
if not sys.argv:
print('*** ERROR: No filenames specified')
exit(1)
for filename in sys.argv:
print(filename) print(filename)
result = add_toc(filename, numeric) result = add_toc(filename)
if result is None: if result is None:
print('*** WARNING: No "## TOC" heading found') print('*** WARNING: No "## TOC" heading found')
elif result is False: elif result is False: