mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-22 15:42:02 +00:00
Some refactoring in concatenate.py
This commit is contained in:
parent
eefff21f6d
commit
5ec05996f4
@ -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])
|
||||||
|
Loading…
Reference in New Issue
Block a user