234 lines
7.1 KiB
Python
234 lines
7.1 KiB
Python
from django.db import models
|
|
from django.conf import settings
|
|
from lang.models import Language
|
|
from glob import glob
|
|
import os
|
|
import os.path
|
|
import git
|
|
from translate.storage import factory
|
|
|
|
from trans.managers import TranslationManager, UnitManager
|
|
from util import is_plural, split_plural, join_plural
|
|
|
|
class Project(models.Model):
|
|
name = models.CharField(max_length = 100)
|
|
slug = models.SlugField(db_index = True)
|
|
web = models.URLField()
|
|
mail = models.EmailField()
|
|
instructions = models.URLField()
|
|
|
|
@models.permalink
|
|
def get_absolute_url(self):
|
|
return ('trans.views.show_project', (), {'project': self.slug})
|
|
|
|
def get_path(self):
|
|
return os.path.join(settings.GIT_ROOT, self.slug)
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs):
|
|
# Create filesystem directory for storing data
|
|
p = self.get_path()
|
|
if not os.path.exists(p):
|
|
os.makedirs(p)
|
|
|
|
super(Project, self).save(*args, **kwargs)
|
|
|
|
class SubProject(models.Model):
|
|
name = models.CharField(max_length = 100)
|
|
slug = models.SlugField(db_index = True)
|
|
project = models.ForeignKey(Project)
|
|
repo = models.CharField(max_length = 200)
|
|
branch = models.CharField(max_length = 50)
|
|
filemask = models.CharField(max_length = 200)
|
|
style_choices = (('po', 'GNU Gettext'), ('ts', 'Qt TS'))
|
|
style = models.CharField(max_length = 10, choices = style_choices)
|
|
|
|
@models.permalink
|
|
def get_absolute_url(self):
|
|
return ('trans.views.show_subproject', (), {'project': self.project.slug, 'subproject': self.slug})
|
|
|
|
def __unicode__(self):
|
|
return '%s/%s' % (self.project.__unicode__(), self.name)
|
|
|
|
def get_path(self):
|
|
return os.path.join(self.project.get_path(), self.slug)
|
|
|
|
def get_repo(self):
|
|
'''
|
|
Gets Git repository object.
|
|
'''
|
|
p = self.get_path()
|
|
try:
|
|
return git.Repo(p)
|
|
except:
|
|
return git.Repo.init(p)
|
|
|
|
def configure_repo(self):
|
|
'''
|
|
Ensures repository is correctly configured and points to current remote.
|
|
'''
|
|
# Create/Open repo
|
|
repo = self.get_repo()
|
|
# Get/Create origin remote
|
|
try:
|
|
origin = repo.remotes.origin
|
|
except:
|
|
repo.git.remote('add', 'origin', self.repo)
|
|
origin = repo.remotes.origin
|
|
# Check remote source
|
|
if origin.url != self.repo:
|
|
repo.git.remote('set-url', 'origin', self.repo)
|
|
# Update
|
|
repo.git.remote('update', 'origin')
|
|
|
|
def configure_branch(self):
|
|
'''
|
|
Ensures local tracking branch exists and is checkouted.
|
|
'''
|
|
repo = self.get_repo()
|
|
try:
|
|
head = repo.heads[self.branch]
|
|
except:
|
|
repo.git.branch('--track', self.branch, 'origin/%s' % self.branch)
|
|
head = repo.heads[self.branch]
|
|
repo.git.checkout(self.branch)
|
|
|
|
def update_branch(self):
|
|
'''
|
|
Updates current branch to match remote (if possible).
|
|
'''
|
|
repo = self.get_repo()
|
|
repo.remotes.origin.pull()
|
|
try:
|
|
repo.git.merge('origin/%s' % self.branch)
|
|
except:
|
|
repo.git.merge('--abort')
|
|
|
|
def get_translation_blobs(self):
|
|
'''
|
|
Scans directory for translation blobs and returns them as list.
|
|
'''
|
|
repo = self.get_repo()
|
|
tree = repo.tree()
|
|
|
|
# Glob files
|
|
files = glob(os.path.join(self.get_path(), self.filemask))
|
|
prefix = os.path.join(self.get_path(), '')
|
|
files = [f.replace(prefix, '') for f in files]
|
|
|
|
# Get blobs for files
|
|
return [(self.get_lang_code(f), f, tree[f]) for f in files]
|
|
|
|
def create_translations(self):
|
|
'''
|
|
Loads translations from git.
|
|
'''
|
|
blobs = self.get_translation_blobs()
|
|
for code, path, blob in blobs:
|
|
Translation.objects.update_from_blob(self, code, path, blob)
|
|
|
|
def get_lang_code(self, path):
|
|
'''
|
|
Parses language code from path.
|
|
'''
|
|
parts = self.filemask.split('*')
|
|
return path[len(parts[0]):-len(parts[1])]
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.configure_repo()
|
|
self.configure_branch()
|
|
self.update_branch()
|
|
self.create_translations()
|
|
|
|
super(SubProject, self).save(*args, **kwargs)
|
|
|
|
class Translation(models.Model):
|
|
subproject = models.ForeignKey(SubProject)
|
|
language = models.ForeignKey(Language)
|
|
translated = models.FloatField(default = 0)
|
|
fuzzy = models.FloatField(default = 0)
|
|
revision = models.CharField(max_length = 40, default = '', blank = True)
|
|
filename = models.CharField(max_length = 200)
|
|
|
|
objects = TranslationManager()
|
|
|
|
@models.permalink
|
|
def get_absolute_url(self):
|
|
return ('trans.views.show_translation', (), {'project': self.subproject.slug, 'subproject': self.subproject.slug, 'lang': self.language.code})
|
|
|
|
def __unicode__(self):
|
|
return '%s@%s' % (self.language.name, self.subproject.__unicode__())
|
|
|
|
def update_from_blob(self, blob):
|
|
'''
|
|
Updates translation data from blob.
|
|
'''
|
|
# Check if we're not already up to date
|
|
if self.revision == blob.hexsha:
|
|
return
|
|
|
|
oldunits = set(self.unit_set.all().values_list('id', flat = True))
|
|
|
|
# Load po file
|
|
store = factory.getobject(os.path.join(self.subproject.get_path(), self.filename))
|
|
for unit in store.units:
|
|
newunit = Unit.objects.update_from_unit(self, unit)
|
|
try:
|
|
oldunits.remove(newunit.id)
|
|
except:
|
|
pass
|
|
|
|
# Delete not used units
|
|
Unit.objects.filter(translation = self, id__in = oldunits).delete()
|
|
|
|
# Update revision
|
|
self.revision = blob.hexsha
|
|
self.save()
|
|
|
|
|
|
class Unit(models.Model):
|
|
translation = models.ForeignKey(Translation)
|
|
location = models.TextField(default = '', blank = True)
|
|
context = models.TextField(default = '', blank = True)
|
|
flags = models.TextField(default = '', blank = True)
|
|
source = models.TextField()
|
|
target = models.TextField(default = '', blank = True)
|
|
fuzzy = models.BooleanField(default = False)
|
|
|
|
objects = UnitManager()
|
|
|
|
def update_from_unit(self, unit):
|
|
location = ', '.join(unit.getlocations())
|
|
flags = '' # FIXME
|
|
target = PLURAL_SEPARATOR.join(unit.target.strings)
|
|
fuzzy = unit.isfuzzy()
|
|
if location == self.location and flags == self.flags and target == self.target and fuzzy == self.fuzzy:
|
|
return
|
|
self.location = location
|
|
self.flags = flags
|
|
self.target = target
|
|
self.fuzzy = fuzzy
|
|
self.save()
|
|
|
|
def is_plural(self):
|
|
return is_plural(self.source)
|
|
|
|
def get_source_plurals(self):
|
|
return split_plural(self.source)
|
|
|
|
def get_target_plurals(self):
|
|
ret = split_plural(self.target)
|
|
plurals = self.translation.language.nplurals
|
|
if len(ret) == plurals:
|
|
return ret
|
|
|
|
while len(ret) < plurals:
|
|
ret.append('')
|
|
|
|
while len(ret) > plurals:
|
|
del(ret[-1])
|
|
|
|
return ret
|