# This file is part of Shuup.
#
# Copyright (c) 2012-2017, Shoop Commerce Ltd. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
"""
Patched version of Django's Makemessages that works with Jinja2.
Works by monkey patching django.utils.translation.trans_real.templatize
with our version.
"""
from __future__ import unicode_literals
import datetime
import os
from io import BytesIO
import babel.messages.extract
from django.core.management.commands import makemessages
from django.utils.six import StringIO
from django.utils.translation import trans_real
KEYWORDS = dict(babel.messages.extract.DEFAULT_KEYWORDS, **{
'_L': None,
'gettext_lazy': None,
'ugettext_lazy': None,
})
JINJA_EXTENSIONS = [
'jinja2.ext.with_',
'jinja2.ext.loopcontrols',
'django_jinja.builtins.extensions.CsrfExtension',
'shuup.xtheme.parsing.LayoutPartExtension',
'shuup.xtheme.parsing.PlaceholderExtension',
'shuup.xtheme.parsing.PluginExtension',
]
BABEL_OPTIONS = {
'silent': 'false',
'newstyle_gettext': 'on',
'extensions': ','.join(JINJA_EXTENSIONS),
}
COMMENT_TAG = 'Translators:'
[docs]class Command(makemessages.Command):
[docs] def add_arguments(self, parser):
super(Command, self).add_arguments(parser)
parser.add_argument(
'--no-pot-date', action='store_true', dest='no_pot_date',
default=False,
help=(
"Don't update POT-Creation-Date if it would be "
"the only change to the PO file"))
[docs] def handle(self, *args, **options):
self.no_pot_date = options.get('no_pot_date')
old_templatize = trans_real.templatize
trans_real.templatize = jinja_messages_to_python
try:
super(Command, self).handle(*args, **options)
finally:
trans_real.templatize = old_templatize
[docs] def build_potfiles(self):
"""
Build PO Template files and return paths to them.
Extends base classes version of this method by adding the
"Remove POT-Creation-Date" feature.
"""
potfiles = super(Command, self).build_potfiles()
if self.no_pot_date:
unique_potfiles = sorted(set(potfiles)) # Remove duplicates
for potfile in unique_potfiles:
_remove_pot_creation_date(potfile)
return potfiles
[docs] def write_po_file(self, potfile, locale):
"""
Writo PO file of given locale to disk.
Extends base classes version of this method by adding a feature
to not change those PO files at all that have only the
"POT-Creation-Date" header changed.
"""
basedir = os.path.join(os.path.dirname(potfile), locale, 'LC_MESSAGES')
pofile = os.path.join(basedir, '%s.po' % str(self.domain))
if self.no_pot_date:
orig_contents = _read_file(pofile)
super(Command, self).write_po_file(potfile, locale)
if self.no_pot_date:
new_contents = _read_file(pofile)
if orig_contents != new_contents and new_contents:
modified_contents = _update_pot_creation_date(new_contents)
with open(pofile, 'wb') as fp:
fp.write(modified_contents)
def _remove_pot_creation_date(filepath):
modified_lines = []
with open(filepath, 'rb') as fp:
for line in fp:
if not line.startswith(b'"POT-Creation-Date: '):
modified_lines.append(line)
with open(filepath, 'wb') as fp:
for line in modified_lines:
fp.write(line)
def _read_file(filepath):
if not os.path.exists(filepath):
return None
with open(filepath, 'rb') as fp:
return fp.read()
def _update_pot_creation_date(po_contents):
pot_line_template = '"POT-Creation-Date: {:%Y-%m-%d %H:%M}+0000\\n"'
now = datetime.datetime.utcnow()
pot_line = pot_line_template.format(now).encode('utf-8')
lines = []
for line in po_contents.splitlines():
if line.startswith(b'"PO-Revision-Date: '):
lines.append(pot_line + b'\n')
if not line.startswith(b'"POT-Creation-Date: '):
lines.append(line + b'\n')
return b''.join(lines)
[docs]def jinja_messages_to_python(src, origin=None):
"""
Convert Jinja2 file to Python preserving only messages.
"""
output = StringIO('')
output_lineno = 1
for (lineno, message, comments, context) in extract_jinja(src, origin):
for comment in comments:
output.write(('# %s %s\n' % (COMMENT_TAG, comment)))
output_lineno += 1
lines_to_add = (lineno - output_lineno)
if lines_to_add > 0: # Try to keep line numbers in sync
output.write(lines_to_add * '\n')
output_lineno += lines_to_add
output.write('gettext(%r),' % (message,))
return output.getvalue()