import re, time, os class Node: def __init__(self, name): self.name = name self.body = "" self.index = "" self.indexname = "" self.tag = "" self.nodes = [] def Sorted(self): names = [] for n in self.nodes: names += [n.name] names.sort() new_list = [] for name in names: for n in self.nodes: if n.name == name: new_list += [n] break self.nodes = new_list return self # tags group_tag = "@GROUP" function_tag = "@FUNCTION" option_tag = "@OPTION" body_tag = "@BODY" tags = [function_tag, option_tag, body_tag] end_tag = "@END" class DocInfo: def __init__(self): self.name = "" self.copyright = "" self.logo = "" class Output: def __init__(self, filename, docinfo): self.filename = filename self.docinfo = docinfo def output_name(self): return self.filename def render_node_index(self, cur): if len(cur.index): print >>self.file, self.index_node_begin(cur) for node in cur.nodes: self.render_node_index(node) if len(cur.index): print >>self.file, self.index_node_end(cur) def render_node(self, cur): if len(cur.index): print >>self.file, self.format_header(cur) print >>self.file, self.format_body(cur) for node in cur.nodes: self.render_node(node) def index_nodes(self, cur, index=""): for i in xrange(0, len(cur.nodes)): if len(index): cur.nodes[i].index = index + "." + str(i+1) else: cur.nodes[i].index = str(i+1) cur.nodes[i].indexname = cur.nodes[i].name if cur.nodes[i].tag == function_tag: cur.nodes[i].indexname = cur.nodes[i].indexname.split("(")[0].strip() self.index_nodes(cur.nodes[i], cur.nodes[i].index) def render(self, rootnode): self.index_nodes(rootnode) print >>self.file, self.render_begin() print >>self.file, self.index_begin() self.render_node_index(rootnode) print >>self.file, self.index_end() self.render_node(rootnode) print >>self.file, self.render_end() class HTMLOutput(Output): def render_begin(self): img = "" if len(self.docinfo.logo): img = '

'%self.docinfo.logo # large part of the style sheet is stolen from luas documentation return ''' Bam Manual

%s

%s ''' %(img, self.docinfo.name, self.docinfo.note, self.docinfo.copyright) def render_end(self): try: now = time.gmtime(int(os.environ['SOURCE_DATE_EPOCH'])) except KeyError: now = time.localtime() return '''

Generated at %s. ''' % (time.asctime(now)) def index_begin(self): return '

' def index_end(self): return '

' def index_node_begin(self, node): return '

%s - %s

'%(node.index,node.index,node.indexname) def index_node_end(self, node): if len(node.index) == 1: return '

' return '' def format_header(self, node): i = (len(node.index)-1)/2 + 1 header = '%s - %s'%(i,node.indexname,node.index,node.index,node.name,node.name,i) if node.tag == function_tag: header = '

' + header return header def format_body(self, node): body = node.body body = re.sub('\^(?P[^\^]+)\^', '\g', body) body = re.sub('\[(?P[^\]]+)\]', '\g', body) body = re.sub('{{{{', '

', body)
		body = re.sub('}}}}', '

', body) body = re.sub('!IMG (?P.+)', '

', body) body = re.sub('\\\\t', ' ', body) body = re.sub('\n\n', '

', body) body = '

' + body + '

\n' body += '\n\n' % (node.indexname) return body def ParseTextFile(rootnode, filename, addbr=False): group = rootnode for line in file(filename): if group_tag in line: group_name = line.split(group_tag)[-1].split(end_tag)[0].strip() group = Node(group_name) rootnode.nodes += [group] else: if addbr: group.body += line.strip() + "
\n" else: group.body += line.strip() + "\n" return rootnode def ParseFile(rootnode, filename): # 0 = scaning for start tag # 1 = scaning for end tag, # 2 = outputting function decl state = 0 group = rootnode for line in file(filename): if state == 0: if group_tag in line: group_name = line.split(group_tag)[-1].split(end_tag)[0].strip() group = Node(group_name) rootnode.nodes += [group] else: for t in tags: if t in line: title = line.split(t)[-1].strip() tag = t body = "" state = 1 break elif state == 1: if end_tag in line: state = 3 elif "@PAUSE" in line: state = 2 else: body += line.strip() + "\n" elif state == 2: if "@RESUME" in line: state = 1 else: if tag == function_tag: if len(title) == 0: title = line.replace("function", "").strip() title = title.replace("(", " (") node = Node(title) node.body = body node.tag = tag group.nodes += [node] state = 0 return rootnode

%s

Contents