def _visit_python(self, fname, api):
"""Visit a Python source."""
fin = file(fname)
buffer = fin.read()
fin.close()
messages = extract_messages(buffer)
return messages
taglib = TranslationTagLibrary(TranslationTagLibrary.COLLECT_MESSAGES)
rewriter.add_tag_library(taglib)
try:
dummy = rewriter.filter(buffer)
except RewriteError, err:
self.error('%s: %s' % (relname, str(err)))
return []
messages = taglib.messages()
# Message type #2: tr() function inside code
try:
buffer = api.parser.parse(relname, opener=api.opener,
mode=Parser.COLLECT_CODE)
except ParseError, err:
self.error('%s: %s' % (relname, str(err)))
return []
messages += extract_messages(buffer)
return messages
def _visit_python(self, fname, api):
"""Visit a Python source."""
fin = file(fname)
buffer = fin.read()
fin.close()
messages = extract_messages(buffer)
return messages
def _visit(self, fname, api):
"""Visit a file."""
basename, ext = os.path.splitext(fname)
extension = '.' + api.options['extension']
if ext == '.py':
def test_extract(self):
for s,ref in self.messages:
messages = util.extract_messages(s)
assert messages == ref
