Refactor asciidoc2html.py.

This commit is contained in:
Florian Bruhin 2015-10-20 06:54:43 +02:00
parent bd189392a8
commit 71a150af22

View File

@ -32,10 +32,61 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), os.pardir))
from scripts import utils
def _get_asciidoc_cmd(args):
class AsciiDoc:
"""Abstraction of an asciidoc subprocess."""
FILES = [
('FAQ.asciidoc', 'qutebrowser/html/doc/FAQ.html'),
('CHANGELOG.asciidoc', 'qutebrowser/html/doc/CHANGELOG.html'),
('doc/quickstart.asciidoc', 'qutebrowser/html/doc/quickstart.html'),
('doc/userscripts.asciidoc', 'qutebrowser/html/doc/userscripts.html'),
]
def __init__(self, args):
self._cmd = None
self._args = args
def prepare(self):
"""Get the asciidoc command and create the homedir to use."""
self._cmd = self._get_asciidoc_cmd()
def build(self):
if self._args.website:
self._build_website()
else:
self._build_docs()
def _build_docs(self):
files = self.FILES[:]
for src in glob.glob('doc/help/*.asciidoc'):
name, _ext = os.path.splitext(os.path.basename(src))
dst = 'qutebrowser/html/doc/{}.html'.format(name)
files.append((src, dst))
for src, dst in files:
self.call(src, dst)
def _build_website(self):
for root, _dirs, files in os.walk(os.getcwd()):
for filename in files:
if os.path.splitext(filename)[1] != '.asciidoc':
continue
src = os.path.join(root, filename)
parts = [self._args.all[0]]
dirname = os.path.dirname(src)
if dirname:
parts.append(os.path.relpath(os.path.dirname(src)))
parts.append(
os.extsep.join((os.path.splitext(os.path.basename(src))[0],
'html')))
dst = os.path.join(*parts)
os.makedirs(os.path.dirname(dst), exist_ok=True)
self.call(src, dst)
def _get_asciidoc_cmd(self):
"""Try to find out what commandline to use to invoke asciidoc."""
if args.asciidoc is not None:
return args.asciidoc
if self._args.asciidoc is not None:
return self._args.asciidoc
try:
subprocess.call(['asciidoc'], stdout=subprocess.DEVNULL,
@ -55,17 +106,16 @@ def _get_asciidoc_cmd(args):
raise FileNotFoundError
def call_asciidoc(args, src, dst):
def call(self, src, dst, theme=None):
"""Call asciidoc for the given files.
Args:
args: The asciidoc binary to use, as a list.
src: The source .asciidoc file.
dst: The destination .html file, or None to auto-guess.
theme: The CSS to use as a theme.
"""
print("Calling asciidoc for {}...".format(os.path.basename(src)))
args = args[:]
args = self._cmd[:]
if dst is not None:
args += ['--out-file', dst]
args.append(src)
@ -81,53 +131,27 @@ def main(colors=False):
utils.change_cwd()
utils.use_color = colors
parser = argparse.ArgumentParser()
parser.add_argument('--all', help="Build all documentation into a given "
parser.add_argument('--website', help="Build website into a given "
"directory.", nargs=1)
parser.add_argument('--asciidoc', help="Full path to python and "
"asciidoc.py. If not given, it's searched in PATH.",
nargs=2, required=False,
metavar=('PYTHON', 'ASCIIDOC'))
args = parser.parse_args()
asciidoc_files = [
('FAQ.asciidoc', 'qutebrowser/html/doc/FAQ.html'),
('CHANGELOG.asciidoc', 'qutebrowser/html/doc/CHANGELOG.html'),
('doc/quickstart.asciidoc', 'qutebrowser/html/doc/quickstart.html'),
('doc/userscripts.asciidoc', 'qutebrowser/html/doc/userscripts.html'),
]
try:
os.mkdir('qutebrowser/html/doc')
except FileExistsError:
pass
asciidoc = AsciiDoc(args)
try:
asciidoc = _get_asciidoc_cmd(args)
asciidoc.prepare()
except FileNotFoundError:
utils.print_col("Could not find asciidoc! Please install it, or use "
"the --asciidoc argument to point this script to the "
"correct python/asciidoc.py location!", 'red')
sys.exit(1)
if args.all:
for root, _dirs, files in os.walk(os.getcwd()):
for filename in files:
if os.path.splitext(filename)[1] != '.asciidoc':
continue
src = os.path.join(root, filename)
parts = [args.all[0]]
dirname = os.path.dirname(src)
if dirname:
parts.append(os.path.relpath(os.path.dirname(src)))
parts.append(
os.extsep.join((os.path.splitext(os.path.basename(src))[0],
'html')))
dst = os.path.join(*parts)
os.makedirs(os.path.dirname(dst), exist_ok=True)
call_asciidoc(asciidoc, src, dst)
else:
for src in glob.glob('doc/help/*.asciidoc'):
name, _ext = os.path.splitext(os.path.basename(src))
dst = 'qutebrowser/html/doc/{}.html'.format(name)
asciidoc_files.append((src, dst))
for src, dst in asciidoc_files:
call_asciidoc(asciidoc, src, dst)
asciidoc.build()
if __name__ == '__main__':