~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-03-16 08:26:29 UTC
  • mfrom: (4145.1.6 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20090316082629-xuzqut3b3ur5bn3b
(robertc) Add a new repository method refresh_data to allow clean
        handling of _real_repositories after inserting a stream via the
        smart server. (Robert Collins, Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2007, 2008 Canonical Ltd
2
2
#   Authors: Robert Collins <robert.collins@canonical.com>
3
3
#            and others
4
4
#
14
14
#
15
15
# You should have received a copy of the GNU General Public License
16
16
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
18
 
19
19
"""Configuration that affects the behaviour of Bazaar.
20
20
 
146
146
class Config(object):
147
147
    """A configuration policy - what username, editor, gpg needs etc."""
148
148
 
149
 
    def __init__(self):
150
 
        super(Config, self).__init__()
151
 
 
152
149
    def get_editor(self):
153
150
        """Get the users pop up editor."""
154
151
        raise NotImplementedError
155
152
 
156
 
    def get_change_editor(self, old_tree, new_tree):
157
 
        from bzrlib import diff
158
 
        cmd = self._get_change_editor()
159
 
        if cmd is None:
160
 
            return None
161
 
        return diff.DiffFromTool.from_string(cmd, old_tree, new_tree,
162
 
                                             sys.stdout)
163
 
 
164
 
 
165
153
    def get_mail_client(self):
166
154
        """Get a mail client to use"""
167
155
        selected_client = self.get_user_option('mail_client')
186
174
        """Get a generic option - no special process, no default."""
187
175
        return self._get_user_option(option_name)
188
176
 
189
 
    def get_user_option_as_bool(self, option_name):
190
 
        """Get a generic option as a boolean - no special process, no default.
191
 
 
192
 
        :return None if the option doesn't exist or its value can't be
193
 
            interpreted as a boolean. Returns True or False otherwise.
194
 
        """
195
 
        s = self._get_user_option(option_name)
196
 
        if s is None:
197
 
            # The option doesn't exist
198
 
            return None
199
 
        val = ui.bool_from_string(s)
200
 
        if val is None:
201
 
            # The value can't be interpreted as a boolean
202
 
            trace.warning('Value "%s" is not a boolean for "%s"',
203
 
                          s, option_name)
204
 
        return val
205
 
 
206
 
    def get_user_option_as_list(self, option_name):
207
 
        """Get a generic option as a list - no special process, no default.
208
 
 
209
 
        :return None if the option doesn't exist. Returns the value as a list
210
 
            otherwise.
211
 
        """
212
 
        l = self._get_user_option(option_name)
213
 
        if isinstance(l, (str, unicode)):
214
 
            # A single value, most probably the user forgot the final ','
215
 
            l = [l]
216
 
        return l
217
 
 
218
177
    def gpg_signing_command(self):
219
178
        """What program should be used to sign signatures?"""
220
179
        result = self._gpg_signing_command()
237
196
        """See log_format()."""
238
197
        return None
239
198
 
 
199
    def __init__(self):
 
200
        super(Config, self).__init__()
 
201
 
240
202
    def post_commit(self):
241
203
        """An ordered list of python functions to call.
242
204
 
333
295
                path = 'bzr'
334
296
            return path
335
297
 
336
 
    def suppress_warning(self, warning):
337
 
        """Should the warning be suppressed or emitted.
338
 
 
339
 
        :param warning: The name of the warning being tested.
340
 
 
341
 
        :returns: True if the warning should be suppressed, False otherwise.
342
 
        """
343
 
        warnings = self.get_user_option_as_list('suppress_warnings')
344
 
        if warnings is None or warning not in warnings:
345
 
            return False
346
 
        else:
347
 
            return True
348
 
 
349
298
 
350
299
class IniBasedConfig(Config):
351
300
    """A configuration policy that draws from ini files."""
352
301
 
353
 
    def __init__(self, get_filename):
354
 
        super(IniBasedConfig, self).__init__()
355
 
        self._get_filename = get_filename
356
 
        self._parser = None
357
 
 
358
302
    def _get_parser(self, file=None):
359
303
        if self._parser is not None:
360
304
            return self._parser
388
332
        """Return the policy for the given (section, option_name) pair."""
389
333
        return POLICY_NONE
390
334
 
391
 
    def _get_change_editor(self):
392
 
        return self.get_user_option('change_editor')
393
 
 
394
335
    def _get_signature_checking(self):
395
336
        """See Config._get_signature_checking."""
396
337
        policy = self._get_user_option('check_signatures')
440
381
        """See Config.log_format."""
441
382
        return self._get_user_option('log_format')
442
383
 
 
384
    def __init__(self, get_filename):
 
385
        super(IniBasedConfig, self).__init__()
 
386
        self._get_filename = get_filename
 
387
        self._parser = None
 
388
 
443
389
    def _post_commit(self):
444
390
        """See Config.post_commit."""
445
391
        return self._get_user_option('post_commit')
518
464
        self._write_config_file()
519
465
 
520
466
    def _write_config_file(self):
521
 
        path = self._get_filename()
522
 
        f = open(path, 'wb')
523
 
        osutils.copy_ownership_from_path(path)
 
467
        f = open(self._get_filename(), 'wb')
524
468
        self._get_parser().write(f)
525
469
        f.close()
526
470
 
726
670
 
727
671
        return self._get_best_value('_get_user_id')
728
672
 
729
 
    def _get_change_editor(self):
730
 
        return self._get_best_value('_get_change_editor')
731
 
 
732
673
    def _get_signature_checking(self):
733
674
        """See Config._get_signature_checking."""
734
675
        return self._get_best_value('_get_signature_checking')
768
709
                        trace.warning('Value "%s" is masked by "%s" from'
769
710
                                      ' branch.conf', value, mask_value)
770
711
 
 
712
 
771
713
    def _gpg_signing_command(self):
772
714
        """See Config.gpg_signing_command."""
773
715
        return self._get_safe_value('_gpg_signing_command')
820
762
            os.mkdir(parent_dir)
821
763
        trace.mutter('creating config directory: %r', path)
822
764
        os.mkdir(path)
823
 
        osutils.copy_ownership_from_path(path)
824
765
 
825
766
 
826
767
def config_dir():
872
813
    return osutils.pathjoin(config_dir(), 'ignore')
873
814
 
874
815
 
875
 
def crash_dir():
876
 
    """Return the directory name to store crash files.
877
 
 
878
 
    This doesn't implicitly create it.
879
 
 
880
 
    On Windows it's in the config directory; elsewhere it's /var/crash
881
 
    which may be monitored by apport.  It can be overridden by
882
 
    $APPORT_CRASH_DIR.
883
 
    """
884
 
    if sys.platform == 'win32':
885
 
        return osutils.pathjoin(config_dir(), 'Crash')
886
 
    else:
887
 
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
888
 
        # 2010-01-31
889
 
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
890
 
 
891
 
 
892
 
def xdg_cache_dir():
893
 
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
894
 
    # Possibly this should be different on Windows?
895
 
    e = os.environ.get('XDG_CACHE_DIR', None)
896
 
    if e:
897
 
        return e
898
 
    else:
899
 
        return os.path.expanduser('~/.cache')
900
 
 
901
 
 
902
816
def _auto_user_id():
903
817
    """Calculate automatic user identification.
904
818
 
1003
917
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
1004
918
 
1005
919
    def __init__(self, branch):
1006
 
        self._config = branch._get_config()
 
920
        # XXX: Really this should be asking the branch for its configuration
 
921
        # data, rather than relying on a Transport, so that it can work
 
922
        # more cleanly with a RemoteBranch that has no transport.
 
923
        self._config = TransportConfig(branch._transport, 'branch.conf')
1007
924
        self.branch = branch
1008
925
 
1009
926
    def _get_parser(self, file=None):
1017
934
            return self._config.get_option(name, section, default)
1018
935
        finally:
1019
936
            self.branch.unlock()
 
937
        return result
1020
938
 
1021
939
    def set_option(self, value, name, section=None):
1022
940
        """Set a per-branch configuration option"""
1075
993
        section[option_name] = value
1076
994
        self._save()
1077
995
 
1078
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1079
 
                        realm=None):
 
996
    def get_credentials(self, scheme, host, port=None, user=None, path=None):
1080
997
        """Returns the matching credentials from authentication.conf file.
1081
998
 
1082
999
        :param scheme: protocol
1088
1005
        :param user: login (optional)
1089
1006
 
1090
1007
        :param path: the absolute path on the server (optional)
1091
 
        
1092
 
        :param realm: the http authentication realm (optional)
1093
1008
 
1094
1009
        :return: A dict containing the matching credentials or None.
1095
1010
           This includes:
1096
1011
           - name: the section name of the credentials in the
1097
1012
             authentication.conf file,
1098
 
           - user: can't be different from the provided user if any,
1099
 
           - scheme: the server protocol,
1100
 
           - host: the server address,
1101
 
           - port: the server port (can be None),
1102
 
           - path: the absolute server path (can be None),
1103
 
           - realm: the http specific authentication realm (can be None),
 
1013
           - user: can't de different from the provided user if any,
1104
1014
           - password: the decoded password, could be None if the credential
1105
1015
             defines only the user
1106
1016
           - verify_certificates: https specific, True if the server
1147
1057
            if a_user is None:
1148
1058
                # Can't find a user
1149
1059
                continue
1150
 
            # Prepare a credentials dictionary with additional keys
1151
 
            # for the credential providers
1152
1060
            credentials = dict(name=auth_def_name,
1153
1061
                               user=a_user,
1154
 
                               scheme=a_scheme,
1155
 
                               host=host,
1156
 
                               port=port,
1157
 
                               path=path,
1158
 
                               realm=realm,
1159
1062
                               password=auth_def.get('password', None),
1160
1063
                               verify_certificates=a_verify_certificates)
1161
 
            # Decode the password in the credentials (or get one)
1162
1064
            self.decode_password(credentials,
1163
1065
                                 auth_def.get('password_encoding', None))
1164
1066
            if 'auth' in debug.debug_flags:
1165
1067
                trace.mutter("Using authentication section: %r", auth_def_name)
1166
1068
            break
1167
1069
 
1168
 
        if credentials is None:
1169
 
            # No credentials were found in authentication.conf, try the fallback
1170
 
            # credentials stores.
1171
 
            credentials = credential_store_registry.get_fallback_credentials(
1172
 
                scheme, host, port, user, path, realm)
1173
 
 
1174
1070
        return credentials
1175
1071
 
1176
1072
    def set_credentials(self, name, host, user, scheme=None, password=None,
1177
 
                        port=None, path=None, verify_certificates=None,
1178
 
                        realm=None):
 
1073
                        port=None, path=None, verify_certificates=None):
1179
1074
        """Set authentication credentials for a host.
1180
1075
 
1181
1076
        Any existing credentials with matching scheme, host, port and path
1192
1087
            apply to.
1193
1088
        :param verify_certificates: On https, verify server certificates if
1194
1089
            True.
1195
 
        :param realm: The http authentication realm (optional).
1196
1090
        """
1197
1091
        values = {'host': host, 'user': user}
1198
1092
        if password is not None:
1205
1099
            values['path'] = path
1206
1100
        if verify_certificates is not None:
1207
1101
            values['verify_certificates'] = str(verify_certificates)
1208
 
        if realm is not None:
1209
 
            values['realm'] = realm
1210
1102
        config = self._get_config()
1211
1103
        for_deletion = []
1212
1104
        for section, existing_values in config.items():
1213
 
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
 
1105
            for key in ('scheme', 'host', 'port', 'path'):
1214
1106
                if existing_values.get(key) != values.get(key):
1215
1107
                    break
1216
1108
            else:
1218
1110
        config.update({name: values})
1219
1111
        self._save()
1220
1112
 
1221
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1222
 
                 prompt=None, ask=False, default=None):
 
