# 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"""
from typing import Any, List, Tuple
__docformat__ = "restructuredtext en"
from logilab.common.textutils import linesep
from logilab.common.ureports import BaseWriter
from logilab.common.ureports.nodes import (
Section,
Title,
Table,
List as NodeList,
Paragraph,
Link,
VerbatimText,
Text,
)
TITLE_UNDERLINES = ["", "=", "-", "`", ".", "~", "^"]
BULLETS = ["*", "-"]
[docs]class TextWriter(BaseWriter):
"""format layouts as text
(ReStructured inspiration but not totally handled yet)
"""
[docs] def begin_format(self, layout: Any) -> None:
super(TextWriter, self).begin_format(layout)
self.list_level = 0
self.pending_urls: List[Tuple[str, str]] = []
[docs] def visit_section(self, layout: Section) -> None:
"""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(f".. _`{label}`: {url}")
self.pending_urls = []
self.section -= 1
self.writeln()
[docs] def visit_title(self, layout: Title) -> None:
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")
[docs] def visit_paragraph(self, layout: "Paragraph") -> None:
"""enter a paragraph"""
self.format_children(layout)
self.writeln()
[docs] def visit_span(self, layout):
"""enter a span"""
self.format_children(layout)
[docs] def visit_table(self, layout: Table) -> None:
"""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()
[docs] def default_table(
self, layout: Table, table_content: List[List[str]], cols_width: List[int]
) -> None:
"""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_list = 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_list[j] % line[j])
self.write("|")
if i == 0 and layout.rheaders:
self.write(headsep)
else:
self.write(table_linesep)
[docs] def field_table(
self, layout: Table, table_content: List[List[str]], cols_width: List[int]
) -> None:
"""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))
[docs] def visit_list(self, layout: NodeList) -> None:
"""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(f"{linesep}{indent}{bullet} ")
child.accept(self)
self.list_level -= 1
[docs] def visit_link(self, layout: Link) -> None:
"""add a hyperlink"""
if layout.label != layout.url:
self.write(f"`{layout.label}`_")
self.pending_urls.append((layout.label, layout.url))
else:
self.write(layout.url)
[docs] def visit_verbatimtext(self, layout: VerbatimText) -> None:
"""display a verbatim layout as text (so difficult ;)"""
self.writeln("::\n")
for line in layout.data.splitlines():
self.writeln(" " + line)
self.writeln()
[docs] def visit_text(self, layout: Text) -> None:
"""add some text"""
self.write(f"{layout.data}")