Implement babel compilemessages command

Author: EnTeQuAk

django_babel/extract.py

@@ -35,18 +35,11 @@ def extract_django(fileobj, keywords, comment_tags, options):
                 pluralmatch = plural_re.match(t.contents)
                 if endbmatch:
                     if inplural:
-                        yield (
-                            lineno,
-                            'ngettext',
-                            (smart_unicode(u''.join(singular)),
-                             smart_unicode(u''.join(plural))),
-                            [])
+                        yield lineno, 'ngettext', (smart_unicode(u''.join(singular)),
+                                                   smart_unicode(u''.join(plural))), []
                     else:
-                        yield (
-                            lineno,
-                            None,
-                            smart_unicode(u''.join(singular)),
-                            [])
+                        yield lineno, None, smart_unicode(u''.join(singular)), []
+
                     intrans = False
                     inplural = False
                     singular = []

django_babel/management/commands/babel.py

@@ -28,6 +28,8 @@ def handle_label(self, command, **options):
 
         if command == 'makemessages':
             self.handle_makemessages(**options)
+        if command == 'compilemessages':
+            self.handle_compilemessages(**options)
 
     def handle_makemessages(self, **options):
         locale_paths = list(settings.LOCALE_PATHS)
@@ -70,3 +72,15 @@ def handle_makemessages(self, **options):
                       '-l', locale]
                 print cmd
                 call(cmd)
+
+    def handle_compilemessages(self, **options):
+        locale_paths = list(settings.LOCALE_PATHS)
+        domain = options.pop('domain')
+        locales = options.pop('locale')
+
+        for path in locale_paths:
+            for locale in locales:
+                po_file = os.path.join(path, locale, 'LC_MESSAGES', domain + '.po')
+                if os.path.exists(po_file):
+                    cmd = ['pybabel', 'compile', '-D', domain, '-d', path, '-l', locale]
+                    call(cmd)

setup.py

@@ -32,7 +32,6 @@ def read(*parts):
 ]
 
 
-
 class PyTest(TestCommand):
 
     def finalize_options(self):