45
43
editor - this option sets the pop up editor to use during commits.
46
44
email - this option sets the user id bzr will use when committing.
47
45
check_signatures - this option controls whether bzr will require good gpg
48
signatures, ignore them, or check them if they are
46
signatures, ignore them, or check them if they are
50
create_signatures - this option controls whether bzr will always create
48
create_signatures - this option controls whether bzr will always create
51
49
gpg signatures, never create them, or create them if the
52
50
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
51
NB: This option is planned, but not implemented yet.
68
from bzrlib.lazy_import import lazy_import
69
lazy_import(globals(), """
71
58
from fnmatch import fnmatch
73
from cStringIO import StringIO
88
from bzrlib.util.configobj import configobj
62
import bzrlib.errors as errors
63
from bzrlib.osutils import pathjoin
64
from bzrlib.trace import mutter
65
import bzrlib.util.configobj.configobj as configobj
66
from StringIO import StringIO
92
68
CHECK_IF_POSSIBLE=0
104
POLICY_APPENDPATH = 2
108
POLICY_NORECURSE: 'norecurse',
109
POLICY_APPENDPATH: 'appendpath',
114
'norecurse': POLICY_NORECURSE,
115
'appendpath': POLICY_APPENDPATH,
119
STORE_LOCATION = POLICY_NONE
120
STORE_LOCATION_NORECURSE = POLICY_NORECURSE
121
STORE_LOCATION_APPENDPATH = POLICY_APPENDPATH
126
def ConfigObj(*args, **kwargs):
128
if _ConfigObj is None:
129
class ConfigObj(configobj.ConfigObj):
131
def get_bool(self, section, key):
132
return self[section].as_bool(key)
134
def get_value(self, section, name):
135
# Try [] for the old DEFAULT section.
136
if section == "DEFAULT":
141
return self[section][name]
142
_ConfigObj = ConfigObj
143
return _ConfigObj(*args, **kwargs)
73
class ConfigObj(configobj.ConfigObj):
75
def get_bool(self, section, key):
76
val = self[section][key].lower()
77
if val in ('1', 'yes', 'true', 'on'):
79
elif val in ('0', 'no', 'false', 'off'):
82
raise ValueError("Value %r is not boolean" % val)
84
def get_value(self, section, name):
85
# Try [] for the old DEFAULT section.
86
if section == "DEFAULT":
91
return self[section][name]
146
94
class Config(object):
147
95
"""A configuration policy - what username, editor, gpg needs etc."""
150
super(Config, self).__init__()
152
97
def get_editor(self):
153
98
"""Get the users pop up editor."""
154
99
raise NotImplementedError
156
def get_change_editor(self, old_tree, new_tree):
157
from bzrlib import diff
158
cmd = self._get_change_editor()
161
return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
165
def get_mail_client(self):
166
"""Get a mail client to use"""
167
selected_client = self.get_user_option('mail_client')
168
_registry = mail_client.mail_client_registry
170
mail_client_class = _registry.get(selected_client)
172
raise errors.UnknownMailClient(selected_client)
173
return mail_client_class(self)
175
101
def _get_signature_checking(self):
176
102
"""Template method to override signature checking policy."""
178
def _get_signing_policy(self):
179
"""Template method to override signature creation policy."""
181
104
def _get_user_option(self, option_name):
182
105
"""Template method to provide a user option."""
478
256
def __init__(self):
479
257
super(GlobalConfig, self).__init__(config_filename)
481
def set_user_option(self, option, value):
482
"""Save option and its value in the configuration."""
483
self._set_option(option, value, 'DEFAULT')
485
def get_aliases(self):
486
"""Return the aliases section."""
487
if 'ALIASES' in self._get_parser():
488
return self._get_parser()['ALIASES']
492
def set_alias(self, alias_name, alias_command):
493
"""Save the alias in the configuration."""
494
self._set_option(alias_name, alias_command, 'ALIASES')
496
def unset_alias(self, alias_name):
497
"""Unset an existing alias."""
498
aliases = self._get_parser().get('ALIASES')
499
if not aliases or alias_name not in aliases:
500
raise errors.NoSuchAlias(alias_name)
501
del aliases[alias_name]
502
self._write_config_file()
504
def _set_option(self, option, value, section):
505
# FIXME: RBC 20051029 This should refresh the parser and also take a
506
# file lock on bazaar.conf.
507
conf_dir = os.path.dirname(self._get_filename())
508
ensure_config_dir_exists(conf_dir)
509
self._get_parser().setdefault(section, {})[option] = value
510
self._write_config_file()
512
def _write_config_file(self):
513
f = open(self._get_filename(), 'wb')
514
self._get_parser().write(f)
518
260
class LocationConfig(IniBasedConfig):
519
261
"""A configuration object that gives the policy for a location."""
521
263
def __init__(self, location):
522
name_generator = locations_config_filename
523
if (not os.path.exists(name_generator()) and
524
os.path.exists(branches_config_filename())):
525
if sys.platform == 'win32':
526
trace.warning('Please rename %s to %s'
527
% (branches_config_filename(),
528
locations_config_filename()))
530
trace.warning('Please rename ~/.bazaar/branches.conf'
531
' to ~/.bazaar/locations.conf')
532
name_generator = branches_config_filename
533
super(LocationConfig, self).__init__(name_generator)
534
# local file locations are looked up by local path, rather than
535
# by file url. This is because the config file is a user
536
# file, and we would rather not expose the user to file urls.
537
if location.startswith('file://'):
538
location = urlutils.local_path_from_url(location)
264
super(LocationConfig, self).__init__(branches_config_filename)
265
self._global_config = None
539
266
self.location = location
541
def _get_matching_sections(self):
542
"""Return an ordered list of section names matching this location."""
268
def _get_global_config(self):
269
if self._global_config is None:
270
self._global_config = GlobalConfig()
271
return self._global_config
273
def _get_section(self):
274
"""Get the section we should look in for config items.
276
Returns None if none exists.
277
TODO: perhaps return a NullSection that thunks through to the
543
280
sections = self._get_parser()
544
281
location_names = self.location.split('/')
545
282
if self.location.endswith('/'):
546
283
del location_names[-1]
548
285
for section in sections:
549
# location is a local path if possible, so we need
550
# to convert 'file://' urls to local paths if necessary.
551
# This also avoids having file:///path be a more exact
552
# match than '/path'.
553
if section.startswith('file://'):
554
section_path = urlutils.local_path_from_url(section)
556
section_path = section
557
section_names = section_path.split('/')
286
section_names = section.split('/')
558
287
if section.endswith('/'):
559
288
del section_names[-1]
560
289
names = zip(location_names, section_names)
569
298
# if section is longer, no match.
570
299
if len(section_names) > len(location_names):
572
matches.append((len(section_names), section,
573
'/'.join(location_names[len(section_names):])))
301
# if path is longer, and recurse is not true, no match
302
if len(section_names) < len(location_names):
304
if not self._get_parser().get_bool(section, 'recurse'):
308
matches.append((len(section_names), section))
574
311
matches.sort(reverse=True)
576
for (length, section, extra_path) in matches:
577
sections.append((section, extra_path))
578
# should we stop looking for parent configs here?
580
if self._get_parser()[section].as_bool('ignore_parents'):
586
def _get_option_policy(self, section, option_name):
587
"""Return the policy for the given (section, option_name) pair."""
588
# check for the old 'recurse=False' flag
590
recurse = self._get_parser()[section].as_bool('recurse')
594
return POLICY_NORECURSE
596
policy_key = option_name + ':policy'
598
policy_name = self._get_parser()[section][policy_key]
602
return _policy_value[policy_name]
604
def _set_option_policy(self, section, option_name, option_policy):
605
"""Set the policy for the given option name in the given section."""
606
# The old recurse=False option affects all options in the
607
# section. To handle multiple policies in the section, we
608
# need to convert it to a policy_norecurse key.
610
recurse = self._get_parser()[section].as_bool('recurse')
614
symbol_versioning.warn(
615
'The recurse option is deprecated as of 0.14. '
616
'The section "%s" has been converted to use policies.'
619
del self._get_parser()[section]['recurse']
621
for key in self._get_parser()[section].keys():
622
if not key.endswith(':policy'):
623
self._get_parser()[section][key +
624
':policy'] = 'norecurse'
626
policy_key = option_name + ':policy'
627
policy_name = _policy_name[option_policy]
628
if policy_name is not None:
629
self._get_parser()[section][policy_key] = policy_name
631
if policy_key in self._get_parser()[section]:
632
del self._get_parser()[section][policy_key]
634
def set_user_option(self, option, value, store=STORE_LOCATION):
314
def _gpg_signing_command(self):
315
"""See Config.gpg_signing_command."""
316
command = super(LocationConfig, self)._gpg_signing_command()
317
if command is not None:
319
return self._get_global_config()._gpg_signing_command()
321
def _get_user_id(self):
322
user_id = super(LocationConfig, self)._get_user_id()
323
if user_id is not None:
325
return self._get_global_config()._get_user_id()
327
def _get_user_option(self, option_name):
328
"""See Config._get_user_option."""
329
option_value = super(LocationConfig,
330
self)._get_user_option(option_name)
331
if option_value is not None:
333
return self._get_global_config()._get_user_option(option_name)
335
def _get_signature_checking(self):
336
"""See Config._get_signature_checking."""
337
check = super(LocationConfig, self)._get_signature_checking()
338
if check is not None:
340
return self._get_global_config()._get_signature_checking()
342
def _post_commit(self):
343
"""See Config.post_commit."""
344
hook = self._get_user_option('post_commit')
347
return self._get_global_config()._post_commit()
349
def set_user_option(self, option, value):
635
350
"""Save option and its value in the configuration."""
636
if store not in [STORE_LOCATION,
637
STORE_LOCATION_NORECURSE,
638
STORE_LOCATION_APPENDPATH]:
639
raise ValueError('bad storage policy %r for %r' %
641
351
# FIXME: RBC 20051029 This should refresh the parser and also take a
642
# file lock on locations.conf.
352
# file lock on branches.conf.
643
353
conf_dir = os.path.dirname(self._get_filename())
644
354
ensure_config_dir_exists(conf_dir)
645
355
location = self.location
651
361
elif location + '/' in self._get_parser():
652
362
location = location + '/'
653
363
self._get_parser()[location][option]=value
654
# the allowed values of store match the config policies
655
self._set_option_policy(location, option, store)
656
self._get_parser().write(file(self._get_filename(), 'wb'))
364
self._get_parser().write()
659
367
class BranchConfig(Config):
660
368
"""A configuration object giving the policy for a branch."""
662
def _get_branch_data_config(self):
663
if self._branch_data_config is None:
664
self._branch_data_config = TreeConfig(self.branch)
665
return self._branch_data_config
667
370
def _get_location_config(self):
668
371
if self._location_config is None:
669
372
self._location_config = LocationConfig(self.branch.base)
670
373
return self._location_config
672
def _get_global_config(self):
673
if self._global_config is None:
674
self._global_config = GlobalConfig()
675
return self._global_config
677
def _get_best_value(self, option_name):
678
"""This returns a user option from local, tree or global config.
680
They are tried in that order. Use get_safe_value if trusted values
683
for source in self.option_sources:
684
value = getattr(source(), option_name)()
685
if value is not None:
689
def _get_safe_value(self, option_name):
690
"""This variant of get_best_value never returns untrusted values.
692
It does not return values from the branch data, because the branch may
693
not be controlled by the user.
695
We may wish to allow locations.conf to control whether branches are
696
trusted in the future.
698
for source in (self._get_location_config, self._get_global_config):
699
value = getattr(source(), option_name)()
700
if value is not None:
704
375
def _get_user_id(self):
705
376
"""Return the full user id for the branch.
707
e.g. "John Hacker <jhacker@example.com>"
378
e.g. "John Hacker <jhacker@foo.org>"
708
379
This is looked up in the email controlfile for the branch.
711
return (self.branch._transport.get_bytes("email")
712
.decode(osutils.get_user_encoding())
382
return (self.branch.controlfile("email", "r")
384
.decode(bzrlib.user_encoding)
714
386
except errors.NoSuchFile, e:
717
return self._get_best_value('_get_user_id')
719
def _get_change_editor(self):
720
return self._get_best_value('_get_change_editor')
389
return self._get_location_config()._get_user_id()
722
391
def _get_signature_checking(self):
723
392
"""See Config._get_signature_checking."""
724
return self._get_best_value('_get_signature_checking')
726
def _get_signing_policy(self):
727
"""See Config._get_signing_policy."""
728
return self._get_best_value('_get_signing_policy')
393
return self._get_location_config()._get_signature_checking()
730
395
def _get_user_option(self, option_name):
731
396
"""See Config._get_user_option."""
732
for source in self.option_sources:
733
value = source()._get_user_option(option_name)
734
if value is not None:
738
def set_user_option(self, name, value, store=STORE_BRANCH,
740
if store == STORE_BRANCH:
741
self._get_branch_data_config().set_option(value, name)
742
elif store == STORE_GLOBAL:
743
self._get_global_config().set_user_option(name, value)
745
self._get_location_config().set_user_option(name, value, store)
748
if store in (STORE_GLOBAL, STORE_BRANCH):
749
mask_value = self._get_location_config().get_user_option(name)
750
if mask_value is not None:
751
trace.warning('Value "%s" is masked by "%s" from'
752
' locations.conf', value, mask_value)
754
if store == STORE_GLOBAL:
755
branch_config = self._get_branch_data_config()
756
mask_value = branch_config.get_user_option(name)
757
if mask_value is not None:
758
trace.warning('Value "%s" is masked by "%s" from'
759
' branch.conf', value, mask_value)
397
return self._get_location_config()._get_user_option(option_name)
761
399
def _gpg_signing_command(self):
762
400
"""See Config.gpg_signing_command."""
763
return self._get_safe_value('_gpg_signing_command')
401
return self._get_location_config()._gpg_signing_command()
765
403
def __init__(self, branch):
766
404
super(BranchConfig, self).__init__()
767
405
self._location_config = None
768
self._branch_data_config = None
769
self._global_config = None
770
406
self.branch = branch
771
self.option_sources = (self._get_location_config,
772
self._get_branch_data_config,
773
self._get_global_config)
775
408
def _post_commit(self):
776
409
"""See Config.post_commit."""
777
return self._get_safe_value('_post_commit')
779
def _get_nickname(self):
780
value = self._get_explicit_nickname()
781
if value is not None:
783
return urlutils.unescape(self.branch.base.split('/')[-2])
785
def has_explicit_nickname(self):
786
"""Return true if a nickname has been explicitly assigned."""
787
return self._get_explicit_nickname() is not None
789
def _get_explicit_nickname(self):
790
return self._get_best_value('_get_nickname')
792
def _log_format(self):
793
"""See Config.log_format."""
794
return self._get_best_value('_log_format')
410
return self._get_location_config()._post_commit()
797
413
def ensure_config_dir_exists(path=None):
952
493
except ImportError:
955
user_encoding = osutils.get_user_encoding()
956
realname = username = getpass.getuser().decode(user_encoding)
957
except UnicodeDecodeError:
958
raise errors.BzrError("Can't decode username as %s." % \
495
realname = username = getpass.getuser().decode(bzrlib.user_encoding)
961
497
return realname, (username + '@' + socket.gethostname())
964
def parse_username(username):
965
"""Parse e-mail username and return a (name, address) tuple."""
966
match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
968
return (username, '')
970
return (match.group(1), match.group(2))
973
500
def extract_email_address(e):
974
501
"""Return just the address part of an email string.
976
That is just the user@domain part, nothing else.
503
That is just the user@domain part, nothing else.
977
504
This part is required to contain only ascii characters.
978
505
If it can't be extracted, raises an error.
980
507
>>> extract_email_address('Jane Tester <jane@test.com>')
983
name, email = parse_username(e)
985
raise errors.NoEmailInUsername(e)
989
class TreeConfig(IniBasedConfig):
510
m = re.search(r'[\w+.-]+@[\w+.-]+', e)
512
raise errors.BzrError("%r doesn't seem to contain "
513
"a reasonable email address" % e)
516
class TreeConfig(object):
990
517
"""Branch configuration data associated with its contents, not location"""
992
# XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
994
518
def __init__(self, branch):
995
self._config = branch._get_config()
996
519
self.branch = branch
998
def _get_parser(self, file=None):
1000
return IniBasedConfig._get_parser(file)
1001
return self._config._get_configobj()
521
def _get_config(self):
523
obj = ConfigObj(self.branch.controlfile('branch.conf',
526
except errors.NoSuchFile:
1003
530
def get_option(self, name, section=None, default=None):
1004
531
self.branch.lock_read()
1006
return self._config.get_option(name, section, default)
533
obj = self._get_config()
535
if section is not None:
1008
541
self.branch.unlock()
1010
544
def set_option(self, value, name, section=None):
1011
545
"""Set a per-branch configuration option"""
1012
546
self.branch.lock_write()
1014
self._config.set_option(value, name, section)
548
cfg_obj = self._get_config()
553
obj = cfg_obj[section]
555
cfg_obj[section] = {}
556
obj = cfg_obj[section]
558
cfg_obj.encode('UTF-8')
559
out_file = StringIO(''.join([l+'\n' for l in cfg_obj.write()]))
561
self.branch.put_controlfile('branch.conf', out_file, encode=False)
1016
563
self.branch.unlock()
1019
class AuthenticationConfig(object):
1020
"""The authentication configuration file based on a ini file.
1022
Implements the authentication.conf file described in
1023
doc/developers/authentication-ring.txt.
1026
def __init__(self, _file=None):
1027
self._config = None # The ConfigObj
1029
self._filename = authentication_config_filename()
1030
self._input = self._filename = authentication_config_filename()
1032
# Tests can provide a string as _file
1033
self._filename = None
1036
def _get_config(self):
1037
if self._config is not None:
1040
# FIXME: Should we validate something here ? Includes: empty
1041
# sections are useless, at least one of
1042
# user/password/password_encoding should be defined, etc.
1044
# Note: the encoding below declares that the file itself is utf-8
1045
# encoded, but the values in the ConfigObj are always Unicode.
1046
self._config = ConfigObj(self._input, encoding='utf-8')
1047
except configobj.ConfigObjError, e:
1048
raise errors.ParseConfigError(e.errors, e.config.filename)
1052
"""Save the config file, only tests should use it for now."""
1053
conf_dir = os.path.dirname(self._filename)
1054
ensure_config_dir_exists(conf_dir)
1055
self._get_config().write(file(self._filename, 'wb'))
1057
def _set_option(self, section_name, option_name, value):
1058
"""Set an authentication configuration option"""
1059
conf = self._get_config()
1060
section = conf.get(section_name)
1063
section = conf[section]
1064
section[option_name] = value
1067
def get_credentials(self, scheme, host, port=None, user=None, path=None,
1069
"""Returns the matching credentials from authentication.conf file.
1071
:param scheme: protocol
1073
:param host: the server address
1075
:param port: the associated port (optional)
1077
:param user: login (optional)
1079
:param path: the absolute path on the server (optional)
1081
:param realm: the http authentication realm (optional)
1083
:return: A dict containing the matching credentials or None.
1085
- name: the section name of the credentials in the
1086
authentication.conf file,
1087
- user: can't be different from the provided user if any,
1088
- scheme: the server protocol,
1089
- host: the server address,
1090
- port: the server port (can be None),
1091
- path: the absolute server path (can be None),
1092
- realm: the http specific authentication realm (can be None),
1093
- password: the decoded password, could be None if the credential
1094
defines only the user
1095
- verify_certificates: https specific, True if the server
1096
certificate should be verified, False otherwise.
1099
for auth_def_name, auth_def in self._get_config().items():
1100
if type(auth_def) is not configobj.Section:
1101
raise ValueError("%s defined outside a section" % auth_def_name)
1103
a_scheme, a_host, a_user, a_path = map(
1104
auth_def.get, ['scheme', 'host', 'user', 'path'])
1107
a_port = auth_def.as_int('port')
1111
raise ValueError("'port' not numeric in %s" % auth_def_name)
1113
a_verify_certificates = auth_def.as_bool('verify_certificates')
1115
a_verify_certificates = True
1118
"'verify_certificates' not boolean in %s" % auth_def_name)
1121
if a_scheme is not None and scheme != a_scheme:
1123
if a_host is not None:
1124
if not (host == a_host
1125
or (a_host.startswith('.') and host.endswith(a_host))):
1127
if a_port is not None and port != a_port:
1129
if (a_path is not None and path is not None
1130
and not path.startswith(a_path)):
1132
if (a_user is not None and user is not None
1133
and a_user != user):
1134
# Never contradict the caller about the user to be used
1139
# Prepare a credentials dictionary with additional keys
1140
# for the credential providers
1141
credentials = dict(name=auth_def_name,
1148
password=auth_def.get('password', None),
1149
verify_certificates=a_verify_certificates)
1150
# Decode the password in the credentials (or get one)
1151
self.decode_password(credentials,
1152
auth_def.get('password_encoding', None))
1153
if 'auth' in debug.debug_flags:
1154
trace.mutter("Using authentication section: %r", auth_def_name)
1157
if credentials is None:
1158
# No credentials were found in authentication.conf, try the fallback
1159
# credentials stores.
1160
credentials = credential_store_registry.get_fallback_credentials(
1161
scheme, host, port, user, path, realm)
1165
def set_credentials(self, name, host, user, scheme=None, password=None,
1166
port=None, path=None, verify_certificates=None,
1168
"""Set authentication credentials for a host.
1170
Any existing credentials with matching scheme, host, port and path
1171
will be deleted, regardless of name.
1173
:param name: An arbitrary name to describe this set of credentials.
1174
:param host: Name of the host that accepts these credentials.
1175
:param user: The username portion of these credentials.
1176
:param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1178
:param password: Password portion of these credentials.
1179
:param port: The IP port on the host that these credentials apply to.
1180
:param path: A filesystem path on the host that these credentials
1182
:param verify_certificates: On https, verify server certificates if
1184
:param realm: The http authentication realm (optional).
1186
values = {'host': host, 'user': user}
1187
if password is not None:
1188
values['password'] = password
1189
if scheme is not None:
1190
values['scheme'] = scheme
1191
if port is not None:
1192
values['port'] = '%d' % port
1193
if path is not None:
1194
values['path'] = path
1195
if verify_certificates is not None:
1196
values['verify_certificates'] = str(verify_certificates)
1197
if realm is not None:
1198
values['realm'] = realm
1199
config = self._get_config()
1201
for section, existing_values in config.items():
1202
for key in ('scheme', 'host', 'port', 'path', 'realm'):
1203
if existing_values.get(key) != values.get(key):
1207
config.update({name: values})
1210
def get_user(self, scheme, host, port=None, realm=None, path=None,
1211
prompt=None, ask=False, default=None):
1212
"""Get a user from authentication file.
1214
:param scheme: protocol
1216
:param host: the server address
1218
:param port: the associated port (optional)
1220
:param realm: the realm sent by the server (optional)
1222
:param path: the absolute path on the server (optional)
1224
:param ask: Ask the user if there is no explicitly configured username
1227
:param default: The username returned if none is defined (optional).
1229
:return: The found user.
1231
credentials = self.get_credentials(scheme, host, port, user=None,
1232
path=path, realm=realm)
1233
if credentials is not None:
1234
user = credentials['user']
1240
# Create a default prompt suitable for most cases
1241
prompt = scheme.upper() + ' %(host)s username'
1242
# Special handling for optional fields in the prompt
1243
if port is not None:
1244
prompt_host = '%s:%d' % (host, port)
1247
user = ui.ui_factory.get_username(prompt, host=prompt_host)
1252
def get_password(self, scheme, host, user, port=None,
1253
realm=None, path=None, prompt=None):
1254
"""Get a password from authentication file or prompt the user for one.
1256
:param scheme: protocol
1258
:param host: the server address
1260
:param port: the associated port (optional)
1264
:param realm: the realm sent by the server (optional)
1266
:param path: the absolute path on the server (optional)
1268
:return: The found password or the one entered by the user.
1270
credentials = self.get_credentials(scheme, host, port, user, path,
1272
if credentials is not None:
1273
password = credentials['password']
1274
if password is not None and scheme is 'ssh':
1275
trace.warning('password ignored in section [%s],'
1276
' use an ssh agent instead'
1277
% credentials['name'])
1281
# Prompt user only if we could't find a password
1282
if password is None:
1284
# Create a default prompt suitable for most cases
1285
prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
1286
# Special handling for optional fields in the prompt
1287
if port is not None:
1288
prompt_host = '%s:%d' % (host, port)
1291
password = ui.ui_factory.get_password(prompt,
1292
host=prompt_host, user=user)
1295
def decode_password(self, credentials, encoding):
1297
cs = credential_store_registry.get_credential_store(encoding)
1299
raise ValueError('%r is not a known password_encoding' % encoding)
1300
credentials['password'] = cs.decode_password(credentials)
1304
class CredentialStoreRegistry(registry.Registry):
1305
"""A class that registers credential stores.
1307
A credential store provides access to credentials via the password_encoding
1308
field in authentication.conf sections.
1310
Except for stores provided by bzr itself, most stores are expected to be
1311
provided by plugins that will therefore use
1312
register_lazy(password_encoding, module_name, member_name, help=help,
1313
fallback=fallback) to install themselves.
1315
A fallback credential store is one that is queried if no credentials can be
1316
found via authentication.conf.
1319
def get_credential_store(self, encoding=None):
1320
cs = self.get(encoding)
1325
def is_fallback(self, name):
1326
"""Check if the named credentials store should be used as fallback."""
1327
return self.get_info(name)
1329
def get_fallback_credentials(self, scheme, host, port=None, user=None,
1330
path=None, realm=None):
1331
"""Request credentials from all fallback credentials stores.
1333
The first credentials store that can provide credentials wins.
1336
for name in self.keys():
1337
if not self.is_fallback(name):
1339
cs = self.get_credential_store(name)
1340
credentials = cs.get_credentials(scheme, host, port, user,
1342
if credentials is not None:
1343
# We found some credentials
1347
def register(self, key, obj, help=None, override_existing=False,
1349
"""Register a new object to a name.
1351
:param key: This is the key to use to request the object later.
1352
:param obj: The object to register.
1353
:param help: Help text for this entry. This may be a string or
1354
a callable. If it is a callable, it should take two
1355
parameters (registry, key): this registry and the key that
1356
the help was registered under.
1357
:param override_existing: Raise KeyErorr if False and something has
1358
already been registered for that key. If True, ignore if there
1359
is an existing key (always register the new value).
1360
:param fallback: Whether this credential store should be
1363
return super(CredentialStoreRegistry,
1364
self).register(key, obj, help, info=fallback,
1365
override_existing=override_existing)
1367
def register_lazy(self, key, module_name, member_name,
1368
help=None, override_existing=False,
1370
"""Register a new credential store to be loaded on request.
1372
:param module_name: The python path to the module. Such as 'os.path'.
1373
:param member_name: The member of the module to return. If empty or
1374
None, get() will return the module itself.
1375
:param help: Help text for this entry. This may be a string or
1377
:param override_existing: If True, replace the existing object
1378
with the new one. If False, if there is already something
1379
registered with the same key, raise a KeyError
1380
:param fallback: Whether this credential store should be
1383
return super(CredentialStoreRegistry, self).register_lazy(
1384
key, module_name, member_name, help,
1385
info=fallback, override_existing=override_existing)
1388
credential_store_registry = CredentialStoreRegistry()
1391
class CredentialStore(object):
1392
"""An abstract class to implement storage for credentials"""
1394
def decode_password(self, credentials):
1395
"""Returns a clear text password for the provided credentials."""
1396
raise NotImplementedError(self.decode_password)
1398
def get_credentials(self, scheme, host, port=None, user=None, path=None,
1400
"""Return the matching credentials from this credential store.
1402
This method is only called on fallback credential stores.
1404
raise NotImplementedError(self.get_credentials)
1408
class PlainTextCredentialStore(CredentialStore):
1409
"""Plain text credential store for the authentication.conf file."""
1411
def decode_password(self, credentials):
1412
"""See CredentialStore.decode_password."""
1413
return credentials['password']
1416
credential_store_registry.register('plain', PlainTextCredentialStore,
1417
help=PlainTextCredentialStore.__doc__)
1418
credential_store_registry.default_key = 'plain'
1421
class BzrDirConfig(object):
1423
def __init__(self, bzrdir):
1424
self._bzrdir = bzrdir
1425
self._config = bzrdir._get_config()
1427
def set_default_stack_on(self, value):
1428
"""Set the default stacking location.
1430
It may be set to a location, or None.
1432
This policy affects all branches contained by this bzrdir, except for
1433
those under repositories.
1435
if self._config is None:
1436
raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1438
self._config.set_option('', 'default_stack_on')
1440
self._config.set_option(value, 'default_stack_on')
1442
def get_default_stack_on(self):
1443
"""Return the default stacking location.
1445
This will either be a location, or None.
1447
This policy affects all branches contained by this bzrdir, except for
1448
those under repositories.
1450
if self._config is None:
1452
value = self._config.get_option('default_stack_on')
1458
class TransportConfig(object):
1459
"""A Config that reads/writes a config file on a Transport.
1461
It is a low-level object that considers config data to be name/value pairs
1462
that may be associated with a section. Assigning meaning to the these
1463
values is done at higher levels like TreeConfig.
1466
def __init__(self, transport, filename):
1467
self._transport = transport
1468
self._filename = filename
1470
def get_option(self, name, section=None, default=None):
1471
"""Return the value associated with a named option.
1473
:param name: The name of the value
1474
:param section: The section the option is in (if any)
1475
:param default: The value to return if the value is not set
1476
:return: The value or default value
1478
configobj = self._get_configobj()
1480
section_obj = configobj
1483
section_obj = configobj[section]
1486
return section_obj.get(name, default)
1488
def set_option(self, value, name, section=None):
1489
"""Set the value associated with a named option.
1491
:param value: The value to set
1492
:param name: The name of the value to set
1493
:param section: The section the option is in (if any)
1495
configobj = self._get_configobj()
1497
configobj[name] = value
1499
configobj.setdefault(section, {})[name] = value
1500
self._set_configobj(configobj)
1502
def _get_config_file(self):
1504
return StringIO(self._transport.get_bytes(self._filename))
1505
except errors.NoSuchFile:
1508
def _get_configobj(self):
1509
return ConfigObj(self._get_config_file(), encoding='utf-8')
1511
def _set_configobj(self, configobj):
1512
out_file = StringIO()
1513
configobj.write(out_file)
1515
self._transport.put_file(self._filename, out_file)