~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/export_pot.py

  • Committer: Patch Queue Manager
  • Date: 2014-04-09 13:36:25 UTC
  • mfrom: (6592.1.2 1303879-py27-issues)
  • Revision ID: pqm@pqm.ubuntu.com-20140409133625-s24spv3kha2w2860
(vila) Fix python-2.7.6 test failures. (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
18
18
# with Python under the Python License, which is GPL compatible.
19
19
 
20
20
"""Extract docstrings from Bazaar commands.
 
21
 
 
22
This module only handles bzrlib objects that use strings not directly wrapped
 
23
by a gettext() call. To generate a complete translation template file, this
 
24
output needs to be combined with that of xgettext or a similar command for
 
25
extracting those strings, as is done in the bzr Makefile. Sorting the output
 
26
is also left to that stage of the process.
21
27
"""
22
28
 
 
29
from __future__ import absolute_import
 
30
 
23
31
import inspect
24
32
import os
25
33
 
27
35
    commands as _mod_commands,
28
36
    errors,
29
37
    help_topics,
 
38
    option,
30
39
    plugin,
31
40
    help,
32
41
    )
62
71
    return s
63
72
 
64
73
 
65
 
_FOUND_MSGID = None # set by entry function.
66
 
 
67
 
def _poentry(outf, path, lineno, s, comment=None):
68
 
    if s in _FOUND_MSGID:
69
 
        return
70
 
    _FOUND_MSGID.add(s)
71
 
    if comment is None:
72
 
        comment = ''
73
 
    else:
74
 
        comment = "# %s\n" % comment
75
 
    mutter("Exporting msg %r at line %d in %r", s[:20], lineno, path)
76
 
    print >>outf, ('#: %s:%d\n' % (path, lineno) +
77
 
           comment+
78
 
           'msgid %s\n' % _normalize(s) +
79
 
           'msgstr ""\n')
80
 
 
81
 
def _poentry_per_paragraph(outf, path, lineno, msgid, filter=lambda x: False):
82
 
    # TODO: How to split long help?
83
 
    paragraphs = msgid.split('\n\n')
84
 
    for p in paragraphs:
85
 
        if filter(p):
86
 
            continue
87
 
        _poentry(outf, path, lineno, p)
88
 
        lineno += p.count('\n') + 2
89
 
 
90
 
_LAST_CACHE = _LAST_CACHED_SRC = None
91
 
 
92
 
def _offsets_of_literal(src):
93
 
    global _LAST_CACHE, _LAST_CACHED_SRC
94
 
    if src == _LAST_CACHED_SRC:
95
 
        return _LAST_CACHE.copy()
96
 
 
 
74
def _parse_source(source_text):
 
75
    """Get object to lineno mappings from given source_text"""
97
76
    import ast
98
 
    root = ast.parse(src)
99
 
    offsets = {}
100
 
    for node in ast.walk(root):
101
 
        if not isinstance(node, ast.Str):
102
 
            continue
103
 
        offsets[node.s] = node.lineno - node.s.count('\n')
104
 
 
105
 
    _LAST_CACHED_SRC = src
106
 
    _LAST_CACHE = offsets.copy()
107
 
    return offsets
108
 
 
109
 
def _standard_options(outf):
110
 
    from bzrlib.option import Option
111
 
    src = inspect.findsource(Option)[0]
112
 
    src = ''.join(src)
113
 
    path = 'bzrlib/option.py'
114
 
    offsets = _offsets_of_literal(src)
115
 
 
116
 
    for name in sorted(Option.OPTIONS.keys()):
117
 
        opt = Option.OPTIONS[name]
118
 
        if getattr(opt, 'hidden', False):
119
 
            continue
120
 
        if getattr(opt, 'title', None):
121
 
            lineno = offsets.get(opt.title, 9999)
122
 
            if lineno == 9999:
123
 
                note(gettext("%r is not found in bzrlib/option.py") % opt.title)
124
 
            _poentry(outf, path, lineno, opt.title,
125
 
                     'title of %r option' % name)
126
 
        if getattr(opt, 'help', None):
127
 
            lineno = offsets.get(opt.help, 9999)
128
 
            if lineno == 9999:
