211 lines
6.8 KiB
Python
211 lines
6.8 KiB
Python
"""Managment of `*.po` files."""
|
||
|
||
import re
|
||
from typing import List
|
||
|
||
import simplelogging
|
||
|
||
log = simplelogging.get_logger()
|
||
|
||
|
||
class PoItem:
|
||
"""Translation item."""
|
||
|
||
def __init__(self, path, lineno):
|
||
"""Initializer."""
|
||
self.path = path[3:]
|
||
self.lineno_start = lineno
|
||
self.lineno_end = lineno
|
||
self.parsing_msgid = None
|
||
self.msgid = []
|
||
self.msgstr = []
|
||
self.fuzzy = False
|
||
self.warnings = []
|
||
self.inside_pull_request = False
|
||
|
||
def append_line(self, line):
|
||
"""Append a line of a `*.po` file to the item."""
|
||
self.lineno_end += 1
|
||
if line.startswith("msgid"):
|
||
self.parsing_msgid = True
|
||
self.msgid.append(line[7:-2])
|
||
elif line.startswith("msgstr"):
|
||
self.parsing_msgid = False
|
||
self.msgstr.append(line[8:-2])
|
||
elif line.startswith("#, fuzzy"):
|
||
self.fuzzy = True
|
||
elif line.startswith('"'):
|
||
if self.parsing_msgid:
|
||
self.msgid.append(line[1:-2])
|
||
elif not self.parsing_msgid is None:
|
||
self.msgstr.append(line[1:-2])
|
||
|
||
def __str__(self):
|
||
"""Return string representation."""
|
||
return (
|
||
f" - {self.msgid_full_content}\n"
|
||
f" => {self.msgstr_full_content}\n"
|
||
f" => {self.msgstr_rst2txt}\n"
|
||
)
|
||
|
||
@property
|
||
def msgid_full_content(self):
|
||
"""Full content of the msgid."""
|
||
return "".join(self.msgid)
|
||
|
||
@property
|
||
def msgstr_full_content(self):
|
||
"""Full content of the msgstr."""
|
||
return "".join(self.msgstr)
|
||
|
||
@property
|
||
def msgid_rst2txt(self):
|
||
"""Full content of the msgid (reStructuredText escaped)."""
|
||
return self.rst2txt(self.msgid_full_content)
|
||
|
||
@property
|
||
def msgstr_rst2txt(self):
|
||
"""Full content of the msgstr (reStructuredText escaped)."""
|
||
return self.rst2txt(self.msgstr_full_content)
|
||
|
||
@staticmethod
|
||
def rst2txt(text):
|
||
"""
|
||
Escape reStructuredText markup.
|
||
|
||
The text is modified to transform reStructuredText markup
|
||
in textual version. For instance:
|
||
|
||
* "::" becomes ":"
|
||
* ":class:`PoFile`" becomes "« PoFile »"
|
||
"""
|
||
text = re.sub(r"::", r":", text)
|
||
text = re.sub(r"``(.*?)``", r"« \1 »", text)
|
||
text = re.sub(r"\"(.*?)\"", r"« \1 »", text)
|
||
text = re.sub(r":pep:`(.*?)`", r"PEP \1", text)
|
||
text = re.sub(r":[a-z:]+:`(.+?)`", r"« \1 »", text)
|
||
text = re.sub(r"\*\*(.*?)\*\*", r"« \1 »", text)
|
||
text = re.sub(
|
||
r"\*(.*?)\*", r"« \1 »", text
|
||
) # TODO sauf si déjà entre «»
|
||
text = re.sub(
|
||
r"`(.*?)\s*<((?:http|https|ftp)://.*?)>`_", r"\1 (« \2 »)", text
|
||
)
|
||
text = re.sub(r"<((?:http|https|ftp)://.*?)>", r"« \1 »", text)
|
||
return text
|
||
|
||
def add_warning(self, checker_name: str, text: str) -> None:
|
||
"""Add a checker warning to the item."""
|
||
self.warnings.append(Warning(checker_name, text))
|
||
|
||
def add_error(self, checker_name: str, text: str) -> None:
|
||
"""Add a checker error to the item."""
|
||
self.warnings.append(Error(checker_name, text))
|
||
|
||
|
||
class PoFile:
|
||
"""A `*.po` file information."""
|
||
|
||
def __init__(self, path=None):
|
||
"""Initializer."""
|
||
self.content: List[PoItem] = []
|
||
self.path = path
|
||
if path:
|
||
self.parse_file(path)
|
||
|
||
def parse_file(self, path):
|
||
"""Parse a `*.po` file according to its path."""
|
||
# TODO assert path is a file, not a dir
|
||
item = None
|
||
with open(path, encoding="utf8") as f:
|
||
for lineno, line in enumerate(f):
|
||
if line.startswith("#: "):
|
||
if item:
|
||
self.content.append(item)
|
||
item = PoItem(line, lineno + 1)
|
||
elif item:
|
||
item.append_line(line)
|
||
if item:
|
||
self.content.append(item)
|
||
|
||
def __str__(self):
|
||
"""Return string representation."""
|
||
ret = f"Po file: {self.path}\n"
|
||
ret += "\n".join(str(item) for item in self.content)
|
||
return ret
|
||
|
||
def rst2txt(self):
|
||
"""Escape reStructuredText markup."""
|
||
return "\n\n".join(item.msgstr_rst2txt for item in self.content)
|
||
|
||
def display_warnings(self, pull_request_info=None):
|
||
"""Log warnings and errors, return errors and warnings lists."""
|
||
self.tag_in_pull_request(pull_request_info)
|
||
errors = []
|
||
warnings = []
|
||
for item in self.content:
|
||
if not item.inside_pull_request:
|
||
continue
|
||
prefix = f"{self.path}:{item.lineno_start:-4} %s"
|
||
log.debug(prefix, "")
|
||
for message in item.warnings:
|
||
if isinstance(message, Error):
|
||
log.error(prefix, message)
|
||
errors.append(message)
|
||
elif isinstance(message, Warning):
|
||
log.warning(prefix, message)
|
||
warnings.append(message)
|
||
return errors, warnings
|
||
|
||
def tag_in_pull_request(self, pull_request_info):
|
||
"""Tag items being part of the pull request."""
|
||
if not pull_request_info:
|
||
for item in self.content:
|
||
item.inside_pull_request = True
|
||
else:
|
||
diff = pull_request_info.diff(self.path)
|
||
for item in self.content:
|
||
item.inside_pull_request = False
|
||
for lineno_diff in self.lines_in_diff(diff):
|
||
for item in self.content:
|
||
if item.lineno_start <= lineno_diff <= item.lineno_end:
|
||
item.inside_pull_request = True
|
||
|
||
@staticmethod
|
||
def lines_in_diff(diff):
|
||
"""Yield line numbers modified in a diff (new line numbers)."""
|
||
for line in diff.splitlines():
|
||
if line.startswith("@@"):
|
||
match = re.search(r"@@\s*\-\d+,\d+\s+\+(\d+),(\d+)\s+@@", line)
|
||
if match:
|
||
line_start = int(match.group(1))
|
||
nb_lines = int(match.group(2))
|
||
# github add 3 extra lines around diff info
|
||
extra_info_lines = 3
|
||
for lineno in range(
|
||
line_start + extra_info_lines,
|
||
line_start + nb_lines - extra_info_lines,
|
||
):
|
||
yield lineno
|
||
|
||
|
||
class Message:
|
||
"""Checker message."""
|
||
|
||
def __init__(self, checker_name: str, text: str):
|
||
"""Initializer."""
|
||
self.checker_name = checker_name
|
||
self.text = text
|
||
|
||
def __str__(self):
|
||
"""Return string representation."""
|
||
return f"[{self.checker_name:^14}] {self.text}"
|
||
|
||
|
||
class Warning(Message):
|
||
"""Checker warning message."""
|
||
|
||
|
||
class Error(Message):
|
||
"""Checker error message."""
|