1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
from os import path
import codecs
from docutils.io import DocTreeInput, StringOutput
from docutils.utils import new_document
from docutils.core import Publisher
from docutils.readers.doctree import Reader as DoctreeReader
from sphinx.builders import Builder
from sphinx.util.osutil import ensuredir, os_path, copyfile
from sphinx.util.pycompat import b
from sphinx.util.console import bold, darkgreen, brown
from ditawriter import DitaWriter
from ditamapwriter import render_map
class DitaBuilder(Builder):
name = "dita"
format = 'dita'
out_suffix = '.dita'
_publisher = None
supported_image_types = ['image/png', 'image/jpeg']
def init(self):
print('DitaBuilder.init()')
def get_outdated_docs(self):
for docname in self.env.found_docs:
if docname not in self.env.all_docs:
yield docname
continue
targetname = self.env.doc2path(docname, self.outdir,
self.out_suffix)
try:
targetmtime = path.getmtime(targetname)
except Exception:
targetmtime = 0
try:
srcmtime = path.getmtime(self.env.doc2path(docname))
if srcmtime > targetmtime:
yield docname
except EnvironmentError:
# source doesn't exist anymore
pass
def prepare_writing(self, docnames):
self.writer = DitaWriter(self)
def get_target_uri(self, docname, typ=None):
return docname + self.out_suffix
def write_doc(self, docname, doctree):
self.docname = docname
destination = StringOutput(encoding='utf-8')
self.writer.write(doctree, destination)
outfilename = path.join(self.outdir, os_path(docname) + self.out_suffix)
ensuredir(path.dirname(outfilename))
try:
f = codecs.open(outfilename, 'w', 'utf-8')
try:
f.write(self.writer.output)
finally:
f.close()
except (IOError, OSError), err:
self.warn("error writing file %s: %s" % (outfilename, err))
self.post_process_images(doctree)
def copy_image_files(self):
# copy image files
if self.images:
for src in self.status_iterator(self.images, 'copying images... ',
brown, len(self.images)):
dest = path.join(self.outdir, src)
ensuredir(path.dirname(dest))
try:
copyfile(path.join(self.srcdir, src), dest)
except Exception, err:
self.warn('cannot copy image file %r: %s' %
(path.join(self.srcdir, src), err))
def finish(self):
node = self.env.get_toctree_for(self.config.master_doc, self, False)
output = render_map(self, node)
outfilename = path.join(self.outdir, 'index.ditamap')
ensuredir(path.dirname(outfilename))
try:
f = codecs.open(outfilename, 'w', 'utf-8', 'xmlcharrefreplace')
try:
f.write(output)
finally:
f.close()
except (IOError, OSError), err:
self.warn("error writing file %s: %s" % (outfilename, err))
self.copy_image_files()
|