1113
    def get_user(self, scheme, host, port=None,
 
1114
                 realm=None, path=None, prompt=None):
1223
1115
        """Get a user from authentication file.
1224
1116
 
1225
1117
        :param scheme: protocol
1232
1124
 
1233
1125
        :param path: the absolute path on the server (optional)
1234
1126
 
1235
 
        :param ask: Ask the user if there is no explicitly configured username 
1236
 
                    (optional)
1237
 
 
1238
 
        :param default: The username returned if none is defined (optional).
1239
 
 
1240
1127
        :return: The found user.
1241
1128
        """
1242
1129
        credentials = self.get_credentials(scheme, host, port, user=None,
1243
 
                                           path=path, realm=realm)
 
1130
                                           path=path)
1244
1131
        if credentials is not None:
1245
1132
            user = credentials['user']
1246
1133
        else:
1247
1134
            user = None
1248
 
        if user is None:
1249
 
            if ask:
1250
 
                if prompt is None:
1251
 
                    # Create a default prompt suitable for most cases
1252
 
                    prompt = scheme.upper() + ' %(host)s username'
1253
 
                # Special handling for optional fields in the prompt
1254
 
                if port is not None:
1255
 
                    prompt_host = '%s:%d' % (host, port)
1256
 
                else:
1257
 
                    prompt_host = host
