Some refactoring in concatenate.py

This commit is contained in:
Ivan Blinkov 2018-07-09 15:35:19 +03:00
parent eefff21f6d
commit 5ec05996f4

View File

@ -14,32 +14,26 @@
# - For not http-links without anchor script logs an error and cuts them from the resulting single-page document. # - For not http-links without anchor script logs an error and cuts them from the resulting single-page document.
import codecs
import sys import sys
import re import re
import os import os
if len(sys.argv) < 2:
print("Usage: concatenate.py language_dir")
print("Example: concatenate.py ru")
sys.exit(1)
if not os.path.exists(sys.argv[1]): def concatenate(lang):
if not os.path.exists(lang):
print("Pass language_dir correctly. For example, 'ru'.") print("Pass language_dir correctly. For example, 'ru'.")
sys.exit(2) sys.exit(2)
# Configuration # Configuration
PROJ_CONFIG = 'mkdocs_' + sys.argv[1] + '.yml' PROJ_CONFIG = 'mkdocs_' + lang + '.yml'
SINGLE_PAGE = sys.argv[1] + '_single_page/index.md' SINGLE_PAGE = lang + '_single_page/index.md'
DOCS_DIR = sys.argv[1] + '/' DOCS_DIR = lang + '/'
# 1. Open mkdocs.yml file and read `pages` configuration to get an ordered list of files # 1. Open mkdocs.yml file and read `pages` configuration to get an ordered list of files
cfg_file = open(PROJ_CONFIG) with open(PROJ_CONFIG) as cfg_file:
files_to_concatenate = [] files_to_concatenate = []
for l in cfg_file: for l in cfg_file:
if('.md' in l) and ('single_page' not in l): if '.md' in l and 'single_page' not in l:
path = (l[l.index(':') + 1:]).strip(" '\n") path = (l[l.index(':') + 1:]).strip(" '\n")
files_to_concatenate.append(path) files_to_concatenate.append(path)
@ -48,7 +42,7 @@ print(files_to_concatenate)
# 2. Concatenate all of the files in the list # 2. Concatenate all of the files in the list
single_page_file = open(SINGLE_PAGE, 'w') with open(SINGLE_PAGE, 'w') as single_page_file:
first_file = True first_file = True
@ -56,7 +50,7 @@ for path in files_to_concatenate:
single_page_file.write('\n\n') single_page_file.write('\n\n')
file = open(DOCS_DIR + path) with open(DOCS_DIR + path) as f:
# function is passed into re.sub() to process links # function is passed into re.sub() to process links
def link_proc(matchObj): def link_proc(matchObj):
@ -68,20 +62,28 @@ for path in files_to_concatenate:
if sharp_pos > -1: if sharp_pos > -1:
return '[' + text + '](' + link[sharp_pos:] + ')' return '[' + text + '](' + link[sharp_pos:] + ')'
else: else:
raise RuntimeError('ERROR: Link [' + text + '](' + link + ') in file ' + path + ' has no anchor. Please provide it.') raise RuntimeError(
'ERROR: Link [' + text + '](' + link + ') in file ' + path + ' has no anchor. Please provide it.')
# return '['+text+'](#'+link.replace('/','-')+')' # return '['+text+'](#'+link.replace('/','-')+')'
for l in file: for l in f:
# Processing links in a string # Processing links in a string
l = re.sub(r'\[.+?\]\(.+?\)', link_proc, l) l = re.sub(r'\[.+?\]\(.+?\)', link_proc, l)
# Correcting headers levels # Correcting headers levels
if not first_file: if not first_file:
if(l.startswith('#')): if l.startswith('#'):
l = '#' + l l = '#' + l
else: else:
first_file = False first_file = False
single_page_file.write(l) single_page_file.write(l)
single_page_file.close()
if __name__ == '__main__':
if len(sys.argv) < 2:
print("Usage: concatenate.py language_dir")
print("Example: concatenate.py ru")
sys.exit(1)
concatenate(sys.argv[1])