129
 
                note(gettext("%r is not found in bzrlib/option.py") % opt.help)
130
 
            _poentry(outf, path, lineno, opt.help,
131
 
                     'help of %r option' % name)
132
 
 
133
 
def _command_options(outf, path, cmd):
134
 
    src, default_lineno = inspect.findsource(cmd.__class__)
135
 
    offsets = _offsets_of_literal(''.join(src))
 
77
    cls_to_lineno = {}
 
78
    str_to_lineno = {}
 
79
    for node in ast.walk(ast.parse(source_text)):
 
80
        # TODO: worry about duplicates?
 
81
        if isinstance(node, ast.ClassDef):
 
82
            # TODO: worry about nesting?
 
83
            cls_to_lineno[node.name] = node.lineno
 
84
        elif isinstance(node, ast.Str):
 
85
            # Python AST gives location of string literal as the line the
 
86
            # string terminates on. It's more useful to have the line the
 
87
            # string begins on. Unfortunately, counting back newlines is
 
88
            # only an approximation as the AST is ignorant of escaping.
 
89
            str_to_lineno[node.s] = node.lineno - node.s.count('\n')
 
90
    return cls_to_lineno, str_to_lineno
 
91
 
 
92
 
 
93
class _ModuleContext(object):
 
94
    """Record of the location within a source tree"""
 
95
 
 
96
    def __init__(self, path, lineno=1, _source_info=None):
 
97
        self.path = path
 
98
        self.lineno = lineno
 
99
        if _source_info is not None:
 
100
            self._cls_to_lineno, self._str_to_lineno = _source_info
 
101
 
 
102
    @classmethod
 
103
    def from_module(cls, module):
 
104
        """Get new context from module object and parse source for linenos"""
 
105
        sourcepath = inspect.getsourcefile(module)
 
106
        # TODO: fix this to do the right thing rather than rely on cwd
 
107
        relpath = os.path.relpath(sourcepath)
 
108
        return cls(relpath,
 
109
            _source_info=_parse_source("".join(inspect.findsource(module)[0])))
 
110
 
 
111
    def from_class(self, cls):
 
112
        """Get new context with same details but lineno of class in source"""
 
113
        try:
 
114
            lineno = self._cls_to_lineno[cls.__name__]
 
115
        except (AttributeError, KeyError):
 
116
            mutter("Definition of %r not found in %r", cls, self.path)
 
117
            return self
 
118
        return self.__class__(self.path, lineno,
 
119
            (self._cls_to_lineno, self._str_to_lineno))
 
120
 
 
121
    def from_string(self, string):
 
122
        """Get new context with same details but lineno of string in source"""
 
123
        try:
 
124
            lineno = self._str_to_lineno[string]
 
125
        except (AttributeError, KeyError):
 
126
            mutter("String %r not found in %r", string[:20], self.path)
 
127
            return self
 
128
        return self.__class__(self.path, lineno,
 
129
            (self._cls_to_lineno, self._str_to_lineno))
 
130
 
 
131
 
 
132
class _PotExporter(object):
 
133
    """Write message details to output stream in .pot file format"""
 
134
 
 
135
    def __init__(self, outf, include_duplicates=False):
 
136
        self.outf = outf
 
137
        if include_duplicates:
 
138
            self._msgids = None
 
139
        else:
 
140
            self._msgids = set()
 
141
        self._module_contexts = {}
 
142
 
 
143
    def poentry(self, path, lineno, s, comment=None):
 
144
        if self._msgids is not None:
 
145
            if s in self._msgids:
 
146
                return
 
147
            self._msgids.add(s)
 
148
        if comment is None:
 
149
            comment = ''
 
150
        else:
 
151
            comment = "# %s\n" % comment
 
152
        mutter("Exporting msg %r at line %d in %r", s[:20], lineno, path)
 
153
        self.outf.write(
 
154
            "#: {path}:{lineno}\n"
 
155
            "{comment}"
 
156
            "msgid {msg}\n"
 
157
            "msgstr \"\"\n"
 
158
            "\n".format(
 
159
                path=path, lineno=lineno, comment=comment, msg=_normalize(s)))
 
160
 
 
161
    def poentry_in_context(self, context, string, comment=None):
 
