Annikka/evennia
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed issues with transformation functions on ANSIString.

Browse source
This commit is contained in:
Jonathan Piacenti 2014-04-07 11:12:59 -05:00 • committed by Griatch
parent 7fd47eb386
commit 834b039a0d
2 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/utils/ansi.py
View file

@ -405,11 +405,13 @@ def _transform(func_name):
def wrapped(self, *args, **kwargs): def wrapped(self, *args, **kwargs):
replacement_string = _query_super(func_name)(self, *args, **kwargs) replacement_string = _query_super(func_name)(self, *args, **kwargs)
to_string = [] to_string = []
char_counter = 0
for index in range(0, len(self._raw_string)): for index in range(0, len(self._raw_string)):
if index in self._code_indexes: if index in self._code_indexes:
to_string.append(self._raw_string[index]) to_string.append(self._raw_string[index])
elif index in self._char_indexes: elif index in self._char_indexes:
to_string.append(replacement_string[index]) to_string.append(replacement_string[char_counter])
char_counter += 1
return ANSIString(''.join(to_string), decoded=True) return ANSIString(''.join(to_string), decoded=True)
return wrapped return wrapped

9
src/utils/tests.py
View file

@ -110,3 +110,12 @@ class ANSIStringTestCase(TestCase):
ANSI codes. ANSI codes.
""" """
self.assertEqual(len(ANSIString('{gTest{n')), 4) self.assertEqual(len(ANSIString('{gTest{n')), 4)
def test_capitalize(self):
"""
Make sure that capitalization works. This is the simplest of the
_transform functions.
"""
target = ANSIString('{gtest{n')
result = u'\x1b[1m\x1b[32mTest\x1b[0m'
self.checker(target.capitalize(), result, u'Test')