1258
 
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
1259
 
            else:
1260
 
                user = default
1261
1135
        return user
1262
1136
 
1263
1137
    def get_password(self, scheme, host, user, port=None,
1278
1152
 
1279
1153
        :return: The found password or the one entered by the user.
1280
1154
        """
1281
 
        credentials = self.get_credentials(scheme, host, port, user, path,
1282
 
                                           realm)
 
1155
        credentials = self.get_credentials(scheme, host, port, user, path)
1283
1156
        if credentials is not None:
1284
1157
            password = credentials['password']
1285
1158
            if password is not None and scheme is 'ssh':
1318
1191
    A credential store provides access to credentials via the password_encoding
1319
1192
    field in authentication.conf sections.
1320
1193
 
1321
 
    Except for stores provided by bzr itself, most stores are expected to be
 
1194
    Except for stores provided by bzr itself,most stores are expected to be
1322
1195
    provided by plugins that will therefore use
1323
1196
    register_lazy(password_encoding, module_name, member_name, help=help,
1324
 
    fallback=fallback) to install themselves.
1325
 
 
1326
 
    A fallback credential store is one that is queried if no credentials can be
1327
 
    found via authentication.conf.
 
1197
    info=info) to install themselves.
1328
1198
    """
1329
1199
 
1330
1200
    def get_credential_store(self, encoding=None):
1333
1203
            cs = cs()
1334
1204
        return cs
1335
1205
 
1336
 
    def is_fallback(self, name):
1337
 
        """Check if the named credentials store should be used as fallback."""
1338
 
        return self.get_info(name)
1339
 
 
1340
 
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
1341
 
                                 path=None, realm=None):
