diff options
Diffstat (limited to 'wikitrans/wikidump.py')
-rw-r--r-- | wikitrans/wikidump.py | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/wikitrans/wikidump.py b/wikitrans/wikidump.py index d5f651c..bc71876 100644 --- a/wikitrans/wikidump.py +++ b/wikitrans/wikidump.py @@ -1,16 +1,16 @@ # Wiki "dump" format. -*- coding: utf-8 -*- # Copyright (C) 2015-2018 Sergey Poznyakoff -# +# # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 3, or (at your option) # any later version. -# +# # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. -# +# # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. @@ -28,6 +28,7 @@ from wikitrans.wikitoken import * import json from wikitrans.wikimarkup import WikiMarkup + class DumpReferences(object): idx = 0 def __len__(self): @@ -43,13 +44,14 @@ class DumpWikiMarkup(WikiMarkup): x = DumpWikiMarkup(file="input.wiki") # Parse the input: x.parse() - # Print a JSON dump of the parse tree + # Print a JSON dump of the parse tree print(str(x)) - + """ - + indent = None references = DumpReferences() + def __init__(self, **kwarg): """Create a DumpWikiMarkup object. @@ -64,14 +66,14 @@ class DumpWikiMarkup(WikiMarkup): indent=N Basic indent offset for JSON objects. """ - n = kwarg.pop('indent', None) if n != None: self.indent = int(n) - super(DumpWikiMarkup,self).__init__(self, **kwarg) + super(DumpWikiMarkup, self).__init__(self, **kwarg) + def __str__(self): return json.dumps(self.tree, cls=WikiNodeEncoder, indent=self.indent, - separators=(',',': '), + separators=(',', ': '), sort_keys=True) |