aeidon/agents/format.py¶
Source code for module aeidon.agents.format from file aeidon/agents/format.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 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 | # -*- coding: utf-8 -*-
# Copyright (C) 2005 Osmo Salomaa
#
# 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 of the License, 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/>.
"""Changing the appearance of texts."""
import aeidon
import re
_ = aeidon.i18n._
class FormatAgent(aeidon.Delegate):
"""Changing the appearance of texts."""
_re_alphanum = re.compile(r"\w")
@aeidon.deco.export
@aeidon.deco.revertable
def add_dialogue_dashes(self, indices, doc, register=-1):
"""Add dialogue dashes to all lines of texts."""
new_texts = []
parser = self.get_parser(doc)
for index in indices:
subtitle = self.subtitles[index]
parser.set_text(subtitle.get_text(doc))
parser.set_regex(r"^[\-\–\—]\s*")
parser.replacement = ""
parser.replace_all()
parser.set_regex(r"^")
parser.replacement = r"- "
parser.replace_all()
new_texts.append(parser.get_text())
self.replace_texts(indices, doc, new_texts, register=register)
self.set_action_description(register, _("Adding dialogue dashes"))
@aeidon.deco.export
@aeidon.deco.revertable
def change_case(self, indices, doc, method, register=-1):
"""
Change the case of texts with `method`.
`method` should be "title", "capitalize", "upper" or "lower", which
correspond to the built-in Python string methods.
"""
new_texts = []
parser = self.get_parser(doc)
for index in indices:
subtitle = self.subtitles[index]
parser.set_text(subtitle.get_text(doc))
self._change_case_first(parser, method)
new_texts.append(parser.get_text())
self.replace_texts(indices, doc, new_texts, register=register)
self.set_action_description(register, _("Changing case"))
def _change_case_first(self, parser, method):
"""Change the case of the alphanumeric substring."""
match = self._re_alphanum.search(parser.text)
if match is None: return
a = match.start()
prefix = parser.text[:a]
text = getattr(parser.text[a:], method)()
parser.text = prefix + text
@aeidon.deco.export
@aeidon.deco.revertable
def italicize(self, indices, doc, register=-1):
"""Surround texts with italic markup."""
new_texts = []
markup = self.get_markup(doc)
re_italic_tag = markup.italic_tag
for index in indices:
text = self.subtitles[index].get_text(doc)
text = re_italic_tag.sub("", text)
text = markup.italicize(text)
new_texts.append(text)
self.replace_texts(indices, doc, new_texts, register=register)
self.set_action_description(register, _("Italicizing"))
@aeidon.deco.export
@aeidon.deco.revertable
def remove_dialogue_dashes(self, indices, doc, register=-1):
"""Remove dialogue dashes from all lines of texts."""
new_texts = []
parser = self.get_parser(doc)
for index in indices:
subtitle = self.subtitles[index]
parser.set_text(subtitle.get_text(doc))
parser.set_regex(r"^[\-\–\—]\s*")
parser.replacement = ""
parser.replace_all()
new_texts.append(parser.get_text())
self.replace_texts(indices, doc, new_texts, register=register)
self.set_action_description(register, _("Removing dialogue dashes"))
def _should_add_dialogue_dashes(self, indices, doc):
"""Return ``True`` if dialogue dashes should be added to texts."""
re_tag = self.get_markup_tag_regex(doc)
for index in indices:
text = self.subtitles[index].get_text(doc)
for line in text.split("\n"):
# Strip all tags from line.
# If leftover doesn't start with "-",
# dialogue dashes should be added.
if re_tag is not None:
line = re_tag.sub("", line)
if not line.startswith("-"):
return True
return False
def _should_italicize(self, indices, doc):
"""Return ``True`` if texts should be italicized."""
re_tag = self.get_markup_tag_regex(doc)
markup = self.get_markup(doc)
re_italic_tag = markup.italic_tag
for index in indices:
text = self.subtitles[index].get_text(doc)
# Remove tags from the start of the text,
# ending after all tags are removed
# or when an italic tag is found.
if re_tag is not None:
while re_tag.match(text) is not None:
if re_italic_tag.match(text) is not None:
break
text = re_tag.sub("", text, 1)
# If there is no italic tag at the start of the text,
# all texts should be italicized.
if re_italic_tag.match(text) is None:
return True
return False
@aeidon.deco.export
@aeidon.deco.revertable
def toggle_dialogue_dashes(self, indices, doc, register=-1):
"""Show or hide dialogue dashes on texts."""
if self._should_add_dialogue_dashes(indices, doc):
return self.add_dialogue_dashes(indices, doc, register=register)
return self.remove_dialogue_dashes(indices, doc, register=register)
@aeidon.deco.export
@aeidon.deco.revertable
def toggle_italicization(self, indices, doc, register=-1):
"""Add or remove italic markup surrounding texts."""
if self._should_italicize(indices, doc):
return self.italicize(indices, doc, register=register)
return self.unitalicize(indices, doc, register=register)
@aeidon.deco.export
@aeidon.deco.revertable
def unitalicize(self, indices, doc, register=-1):
"""Remove any italic markup surrounding texts."""
new_texts = []
markup = self.get_markup(doc)
re_italic_tag = markup.italic_tag
for index in indices:
text = self.subtitles[index].get_text(doc)
text = re_italic_tag.sub("", text)
new_texts.append(text)
self.replace_texts(indices, doc, new_texts, register=register)
self.set_action_description(register, _("Unitalicizing"))
|