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
import bzrlib.util.configobj.configobj as configobj
64
from StringIO import StringIO
92
66
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)
71
class ConfigObj(configobj.ConfigObj):
73
def get_bool(self, section, key):
74
val = self[section][key].lower()
75
if val in ('1', 'yes', 'true', 'on'):
77
elif val in ('0', 'no', 'false', 'off'):
80
raise ValueError("Value %r is not boolean" % val)
82
def get_value(self, section, name):
83
# Try [] for the old DEFAULT section.
84
if section == "DEFAULT":
89
return self[section][name]
146
92
class Config(object):
147
93
"""A configuration policy - what username, editor, gpg needs etc."""
150
super(Config, self).__init__()
152
95
def get_editor(self):
153
96
"""Get the users pop up editor."""
154
97
raise NotImplementedError
156
def get_mail_client(self):
157
"""Get a mail client to use"""
158
selected_client = self.get_user_option('mail_client')
159
_registry = mail_client.mail_client_registry
161
mail_client_class = _registry.get(selected_client)
163
raise errors.UnknownMailClient(selected_client)
164
return mail_client_class(self)
166
99
def _get_signature_checking(self):
167
100
"""Template method to override signature checking policy."""
169
def _get_signing_policy(self):
170
"""Template method to override signature creation policy."""
172
102
def _get_user_option(self, option_name):
173
103
"""Template method to provide a user option."""
441
254
def __init__(self):
442
255
super(GlobalConfig, self).__init__(config_filename)
444
def set_user_option(self, option, value):
445
"""Save option and its value in the configuration."""
446
self._set_option(option, value, 'DEFAULT')
448
def get_aliases(self):
449
"""Return the aliases section."""
450
if 'ALIASES' in self._get_parser():
451
return self._get_parser()['ALIASES']
455
def set_alias(self, alias_name, alias_command):
456
"""Save the alias in the configuration."""
457
self._set_option(alias_name, alias_command, 'ALIASES')
459
def unset_alias(self, alias_name):
460
"""Unset an existing alias."""
461
aliases = self._get_parser().get('ALIASES')
462
if not aliases or alias_name not in aliases:
463
raise errors.NoSuchAlias(alias_name)
464
del aliases[alias_name]
465
self._write_config_file()
467
def _set_option(self, option, value, section):
468
# FIXME: RBC 20051029 This should refresh the parser and also take a
469
# file lock on bazaar.conf.
470
conf_dir = os.path.dirname(self._get_filename())
471
ensure_config_dir_exists(conf_dir)
472
self._get_parser().setdefault(section, {})[option] = value
473
self._write_config_file()
475
def _write_config_file(self):
476
f = open(self._get_filename(), 'wb')
477
self._get_parser().write(f)
481
258
class LocationConfig(IniBasedConfig):
482
259
"""A configuration object that gives the policy for a location."""
484
261
def __init__(self, location):
485
name_generator = locations_config_filename
486
if (not os.path.exists(name_generator()) and
487
os.path.exists(branches_config_filename())):
488
if sys.platform == 'win32':
489
trace.warning('Please rename %s to %s'
490
% (branches_config_filename(),
491
locations_config_filename()))
493
trace.warning('Please rename ~/.bazaar/branches.conf'
494
' to ~/.bazaar/locations.conf')
495
name_generator = branches_config_filename
496
super(LocationConfig, self).__init__(name_generator)
497
# local file locations are looked up by local path, rather than
498
# by file url. This is because the config file is a user
499
# file, and we would rather not expose the user to file urls.
500
if location.startswith('file://'):
501
location = urlutils.local_path_from_url(location)
262
super(LocationConfig, self).__init__(branches_config_filename)
263
self._global_config = None
502
264
self.location = location
504
def _get_matching_sections(self):
505
"""Return an ordered list of section names matching this location."""
266
def _get_global_config(self):
267
if self._global_config is None:
268
self._global_config = GlobalConfig()
269
return self._global_config
271
def _get_section(self):
272
"""Get the section we should look in for config items.
274
Returns None if none exists.
275
TODO: perhaps return a NullSection that thunks through to the
506
278
sections = self._get_parser()
507
279
location_names = self.location.split('/')
508
280
if self.location.endswith('/'):
509
281
del location_names[-1]
511
283
for section in sections:
512
# location is a local path if possible, so we need
513
# to convert 'file://' urls to local paths if necessary.
514
# This also avoids having file:///path be a more exact
515
# match than '/path'.
516
if section.startswith('file://'):
517
section_path = urlutils.local_path_from_url(section)
519
section_path = section
520
section_names = section_path.split('/')
284
section_names = section.split('/')
521
285
if section.endswith('/'):
522
286
del section_names[-1]
523
287
names = zip(location_names, section_names)
532
296
# if section is longer, no match.
533
297
if len(section_names) > len(location_names):
535
matches.append((len(section_names), section,
536
'/'.join(location_names[len(section_names):])))
299
# if path is longer, and recurse is not true, no match
300
if len(section_names) < len(location_names):
302
if not self._get_parser().get_bool(section, 'recurse'):
306
matches.append((len(section_names), section))
537
309
matches.sort(reverse=True)
539
for (length, section, extra_path) in matches:
540
sections.append((section, extra_path))
541
# should we stop looking for parent configs here?
543
if self._get_parser()[section].as_bool('ignore_parents'):
549
def _get_option_policy(self, section, option_name):
550
"""Return the policy for the given (section, option_name) pair."""
551
# check for the old 'recurse=False' flag
553
recurse = self._get_parser()[section].as_bool('recurse')
557
return POLICY_NORECURSE
559
policy_key = option_name + ':policy'
561
policy_name = self._get_parser()[section][policy_key]
565
return _policy_value[policy_name]
567
def _set_option_policy(self, section, option_name, option_policy):
568
"""Set the policy for the given option name in the given section."""
569
# The old recurse=False option affects all options in the
570
# section. To handle multiple policies in the section, we
571
# need to convert it to a policy_norecurse key.
573
recurse = self._get_parser()[section].as_bool('recurse')
577
symbol_versioning.warn(
578
'The recurse option is deprecated as of 0.14. '
579
'The section "%s" has been converted to use policies.'
582
del self._get_parser()[section]['recurse']
584
for key in self._get_parser()[section].keys():
585
if not key.endswith(':policy'):
586
self._get_parser()[section][key +
587
':policy'] = 'norecurse'
589
policy_key = option_name + ':policy'
590
policy_name = _policy_name[option_policy]
591
if policy_name is not None:
592
self._get_parser()[section][policy_key] = policy_name
594
if policy_key in self._get_parser()[section]:
595
del self._get_parser()[section][policy_key]
597
def set_user_option(self, option, value, store=STORE_LOCATION):
312
def _gpg_signing_command(self):
313
"""See Config.gpg_signing_command."""
314
command = super(LocationConfig, self)._gpg_signing_command()
315
if command is not None:
317
return self._get_global_config()._gpg_signing_command()
319
def _get_user_id(self):
320
user_id = super(LocationConfig, self)._get_user_id()
321
if user_id is not None:
323
return self._get_global_config()._get_user_id()
325
def _get_user_option(self, option_name):
326
"""See Config._get_user_option."""
327
option_value = super(LocationConfig,
328
self)._get_user_option(option_name)
329
if option_value is not None:
331
return self._get_global_config()._get_user_option(option_name)
333
def _get_signature_checking(self):
334
"""See Config._get_signature_checking."""
335
check = super(LocationConfig, self)._get_signature_checking()
336
if check is not None:
338
return self._get_global_config()._get_signature_checking()
340
def _post_commit(self):
341
"""See Config.post_commit."""
342
hook = self._get_user_option('post_commit')
345
return self._get_global_config()._post_commit()
347
def set_user_option(self, option, value):
598
348
"""Save option and its value in the configuration."""
599
if store not in [STORE_LOCATION,
600
STORE_LOCATION_NORECURSE,
601
STORE_LOCATION_APPENDPATH]:
602
raise ValueError('bad storage policy %r for %r' %
604
349
# FIXME: RBC 20051029 This should refresh the parser and also take a
605
# file lock on locations.conf.
606
conf_dir = os.path.dirname(self._get_filename())
607
ensure_config_dir_exists(conf_dir)
350
# file lock on branches.conf.
351
if not os.path.isdir(os.path.dirname(self._get_filename())):
352
os.mkdir(os.path.dirname(self._get_filename()))
608
353
location = self.location
609
354
if location.endswith('/'):
610
355
location = location[:-1]
614
359
elif location + '/' in self._get_parser():
615
360
location = location + '/'
616
361
self._get_parser()[location][option]=value
617
# the allowed values of store match the config policies
618
self._set_option_policy(location, option, store)
619
self._get_parser().write(file(self._get_filename(), 'wb'))
362
self._get_parser().write()
622
365
class BranchConfig(Config):
623
366
"""A configuration object giving the policy for a branch."""
625
def _get_branch_data_config(self):
626
if self._branch_data_config is None:
627
self._branch_data_config = TreeConfig(self.branch)
628
return self._branch_data_config
630
368
def _get_location_config(self):
631
369
if self._location_config is None:
632
370
self._location_config = LocationConfig(self.branch.base)
633
371
return self._location_config
635
def _get_global_config(self):
636
if self._global_config is None:
637
self._global_config = GlobalConfig()
638
return self._global_config
640
def _get_best_value(self, option_name):
641
"""This returns a user option from local, tree or global config.
643
They are tried in that order. Use get_safe_value if trusted values
646
for source in self.option_sources:
647
value = getattr(source(), option_name)()
648
if value is not None:
652
def _get_safe_value(self, option_name):
653
"""This variant of get_best_value never returns untrusted values.
655
It does not return values from the branch data, because the branch may
656
not be controlled by the user.
658
We may wish to allow locations.conf to control whether branches are
659
trusted in the future.
661
for source in (self._get_location_config, self._get_global_config):
662
value = getattr(source(), option_name)()
663
if value is not None:
667
373
def _get_user_id(self):
668
374
"""Return the full user id for the branch.
670
e.g. "John Hacker <jhacker@example.com>"
376
e.g. "John Hacker <jhacker@foo.org>"
671
377
This is looked up in the email controlfile for the branch.
674
return (self.branch._transport.get_bytes("email")
675
.decode(osutils.get_user_encoding())
380
return (self.branch.controlfile("email", "r")
382
.decode(bzrlib.user_encoding)
677
384
except errors.NoSuchFile, e:
680
return self._get_best_value('_get_user_id')
387
return self._get_location_config()._get_user_id()
682
389
def _get_signature_checking(self):
683
390
"""See Config._get_signature_checking."""
684
return self._get_best_value('_get_signature_checking')
686
def _get_signing_policy(self):
687
"""See Config._get_signing_policy."""
688
return self._get_best_value('_get_signing_policy')
391
return self._get_location_config()._get_signature_checking()
690
393
def _get_user_option(self, option_name):
691
394
"""See Config._get_user_option."""
692
for source in self.option_sources:
693
value = source()._get_user_option(option_name)
694
if value is not None:
698
def set_user_option(self, name, value, store=STORE_BRANCH,
700
if store == STORE_BRANCH:
701
self._get_branch_data_config().set_option(value, name)
702
elif store == STORE_GLOBAL:
703
self._get_global_config().set_user_option(name, value)
705
self._get_location_config().set_user_option(name, value, store)
708
if store in (STORE_GLOBAL, STORE_BRANCH):
709
mask_value = self._get_location_config().get_user_option(name)
710
if mask_value is not None:
711
trace.warning('Value "%s" is masked by "%s" from'
712
' locations.conf', value, mask_value)
714
if store == STORE_GLOBAL:
715
branch_config = self._get_branch_data_config()
716
mask_value = branch_config.get_user_option(name)
717
if mask_value is not None:
718
trace.warning('Value "%s" is masked by "%s" from'
719
' branch.conf', value, mask_value)
395
return self._get_location_config()._get_user_option(option_name)
721
397
def _gpg_signing_command(self):
722
398
"""See Config.gpg_signing_command."""
723
return self._get_safe_value('_gpg_signing_command')
399
return self._get_location_config()._gpg_signing_command()
725
401
def __init__(self, branch):
726
402
super(BranchConfig, self).__init__()
727
403
self._location_config = None
728
self._branch_data_config = None
729
self._global_config = None
730
404
self.branch = branch
731
self.option_sources = (self._get_location_config,
732
self._get_branch_data_config,
733
self._get_global_config)
735
406
def _post_commit(self):
736
407
"""See Config.post_commit."""
737
return self._get_safe_value('_post_commit')
739
def _get_nickname(self):
740
value = self._get_explicit_nickname()
741
if value is not None:
743
return urlutils.unescape(self.branch.base.split('/')[-2])
745
def has_explicit_nickname(self):
746
"""Return true if a nickname has been explicitly assigned."""
747
return self._get_explicit_nickname() is not None
749
def _get_explicit_nickname(self):
750
return self._get_best_value('_get_nickname')
752
def _log_format(self):
753
"""See Config.log_format."""
754
return self._get_best_value('_log_format')
757
def ensure_config_dir_exists(path=None):
758
"""Make sure a configuration directory exists.
759
This makes sure that the directory exists.
760
On windows, since configuration directories are 2 levels deep,
761
it makes sure both the directory and the parent directory exists.
765
if not os.path.isdir(path):
766
if sys.platform == 'win32':
767
parent_dir = os.path.dirname(path)
768
if not os.path.isdir(parent_dir):
769
trace.mutter('creating config parent directory: %r', parent_dir)
771
trace.mutter('creating config directory: %r', path)
408
return self._get_location_config()._post_commit()
775
411
def config_dir():
776
412
"""Return per-user configuration directory.
778
414
By default this is ~/.bazaar/
780
416
TODO: Global option --config-dir to override this.
782
418
base = os.environ.get('BZR_HOME', None)
783
419
if sys.platform == 'win32':
785
base = win32utils.get_appdata_location_unicode()
421
base = os.environ.get('APPDATA', None)
787
423
base = os.environ.get('HOME', None)
789
raise errors.BzrError('You must have one of BZR_HOME, APPDATA,'
791
return osutils.pathjoin(base, 'bazaar', '2.0')
425
raise BzrError('You must have one of BZR_HOME, APPDATA, or HOME set')
426
return os.path.join(base, 'bazaar', '2.0')
793
428
# cygwin, linux, and darwin all have a $HOME directory
795
430
base = os.path.expanduser("~")
796
return osutils.pathjoin(base, ".bazaar")
431
return os.path.join(base, ".bazaar")
799
434
def config_filename():
800
435
"""Return per-user configuration ini file filename."""
801
return osutils.pathjoin(config_dir(), 'bazaar.conf')
436
return os.path.join(config_dir(), 'bazaar.conf')
804
439
def branches_config_filename():
805
440
"""Return per-user configuration ini file filename."""
806
return osutils.pathjoin(config_dir(), 'branches.conf')
809
def locations_config_filename():
810
"""Return per-user configuration ini file filename."""
811
return osutils.pathjoin(config_dir(), 'locations.conf')
814
def authentication_config_filename():
815
"""Return per-user authentication ini file filename."""
816
return osutils.pathjoin(config_dir(), 'authentication.conf')
819
def user_ignore_config_filename():
820
"""Return the user default ignore filename"""
821
return osutils.pathjoin(config_dir(), 'ignore')
441
return os.path.join(config_dir(), 'branches.conf')
824
444
def _auto_user_id():
885
473
except ImportError:
888
user_encoding = osutils.get_user_encoding()
889
realname = username = getpass.getuser().decode(user_encoding)
890
except UnicodeDecodeError:
891
raise errors.BzrError("Can't decode username as %s." % \
475
realname = username = getpass.getuser().decode(bzrlib.user_encoding)
894
477
return realname, (username + '@' + socket.gethostname())
897
def parse_username(username):
898
"""Parse e-mail username and return a (name, address) tuple."""
899
match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
901
return (username, '')
903
return (match.group(1), match.group(2))
906
480
def extract_email_address(e):
907
481
"""Return just the address part of an email string.
909
That is just the user@domain part, nothing else.
483
That is just the user@domain part, nothing else.
910
484
This part is required to contain only ascii characters.
911
485
If it can't be extracted, raises an error.
913
487
>>> extract_email_address('Jane Tester <jane@test.com>')
916
name, email = parse_username(e)
918
raise errors.NoEmailInUsername(e)
922
class TreeConfig(IniBasedConfig):
490
m = re.search(r'[\w+.-]+@[\w+.-]+', e)
492
raise errors.BzrError("%r doesn't seem to contain "
493
"a reasonable email address" % e)
496
class TreeConfig(object):
923
497
"""Branch configuration data associated with its contents, not location"""
925
# XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
927
498
def __init__(self, branch):
928
self._config = branch._get_config()
929
499
self.branch = branch
931
def _get_parser(self, file=None):
933
return IniBasedConfig._get_parser(file)
934
return self._config._get_configobj()
501
def _get_config(self):
503
obj = ConfigObj(self.branch.controlfile('branch.conf',
506
except errors.NoSuchFile:
936
510
def get_option(self, name, section=None, default=None):
937
511
self.branch.lock_read()
939
return self._config.get_option(name, section, default)
513
obj = self._get_config()
515
if section is not None:
941
521
self.branch.unlock()
943
524
def set_option(self, value, name, section=None):
944
525
"""Set a per-branch configuration option"""
945
526
self.branch.lock_write()
947
self._config.set_option(value, name, section)
528
cfg_obj = self._get_config()
533
obj = cfg_obj[section]
535
cfg_obj[section] = {}
536
obj = cfg_obj[section]
538
cfg_obj.encode('UTF-8')
539
out_file = StringIO(''.join([l+'\n' for l in cfg_obj.write()]))
541
self.branch.put_controlfile('branch.conf', out_file, encode=False)
949
543
self.branch.unlock()
952
class AuthenticationConfig(object):
953
"""The authentication configuration file based on a ini file.
955
Implements the authentication.conf file described in
956
doc/developers/authentication-ring.txt.
959
def __init__(self, _file=None):
960
self._config = None # The ConfigObj
962
self._filename = authentication_config_filename()
963
self._input = self._filename = authentication_config_filename()
965
# Tests can provide a string as _file
966
self._filename = None
969
def _get_config(self):
970
if self._config is not None:
973
# FIXME: Should we validate something here ? Includes: empty
974
# sections are useless, at least one of
975
# user/password/password_encoding should be defined, etc.
977
# Note: the encoding below declares that the file itself is utf-8
978
# encoded, but the values in the ConfigObj are always Unicode.
979
self._config = ConfigObj(self._input, encoding='utf-8')
980
except configobj.ConfigObjError, e:
981
raise errors.ParseConfigError(e.errors, e.config.filename)
985
"""Save the config file, only tests should use it for now."""
986
conf_dir = os.path.dirname(self._filename)
987
ensure_config_dir_exists(conf_dir)
988
self._get_config().write(file(self._filename, 'wb'))
990
def _set_option(self, section_name, option_name, value):
991
"""Set an authentication configuration option"""
992
conf = self._get_config()
993
section = conf.get(section_name)
996
section = conf[section]
997
section[option_name] = value
1000
def get_credentials(self, scheme, host, port=None, user=None, path=None,
1002
"""Returns the matching credentials from authentication.conf file.
1004
:param scheme: protocol
1006
:param host: the server address
1008
:param port: the associated port (optional)
1010
:param user: login (optional)
1012
:param path: the absolute path on the server (optional)
1014
:param realm: the http authentication realm (optional)
1016
:return: A dict containing the matching credentials or None.
1018
- name: the section name of the credentials in the
1019
authentication.conf file,
1020
- user: can't be different from the provided user if any,
1021
- scheme: the server protocol,
1022
- host: the server address,
1023
- port: the server port (can be None),
1024
- path: the absolute server path (can be None),
1025
- realm: the http specific authentication realm (can be None),
1026
- password: the decoded password, could be None if the credential
1027
defines only the user
1028
- verify_certificates: https specific, True if the server
1029
certificate should be verified, False otherwise.
1032
for auth_def_name, auth_def in self._get_config().items():
1033
if type(auth_def) is not configobj.Section:
1034
raise ValueError("%s defined outside a section" % auth_def_name)
1036
a_scheme, a_host, a_user, a_path = map(
1037
auth_def.get, ['scheme', 'host', 'user', 'path'])
1040
a_port = auth_def.as_int('port')
1044
raise ValueError("'port' not numeric in %s" % auth_def_name)
1046
a_verify_certificates = auth_def.as_bool('verify_certificates')
1048
a_verify_certificates = True
1051
"'verify_certificates' not boolean in %s" % auth_def_name)
1054
if a_scheme is not None and scheme != a_scheme:
1056
if a_host is not None:
1057
if not (host == a_host
1058
or (a_host.startswith('.') and host.endswith(a_host))):
1060
if a_port is not None and port != a_port:
1062
if (a_path is not None and path is not None
1063
and not path.startswith(a_path)):
1065
if (a_user is not None and user is not None
1066
and a_user != user):
1067
# Never contradict the caller about the user to be used
1072
# Prepare a credentials dictionary with additional keys
1073
# for the credential providers
1074
credentials = dict(name=auth_def_name,
1081
password=auth_def.get('password', None),
1082
verify_certificates=a_verify_certificates)
1083
# Decode the password in the credentials (or get one)
1084
self.decode_password(credentials,
1085
auth_def.get('password_encoding', None))
1086
if 'auth' in debug.debug_flags:
1087
trace.mutter("Using authentication section: %r", auth_def_name)
1090
if credentials is None:
1091
# No credentials were found in authentication.conf, try the fallback
1092
# credentials stores.
1093
credentials = credential_store_registry.get_fallback_credentials(
1094
scheme, host, port, user, path, realm)
1098
def set_credentials(self, name, host, user, scheme=None, password=None,
1099
port=None, path=None, verify_certificates=None,
1101
"""Set authentication credentials for a host.
1103
Any existing credentials with matching scheme, host, port and path
1104
will be deleted, regardless of name.
1106
:param name: An arbitrary name to describe this set of credentials.
1107
:param host: Name of the host that accepts these credentials.
1108
:param user: The username portion of these credentials.
1109
:param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1111
:param password: Password portion of these credentials.
1112
:param port: The IP port on the host that these credentials apply to.
1113
:param path: A filesystem path on the host that these credentials
1115
:param verify_certificates: On https, verify server certificates if
1117
:param realm: The http authentication realm (optional).
1119
values = {'host': host, 'user': user}
1120
if password is not None:
1121
values['password'] = password
1122
if scheme is not None:
1123
values['scheme'] = scheme
1124
if port is not None:
1125
values['port'] = '%d' % port
1126
if path is not None:
1127
values['path'] = path
1128
if verify_certificates is not None:
1129
values['verify_certificates'] = str(verify_certificates)
1130
if realm is not None:
1131
values['realm'] = realm
1132
config = self._get_config()
1134
for section, existing_values in config.items():
1135
for key in ('scheme', 'host', 'port', 'path', 'realm'):
1136
if existing_values.get(key) != values.get(key):
1140
config.update({name: values})
1143
def get_user(self, scheme, host, port=None, realm=None, path=None,
1144
prompt=None, ask=False, default=None):
1145
"""Get a user from authentication file.
1147
:param scheme: protocol
1149
:param host: the server address
1151
:param port: the associated port (optional)
1153
:param realm: the realm sent by the server (optional)
1155
:param path: the absolute path on the server (optional)
1157
:param ask: Ask the user if there is no explicitly configured username
1160
:param default: The username returned if none is defined (optional).
1162
:return: The found user.
1164
credentials = self.get_credentials(scheme, host, port, user=None,
1165
path=path, realm=realm)
1166
if credentials is not None:
1167
user = credentials['user']
1173
# Create a default prompt suitable for most cases
1174
prompt = scheme.upper() + ' %(host)s username'
1175
# Special handling for optional fields in the prompt
1176
if port is not None:
1177
prompt_host = '%s:%d' % (host, port)
1180
user = ui.ui_factory.get_username(prompt, host=prompt_host)
1185
def get_password(self, scheme, host, user, port=None,
1186
realm=None, path=None, prompt=None):
1187
"""Get a password from authentication file or prompt the user for one.
1189
:param scheme: protocol
1191
:param host: the server address
1193
:param port: the associated port (optional)
1197
:param realm: the realm sent by the server (optional)
1199
:param path: the absolute path on the server (optional)
1201
:return: The found password or the one entered by the user.
1203
credentials = self.get_credentials(scheme, host, port, user, path,
1205
if credentials is not None:
1206
password = credentials['password']
1207
if password is not None and scheme is 'ssh':
1208
trace.warning('password ignored in section [%s],'
1209
' use an ssh agent instead'
1210
% credentials['name'])
1214
# Prompt user only if we could't find a password
1215
if password is None:
1217
# Create a default prompt suitable for most cases
1218
prompt = '%s' % scheme.upper() + ' %(user)s@%(host)s password'
1219
# Special handling for optional fields in the prompt
1220
if port is not None:
1221
prompt_host = '%s:%d' % (host, port)
1224
password = ui.ui_factory.get_password(prompt,
1225
host=prompt_host, user=user)
1228
def decode_password(self, credentials, encoding):
1230
cs = credential_store_registry.get_credential_store(encoding)
1232
raise ValueError('%r is not a known password_encoding' % encoding)
1233
credentials['password'] = cs.decode_password(credentials)
1237
class CredentialStoreRegistry(registry.Registry):
1238
"""A class that registers credential stores.
1240
A credential store provides access to credentials via the password_encoding
1241
field in authentication.conf sections.
1243
Except for stores provided by bzr itself, most stores are expected to be
1244
provided by plugins that will therefore use
1245
register_lazy(password_encoding, module_name, member_name, help=help,
1246
fallback=fallback) to install themselves.
1248
A fallback credential store is one that is queried if no credentials can be
1249
found via authentication.conf.
1252
def get_credential_store(self, encoding=None):
1253
cs = self.get(encoding)
1258
def is_fallback(self, name):
1259
"""Check if the named credentials store should be used as fallback."""
1260
return self.get_info(name)
1262
def get_fallback_credentials(self, scheme, host, port=None, user=None,
1263
path=None, realm=None):
1264
"""Request credentials from all fallback credentials stores.
1266
The first credentials store that can provide credentials wins.
1269
for name in self.keys():
1270
if not self.is_fallback(name):
1272
cs = self.get_credential_store(name)
1273
credentials = cs.get_credentials(scheme, host, port, user,
1275
if credentials is not None:
1276
# We found some credentials
1280
def register(self, key, obj, help=None, override_existing=False,
1282
"""Register a new object to a name.
1284
:param key: This is the key to use to request the object later.
1285
:param obj: The object to register.
1286
:param help: Help text for this entry. This may be a string or
1287
a callable. If it is a callable, it should take two
1288
parameters (registry, key): this registry and the key that
1289
the help was registered under.
1290
:param override_existing: Raise KeyErorr if False and something has
1291
already been registered for that key. If True, ignore if there
1292
is an existing key (always register the new value).
1293
:param fallback: Whether this credential store should be
1296
return super(CredentialStoreRegistry,
1297
self).register(key, obj, help, info=fallback,
1298
override_existing=override_existing)
1300
def register_lazy(self, key, module_name, member_name,
1301
help=None, override_existing=False,
1303
"""Register a new credential store to be loaded on request.
1305
:param module_name: The python path to the module. Such as 'os.path'.
1306
:param member_name: The member of the module to return. If empty or
1307
None, get() will return the module itself.
1308
:param help: Help text for this entry. This may be a string or
1310
:param override_existing: If True, replace the existing object
1311
with the new one. If False, if there is already something
1312
registered with the same key, raise a KeyError
1313
:param fallback: Whether this credential store should be
1316
return super(CredentialStoreRegistry, self).register_lazy(
1317
key, module_name, member_name, help,
1318
info=fallback, override_existing=override_existing)
1321
credential_store_registry = CredentialStoreRegistry()
1324
class CredentialStore(object):
1325
"""An abstract class to implement storage for credentials"""
1327
def decode_password(self, credentials):
1328
"""Returns a clear text password for the provided credentials."""
1329
raise NotImplementedError(self.decode_password)
1331
def get_credentials(self, scheme, host, port=None, user=None, path=None,
1333
"""Return the matching credentials from this credential store.
1335
This method is only called on fallback credential stores.
1337
raise NotImplementedError(self.get_credentials)
1341
class PlainTextCredentialStore(CredentialStore):
1342
"""Plain text credential store for the authentication.conf file."""
1344
def decode_password(self, credentials):
1345
"""See CredentialStore.decode_password."""
1346
return credentials['password']
1349
credential_store_registry.register('plain', PlainTextCredentialStore,
1350
help=PlainTextCredentialStore.__doc__)
1351
credential_store_registry.default_key = 'plain'
1354
class BzrDirConfig(object):
1356
def __init__(self, bzrdir):
1357
self._bzrdir = bzrdir
1358
self._config = bzrdir._get_config()
1360
def set_default_stack_on(self, value):
1361
"""Set the default stacking location.
1363
It may be set to a location, or None.
1365
This policy affects all branches contained by this bzrdir, except for
1366
those under repositories.
1368
if self._config is None:
1369
raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1371
self._config.set_option('', 'default_stack_on')
1373
self._config.set_option(value, 'default_stack_on')
1375
def get_default_stack_on(self):
1376
"""Return the default stacking location.
1378
This will either be a location, or None.
1380
This policy affects all branches contained by this bzrdir, except for
1381
those under repositories.
1383
if self._config is None:
1385
value = self._config.get_option('default_stack_on')
1391
class TransportConfig(object):
1392
"""A Config that reads/writes a config file on a Transport.
1394
It is a low-level object that considers config data to be name/value pairs
1395
that may be associated with a section. Assigning meaning to the these
1396
values is done at higher levels like TreeConfig.
1399
def __init__(self, transport, filename):
1400
self._transport = transport
1401
self._filename = filename
1403
def get_option(self, name, section=None, default=None):
1404
"""Return the value associated with a named option.
1406
:param name: The name of the value
1407
:param section: The section the option is in (if any)
1408
:param default: The value to return if the value is not set
1409
:return: The value or default value
1411
configobj = self._get_configobj()
1413
section_obj = configobj
1416
section_obj = configobj[section]
1419
return section_obj.get(name, default)
1421
def set_option(self, value, name, section=None):
1422
"""Set the value associated with a named option.
1424
:param value: The value to set
1425
:param name: The name of the value to set
1426
:param section: The section the option is in (if any)
1428
configobj = self._get_configobj()
1430
configobj[name] = value
1432
configobj.setdefault(section, {})[name] = value
1433
self._set_configobj(configobj)
1435
def _get_config_file(self):
1437
return self._transport.get(self._filename)
1438
except errors.NoSuchFile:
1441
def _get_configobj(self):
1442
return ConfigObj(self._get_config_file(), encoding='utf-8')
1444
def _set_configobj(self, configobj):
1445
out_file = StringIO()
1446
configobj.write(out_file)
1448
self._transport.put_file(self._filename, out_file)