~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Andrew Bennetts
  • Date: 2009-04-02 05:53:12 UTC
  • mto: This revision was merged to the branch mainline in revision 4242.
  • Revision ID: andrew.bennetts@canonical.com-20090402055312-h7mvgumvm7e620mj
Fix nits in spelling and naming.

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
#
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"""
1165
1083
                trace.mutter("Using authentication section: %r", auth_def_name)
1166
1084
            break
1167
1085
 
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
1086
        return credentials
1175
1087
 
1176
1088
    def set_credentials(self, name, host, user, scheme=None, password=None,
1218
1130
        config.update({name: values})
1219
1131
        self._save()
1220
1132
 
1221
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1222
 
                 prompt=None, ask=False, default=None):
 
1133
    def get_user(self, scheme, host, port=None,
 
1134
                 realm=None, path=None, prompt=None):
1223
1135
        """Get a user from authentication file.
1224
1136
 
1225
1137
        :param scheme: protocol
1232
1144
 
1233
1145
        :param path: the absolute path on the server (optional)
1234
1146
 
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
1147
        :return: The found user.
1241
1148
        """
1242
1149
        credentials = self.get_credentials(scheme, host, port, user=None,
1245
1152
            user = credentials['user']
1246
1153
        else:
1247
1154
            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
1155
        return user
1262
1156
 
1263
1157
    def get_password(self, scheme, host, user, port=None,
1318
1212
    A credential store provides access to credentials via the password_encoding
1319
1213
    field in authentication.conf sections.
1320
1214
 
1321
 
    Except for stores provided by bzr itself, most stores are expected to be
 
1215
    Except for stores provided by bzr itself,most stores are expected to be
1322
1216
    provided by plugins that will therefore use
1323
1217
    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.
 
1218
    info=info) to install themselves.
1328
1219
    """
1329
1220
 
1330
1221
    def get_credential_store(self, encoding=None):
1333
1224
            cs = cs()
1334
1225
        return cs
1335
1226
 
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
1227
 
1399
1228
credential_store_registry = CredentialStoreRegistry()
1400
1229
 
1403
1232
    """An abstract class to implement storage for credentials"""
1404
1233
 
1405
1234
    def decode_password(self, credentials):
1406
 
        """Returns a clear text password for the provided credentials."""
 
1235
        """Returns a password for the provided credentials in clear text."""
1407
1236
        raise NotImplementedError(self.decode_password)
1408
1237
 
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
1238
 
1419
1239
class PlainTextCredentialStore(CredentialStore):
1420
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1240
    """Plain text credential store for the authentication.conf file."""
1421
1241
 
1422
1242
    def decode_password(self, credentials):
1423
1243
        """See CredentialStore.decode_password."""
1431
1251
 
1432
1252
class BzrDirConfig(object):
1433
1253
 
1434
 
    def __init__(self, bzrdir):
1435
 
        self._bzrdir = bzrdir
1436
 
        self._config = bzrdir._get_config()
 
1254
    def __init__(self, transport):
 
1255
        self._config = TransportConfig(transport, 'control.conf')
1437
1256
 
1438
1257
    def set_default_stack_on(self, value):
1439
1258
        """Set the default stacking location.
1443
1262
        This policy affects all branches contained by this bzrdir, except for
1444
1263
        those under repositories.
1445
1264
        """
1446
 
        if self._config is None:
1447
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1448
1265
        if value is None:
1449
1266
            self._config.set_option('', 'default_stack_on')
1450
1267
        else:
1458
1275
        This policy affects all branches contained by this bzrdir, except for
1459
1276
        those under repositories.
1460
1277
        """
1461
 
        if self._config is None:
1462
 
            return None
1463
1278
        value = self._config.get_option('default_stack_on')
1464
1279
        if value == '':
1465
1280
            value = None
1510
1325
            configobj.setdefault(section, {})[name] = value
1511
1326
        self._set_configobj(configobj)
1512
1327
 
1513
 
    def _get_config_file(self):
 
1328
    def _get_configobj(self):
1514
1329
        try:
1515
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1330
            return ConfigObj(self._transport.get(self._filename),
 
1331
                             encoding='utf-8')
1516
1332
        except errors.NoSuchFile:
1517
 
            return StringIO()
1518
 
 
1519
 
    def _get_configobj(self):
1520
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1333
            return ConfigObj(encoding='utf-8')
1521
1334
 
1522
1335
    def _set_configobj(self, configobj):
1523
1336
        out_file = StringIO()