1 import os

     2 import sys

     3 from optparse import make_option

     4 from django.core.management.base import BaseCommand, CommandError

     5 

     6 try:

     7     set

     8 except NameError:

     9     from sets import Set as set     # For Python 2.3

    10 

    11 def compile_messages(locale=None):

    12     basedirs = [os.path.join('conf', 'locale'), 'locale']

    13     if os.environ.get('DJANGO_SETTINGS_MODULE'):

    14         from django.conf import settings

    15         basedirs.extend(settings.LOCALE_PATHS)

    16 

    17     # Gather existing directories.

    18     basedirs = set(map(os.path.abspath, filter(os.path.isdir, basedirs)))

    19 

    20     if not basedirs:

    21         raise CommandError("This script should be run from the Django SVN tree or your project or app tree, or with the settings module specified.")

    22 

    23     for basedir in basedirs:

    24         if locale:

    25             basedir = os.path.join(basedir, locale, 'LC_MESSAGES')

    26         for dirpath, dirnames, filenames in os.walk(basedir):

    27             for f in filenames:

    28                 if f.endswith('.po'):

    29                     sys.stderr.write('processing file %s in %s\n' % (f, dirpath))

    30                     pf = os.path.splitext(os.path.join(dirpath, f))[0]

    31                     # Store the names of the .mo and .po files in an environment

    32                     # variable, rather than doing a string replacement into the

    33                     # command, so that we can take advantage of shell quoting, to

    34                     # quote any malicious characters/escaping.

    35                     # See http://cyberelk.net/tim/articles/cmdline/ar01s02.html

    36                     os.environ['djangocompilemo'] = pf + '.mo'

    37                     os.environ['djangocompilepo'] = pf + '.po'

    38                     if sys.platform == 'win32': # Different shell-variable syntax

    39                         cmd = 'msgfmt --check-format -o "%djangocompilemo%" "%djangocompilepo%"'

    40                     else:

    41                         cmd = 'msgfmt --check-format -o "$djangocompilemo" "$djangocompilepo"'

    42                     os.system(cmd)

    43 

    44 

    45 class Command(BaseCommand):

    46     option_list = BaseCommand.option_list + (

    47         make_option('--locale', '-l', dest='locale',

    48             help='The locale to process. Default is to process all.'),

    49     )

    50     help = 'Compiles .po files to .mo files for use with builtin gettext support.'

    51 

    52     requires_model_validation = False

    53     can_import_settings = False

    54 

    55     def handle(self, **options):

    56         locale = options.get('locale')

    57         compile_messages(locale)