1342
 
        """Request credentials from all fallback credentials stores.
1343
 
 
1344
 
        The first credentials store that can provide credentials wins.
1345
 
        """
1346
 
        credentials = None
1347
 
        for name in self.keys():
1348
 
            if not self.is_fallback(name):
1349
 
                continue
1350
 
            cs = self.get_credential_store(name)
1351
 
            credentials = cs.get_credentials(scheme, host, port, user,
1352
 
                                             path, realm)
1353
 
            if credentials is not None:
1354
 
                # We found some credentials
1355
 
                break
1356
 
        return credentials
1357
 
 
1358
 
    def register(self, key, obj, help=None, override_existing=False,
1359
 
                 fallback=False):
1360
 
        """Register a new object to a name.
1361
 
 
1362
 
        :param key: This is the key to use to request the object later.
1363
 
        :param obj: The object to register.
1364
 
        :param help: Help text for this entry. This may be a string or
1365
 
                a callable. If it is a callable, it should take two
1366
 
                parameters (registry, key): this registry and the key that
1367
 
                the help was registered under.
1368
 
        :param override_existing: Raise KeyErorr if False and something has
1369
 
                already been registered for that key. If True, ignore if there
1370
 
                is an existing key (always register the new value).
1371
 
        :param fallback: Whether this credential store should be 
1372
 
                used as fallback.
1373
 
        """
