1
# Copyright (C) 2005, 2007 Canonical Ltd
2
# Authors: Robert Collins <robert.collins@canonical.com>
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.
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.
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
19
"""Configuration that affects the behaviour of Bazaar.
21
Currently this configuration resides in ~/.bazaar/bazaar.conf
22
and ~/.bazaar/locations.conf, which is written to by bzr.
24
In bazaar.conf the following options may be set:
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
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
37
[/home/robertc/source]
38
recurse=False|True(default)
40
check_signatures= as above
41
create_signatures= as above.
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
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.
56
In bazaar.conf you can also define aliases in the ALIASES sections, example
59
lastlog=log --line -r-10..-1
60
ll=log --line -r-10..-1
68
from bzrlib.lazy_import import lazy_import
69
lazy_import(globals(), """
71
from fnmatch import fnmatch
73
from cStringIO import StringIO
87
from bzrlib.util.configobj import configobj
103
POLICY_APPENDPATH = 2
107
POLICY_NORECURSE: 'norecurse',
108
POLICY_APPENDPATH: 'appendpath',
113
'norecurse': POLICY_NORECURSE,
114
'appendpath': POLICY_APPENDPATH,
118
STORE_LOCATION = POLICY_NONE
119
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
120
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
125
class ConfigObj(configobj.ConfigObj):
127
def get_bool(self, section, key):
128
return self[section].as_bool(key)
130
def get_value(self, section, name):
131
# Try [] for the old DEFAULT section.
132
if section == "DEFAULT":
137
return self[section][name]
140
class Config(object):
141
"""A configuration policy - what username, editor, gpg needs etc."""
143
def get_editor(self):
144
"""Get the users pop up editor."""
145
raise NotImplementedError
147
def get_mail_client(self):
148
"""Get a mail client to use"""
149
selected_client = self.get_user_option('mail_client')
150
_registry = mail_client.mail_client_registry
152
mail_client_class = _registry.get(selected_client)
154
raise errors.UnknownMailClient(selected_client)
155
return mail_client_class(self)
157
def _get_signature_checking(self):
158
"""Template method to override signature checking policy."""
160
def _get_signing_policy(self):
161
"""Template method to override signature creation policy."""
163
def _get_user_option(self, option_name):
164
"""Template method to provide a user option."""
167
def get_user_option(self, option_name):
168
"""Get a generic option - no special process, no default."""
169
return self._get_user_option(option_name)
171
def gpg_signing_command(self):
172
"""What program should be used to sign signatures?"""
173
result = self._gpg_signing_command()
178
def _gpg_signing_command(self):
179
"""See gpg_signing_command()."""
182
def log_format(self):
183
"""What log format should be used"""
184
result = self._log_format()
189
def _log_format(self):
190
"""See log_format()."""
194
super(Config, self).__init__()
196
def post_commit(self):
197
"""An ordered list of python functions to call.
199
Each function takes branch, rev_id as parameters.
201
return self._post_commit()
203
def _post_commit(self):
204
"""See Config.post_commit."""
207
def user_email(self):
208
"""Return just the email component of a username."""
209
return extract_email_address(self.username())
212
"""Return email-style username.
214
Something similar to 'Martin Pool <mbp@sourcefrog.net>'
216
$BZR_EMAIL can be set to override this (as well as the
217
deprecated $BZREMAIL), then
218
the concrete policy type is checked, and finally
220
If none is found, a reasonable default is (hopefully)
223
TODO: Check it's reasonably well-formed.
225
v = os.environ.get('BZR_EMAIL')
227
return v.decode(bzrlib.user_encoding)
229
v = self._get_user_id()
233
v = os.environ.get('EMAIL')
235
return v.decode(bzrlib.user_encoding)
237
name, email = _auto_user_id()
239
return '%s <%s>' % (name, email)
243
def signature_checking(self):
244
"""What is the current policy for signature checking?."""
245
policy = self._get_signature_checking()
246
if policy is not None:
248
return CHECK_IF_POSSIBLE
250
def signing_policy(self):
251
"""What is the current policy for signature checking?."""
252
policy = self._get_signing_policy()
253
if policy is not None:
255
return SIGN_WHEN_REQUIRED
257
def signature_needed(self):
258
"""Is a signature needed when committing ?."""
259
policy = self._get_signing_policy()
261
policy = self._get_signature_checking()
262
if policy is not None:
263
trace.warning("Please use create_signatures,"
264
" not check_signatures to set signing policy.")
265
if policy == CHECK_ALWAYS:
267
elif policy == SIGN_ALWAYS:
271
def get_alias(self, value):
272
return self._get_alias(value)
274
def _get_alias(self, value):
277
def get_nickname(self):
278
return self._get_nickname()
280
def _get_nickname(self):
283
def get_bzr_remote_path(self):
285
return os.environ['BZR_REMOTE_PATH']
287
path = self.get_user_option("bzr_remote_path")
293
class IniBasedConfig(Config):
294
"""A configuration policy that draws from ini files."""
296
def _get_parser(self, file=None):
297
if self._parser is not None:
300
input = self._get_filename()
304
self._parser = ConfigObj(input, encoding='utf-8')
305
except configobj.ConfigObjError, e:
306
raise errors.ParseConfigError(e.errors, e.config.filename)
309
def _get_matching_sections(self):
310
"""Return an ordered list of (section_name, extra_path) pairs.
312
If the section contains inherited configuration, extra_path is
313
a string containing the additional path components.
315
section = self._get_section()
316
if section is not None:
317
return [(section, '')]
321
def _get_section(self):
322
"""Override this to define the section used by the config."""
325
def _get_option_policy(self, section, option_name):
326
"""Return the policy for the given (section, option_name) pair."""
329
def _get_signature_checking(self):
330
"""See Config._get_signature_checking."""
331
policy = self._get_user_option('check_signatures')
333
return self._string_to_signature_policy(policy)
335
def _get_signing_policy(self):
336
"""See Config._get_signing_policy"""
337
policy = self._get_user_option('create_signatures')
339
return self._string_to_signing_policy(policy)
341
def _get_user_id(self):
342
"""Get the user id from the 'email' key in the current section."""
343
return self._get_user_option('email')
345
def _get_user_option(self, option_name):
346
"""See Config._get_user_option."""
347
for (section, extra_path) in self._get_matching_sections():
349
value = self._get_parser().get_value(section, option_name)
352
policy = self._get_option_policy(section, option_name)
353
if policy == POLICY_NONE:
355
elif policy == POLICY_NORECURSE:
356
# norecurse items only apply to the exact path
361
elif policy == POLICY_APPENDPATH:
363
value = urlutils.join(value, extra_path)
366
raise AssertionError('Unexpected config policy %r' % policy)
370
def _gpg_signing_command(self):
371
"""See Config.gpg_signing_command."""
372
return self._get_user_option('gpg_signing_command')
374
def _log_format(self):
375
"""See Config.log_format."""
376
return self._get_user_option('log_format')
378
def __init__(self, get_filename):
379
super(IniBasedConfig, self).__init__()
380
self._get_filename = get_filename
383
def _post_commit(self):
384
"""See Config.post_commit."""
385
return self._get_user_option('post_commit')
387
def _string_to_signature_policy(self, signature_string):
388
"""Convert a string to a signing policy."""
389
if signature_string.lower() == 'check-available':
390
return CHECK_IF_POSSIBLE
391
if signature_string.lower() == 'ignore':
393
if signature_string.lower() == 'require':
395
raise errors.BzrError("Invalid signatures policy '%s'"
398
def _string_to_signing_policy(self, signature_string):
399
"""Convert a string to a signing policy."""
400
if signature_string.lower() == 'when-required':
401
return SIGN_WHEN_REQUIRED
402
if signature_string.lower() == 'never':
404
if signature_string.lower() == 'always':
406
raise errors.BzrError("Invalid signing policy '%s'"
409
def _get_alias(self, value):
411
return self._get_parser().get_value("ALIASES",
416
def _get_nickname(self):
417
return self.get_user_option('nickname')
420
class GlobalConfig(IniBasedConfig):
421
"""The configuration that should be used for a specific location."""
423
def get_editor(self):
424
return self._get_user_option('editor')
427
super(GlobalConfig, self).__init__(config_filename)
429
def set_user_option(self, option, value):
430
"""Save option and its value in the configuration."""
431
self._set_option(option, value, 'DEFAULT')
433
def get_aliases(self):
434
"""Return the aliases section."""
435
if 'ALIASES' in self._get_parser():
436
return self._get_parser()['ALIASES']
440
def set_alias(self, alias_name, alias_command):
441
"""Save the alias in the configuration."""
442
self._set_option(alias_name, alias_command, 'ALIASES')
444
def unset_alias(self, alias_name):
445
"""Unset an existing alias."""
446
aliases = self._get_parser().get('ALIASES')
447
if not aliases or alias_name not in aliases:
448
raise errors.NoSuchAlias(alias_name)
449
del aliases[alias_name]
450
self._write_config_file()
452
def _set_option(self, option, value, section):
453
# FIXME: RBC 20051029 This should refresh the parser and also take a
454
# file lock on bazaar.conf.
455
conf_dir = os.path.dirname(self._get_filename())
456
ensure_config_dir_exists(conf_dir)
457
self._get_parser().setdefault(section, {})[option] = value
458
self._write_config_file()
460
def _write_config_file(self):
461
f = open(self._get_filename(), 'wb')
462
self._get_parser().write(f)
466
class LocationConfig(IniBasedConfig):
467
"""A configuration object that gives the policy for a location."""
469
def __init__(self, location):
470
name_generator = locations_config_filename
471
if (not os.path.exists(name_generator()) and
472
os.path.exists(branches_config_filename())):
473
if sys.platform == 'win32':
474
trace.warning('Please rename %s to %s'
475
% (branches_config_filename(),
476
locations_config_filename()))
478
trace.warning('Please rename ~/.bazaar/branches.conf'
479
' to ~/.bazaar/locations.conf')
480
name_generator = branches_config_filename
481
super(LocationConfig, self).__init__(name_generator)
482
# local file locations are looked up by local path, rather than
483
# by file url. This is because the config file is a user
484
# file, and we would rather not expose the user to file urls.
485
if location.startswith('file://'):
486
location = urlutils.local_path_from_url(location)
487
self.location = location
489
def _get_matching_sections(self):
490
"""Return an ordered list of section names matching this location."""
491
sections = self._get_parser()
492
location_names = self.location.split('/')
493
if self.location.endswith('/'):
494
del location_names[-1]
496
for section in sections:
497
# location is a local path if possible, so we need
498
# to convert 'file://' urls to local paths if necessary.
499
# This also avoids having file:///path be a more exact
500
# match than '/path'.
501
if section.startswith('file://'):
502
section_path = urlutils.local_path_from_url(section)
504
section_path = section
505
section_names = section_path.split('/')
506
if section.endswith('/'):
507
del section_names[-1]
508
names = zip(location_names, section_names)
511
if not fnmatch(name[0], name[1]):
516
# so, for the common prefix they matched.
517
# if section is longer, no match.
518
if len(section_names) > len(location_names):
520
matches.append((len(section_names), section,
521
'/'.join(location_names[len(section_names):])))
522
matches.sort(reverse=True)
524
for (length, section, extra_path) in matches:
525
sections.append((section, extra_path))
526
# should we stop looking for parent configs here?
528
if self._get_parser()[section].as_bool('ignore_parents'):
534
def _get_option_policy(self, section, option_name):
535
"""Return the policy for the given (section, option_name) pair."""
536
# check for the old 'recurse=False' flag
538
recurse = self._get_parser()[section].as_bool('recurse')
542
return POLICY_NORECURSE
544
policy_key = option_name + ':policy'
546
policy_name = self._get_parser()[section][policy_key]
550
return _policy_value[policy_name]
552
def _set_option_policy(self, section, option_name, option_policy):
553
"""Set the policy for the given option name in the given section."""
554
# The old recurse=False option affects all options in the
555
# section. To handle multiple policies in the section, we
556
# need to convert it to a policy_norecurse key.
558
recurse = self._get_parser()[section].as_bool('recurse')
562
symbol_versioning.warn(
563
'The recurse option is deprecated as of 0.14. '
564
'The section "%s" has been converted to use policies.'
567
del self._get_parser()[section]['recurse']
569
for key in self._get_parser()[section].keys():
570
if not key.endswith(':policy'):
571
self._get_parser()[section][key +
572
':policy'] = 'norecurse'
574
policy_key = option_name + ':policy'
575
policy_name = _policy_name[option_policy]
576
if policy_name is not None:
577
self._get_parser()[section][policy_key] = policy_name
579
if policy_key in self._get_parser()[section]:
580
del self._get_parser()[section][policy_key]
582
def set_user_option(self, option, value, store=STORE_LOCATION):
583
"""Save option and its value in the configuration."""
584
if store not in [STORE_LOCATION,
585
STORE_LOCATION_NORECURSE,
586
STORE_LOCATION_APPENDPATH]:
587
raise ValueError('bad storage policy %r for %r' %
589
# FIXME: RBC 20051029 This should refresh the parser and also take a
590
# file lock on locations.conf.
591
conf_dir = os.path.dirname(self._get_filename())
592
ensure_config_dir_exists(conf_dir)
593
location = self.location
594
if location.endswith('/'):
595
location = location[:-1]
596
if (not location in self._get_parser() and
597
not location + '/' in self._get_parser()):
598
self._get_parser()[location]={}
599
elif location + '/' in self._get_parser():
600
location = location + '/'
601
self._get_parser()[location][option]=value
602
# the allowed values of store match the config policies
603
self._set_option_policy(location, option, store)
604
self._get_parser().write(file(self._get_filename(), 'wb'))
607
class BranchConfig(Config):
608
"""A configuration object giving the policy for a branch."""
610
def _get_branch_data_config(self):
611
if self._branch_data_config is None:
612
self._branch_data_config = TreeConfig(self.branch)
613
return self._branch_data_config
615
def _get_location_config(self):
616
if self._location_config is None:
617
self._location_config = LocationConfig(self.branch.base)
618
return self._location_config
620
def _get_global_config(self):
621
if self._global_config is None:
622
self._global_config = GlobalConfig()
623
return self._global_config
625
def _get_best_value(self, option_name):
626
"""This returns a user option from local, tree or global config.
628
They are tried in that order. Use get_safe_value if trusted values
631
for source in self.option_sources:
632
value = getattr(source(), option_name)()
633
if value is not None:
637
def _get_safe_value(self, option_name):
638
"""This variant of get_best_value never returns untrusted values.
640
It does not return values from the branch data, because the branch may
641
not be controlled by the user.
643
We may wish to allow locations.conf to control whether branches are
644
trusted in the future.
646
for source in (self._get_location_config, self._get_global_config):
647
value = getattr(source(), option_name)()
648
if value is not None:
652
def _get_user_id(self):
653
"""Return the full user id for the branch.
655
e.g. "John Hacker <jhacker@example.com>"
656
This is looked up in the email controlfile for the branch.
659
return (self.branch._transport.get_bytes("email")
660
.decode(bzrlib.user_encoding)
662
except errors.NoSuchFile, e:
665
return self._get_best_value('_get_user_id')
667
def _get_signature_checking(self):
668
"""See Config._get_signature_checking."""
669
return self._get_best_value('_get_signature_checking')
671
def _get_signing_policy(self):
672
"""See Config._get_signing_policy."""
673
return self._get_best_value('_get_signing_policy')
675
def _get_user_option(self, option_name):
676
"""See Config._get_user_option."""
677
for source in self.option_sources:
678
value = source()._get_user_option(option_name)
679
if value is not None:
683
def set_user_option(self, name, value, store=STORE_BRANCH,
685
if store == STORE_BRANCH:
686
self._get_branch_data_config().set_option(value, name)
687
elif store == STORE_GLOBAL:
688
self._get_global_config().set_user_option(name, value)
690
self._get_location_config().set_user_option(name, value, store)
693
if store in (STORE_GLOBAL, STORE_BRANCH):
694
mask_value = self._get_location_config().get_user_option(name)
695
if mask_value is not None:
696
trace.warning('Value "%s" is masked by "%s" from'
697
' locations.conf', value, mask_value)
699
if store == STORE_GLOBAL:
700
branch_config = self._get_branch_data_config()
701
mask_value = branch_config.get_user_option(name)
702
if mask_value is not None:
703
trace.warning('Value "%s" is masked by "%s" from'
704
' branch.conf', value, mask_value)
707
def _gpg_signing_command(self):
708
"""See Config.gpg_signing_command."""
709
return self._get_safe_value('_gpg_signing_command')
711
def __init__(self, branch):
712
super(BranchConfig, self).__init__()
713
self._location_config = None
714
self._branch_data_config = None
715
self._global_config = None
717
self.option_sources = (self._get_location_config,
718
self._get_branch_data_config,
719
self._get_global_config)
721
def _post_commit(self):
722
"""See Config.post_commit."""
723
return self._get_safe_value('_post_commit')
725
def _get_nickname(self):
726
value = self._get_explicit_nickname()
727
if value is not None:
729
return urlutils.unescape(self.branch.base.split('/')[-2])
731
def has_explicit_nickname(self):
732
"""Return true if a nickname has been explicitly assigned."""
733
return self._get_explicit_nickname() is not None
735
def _get_explicit_nickname(self):
736
return self._get_best_value('_get_nickname')
738
def _log_format(self):
739
"""See Config.log_format."""
740
return self._get_best_value('_log_format')
743
def ensure_config_dir_exists(path=None):
744
"""Make sure a configuration directory exists.
745
This makes sure that the directory exists.
746
On windows, since configuration directories are 2 levels deep,
747
it makes sure both the directory and the parent directory exists.
751
if not os.path.isdir(path):
752
if sys.platform == 'win32':
753
parent_dir = os.path.dirname(path)
754
if not os.path.isdir(parent_dir):
755
trace.mutter('creating config parent directory: %r', parent_dir)
757
trace.mutter('creating config directory: %r', path)
762
"""Return per-user configuration directory.
764
By default this is ~/.bazaar/
766
TODO: Global option --config-dir to override this.
768
base = os.environ.get('BZR_HOME', None)
769
if sys.platform == 'win32':
771
base = win32utils.get_appdata_location_unicode()
773
base = os.environ.get('HOME', None)
775
raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
777
return osutils.pathjoin(base, 'bazaar', '2.0')
779
# cygwin, linux, and darwin all have a $HOME directory
781
base = os.path.expanduser("~")
782
return osutils.pathjoin(base, ".bazaar")
785
def config_filename():
786
"""Return per-user configuration ini file filename."""
787
return osutils.pathjoin(config_dir(), 'bazaar.conf')
790
def branches_config_filename():
791
"""Return per-user configuration ini file filename."""
792
return osutils.pathjoin(config_dir(), 'branches.conf')
795
def locations_config_filename():
796
"""Return per-user configuration ini file filename."""
797
return osutils.pathjoin(config_dir(), 'locations.conf')
800
def authentication_config_filename():
801
"""Return per-user authentication ini file filename."""
802
return osutils.pathjoin(config_dir(), 'authentication.conf')
805
def user_ignore_config_filename():
806
"""Return the user default ignore filename"""
807
return osutils.pathjoin(config_dir(), 'ignore')
811
"""Calculate automatic user identification.
813
Returns (realname, email).
815
Only used when none is set in the environment or the id file.
817
This previously used the FQDN as the default domain, but that can
818
be very slow on machines where DNS is broken. So now we simply
823
if sys.platform == 'win32':
824
name = win32utils.get_user_name_unicode()
826
raise errors.BzrError("Cannot autodetect user name.\n"
827
"Please, set your name with command like:\n"
828
'bzr whoami "Your Name <name@domain.com>"')
829
host = win32utils.get_host_name_unicode()
831
host = socket.gethostname()
832
return name, (name + '@' + host)
837
w = pwd.getpwuid(uid)
839
# we try utf-8 first, because on many variants (like Linux),
840
# /etc/passwd "should" be in utf-8, and because it's unlikely to give
841
# false positives. (many users will have their user encoding set to
842
# latin-1, which cannot raise UnicodeError.)
844
gecos = w.pw_gecos.decode('utf-8')
848
gecos = w.pw_gecos.decode(bzrlib.user_encoding)
849
encoding = bzrlib.user_encoding
851
raise errors.BzrCommandError('Unable to determine your name. '
852
'Use "bzr whoami" to set it.')
854
username = w.pw_name.decode(encoding)
856
raise errors.BzrCommandError('Unable to determine your name. '
857
'Use "bzr whoami" to set it.')
859
comma = gecos.find(',')
863
realname = gecos[:comma]
870
realname = username = getpass.getuser().decode(bzrlib.user_encoding)
871
except UnicodeDecodeError:
872
raise errors.BzrError("Can't decode username as %s." % \
873
bzrlib.user_encoding)
875
return realname, (username + '@' + socket.gethostname())
878
def parse_username(username):
879
"""Parse e-mail username and return a (name, address) tuple."""
880
match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
882
return (username, '')
884
return (match.group(1), match.group(2))
887
def extract_email_address(e):
888
"""Return just the address part of an email string.
890
That is just the user@domain part, nothing else.
891
This part is required to contain only ascii characters.
892
If it can't be extracted, raises an error.
894
>>> extract_email_address('Jane Tester <jane@test.com>')
897
name, email = parse_username(e)
899
raise errors.NoEmailInUsername(e)
903
class TreeConfig(IniBasedConfig):
904
"""Branch configuration data associated with its contents, not location"""
906
# XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
908
def __init__(self, branch):
909
# XXX: Really this should be asking the branch for its configuration
910
# data, rather than relying on a Transport, so that it can work
911
# more cleanly with a RemoteBranch that has no transport.
912
self._config = TransportConfig(branch._transport, 'branch.conf')
915
def _get_parser(self, file=None):
917
return IniBasedConfig._get_parser(file)
918
return self._config._get_configobj()
920
def get_option(self, name, section=None, default=None):
921
self.branch.lock_read()
923
return self._config.get_option(name, section, default)
928
def set_option(self, value, name, section=None):
929
"""Set a per-branch configuration option"""
930
self.branch.lock_write()
932
self._config.set_option(value, name, section)
937
class AuthenticationConfig(object):
938
"""The authentication configuration file based on a ini file.
940
Implements the authentication.conf file described in
941
doc/developers/authentication-ring.txt.
944
def __init__(self, _file=None):
945
self._config = None # The ConfigObj
947
self._filename = authentication_config_filename()
948
self._input = self._filename = authentication_config_filename()
950
# Tests can provide a string as _file
951
self._filename = None
954
def _get_config(self):
955
if self._config is not None:
958
# FIXME: Should we validate something here ? Includes: empty
959
# sections are useless, at least one of
960
# user/password/password_encoding should be defined, etc.
962
# Note: the encoding below declares that the file itself is utf-8
963
# encoded, but the values in the ConfigObj are always Unicode.
964
self._config = ConfigObj(self._input, encoding='utf-8')
965
except configobj.ConfigObjError, e:
966
raise errors.ParseConfigError(e.errors, e.config.filename)
970
"""Save the config file, only tests should use it for now."""
971
conf_dir = os.path.dirname(self._filename)
972
ensure_config_dir_exists(conf_dir)
973
self._get_config().write(file(self._filename, 'wb'))
975
def _set_option(self, section_name, option_name, value):
976
"""Set an authentication configuration option"""
977
conf = self._get_config()
978
section = conf.get(section_name)
981
section = conf[section]
982
section[option_name] = value
985
def get_credentials(self, scheme, host, port=None, user=None, path=None):
986
"""Returns the matching credentials from authentication.conf file.
988
:param scheme: protocol
990
:param host: the server address
992
:param port: the associated port (optional)
994
:param user: login (optional)
996
:param path: the absolute path on the server (optional)
998
:return: A dict containing the matching credentials or None.
1000
- name: the section name of the credentials in the
1001
authentication.conf file,
1002
- user: can't de different from the provided user if any,
1003
- password: the decoded password, could be None if the credential
1004
defines only the user
1005
- verify_certificates: https specific, True if the server
1006
certificate should be verified, False otherwise.
1009
for auth_def_name, auth_def in self._get_config().items():
1010
if type(auth_def) is not configobj.Section:
1011
raise ValueError("%s defined outside a section" % auth_def_name)
1013
a_scheme, a_host, a_user, a_path = map(
1014
auth_def.get, ['scheme', 'host', 'user', 'path'])
1017
a_port = auth_def.as_int('port')
1021
raise ValueError("'port' not numeric in %s" % auth_def_name)
1023
a_verify_certificates = auth_def.as_bool('verify_certificates')
1025
a_verify_certificates = True
1028
"'verify_certificates' not boolean in %s" % auth_def_name)
1031
if a_scheme is not None and scheme != a_scheme:
1033
if a_host is not None:
1034
if not (host == a_host
1035
or (a_host.startswith('.') and host.endswith(a_host))):
1037
if a_port is not None and port != a_port:
1039
if (a_path is not None and path is not None
1040
and not path.startswith(a_path)):
1042
if (a_user is not None and user is not None
1043
and a_user != user):
1044
# Never contradict the caller about the user to be used
1049
credentials = dict(name=auth_def_name,
1051
password=auth_def.get('password', None),
1052
verify_certificates=a_verify_certificates)
1053
self.decode_password(credentials,
1054
auth_def.get('password_encoding', None))
1055
if 'auth' in debug.debug_flags:
1056
trace.mutter("Using authentication section: %r", auth_def_name)
1061
def get_user(self, scheme, host, port=None,
1062
realm=None, path=None, prompt=None):
1063
"""Get a user from authentication file.
1065
:param scheme: protocol
1067
:param host: the server address
1069
:param port: the associated port (optional)
1071
:param realm: the realm sent by the server (optional)
1073
:param path: the absolute path on the server (optional)
1075
:return: The found user.
1077
credentials = self.get_credentials(scheme, host, port, user=None,
1079
if credentials is not None:
1080
user = credentials['user']
1085
def get_password(self, scheme, host, user, port=None,
1086
realm=None, path=None, prompt=None):
1087
"""Get a password from authentication file or prompt the user for one.
1089
:param scheme: protocol
1091
:param host: the server address
1093
:param port: the associated port (optional)
1097
:param realm: the realm sent by the server (optional)
1099
:param path: the absolute path on the server (optional)
1101
:return: The found password or the one entered by the user.
1103
credentials = self.get_credentials(scheme, host, port, user, path)
1104
if credentials is not None:
1105
password = credentials['password']
1106
if password is not None and scheme is 'ssh':
1107
trace.warning('password ignored in section [%s],'
1108
' use an ssh agent instead'
1109
% credentials['name'])
1113
# Prompt user only if we could't find a password
1114
if password is None:
1116
# Create a default prompt suitable for most cases
1117
prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
1118
# Special handling for optional fields in the prompt
1119
if port is not None:
1120
prompt_host = '%s:%d' % (host, port)
1123
password = ui.ui_factory.get_password(prompt,
1124
host=prompt_host, user=user)
1127
def decode_password(self, credentials, encoding):
1131
class BzrDirConfig(object):
1133
def __init__(self, transport):
1134
self._config = TransportConfig(transport, 'control.conf')
1136
def set_default_stack_on(self, value):
1137
"""Set the default stacking location.
1139
It may be set to a location, or None.
1141
This policy affects all branches contained by this bzrdir, except for
1142
those under repositories.
1145
self._config.set_option('', 'default_stack_on')
1147
self._config.set_option(value, 'default_stack_on')
1149
def get_default_stack_on(self):
1150
"""Return the default stacking location.
1152
This will either be a location, or None.
1154
This policy affects all branches contained by this bzrdir, except for
1155
those under repositories.
1157
value = self._config.get_option('default_stack_on')
1163
class TransportConfig(object):
1164
"""A Config that reads/writes a config file on a Transport.
1166
It is a low-level object that considers config data to be name/value pairs
1167
that may be associated with a section. Assigning meaning to the these
1168
values is done at higher levels like TreeConfig.
1171
def __init__(self, transport, filename):
1172
self._transport = transport
1173
self._filename = filename
1175
def get_option(self, name, section=None, default=None):
1176
"""Return the value associated with a named option.
1178
:param name: The name of the value
1179
:param section: The section the option is in (if any)
1180
:param default: The value to return if the value is not set
1181
:return: The value or default value
1183
configobj = self._get_configobj()
1185
section_obj = configobj
1188
section_obj = configobj[section]
1191
return section_obj.get(name, default)
1193
def set_option(self, value, name, section=None):
1194
"""Set the value associated with a named option.
1196
:param value: The value to set
1197
:param name: The name of the value to set
1198
:param section: The section the option is in (if any)
1200
configobj = self._get_configobj()
1202
configobj[name] = value
1204
configobj.setdefault(section, {})[name] = value
1205
self._set_configobj(configobj)
1207
def _get_configobj(self):
1209
return ConfigObj(self._transport.get(self._filename),
1211
except errors.NoSuchFile:
1212
return ConfigObj(encoding='utf-8')
1214
def _set_configobj(self, configobj):
1215
out_file = StringIO()
1216
configobj.write(out_file)
1218
self._transport.put_file(self._filename, out_file)