Index: third_party/logilab/common/ureports/text_writer.py |
diff --git a/third_party/logilab/common/ureports/text_writer.py b/third_party/logilab/common/ureports/text_writer.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..04c8f263c9a96d36cc632fbc2417e158837d1842 |
--- /dev/null |
+++ b/third_party/logilab/common/ureports/text_writer.py |
@@ -0,0 +1,140 @@ |
+# copyright 2003-2011 LOGILAB S.A. (Paris, FRANCE), all rights reserved. |
+# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr |
+# |
+# This file is part of logilab-common. |
+# |
+# logilab-common is free software: you can redistribute it and/or modify it under |
+# the terms of the GNU Lesser General Public License as published by the Free |
+# Software Foundation, either version 2.1 of the License, or (at your option) any |
+# later version. |
+# |
+# logilab-common 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 Lesser General Public License for more |
+# details. |
+# |
+# You should have received a copy of the GNU Lesser General Public License along |
+# with logilab-common. If not, see <http://www.gnu.org/licenses/>. |
+"""Text formatting drivers for ureports""" |
+__docformat__ = "restructuredtext en" |
+ |
+from logilab.common.textutils import linesep |
+from logilab.common.ureports import BaseWriter |
+ |
+ |
+TITLE_UNDERLINES = ['', '=', '-', '`', '.', '~', '^'] |
+BULLETS = ['*', '-'] |
+ |
+class TextWriter(BaseWriter): |
+ """format layouts as text |
+ (ReStructured inspiration but not totally handled yet) |
+ """ |
+ def begin_format(self, layout): |
+ super(TextWriter, self).begin_format(layout) |
+ self.list_level = 0 |
+ self.pending_urls = [] |
+ |
+ def visit_section(self, layout): |
+ """display a section as text |
+ """ |
+ self.section += 1 |
+ self.writeln() |
+ self.format_children(layout) |
+ if self.pending_urls: |
+ self.writeln() |
+ for label, url in self.pending_urls: |
+ self.writeln('.. _`%s`: %s' % (label, url)) |
+ self.pending_urls = [] |
+ self.section -= 1 |
+ self.writeln() |
+ |
+ def visit_title(self, layout): |
+ title = ''.join(list(self.compute_content(layout))) |
+ self.writeln(title) |
+ try: |
+ self.writeln(TITLE_UNDERLINES[self.section] * len(title)) |
+ except IndexError: |
+ print "FIXME TITLE TOO DEEP. TURNING TITLE INTO TEXT" |
+ |
+ def visit_paragraph(self, layout): |
+ """enter a paragraph""" |
+ self.format_children(layout) |
+ self.writeln() |
+ |
+ def visit_span(self, layout): |
+ """enter a span""" |
+ self.format_children(layout) |
+ |
+ def visit_table(self, layout): |
+ """display a table as text""" |
+ table_content = self.get_table_content(layout) |
+ # get columns width |
+ cols_width = [0]*len(table_content[0]) |
+ for row in table_content: |
+ for index in range(len(row)): |
+ col = row[index] |
+ cols_width[index] = max(cols_width[index], len(col)) |
+ if layout.klass == 'field': |
+ self.field_table(layout, table_content, cols_width) |
+ else: |
+ self.default_table(layout, table_content, cols_width) |
+ self.writeln() |
+ |
+ def default_table(self, layout, table_content, cols_width): |
+ """format a table""" |
+ cols_width = [size+1 for size in cols_width] |
+ format_strings = ' '.join(['%%-%ss'] * len(cols_width)) |
+ format_strings = format_strings % tuple(cols_width) |
+ format_strings = format_strings.split(' ') |
+ table_linesep = '\n+' + '+'.join(['-'*w for w in cols_width]) + '+\n' |
+ headsep = '\n+' + '+'.join(['='*w for w in cols_width]) + '+\n' |
+ # FIXME: layout.cheaders |
+ self.write(table_linesep) |
+ for i in range(len(table_content)): |
+ self.write('|') |
+ line = table_content[i] |
+ for j in range(len(line)): |
+ self.write(format_strings[j] % line[j]) |
+ self.write('|') |
+ if i == 0 and layout.rheaders: |
+ self.write(headsep) |
+ else: |
+ self.write(table_linesep) |
+ |
+ def field_table(self, layout, table_content, cols_width): |
+ """special case for field table""" |
+ assert layout.cols == 2 |
+ format_string = '%s%%-%ss: %%s' % (linesep, cols_width[0]) |
+ for field, value in table_content: |
+ self.write(format_string % (field, value)) |
+ |
+ |
+ def visit_list(self, layout): |
+ """display a list layout as text""" |
+ bullet = BULLETS[self.list_level % len(BULLETS)] |
+ indent = ' ' * self.list_level |
+ self.list_level += 1 |
+ for child in layout.children: |
+ self.write('%s%s%s ' % (linesep, indent, bullet)) |
+ child.accept(self) |
+ self.list_level -= 1 |
+ |
+ def visit_link(self, layout): |
+ """add a hyperlink""" |
+ if layout.label != layout.url: |
+ self.write('`%s`_' % layout.label) |
+ self.pending_urls.append( (layout.label, layout.url) ) |
+ else: |
+ self.write(layout.url) |
+ |
+ def visit_verbatimtext(self, layout): |
+ """display a verbatim layout as text (so difficult ;) |
+ """ |
+ self.writeln('::\n') |
+ for line in layout.data.splitlines(): |
+ self.writeln(' ' + line) |
+ self.writeln() |
+ |
+ def visit_text(self, layout): |
+ """add some text""" |
+ self.write(layout.data) |