1374
 
        return super(CredentialStoreRegistry,
1375
 
                     self).register(key, obj, help, info=fallback,
1376
 
                                    override_existing=override_existing)
1377
 
 
1378
 
    def register_lazy(self, key, module_name, member_name,
1379
 
                      help=None, override_existing=False,
1380
 
                      fallback=False):
1381
 
        """Register a new credential store to be loaded on request.
1382
 
 
1383
 
        :param module_name: The python path to the module. Such as 'os.path'.
1384
 
        :param member_name: The member of the module to return.  If empty or
1385
 
                None, get() will return the module itself.
1386
 
        :param help: Help text for this entry. This may be a string or
1387
 
                a callable.
1388
 
        :param override_existing: If True, replace the existing object
1389
 
                with the new one. If False, if there is already something
1390
 
                registered with the same key, raise a KeyError
1391
 
        :param fallback: Whether this credential store should be 
1392
 
                used as fallback.
1393
 
        """
1394
 
        return super(CredentialStoreRegistry, self).register_lazy(
1395
 
            key, module_name, member_name, help,
1396
 
            info=fallback, override_existing=override_existing)
1397
 
 
1398
1206
 
1399
1207
credential_store_registry = CredentialStoreRegistry()
1400
1208
 
1403
1211
    """An abstract class to implement storage for credentials"""
1404
1212
 
1405
1213
    def decode_password(self, credentials):
1406
 
        """Returns a clear text password for the provided credentials."""
 
1214
        """Returns a password for the provided credentials in clear text."""
1407
1215
        raise NotImplementedError(self.decode_password)
1408
1216
 
1409
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
1410
 
                        realm=None):
1411
 
        """Return the matching credentials from this credential store.
1412
 
 
1413
 
        This method is only called on fallback credential stores.
1414
 
        """
1415
 
        raise NotImplementedError(self.get_credentials)
1416
 
 
1417
 
 
1418
1217
 
1419
1218
class PlainTextCredentialStore(CredentialStore):
1420
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1219
    """Plain text credential store for the authentication.conf file."""
1421
1220
 
1422
1221
    def decode_password(self, credentials):
1423
1222
        """See CredentialStore.decode_password."""
1431
1230
 
1432
1231
class BzrDirConfig(object):
1433
1232
 
1434
 
    def __init__(self, bzrdir):
1435
 
        self._bzrdir = bzrdir
1436
 
        self._config = bzrdir._get_config()
 
1233
    def __init__(self, transport):
 
1234
        self._config = TransportConfig(transport, 'control.conf')
1437
1235
 
1438
1236
    def set_default_stack_on(self, value):
1439
1237
        """Set the default stacking location.
1443
1241
        This policy affects all branches contained by this bzrdir, except for
1444
1242
        those under repositories.
1445
1243
        """
1446
 
        if self._config is None:
1447
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1448
1244
        if value is None:
1449
1245
            self._config.set_option('', 'default_stack_on')
1450
1246
        else:
1458
1254
        This policy affects all branches contained by this bzrdir, except for
1459
1255
        those under repositories.
1460
1256
        """
1461
 
        if self._config is None:
1462
 
            return None
1463
1257
        value = self._config.get_option('default_stack_on')
1464
1258
        if value == '':
1465
1259
            value = None
1510
1304
            configobj.setdefault(section, {})[name] = value
1511
1305
        self._set_configobj(configobj)
1512
1306
 
1513
 
    def _get_config_file(self):
 
1307
    def _get_configobj(self):
1514
1308
        try:
1515
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1309
            return ConfigObj(self._transport.get(self._filename),
 
1310
                             encoding='utf-8')
1516
1311
        except errors.NoSuchFile:
1517
 
            return StringIO()
1518
 
 
1519
 
    def _get_configobj(self):
1520
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1312
            return ConfigObj(encoding='utf-8')
1521
1313
 
1522
1314
    def _set_configobj(self, configobj):
1523
1315
        out_file = StringIO()