diff options
Diffstat (limited to 'wikitrans/wikimarkup.py')
-rw-r--r-- | wikitrans/wikimarkup.py | 35 |
1 files changed, 30 insertions, 5 deletions
diff --git a/wikitrans/wikimarkup.py b/wikitrans/wikimarkup.py index 19f69e6..1e2429f 100644 --- a/wikitrans/wikimarkup.py +++ b/wikitrans/wikimarkup.py @@ -145,6 +145,13 @@ class WikiMarkupParser(object): Public attributes: + Input: + debug_level -- debug verbosity level (0 - no debug info, 100 - excessively + copious debug messages). Default is 0. + strict -- if True, parser will throw exception upon encountering + invalid markup tag (mostly for future use) + + Output: tree -- constructed parse tree (a subclass of WikiNode) """ @@ -175,6 +182,7 @@ class WikiMarkupParser(object): tags = [ 'code', 'nowiki', 'tt', 'div', 'ref', 'references' ] debug_level = 0 + strict = False def dprint(self, lev, fmt, *argv): """If current debug level is greater than or equal to lev, print *argv @@ -187,7 +195,7 @@ class WikiMarkupParser(object): inline_delims = [ "''", "'''", "[", "]", "[[", "]]", "{{", "}}", "|" ] token_class = {} - + def _new_node(self, **kwarg): return self.token_class[kwarg['type']](self, **kwarg) @@ -374,7 +382,7 @@ class WikiMarkupParser(object): stack.append(i) # Redefine all non-matched tokens as TEXT for i in stack: - # FIXME + # FIXME: How to convert node to TEXT? self.toklist[i] = self._new_node(type='TEXT', content=str(self.toklist[i])) @@ -492,7 +500,10 @@ class WikiMarkupParser(object): flush() acc['seq'].append(self.parse_inline_delim(tok)) else: - raise UnexpectedTokenError(tok) + if self.strict: + raise UnexpectedTokenError(tok) + # FIXME: Another possible variant of handling this case is to + # convert tok to TEXT node and append it to acc['seq'] tok = self.getkn() flush() if acc['seq']: @@ -976,6 +987,13 @@ class WikiMarkup(WikiMarkupParser): media_base=URL Base URL for media files. Default is 'http://www.mediawiki.org/xml/export-0.3' + + debug_level=INT + debug verbosity level (0 - no debug info, 100 - excessively + copious debug messages). Default is 0. + strict=BOOL + Strict parsing mode. Throw exceptions on syntax errors. Default + is False. """ self.token_class = { 'NIL': WikiNode, @@ -999,14 +1017,17 @@ class WikiMarkup(WikiMarkupParser): 'LINK': WikiSeqNode, 'HDR': WikiHdrNode } - + for kw in keywords: if kw == 'file': self.file = keywords[kw] elif kw == 'filename': self.file = open(keywords[kw]) elif kw == 'text': - self.text = keywords[kw].split("\n") + if sys.version_info[0] > 2: + self.text = keywords[kw].decode('utf-8').split("\n") + else: + self.text = keywords[kw].split("\n") elif kw == 'lang': self.lang = keywords[kw] elif kw == 'html_base': @@ -1015,6 +1036,10 @@ class WikiMarkup(WikiMarkupParser): self.image_base = keywords[kw] elif kw == 'media_base': self.media_base = keywords[kw] + elif kw == 'strict': + self.strict = keywords[kw] + elif kw == 'debug_level': + self.debug_level = keywords[kw] def __del__(self): if self.file: |