162
        context = context.from_string(string)
 
163
        self.poentry(context.path, context.lineno, string, comment)
 
164
 
 
165
    def poentry_per_paragraph(self, path, lineno, msgid, include=None):
 
166
        # TODO: How to split long help?
 
167
        paragraphs = msgid.split('\n\n')
 
168
        if include is not None:
 
169
            paragraphs = filter(include, paragraphs)
 
170
        for p in paragraphs:
 
171
            self.poentry(path, lineno, p)
 
172
            lineno += p.count('\n') + 2
 
173
 
 
174
    def get_context(self, obj):
 
175
        module = inspect.getmodule(obj)
 
176
        try:
 
177
            context = self._module_contexts[module.__name__]
 
178
        except KeyError:
 
179
            context = _ModuleContext.from_module(module)
 
180
            self._module_contexts[module.__name__] = context
 
181
        if inspect.isclass(obj):
 
182
            context = context.from_class(obj)
 
183
        return context
 
184
 
 
185
 
 
186
def _write_option(exporter, context, opt, note):
 
187
    if getattr(opt, 'hidden', False):
 
188
        return   
 
189
    optname = opt.name
 
190
    if getattr(opt, 'title', None):
 
191
        exporter.poentry_in_context(context, opt.title,
 
192
            "title of {name!r} {what}".format(name=optname, what=note))
 
193
    for name, _, _, helptxt in opt.iter_switches():
 
194
        if name != optname:
 
195
            if opt.is_hidden(name):
 
196
                continue
 
197
            name = "=".join([optname, name])
 
198
        if helptxt:
 
199
            exporter.poentry_in_context(context, helptxt,
 
200
                "help of {name!r} {what}".format(name=name, what=note))
 
201
 
 
202
 
 
203
def _standard_options(exporter):
 
204
    OPTIONS = option.Option.OPTIONS
 
205
    context = exporter.get_context(option)
 
206
    for name in sorted(OPTIONS.keys()):
 
207
        opt = OPTIONS[name]
 
208
        _write_option(exporter, context.from_string(name), opt, "option")
 
209
 
 
210
 
 
211
def _command_options(exporter, context, cmd):
 
212
    note = "option of {0!r} command".format(cmd.name())
136
213
    for opt in cmd.takes_options:
137
 
        if isinstance(opt, str):
138
 
            continue
139
 
        if getattr(opt, 'hidden', False):
140
 
            continue
141
 
        name = opt.name
142
 
        if getattr(opt, 'title', None):
143
 
            lineno = offsets.get(opt.title, default_lineno)
144
 
            _poentry(outf, path, lineno, opt.title,
145
 
                     'title of %r option of %r command' % (name, cmd.name()))
146
 
        if getattr(opt, 'help', None):
147
 
            lineno = offsets.get(opt.help, default_lineno)
148
 
            _poentry(outf, path, lineno, opt.help,
149
 
                     'help of %r option of %r command' % (name, cmd.name()))
150
 
 
151
 
 
152
 
def _write_command_help(outf, cmd):
153
 
    path = inspect.getfile(cmd.__class__)
154
 
    if path.endswith('.pyc'):
155
 
        path = path[:-1]
156
 
    path = os.path.relpath(path)
157
 
    src, lineno = inspect.findsource(cmd.__class__)
158
 
    offsets = _offsets_of_literal(''.join(src))
159
 
    lineno = offsets[cmd.__doc__]
160
 
    doc = inspect.getdoc(cmd)
161
 
 
162
 
    def filter(p):
 
214
        # String values in Command option lists are for global options
 
215
        if not isinstance(opt, str):
 
216
            _write_option(exporter, context, opt, note)
 
217
 
 
218
 
 
219
def _write_command_help(exporter, cmd):
 
220
    context = exporter.get_context(cmd.__class__)
 
221
    rawdoc = cmd.__doc__
 
222
    dcontext = context.from_string(rawdoc)
 
223
    doc = inspect.cleandoc(rawdoc)
 
224
 
 
225
    def exclude_usage(p):
163
226
        # ':Usage:' has special meaning in help topics.
164
227
        # This is usage example of command and should not be translated.
165
 
        if p.splitlines()[0] == ':Usage:':
 
