/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Robert Collins
  • Date: 2007-08-22 22:32:23 UTC
  • mto: This revision was merged to the branch mainline in revision 2744.
  • Revision ID: robertc@robertcollins.net-20070822223223-0heetwybcuonp00g
Fix an off-by-one in check.py that leads to the progress bar showing
complete for a phase but not actually being complete; if the last file id
checked is big this is confusing. (Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
 
2
#   Authors: Robert Collins <robert.collins@canonical.com>
 
3
#            and others
 
4
#
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
#
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
#
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 
 
19
"""Configuration that affects the behaviour of Bazaar.
 
20
 
 
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
 
22
and ~/.bazaar/locations.conf, which is written to by bzr.
 
23
 
 
24
In bazaar.conf the following options may be set:
 
25
[DEFAULT]
 
26
editor=name-of-program
 
27
email=Your Name <your@email.address>
 
28
check_signatures=require|ignore|check-available(default)
 
29
create_signatures=always|never|when-required(default)
 
30
gpg_signing_command=name-of-program
 
31
log_format=name-of-format
 
32
 
 
33
in locations.conf, you specify the url of a branch and options for it.
 
34
Wildcards may be used - * and ? as normal in shell completion. Options
 
35
set in both bazaar.conf and locations.conf are overridden by the locations.conf
 
36
setting.
 
37
[/home/robertc/source]
 
38
recurse=False|True(default)
 
39
email= as above
 
40
check_signatures= as above 
 
41
create_signatures= as above.
 
42
 
 
43
explanation of options
 
44
----------------------
 
45
editor - this option sets the pop up editor to use during commits.
 
46
email - this option sets the user id bzr will use when committing.
 
47
check_signatures - this option controls whether bzr will require good gpg
 
48
                   signatures, ignore them, or check them if they are 
 
49
                   present.
 
50
create_signatures - this option controls whether bzr will always create 
 
51
                    gpg signatures, never create them, or create them if the
 
52
                    branch is configured to require them.
 
53
log_format - this option sets the default log format.  Possible values are
 
54
             long, short, line, or a plugin can register new formats.
 
55
 
 
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
 
57
 
 
58
[ALIASES]
 
59
lastlog=log --line -r-10..-1
 
60
ll=log --line -r-10..-1
 
61
h=help
 
62
up=pull
 
63
"""
 
64
 
 
65
import os
 
66
import sys
 
67
 
 
68
from bzrlib.lazy_import import lazy_import
 
69
lazy_import(globals(), """
 
70
import errno
 
71
from fnmatch import fnmatch
 
72
import re
 
73
from StringIO import StringIO
 
74
 
 
75
import bzrlib
 
76
from bzrlib import (
 
77
    errors,
 
78
    mail_client,
 
79
    osutils,
 
80
    symbol_versioning,
 
81
    trace,
 
82
    urlutils,
 
83
    win32utils,
 
84
    )
 
85
import bzrlib.util.configobj.configobj as configobj
 
86
""")
 
87
 
 
88
from bzrlib.trace import mutter, warning
 
89
 
 
90
 
 
91
CHECK_IF_POSSIBLE=0
 
92
CHECK_ALWAYS=1
 
93
CHECK_NEVER=2
 
94
 
 
95
 
 
96
SIGN_WHEN_REQUIRED=0
 
97
SIGN_ALWAYS=1
 
98
SIGN_NEVER=2
 
99
 
 
100
 
 
101
POLICY_NONE = 0
 
102
POLICY_NORECURSE = 1
 
103
POLICY_APPENDPATH = 2
 
104
 
 
105
_policy_name = {
 
106
    POLICY_NONE: None,
 
107
    POLICY_NORECURSE: 'norecurse',
 
108
    POLICY_APPENDPATH: 'appendpath',
 
109
    }
 
110
_policy_value = {
 
111
    None: POLICY_NONE,
 
112
    'none': POLICY_NONE,
 
113
    'norecurse': POLICY_NORECURSE,
 
114
    'appendpath': POLICY_APPENDPATH,
 
115
    }
 
116
 
 
117
 
 
118
STORE_LOCATION = POLICY_NONE
 
119
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
 
120
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
 
121
STORE_BRANCH = 3
 
122
STORE_GLOBAL = 4
 
123
 
 
124
 
 
125
class ConfigObj(configobj.ConfigObj):
 
126
 
 
127
    def get_bool(self, section, key):
 
128
        return self[section].as_bool(key)
 
129
 
 
130
    def get_value(self, section, name):
 
131
        # Try [] for the old DEFAULT section.
 
132
        if section == "DEFAULT":
 
133
            try:
 
134
                return self[name]
 
135
            except KeyError:
 
136
                pass
 
137
        return self[section][name]
 
138
 
 
139
 
 
140
class Config(object):
 
141
    """A configuration policy - what username, editor, gpg needs etc."""
 
142
 
 
143
    def get_editor(self):
 
144
        """Get the users pop up editor."""
 
145
        raise NotImplementedError
 
146
 
 
147
    def get_mail_client(self):
 
148
        """Get a mail client to use"""
 
149
        selected_client = self.get_user_option('mail_client')
 
150
        try:
 
151
            mail_client_class = {
 
152
                None: mail_client.DefaultMail,
 
153
                'default': mail_client.DefaultMail,
 
154
                'editor': mail_client.Editor,
 
155
                'thunderbird': mail_client.Thunderbird,
 
156
                'evolution': mail_client.Evolution,
 
157
                'mapi': mail_client.MAPIClient,
 
158
                'xdg-email': mail_client.XDGEmail,
 
159
                'kmail': mail_client.KMail,
 
160
            }[selected_client]
 
161
        except KeyError:
 
162
            raise errors.UnknownMailClient(selected_client)
 
163
        return mail_client_class(self)
 
164
 
 
165
    def _get_signature_checking(self):
 
166
        """Template method to override signature checking policy."""
 
167
 
 
168
    def _get_signing_policy(self):
 
169
        """Template method to override signature creation policy."""
 
170
 
 
171
    def _get_user_option(self, option_name):
 
172
        """Template method to provide a user option."""
 
173
        return None
 
174
 
 
175
    def get_user_option(self, option_name):
 
176
        """Get a generic option - no special process, no default."""
 
177
        return self._get_user_option(option_name)
 
178
 
 
179
    def gpg_signing_command(self):
 
180
        """What program should be used to sign signatures?"""
 
181
        result = self._gpg_signing_command()
 
182
        if result is None:
 
183
            result = "gpg"
 
184
        return result
 
185
 
 
186
    def _gpg_signing_command(self):
 
187
        """See gpg_signing_command()."""
 
188
        return None
 
189
 
 
190
    def log_format(self):
 
191
        """What log format should be used"""
 
192
        result = self._log_format()
 
193
        if result is None:
 
194
            result = "long"
 
195
        return result
 
196
 
 
197
    def _log_format(self):
 
198
        """See log_format()."""
 
199
        return None
 
200
 
 
201
    def __init__(self):
 
202
        super(Config, self).__init__()
 
203
 
 
204
    def post_commit(self):
 
205
        """An ordered list of python functions to call.
 
206
 
 
207
        Each function takes branch, rev_id as parameters.
 
208
        """
 
209
        return self._post_commit()
 
210
 
 
211
    def _post_commit(self):
 
212
        """See Config.post_commit."""
 
213
        return None
 
214
 
 
215
    def user_email(self):
 
216
        """Return just the email component of a username."""
 
217
        return extract_email_address(self.username())
 
218
 
 
219
    def username(self):
 
220
        """Return email-style username.
 
221
    
 
222
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
 
223
        
 
224
        $BZR_EMAIL can be set to override this (as well as the
 
225
        deprecated $BZREMAIL), then
 
226
        the concrete policy type is checked, and finally
 
227
        $EMAIL is examined.
 
228
        If none is found, a reasonable default is (hopefully)
 
229
        created.
 
230
    
 
231
        TODO: Check it's reasonably well-formed.
 
232
        """
 
233
        v = os.environ.get('BZR_EMAIL')
 
234
        if v:
 
235
            return v.decode(bzrlib.user_encoding)
 
236
        v = os.environ.get('BZREMAIL')
 
237
        if v:
 
238
            warning('BZREMAIL is deprecated in favor of BZR_EMAIL. Please update your configuration.')
 
239
            return v.decode(bzrlib.user_encoding)
 
240
    
 
241
        v = self._get_user_id()
 
242
        if v:
 
243
            return v
 
244
        
 
245
        v = os.environ.get('EMAIL')
 
246
        if v:
 
247
            return v.decode(bzrlib.user_encoding)
 
248
 
 
249
        name, email = _auto_user_id()
 
250
        if name:
 
251
            return '%s <%s>' % (name, email)
 
252
        else:
 
253
            return email
 
254
 
 
255
    def signature_checking(self):
 
256
        """What is the current policy for signature checking?."""
 
257
        policy = self._get_signature_checking()
 
258
        if policy is not None:
 
259
            return policy
 
260
        return CHECK_IF_POSSIBLE
 
261
 
 
262
    def signing_policy(self):
 
263
        """What is the current policy for signature checking?."""
 
264
        policy = self._get_signing_policy()
 
265
        if policy is not None:
 
266
            return policy
 
267
        return SIGN_WHEN_REQUIRED
 
268
 
 
269
    def signature_needed(self):
 
270
        """Is a signature needed when committing ?."""
 
271
        policy = self._get_signing_policy()
 
272
        if policy is None:
 
273
            policy = self._get_signature_checking()
 
274
            if policy is not None:
 
275
                warning("Please use create_signatures, not check_signatures "
 
276
                        "to set signing policy.")
 
277
            if policy == CHECK_ALWAYS:
 
278
                return True
 
279
        elif policy == SIGN_ALWAYS:
 
280
            return True
 
281
        return False
 
282
 
 
283
    def get_alias(self, value):
 
284
        return self._get_alias(value)
 
285
 
 
286
    def _get_alias(self, value):
 
287
        pass
 
288
 
 
289
    def get_nickname(self):
 
290
        return self._get_nickname()
 
291
 
 
292
    def _get_nickname(self):
 
293
        return None
 
294
 
 
295
 
 
296
class IniBasedConfig(Config):
 
297
    """A configuration policy that draws from ini files."""
 
298
 
 
299
    def _get_parser(self, file=None):
 
300
        if self._parser is not None:
 
301
            return self._parser
 
302
        if file is None:
 
303
            input = self._get_filename()
 
304
        else:
 
305
            input = file
 
306
        try:
 
307
            self._parser = ConfigObj(input, encoding='utf-8')
 
308
        except configobj.ConfigObjError, e:
 
309
            raise errors.ParseConfigError(e.errors, e.config.filename)
 
310
        return self._parser
 
311
 
 
312
    def _get_matching_sections(self):
 
313
        """Return an ordered list of (section_name, extra_path) pairs.
 
314
 
 
315
        If the section contains inherited configuration, extra_path is
 
316
        a string containing the additional path components.
 
317
        """
 
318
        section = self._get_section()
 
319
        if section is not None:
 
320
            return [(section, '')]
 
321
        else:
 
322
            return []
 
323
 
 
324
    def _get_section(self):
 
325
        """Override this to define the section used by the config."""
 
326
        return "DEFAULT"
 
327
 
 
328
    def _get_option_policy(self, section, option_name):
 
329
        """Return the policy for the given (section, option_name) pair."""
 
330
        return POLICY_NONE
 
331
 
 
332
    def _get_signature_checking(self):
 
333
        """See Config._get_signature_checking."""
 
334
        policy = self._get_user_option('check_signatures')
 
335
        if policy:
 
336
            return self._string_to_signature_policy(policy)
 
337
 
 
338
    def _get_signing_policy(self):
 
339
        """See Config._get_signing_policy"""
 
340
        policy = self._get_user_option('create_signatures')
 
341
        if policy:
 
342
            return self._string_to_signing_policy(policy)
 
343
 
 
344
    def _get_user_id(self):
 
345
        """Get the user id from the 'email' key in the current section."""
 
346
        return self._get_user_option('email')
 
347
 
 
348
    def _get_user_option(self, option_name):
 
349
        """See Config._get_user_option."""
 
350
        for (section, extra_path) in self._get_matching_sections():
 
351
            try:
 
352
                value = self._get_parser().get_value(section, option_name)
 
353
            except KeyError:
 
354
                continue
 
355
            policy = self._get_option_policy(section, option_name)
 
356
            if policy == POLICY_NONE:
 
357
                return value
 
358
            elif policy == POLICY_NORECURSE:
 
359
                # norecurse items only apply to the exact path
 
360
                if extra_path:
 
361
                    continue
 
362
                else:
 
363
                    return value
 
364
            elif policy == POLICY_APPENDPATH:
 
365
                if extra_path:
 
366
                    value = urlutils.join(value, extra_path)
 
367
                return value
 
368
            else:
 
369
                raise AssertionError('Unexpected config policy %r' % policy)
 
370
        else:
 
371
            return None
 
372
 
 
373
    def _gpg_signing_command(self):
 
374
        """See Config.gpg_signing_command."""
 
375
        return self._get_user_option('gpg_signing_command')
 
376
 
 
377
    def _log_format(self):
 
378
        """See Config.log_format."""
 
379
        return self._get_user_option('log_format')
 
380
 
 
381
    def __init__(self, get_filename):
 
382
        super(IniBasedConfig, self).__init__()
 
383
        self._get_filename = get_filename
 
384
        self._parser = None
 
385
        
 
386
    def _post_commit(self):
 
387
        """See Config.post_commit."""
 
388
        return self._get_user_option('post_commit')
 
389
 
 
390
    def _string_to_signature_policy(self, signature_string):
 
391
        """Convert a string to a signing policy."""
 
392
        if signature_string.lower() == 'check-available':
 
393
            return CHECK_IF_POSSIBLE
 
394
        if signature_string.lower() == 'ignore':
 
395
            return CHECK_NEVER
 
396
        if signature_string.lower() == 'require':
 
397
            return CHECK_ALWAYS
 
398
        raise errors.BzrError("Invalid signatures policy '%s'"
 
399
                              % signature_string)
 
400
 
 
401
    def _string_to_signing_policy(self, signature_string):
 
402
        """Convert a string to a signing policy."""
 
403
        if signature_string.lower() == 'when-required':
 
404
            return SIGN_WHEN_REQUIRED
 
405
        if signature_string.lower() == 'never':
 
406
            return SIGN_NEVER
 
407
        if signature_string.lower() == 'always':
 
408
            return SIGN_ALWAYS
 
409
        raise errors.BzrError("Invalid signing policy '%s'"
 
410
                              % signature_string)
 
411
 
 
412
    def _get_alias(self, value):
 
413
        try:
 
414
            return self._get_parser().get_value("ALIASES", 
 
415
                                                value)
 
416
        except KeyError:
 
417
            pass
 
418
 
 
419
    def _get_nickname(self):
 
420
        return self.get_user_option('nickname')
 
421
 
 
422
 
 
423
class GlobalConfig(IniBasedConfig):
 
424
    """The configuration that should be used for a specific location."""
 
425
 
 
426
    def get_editor(self):
 
427
        return self._get_user_option('editor')
 
428
 
 
429
    def __init__(self):
 
430
        super(GlobalConfig, self).__init__(config_filename)
 
431
 
 
432
    def set_user_option(self, option, value):
 
433
        """Save option and its value in the configuration."""
 
434
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
435
        # file lock on bazaar.conf.
 
436
        conf_dir = os.path.dirname(self._get_filename())
 
437
        ensure_config_dir_exists(conf_dir)
 
438
        if 'DEFAULT' not in self._get_parser():
 
439
            self._get_parser()['DEFAULT'] = {}
 
440
        self._get_parser()['DEFAULT'][option] = value
 
441
        f = open(self._get_filename(), 'wb')
 
442
        self._get_parser().write(f)
 
443
        f.close()
 
444
 
 
445
 
 
446
class LocationConfig(IniBasedConfig):
 
447
    """A configuration object that gives the policy for a location."""
 
448
 
 
449
    def __init__(self, location):
 
450
        name_generator = locations_config_filename
 
451
        if (not os.path.exists(name_generator()) and 
 
452
                os.path.exists(branches_config_filename())):
 
453
            if sys.platform == 'win32':
 
454
                warning('Please rename %s to %s' 
 
455
                         % (branches_config_filename(),
 
456
                            locations_config_filename()))
 
457
            else:
 
458
                warning('Please rename ~/.bazaar/branches.conf'
 
459
                        ' to ~/.bazaar/locations.conf')
 
460
            name_generator = branches_config_filename
 
461
        super(LocationConfig, self).__init__(name_generator)
 
462
        # local file locations are looked up by local path, rather than
 
463
        # by file url. This is because the config file is a user
 
464
        # file, and we would rather not expose the user to file urls.
 
465
        if location.startswith('file://'):
 
466
            location = urlutils.local_path_from_url(location)
 
467
        self.location = location
 
468
 
 
469
    def _get_matching_sections(self):
 
470
        """Return an ordered list of section names matching this location."""
 
471
        sections = self._get_parser()
 
472
        location_names = self.location.split('/')
 
473
        if self.location.endswith('/'):
 
474
            del location_names[-1]
 
475
        matches=[]
 
476
        for section in sections:
 
477
            # location is a local path if possible, so we need
 
478
            # to convert 'file://' urls to local paths if necessary.
 
479
            # This also avoids having file:///path be a more exact
 
480
            # match than '/path'.
 
481
            if section.startswith('file://'):
 
482
                section_path = urlutils.local_path_from_url(section)
 
483
            else:
 
484
                section_path = section
 
485
            section_names = section_path.split('/')
 
486
            if section.endswith('/'):
 
487
                del section_names[-1]
 
488
            names = zip(location_names, section_names)
 
489
            matched = True
 
490
            for name in names:
 
491
                if not fnmatch(name[0], name[1]):
 
492
                    matched = False
 
493
                    break
 
494
            if not matched:
 
495
                continue
 
496
            # so, for the common prefix they matched.
 
497
            # if section is longer, no match.
 
498
            if len(section_names) > len(location_names):
 
499
                continue
 
500
            matches.append((len(section_names), section,
 
501
                            '/'.join(location_names[len(section_names):])))
 
502
        matches.sort(reverse=True)
 
503
        sections = []
 
504
        for (length, section, extra_path) in matches:
 
505
            sections.append((section, extra_path))
 
506
            # should we stop looking for parent configs here?
 
507
            try:
 
508
                if self._get_parser()[section].as_bool('ignore_parents'):
 
509
                    break
 
510
            except KeyError:
 
511
                pass
 
512
        return sections
 
513
 
 
514
    def _get_option_policy(self, section, option_name):
 
515
        """Return the policy for the given (section, option_name) pair."""
 
516
        # check for the old 'recurse=False' flag
 
517
        try:
 
518
            recurse = self._get_parser()[section].as_bool('recurse')
 
519
        except KeyError:
 
520
            recurse = True
 
521
        if not recurse:
 
522
            return POLICY_NORECURSE
 
523
 
 
524
        policy_key = option_name + ':policy'
 
525
        try:
 
526
            policy_name = self._get_parser()[section][policy_key]
 
527
        except KeyError:
 
528
            policy_name = None
 
529
 
 
530
        return _policy_value[policy_name]
 
531
 
 
532
    def _set_option_policy(self, section, option_name, option_policy):
 
533
        """Set the policy for the given option name in the given section."""
 
534
        # The old recurse=False option affects all options in the
 
535
        # section.  To handle multiple policies in the section, we
 
536
        # need to convert it to a policy_norecurse key.
 
537
        try:
 
538
            recurse = self._get_parser()[section].as_bool('recurse')
 
539
        except KeyError:
 
540
            pass
 
541
        else:
 
542
            symbol_versioning.warn(
 
543
                'The recurse option is deprecated as of 0.14.  '
 
544
                'The section "%s" has been converted to use policies.'
 
545
                % section,
 
546
                DeprecationWarning)
 
547
            del self._get_parser()[section]['recurse']
 
548
            if not recurse:
 
549
                for key in self._get_parser()[section].keys():
 
550
                    if not key.endswith(':policy'):
 
551
                        self._get_parser()[section][key +
 
552
                                                    ':policy'] = 'norecurse'
 
553
 
 
554
        policy_key = option_name + ':policy'
 
555
        policy_name = _policy_name[option_policy]
 
556
        if policy_name is not None:
 
557
            self._get_parser()[section][policy_key] = policy_name
 
558
        else:
 
559
            if policy_key in self._get_parser()[section]:
 
560
                del self._get_parser()[section][policy_key]
 
561
 
 
562
    def set_user_option(self, option, value, store=STORE_LOCATION):
 
563
        """Save option and its value in the configuration."""
 
564
        assert store in [STORE_LOCATION,
 
565
                         STORE_LOCATION_NORECURSE,
 
566
                         STORE_LOCATION_APPENDPATH], 'bad storage policy'
 
567
        # FIXME: RBC 20051029 This should refresh the parser and also take a
 
568
        # file lock on locations.conf.
 
569
        conf_dir = os.path.dirname(self._get_filename())
 
570
        ensure_config_dir_exists(conf_dir)
 
571
        location = self.location
 
572
        if location.endswith('/'):
 
573
            location = location[:-1]
 
574
        if (not location in self._get_parser() and
 
575
            not location + '/' in self._get_parser()):
 
576
            self._get_parser()[location]={}
 
577
        elif location + '/' in self._get_parser():
 
578
            location = location + '/'
 
579
        self._get_parser()[location][option]=value
 
580
        # the allowed values of store match the config policies
 
581
        self._set_option_policy(location, option, store)
 
582
        self._get_parser().write(file(self._get_filename(), 'wb'))
 
583
 
 
584
 
 
585
class BranchConfig(Config):
 
586
    """A configuration object giving the policy for a branch."""
 
587
 
 
588
    def _get_branch_data_config(self):
 
589
        if self._branch_data_config is None:
 
590
            self._branch_data_config = TreeConfig(self.branch)
 
591
        return self._branch_data_config
 
592
 
 
593
    def _get_location_config(self):
 
594
        if self._location_config is None:
 
595
            self._location_config = LocationConfig(self.branch.base)
 
596
        return self._location_config
 
597
 
 
598
    def _get_global_config(self):
 
599
        if self._global_config is None:
 
600
            self._global_config = GlobalConfig()
 
601
        return self._global_config
 
602
 
 
603
    def _get_best_value(self, option_name):
 
604
        """This returns a user option from local, tree or global config.
 
605
 
 
606
        They are tried in that order.  Use get_safe_value if trusted values
 
607
        are necessary.
 
608
        """
 
609
        for source in self.option_sources:
 
610
            value = getattr(source(), option_name)()
 
611
            if value is not None:
 
612
                return value
 
613
        return None
 
614
 
 
615
    def _get_safe_value(self, option_name):
 
616
        """This variant of get_best_value never returns untrusted values.
 
617
        
 
618
        It does not return values from the branch data, because the branch may
 
619
        not be controlled by the user.
 
620
 
 
621
        We may wish to allow locations.conf to control whether branches are
 
622
        trusted in the future.
 
623
        """
 
624
        for source in (self._get_location_config, self._get_global_config):
 
625
            value = getattr(source(), option_name)()
 
626
            if value is not None:
 
627
                return value
 
628
        return None
 
629
 
 
630
    def _get_user_id(self):
 
631
        """Return the full user id for the branch.
 
632
    
 
633
        e.g. "John Hacker <jhacker@foo.org>"
 
634
        This is looked up in the email controlfile for the branch.
 
635
        """
 
636
        try:
 
637
            return (self.branch.control_files.get_utf8("email") 
 
638
                    .read()
 
639
                    .decode(bzrlib.user_encoding)
 
640
                    .rstrip("\r\n"))
 
641
        except errors.NoSuchFile, e:
 
642
            pass
 
643
        
 
644
        return self._get_best_value('_get_user_id')
 
645
 
 
646
    def _get_signature_checking(self):
 
647
        """See Config._get_signature_checking."""
 
648
        return self._get_best_value('_get_signature_checking')
 
649
 
 
650
    def _get_signing_policy(self):
 
651
        """See Config._get_signing_policy."""
 
652
        return self._get_best_value('_get_signing_policy')
 
653
 
 
654
    def _get_user_option(self, option_name):
 
655
        """See Config._get_user_option."""
 
656
        for source in self.option_sources:
 
657
            value = source()._get_user_option(option_name)
 
658
            if value is not None:
 
659
                return value
 
660
        return None
 
661
 
 
662
    def set_user_option(self, name, value, store=STORE_BRANCH,
 
663
        warn_masked=False):
 
664
        if store == STORE_BRANCH:
 
665
            self._get_branch_data_config().set_option(value, name)
 
666
        elif store == STORE_GLOBAL:
 
667
            self._get_global_config().set_user_option(name, value)
 
668
        else:
 
669
            self._get_location_config().set_user_option(name, value, store)
 
670
        if not warn_masked:
 
671
            return
 
672
        if store in (STORE_GLOBAL, STORE_BRANCH):
 
673
            mask_value = self._get_location_config().get_user_option(name)
 
674
            if mask_value is not None:
 
675
                trace.warning('Value "%s" is masked by "%s" from'
 
676
                              ' locations.conf', value, mask_value)
 
677
            else:
 
678
                if store == STORE_GLOBAL:
 
679
                    branch_config = self._get_branch_data_config()
 
680
                    mask_value = branch_config.get_user_option(name)
 
681
                    if mask_value is not None:
 
682
                        trace.warning('Value "%s" is masked by "%s" from'
 
683
                                      ' branch.conf', value, mask_value)
 
684
 
 
685
 
 
686
    def _gpg_signing_command(self):
 
687
        """See Config.gpg_signing_command."""
 
688
        return self._get_safe_value('_gpg_signing_command')
 
689
        
 
690
    def __init__(self, branch):
 
691
        super(BranchConfig, self).__init__()
 
692
        self._location_config = None
 
693
        self._branch_data_config = None
 
694
        self._global_config = None
 
695
        self.branch = branch
 
696
        self.option_sources = (self._get_location_config, 
 
697
                               self._get_branch_data_config,
 
698
                               self._get_global_config)
 
699
 
 
700
    def _post_commit(self):
 
701
        """See Config.post_commit."""
 
702
        return self._get_safe_value('_post_commit')
 
703
 
 
704
    def _get_nickname(self):
 
705
        value = self._get_explicit_nickname()
 
706
        if value is not None:
 
707
            return value
 
708
        return urlutils.unescape(self.branch.base.split('/')[-2])
 
709
 
 
710
    def has_explicit_nickname(self):
 
711
        """Return true if a nickname has been explicitly assigned."""
 
712
        return self._get_explicit_nickname() is not None
 
713
 
 
714
    def _get_explicit_nickname(self):
 
715
        return self._get_best_value('_get_nickname')
 
716
 
 
717
    def _log_format(self):
 
718
        """See Config.log_format."""
 
719
        return self._get_best_value('_log_format')
 
720
 
 
721
 
 
722
def ensure_config_dir_exists(path=None):
 
723
    """Make sure a configuration directory exists.
 
724
    This makes sure that the directory exists.
 
725
    On windows, since configuration directories are 2 levels deep,
 
726
    it makes sure both the directory and the parent directory exists.
 
727
    """
 
728
    if path is None:
 
729
        path = config_dir()
 
730
    if not os.path.isdir(path):
 
731
        if sys.platform == 'win32':
 
732
            parent_dir = os.path.dirname(path)
 
733
            if not os.path.isdir(parent_dir):
 
734
                mutter('creating config parent directory: %r', parent_dir)
 
735
            os.mkdir(parent_dir)
 
736
        mutter('creating config directory: %r', path)
 
737
        os.mkdir(path)
 
738
 
 
739
 
 
740
def config_dir():
 
741
    """Return per-user configuration directory.
 
742
 
 
743
    By default this is ~/.bazaar/
 
744
    
 
745
    TODO: Global option --config-dir to override this.
 
746
    """
 
747
    base = os.environ.get('BZR_HOME', None)
 
748
    if sys.platform == 'win32':
 
749
        if base is None:
 
750
            base = win32utils.get_appdata_location_unicode()
 
751
        if base is None:
 
752
            base = os.environ.get('HOME', None)
 
753
        if base is None:
 
754
            raise errors.BzrError('You must have one of BZR_HOME, APPDATA, or HOME set')
 
755
        return osutils.pathjoin(base, 'bazaar', '2.0')
 
756
    else:
 
757
        # cygwin, linux, and darwin all have a $HOME directory
 
758
        if base is None:
 
759
            base = os.path.expanduser("~")
 
760
        return osutils.pathjoin(base, ".bazaar")
 
761
 
 
762
 
 
763
def config_filename():
 
764
    """Return per-user configuration ini file filename."""
 
765
    return osutils.pathjoin(config_dir(), 'bazaar.conf')
 
766
 
 
767
 
 
768
def branches_config_filename():
 
769
    """Return per-user configuration ini file filename."""
 
770
    return osutils.pathjoin(config_dir(), 'branches.conf')
 
771
 
 
772
 
 
773
def locations_config_filename():
 
774
    """Return per-user configuration ini file filename."""
 
775
    return osutils.pathjoin(config_dir(), 'locations.conf')
 
776
 
 
777
 
 
778
def user_ignore_config_filename():
 
779
    """Return the user default ignore filename"""
 
780
    return osutils.pathjoin(config_dir(), 'ignore')
 
781
 
 
782
 
 
783
def _auto_user_id():
 
784
    """Calculate automatic user identification.
 
785
 
 
786
    Returns (realname, email).
 
787
 
 
788
    Only used when none is set in the environment or the id file.
 
789
 
 
790
    This previously used the FQDN as the default domain, but that can
 
791
    be very slow on machines where DNS is broken.  So now we simply
 
792
    use the hostname.
 
793
    """
 
794
    import socket
 
795
 
 
796
    if sys.platform == 'win32':
 
797
        name = win32utils.get_user_name_unicode()
 
798
        if name is None:
 
799
            raise errors.BzrError("Cannot autodetect user name.\n"
 
800
                                  "Please, set your name with command like:\n"
 
801
                                  'bzr whoami "Your Name <name@domain.com>"')
 
802
        host = win32utils.get_host_name_unicode()
 
803
        if host is None:
 
804
            host = socket.gethostname()
 
805
        return name, (name + '@' + host)
 
806
 
 
807
    try:
 
808
        import pwd
 
809
        uid = os.getuid()
 
810
        w = pwd.getpwuid(uid)
 
811
 
 
812
        # we try utf-8 first, because on many variants (like Linux),
 
813
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
814
        # false positives.  (many users will have their user encoding set to
 
815
        # latin-1, which cannot raise UnicodeError.)
 
816
        try:
 
817
            gecos = w.pw_gecos.decode('utf-8')
 
818
            encoding = 'utf-8'
 
819
        except UnicodeError:
 
820
            try:
 
821
                gecos = w.pw_gecos.decode(bzrlib.user_encoding)
 
822
                encoding = bzrlib.user_encoding
 
823
            except UnicodeError:
 
824
                raise errors.BzrCommandError('Unable to determine your name.  '
 
825
                   'Use "bzr whoami" to set it.')
 
826
        try:
 
827
            username = w.pw_name.decode(encoding)
 
828
        except UnicodeError:
 
829
            raise errors.BzrCommandError('Unable to determine your name.  '
 
830
                'Use "bzr whoami" to set it.')
 
831
 
 
832
        comma = gecos.find(',')
 
833
        if comma == -1:
 
834
            realname = gecos
 
835
        else:
 
836
            realname = gecos[:comma]
 
837
        if not realname:
 
838
            realname = username
 
839
 
 
840
    except ImportError:
 
841
        import getpass
 
842
        try:
 
843
            realname = username = getpass.getuser().decode(bzrlib.user_encoding)
 
844
        except UnicodeDecodeError:
 
845
            raise errors.BzrError("Can't decode username as %s." % \
 
846
                    bzrlib.user_encoding)
 
847
 
 
848
    return realname, (username + '@' + socket.gethostname())
 
849
 
 
850
 
 
851
def extract_email_address(e):
 
852
    """Return just the address part of an email string.
 
853
    
 
854
    That is just the user@domain part, nothing else. 
 
855
    This part is required to contain only ascii characters.
 
856
    If it can't be extracted, raises an error.
 
857
    
 
858
    >>> extract_email_address('Jane Tester <jane@test.com>')
 
859
    "jane@test.com"
 
860
    """
 
861
    m = re.search(r'[\w+.-]+@[\w+.-]+', e)
 
862
    if not m:
 
863
        raise errors.NoEmailInUsername(e)
 
864
    return m.group(0)
 
865
 
 
866
 
 
867
class TreeConfig(IniBasedConfig):
 
868
    """Branch configuration data associated with its contents, not location"""
 
869
    def __init__(self, branch):
 
870
        self.branch = branch
 
871
 
 
872
    def _get_parser(self, file=None):
 
873
        if file is not None:
 
874
            return IniBasedConfig._get_parser(file)
 
875
        return self._get_config()
 
876
 
 
877
    def _get_config(self):
 
878
        try:
 
879
            obj = ConfigObj(self.branch.control_files.get('branch.conf'),
 
880
                            encoding='utf-8')
 
881
        except errors.NoSuchFile:
 
882
            obj = ConfigObj(encoding='utf=8')
 
883
        return obj
 
884
 
 
885
    def get_option(self, name, section=None, default=None):
 
886
        self.branch.lock_read()
 
887
        try:
 
888
            obj = self._get_config()
 
889
            try:
 
890
                if section is not None:
 
891
                    obj = obj[section]
 
892
                result = obj[name]
 
893
            except KeyError:
 
894
                result = default
 
895
        finally:
 
896
            self.branch.unlock()
 
897
        return result
 
898
 
 
899
    def set_option(self, value, name, section=None):
 
900
        """Set a per-branch configuration option"""
 
901
        self.branch.lock_write()
 
902
        try:
 
903
            cfg_obj = self._get_config()
 
904
            if section is None:
 
905
                obj = cfg_obj
 
906
            else:
 
907
                try:
 
908
                    obj = cfg_obj[section]
 
909
                except KeyError:
 
910
                    cfg_obj[section] = {}
 
911
                    obj = cfg_obj[section]
 
912
            obj[name] = value
 
913
            out_file = StringIO()
 
914
            cfg_obj.write(out_file)
 
915
            out_file.seek(0)
 
916
            self.branch.control_files.put('branch.conf', out_file)
 
917
        finally:
 
918
            self.branch.unlock()