django-orchestra-test/orchestra/management/commands/makemessages.py

import os

from django.core.management.commands import makemessages

from orchestra.core.translations import ModelTranslation
from orchestra.utils.paths import get_site_dir


class Command(makemessages.Command):
    """ Provides database translations support """
    def handle(self, *args, **options):
        self.database_files = []
        try:
            if os.getcwd() == get_site_dir():
                self.generate_database_files()
            super(Command, self).handle(*args, **options)
        finally:
            self.remove_database_files()
    
    def get_contents(self):
        for model, fields in ModelTranslation._registry.items():
            for field in fields:
                contents = []
                for content in model.objects.values_list('id', field):
                    pk, value = content
                    contents.append(
                        (pk, "_(u'%s')" % value)
                    )
                if contents:
                    yield ('_'.join((model._meta.db_table, field)), contents)
    
    def generate_database_files(self):
        """
        Tmp files are generated because:
            1) having a nice gettext location
                # database_db_table_field.sql.py:id
            
            2) Django's makemessages will work with no modifications
        """
        for name, contents in self.get_contents():
            name = str(name)
            maximum = None
            content = {}
            for pk, value in contents:
                if not maximum or pk > maximum:
                    maximum = pk
                content[pk] = value
            tmpcontent = []
            for ix in range(maximum+1):
                tmpcontent.append(content.get(ix, ''))
            tmpcontent = '\n'.join(tmpcontent) + '\n'
            filename = 'database_%s.sql.py' % name
            self.database_files.append(filename)
            with open(filename, 'w') as tmpfile:
                tmpfile.write(tmpcontent.encode('utf-8'))
    
    def remove_database_files(self):
        for path in self.database_files:
            os.unlink(path)