228
        if p.splitlines()[0] != ':Usage:':
166
229
            return True
167
230
 
168
 
    _poentry_per_paragraph(outf, path, lineno, doc, filter)
169
 
    _command_options(outf, path, cmd)
170
 
 
171
 
 
172
 
def _command_helps(outf, plugin_name=None):
 
231
    exporter.poentry_per_paragraph(dcontext.path, dcontext.lineno, doc,
 
232
        exclude_usage)
 
233
    _command_options(exporter, context, cmd)
 
234
 
 
235
 
 
236
def _command_helps(exporter, plugin_name=None):
173
237
    """Extract docstrings from path.
174
238
 
175
239
    This respects the Bazaar cmdtable/table convention and will
186
250
            # only export builtins if we are not exporting plugin commands
187
251
            continue
188
252
        note(gettext("Exporting messages from builtin command: %s"), cmd_name)
189
 
        _write_command_help(outf, command)
 
253
        _write_command_help(exporter, command)
190
254
 
191
255
    plugin_path = plugin.get_core_plugin_path()
192
256
    core_plugins = glob(plugin_path + '/*/__init__.py')
206
270
            continue
207
271
        note(gettext("Exporting messages from plugin command: {0} in {1}").format(
208
272
             cmd_name, command.plugin_name() ))
209
 
        _write_command_help(outf, command)
210
 
 
211
 
 
212
 
def _error_messages(outf):
 
273
        _write_command_help(exporter, command)
 
274
 
 
275
 
 
276
def _error_messages(exporter):
213
277
    """Extract fmt string from bzrlib.errors."""
214
 
    path = errors.__file__
215
 
    if path.endswith('.pyc'):
216
 
        path = path[:-1]
217
 
    offsets = _offsets_of_literal(open(path).read())
218
 
 
 
278
    context = exporter.get_context(errors)
219
279
    base_klass = errors.BzrError
220
280
    for name in dir(errors):
221
281
        klass = getattr(errors, name)
230
290
        fmt = getattr(klass, "_fmt", None)
231
291
        if fmt:
232
292
            note(gettext("Exporting message from error: %s"), name)
233
 
            _poentry(outf, 'bzrlib/errors.py',
234
 
                     offsets.get(fmt, 9999), fmt)
235
 
 
236
 
def _help_topics(outf):
 
293
            exporter.poentry_in_context(context, fmt)
 
294
 
 
295
 
 
296
def _help_topics(exporter):
237
297
    topic_registry = help_topics.topic_registry
238
298
    for key in topic_registry.keys():
239
299
        doc = topic_registry.get(key)
240
300
        if isinstance(doc, str):
241
 
            _poentry_per_paragraph(
242
 
                    outf,
 
301
            exporter.poentry_per_paragraph(
243
302
                    'dummy/help_topics/'+key+'/detail.txt',
244
303
                    1, doc)
245
304
        elif callable(doc): # help topics from files
246
 
            _poentry_per_paragraph(
247
 
                    outf,
 
305
            exporter.poentry_per_paragraph(
248
306
                    'en/help_topics/'+key+'.txt',
249
307
                    1, doc(key))
250
308
        summary = topic_registry.get_summary(key)
251
309
        if summary is not None:
252
 
            _poentry(outf, 'dummy/help_topics/'+key+'/summary.txt',
 
310
            exporter.poentry('dummy/help_topics/'+key+'/summary.txt',
253
311
                     1, summary)
254
312
 
255
 
def export_pot(outf, plugin=None):
256
 
    global _FOUND_MSGID
257
 
    _FOUND_MSGID = set()
 
313
 
 
314
def export_pot(outf, plugin=None, include_duplicates=False):
 
315
    exporter = _PotExporter(outf, include_duplicates)
258
316
    if plugin is None:
259
 
        _standard_options(outf)
260
 
        _command_helps(outf)
261
 
        _error_messages(outf)
262
 
        _help_topics(outf)
 
317
        _standard_options(exporter)
 
318
        _command_helps(exporter)
 
319
        _error_messages(exporter)
 
320
        _help_topics(exporter)
263
321
    else:
264
 
        _command_helps(outf, plugin)
 
322
        _command_helps(exporter, plugin)