/
_mediawiki.py
93 lines (67 loc) · 2.53 KB
/
_mediawiki.py
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
# encoding: utf-8
from __future__ import absolute_import, unicode_literals
import copy
import re
import dataproperty as dp
import typepy
from mbstrdecoder import MultiByteStrDecoder
from pytablewriter import Align
from six.moves import zip
from ._text_writer import TextTableWriter
class MediaWikiTableWriter(TextTableWriter):
"""
A table writer class for MediaWiki format.
:Example:
:ref:`example-mediawiki-table-writer`
"""
FORMAT_NAME = "mediawiki"
__RE_TABLE_SEQUENCE = re.compile(r"^[\s]+[*|#]+")
@property
def format_name(self):
return self.FORMAT_NAME
@property
def support_split_write(self):
return True
def __init__(self):
super(MediaWikiTableWriter, self).__init__()
self.column_delimiter = "\n"
self.is_padding = False
self.is_write_header_separator_row = True
self.is_write_value_separator_row = True
self.is_write_opening_row = True
self.is_write_closing_row = True
self._is_remove_line_break = False
self._quoting_flags = copy.deepcopy(dp.NOT_QUOTING_FLAGS)
def _write_header(self):
if not self.is_write_header:
return
if typepy.is_not_null_string(self.table_name):
self._write_line("|+" + MultiByteStrDecoder(self.table_name).unicode_str)
super(MediaWikiTableWriter, self)._write_header()
def _write_value_row(self, value_list, value_dp_list):
self._write_row(
[
self.__modify_table_element(value, value_dp)
for value, value_dp, in zip(value_list, value_dp_list)
]
)
def _get_opening_row_item_list(self):
return ['{| class="wikitable"']
def _get_header_row_separator_item_list(self):
return ["|-"]
def _get_value_row_separator_item_list(self):
return self._get_header_row_separator_item_list()
def _get_closing_row_item_list(self):
return ["|}"]
def _get_header_format_string(self, col_dp, value_dp):
return "! {{:{:s}{:s}}}".format(
self._get_align_char(Align.CENTER), str(self._get_padding_len(col_dp, value_dp))
)
def __modify_table_element(self, value, value_dp):
if value_dp.align is Align.LEFT:
forma_stirng = "| {1:s}"
else:
forma_stirng = '| style="text-align:{0:s}"| {1:s}'
if self.__RE_TABLE_SEQUENCE.search(value) is not None:
value = "\n" + value.lstrip()
return forma_stirng.format(value_dp.align.align_string, value)