1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# -*- coding: utf-8 -*-
#
# QBzr - Qt frontend to Bazaar commands
# Copyright (C) 2007 Lukáš Lalinský <lalinsky@gmail.com>
# Copyright (C) 2007,2009 Alexander Belchenko <bialix@ukr.net>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
"""build_pot command for setup.py"""
import glob
from distutils import log
from distutils.core import Command
from distutils.errors import DistutilsOptionError
from .en_po import regenerate_en
class build_pot(Command):
"""Distutils command build_pot"""
description = 'extract strings from python sources for translation'
# List of options:
# - long name,
# - short name (None if no short name),
# - help string.
user_options = [('build-dir=', 'd', 'Directory to put POT file'),
('output=', 'o', 'POT filename'),
('lang=', None, 'Comma-separated list of languages to update po-files'),
('no-lang', 'N', "Don't update po-files"),
('english', 'E', 'Regenerate English PO file'),
]
boolean_options = ['no-lang', 'english']
def initialize_options(self):
self.build_dir = None
self.output = None
self.lang = None
self.no_lang = False
self.english = False
def finalize_options(self):
if self.build_dir is None:
self.build_dir = 'po'
if not self.output:
self.output = (self.distribution.get_name() or 'messages')+ '.pot'
if self.lang is not None:
self.lang = [i.strip() for i in self.lang.split(',') if i.strip()]
if self.lang and self.no_lang:
raise DistutilsOptionError("You can't use options --lang=XXX and --no-lang in the same time.")
def _force_LF(self, src, dst=None):
f = open(src, 'rU')
try:
content = f.read()
finally:
f.close()
if dst is None:
dst = src
f = open(dst, 'wb')
try:
f.write(content)
finally:
f.close()
def run(self):
"""Run xgettext for project sources"""
import glob
import os
# project name based on `name` argument in setup() call
prj_name = self.distribution.get_name()
# output file
if self.build_dir != '.':
fullname = os.path.join(self.build_dir, self.output)
else:
fullname = self.output
log.info('Generate POT file: ' + fullname)
if not os.path.isdir(self.build_dir):
log.info('Make directory: ' + self.build_dir)
os.makedirs(self.build_dir)
# RJLRJL: TODO: bypassed for now, to not shred what we have
# self.spawn(['xgettext',
# '--keyword=N_',
# '-p', self.build_dir,
# '-o', self.output,
# '__init__.py',
# ] + glob.glob('lib/*.py') + glob.glob('lib/widgets/*.py')
# )
# self._force_LF(fullname)
# regenerate english PO
print('self.english is ', self.english, 'project is ', prj_name)
if self.english:
log.info('Regenerating English PO file...')
log.info(' project: {0}, build_dir {1}, output {2}'.format(prj_name, self.build_dir, self.output))
print('Regenerating English PO file...')
print(' project: {0}, build_dir {1}, output {2}'.format(prj_name, self.build_dir, self.output))
regenerate_en(prj_name, self.build_dir, self.output, self.spawn)
# search and update all po-files
if self.no_lang:
print('No language: stopping')
return
for po in glob.glob(os.path.join(self.build_dir,'*.po')):
print('\tworking on ', po)
if self.lang is not None:
po_lang = os.path.splitext(os.path.basename(po))[0]
if prj_name and po_lang.startswith(prj_name + '-'):
po_lang = po_lang[5:]
if po_lang not in self.lang:
continue
new_po = po + ".new"
cmd = "msgmerge %s %s -o %s" % (po, fullname, new_po)
self.spawn(cmd.split())
# force LF line-endings
log.info("%s --> %s" % (new_po, po))
self._force_LF(new_po, po)
os.unlink(new_po)
|