71 lines
3.1 KiB
Python
71 lines
3.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import codecs
|
|
import os
|
|
import sys
|
|
from optparse import make_option
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils._os import npath
|
|
|
|
def has_bom(fn):
|
|
with open(fn, 'rb') as f:
|
|
sample = f.read(4)
|
|
return sample[:3] == b'\xef\xbb\xbf' or \
|
|
sample.startswith(codecs.BOM_UTF16_LE) or \
|
|
sample.startswith(codecs.BOM_UTF16_BE)
|
|
|
|
def compile_messages(stderr, locale=None):
|
|
basedirs = [os.path.join('conf', 'locale'), 'locale']
|
|
if os.environ.get('DJANGO_SETTINGS_MODULE'):
|
|
from django.conf import settings
|
|
basedirs.extend(settings.LOCALE_PATHS)
|
|
|
|
# Gather existing directories.
|
|
basedirs = set(map(os.path.abspath, filter(os.path.isdir, basedirs)))
|
|
|
|
if not basedirs:
|
|
raise CommandError("This script should be run from the Django Git checkout or your project or app tree, or with the settings module specified.")
|
|
|
|
for basedir in basedirs:
|
|
if locale:
|
|
dirs = [os.path.join(basedir, l, 'LC_MESSAGES') for l in (locale if isinstance(locale, list) else [locale])]
|
|
else:
|
|
dirs = [basedir]
|
|
for ldir in dirs:
|
|
for dirpath, dirnames, filenames in os.walk(ldir):
|
|
for f in filenames:
|
|
if not f.endswith('.po'):
|
|
continue
|
|
stderr.write('processing file %s in %s\n' % (f, dirpath))
|
|
fn = os.path.join(dirpath, f)
|
|
if has_bom(fn):
|
|
raise CommandError("The %s file has a BOM (Byte Order Mark). Django only supports .po files encoded in UTF-8 and without any BOM." % fn)
|
|
pf = os.path.splitext(fn)[0]
|
|
# Store the names of the .mo and .po files in an environment
|
|
# variable, rather than doing a string replacement into the
|
|
# command, so that we can take advantage of shell quoting, to
|
|
# quote any malicious characters/escaping.
|
|
# See http://cyberelk.net/tim/articles/cmdline/ar01s02.html
|
|
os.environ['djangocompilemo'] = npath(pf + '.mo')
|
|
os.environ['djangocompilepo'] = npath(pf + '.po')
|
|
if sys.platform == 'win32': # Different shell-variable syntax
|
|
cmd = 'msgfmt --check-format -o "%djangocompilemo%" "%djangocompilepo%"'
|
|
else:
|
|
cmd = 'msgfmt --check-format -o "$djangocompilemo" "$djangocompilepo"'
|
|
os.system(cmd)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
option_list = BaseCommand.option_list + (
|
|
make_option('--locale', '-l', dest='locale', action='append',
|
|
help='locale(s) to process (e.g. de_AT). Default is to process all. Can be used multiple times, accepts a comma-separated list of locale names.'),
|
|
)
|
|
help = 'Compiles .po files to .mo files for use with builtin gettext support.'
|
|
|
|
requires_model_validation = False
|
|
can_import_settings = False
|
|
|
|
def handle(self, **options):
|
|
locale = options.get('locale')
|
|
compile_messages(self.stderr, locale=locale)
|