mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-22 15:42:02 +00:00
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import logging
|
|
import os
|
|
import subprocess
|
|
|
|
import bs4
|
|
import cssmin
|
|
import jinja2
|
|
import mkdocs.commands.build
|
|
|
|
import mdx_clickhouse
|
|
import test
|
|
import util
|
|
import website
|
|
|
|
|
|
def prepare_amp_html(lang, args, root, site_temp, main_site_dir):
|
|
src_path = root
|
|
src_index = os.path.join(src_path, 'index.html')
|
|
rel_path = os.path.relpath(src_path, site_temp)
|
|
dst_path = os.path.join(main_site_dir, rel_path, 'amp')
|
|
dst_index = os.path.join(dst_path, 'index.html')
|
|
|
|
logging.debug(f'Generating AMP version for {rel_path} ({lang})')
|
|
os.makedirs(dst_path)
|
|
with open(src_index, 'r') as f:
|
|
content = f.read()
|
|
css_in = ' '.join(website.get_css_in(args))
|
|
command = f"purifycss --min {css_in} '{src_index}'"
|
|
logging.debug(command)
|
|
inline_css = subprocess.check_output(command, shell=True).decode('utf-8')
|
|
inline_css = inline_css.replace('!important', '').replace('/*!', '/*')
|
|
inline_css = cssmin.cssmin(inline_css)
|
|
content = content.replace('CUSTOM_CSS_PLACEHOLDER', inline_css)
|
|
|
|
with open(dst_index, 'w') as f:
|
|
f.write(content)
|
|
|
|
return dst_index
|
|
|
|
|
|
def build_amp(lang, args, cfg):
|
|
# AMP docs: https://amp.dev/documentation/
|
|
logging.info(f'Building AMP version for {lang}')
|
|
with util.temp_dir() as site_temp:
|
|
extra = cfg.data['extra']
|
|
main_site_dir = cfg.data['site_dir']
|
|
extra['is_amp'] = True
|
|
cfg.load_dict({
|
|
'site_dir': site_temp,
|
|
'extra': extra
|
|
})
|
|
|
|
try:
|
|
mkdocs.commands.build.build(cfg)
|
|
except jinja2.exceptions.TemplateError:
|
|
if not args.version_prefix:
|
|
raise
|
|
mdx_clickhouse.PatchedMacrosPlugin.disabled = True
|
|
mkdocs.commands.build.build(cfg)
|
|
|
|
paths = []
|
|
for root, _, filenames in os.walk(site_temp):
|
|
if 'index.html' in filenames:
|
|
paths.append(prepare_amp_html(lang, args, root, site_temp, main_site_dir))
|
|
test.test_amp(paths, lang)
|
|
logging.info(f'Finished building AMP version for {lang}')
|
|
|
|
|
|
def html_to_amp(content):
|
|
soup = bs4.BeautifulSoup(
|
|
content,
|
|
features='html.parser'
|
|
)
|
|
|
|
for tag in soup.find_all():
|
|
if tag.attrs.get('id') == 'tostring':
|
|
tag.attrs['id'] = '_tostring'
|
|
if tag.name == 'img':
|
|
tag.name = 'amp-img'
|
|
tag.attrs['layout'] = 'responsive'
|
|
src = tag.attrs['src']
|
|
if not (src.startswith('/') or src.startswith('http')):
|
|
tag.attrs['src'] = f'../{src}'
|
|
if not tag.attrs.get('width'):
|
|
tag.attrs['width'] = '640'
|
|
if not tag.attrs.get('height'):
|
|
tag.attrs['height'] = '320'
|
|
elif tag.name == 'a':
|
|
href = tag.attrs.get('href')
|
|
if href:
|
|
if not (href.startswith('/') or href.startswith('http')):
|
|
if '#' in href:
|
|
href, anchor = href.split('#')
|
|
else:
|
|
anchor = None
|
|
href = f'../{href}amp/'
|
|
if anchor:
|
|
href = f'{href}#{anchor}'
|
|
tag.attrs['href'] = href
|
|
content = str(soup)
|
|
return website.minify_html(content)
|