~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Aaron Bentley
  • Date: 2009-04-14 15:02:46 UTC
  • mto: This revision was merged to the branch mainline in revision 4302.
  • Revision ID: aaron@aaronbentley.com-20090414150246-6xfd3bwwf2m8v4px
Cleanup

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')
820
761
            os.mkdir(parent_dir)
821
762
        trace.mutter('creating config directory: %r', path)
822
763
        os.mkdir(path)
823
 
        osutils.copy_ownership_from_path(path)
824
764
 
825
765
 
826
766
def config_dir():
872
812
    return osutils.pathjoin(config_dir(), 'ignore')
873
813
 
874
814
 
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
815
def _auto_user_id():
903
816
    """Calculate automatic user identification.
904
817
 
1017
930
            return self._config.get_option(name, section, default)
1018
931
        finally:
1019
932
            self.branch.unlock()
 
933
        return result
1020
934
 
1021
935
    def set_option(self, value, name, section=None):
1022
936
        """Set a per-branch configuration option"""
1165
1079
                trace.mutter("Using authentication section: %r", auth_def_name)
1166
1080
            break
1167
1081
 
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
1082
        return credentials
1175
1083
 
1176
1084
    def set_credentials(self, name, host, user, scheme=None, password=None,
1218
1126
        config.update({name: values})
1219
1127
        self._save()
1220
1128
 
1221
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1222
 
                 prompt=None, ask=False, default=None):
 
1129
    def get_user(self, scheme, host, port=None,
 
1130
                 realm=None, path=None, prompt=None):
1223
1131
        """Get a user from authentication file.
1224
1132
 
1225
1133
        :param scheme: protocol
1232
1140
 
1233
1141
        :param path: the absolute path on the server (optional)
1234
1142
 
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
1143
        :return: The found user.
1241
1144
        """
1242
1145
        credentials = self.get_credentials(scheme, host, port, user=None,
1245
1148
            user = credentials['user']
1246
1149
        else:
1247
1150
            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
1151
        return user
1262
1152
 
1263
1153
    def get_password(self, scheme, host, user, port=None,
1318
1208
    A credential store provides access to credentials via the password_encoding
1319
1209
    field in authentication.conf sections.
1320
1210
 
1321
 
    Except for stores provided by bzr itself, most stores are expected to be
 
1211
    Except for stores provided by bzr itself,most stores are expected to be
1322
1212
    provided by plugins that will therefore use
1323
1213
    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.
 
1214
    info=info) to install themselves.
1328
1215
    """
1329
1216
 
1330
1217
    def get_credential_store(self, encoding=None):
1333
1220
            cs = cs()
1334
1221
        return cs
1335
1222
 
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
1223
 
1399
1224
credential_store_registry = CredentialStoreRegistry()
1400
1225
 
1403
1228
    """An abstract class to implement storage for credentials"""
1404
1229
 
1405
1230
    def decode_password(self, credentials):
1406
 
        """Returns a clear text password for the provided credentials."""
 
1231
        """Returns a password for the provided credentials in clear text."""
1407
1232
        raise NotImplementedError(self.decode_password)
1408
1233
 
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
1234
 
1419
1235
class PlainTextCredentialStore(CredentialStore):
1420
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1236
    """Plain text credential store for the authentication.conf file."""
1421
1237
 
1422
1238
    def decode_password(self, credentials):
1423
1239
        """See CredentialStore.decode_password."""
1431
1247
 
1432
1248
class BzrDirConfig(object):
1433
1249
 
1434
 
    def __init__(self, bzrdir):
1435
 
        self._bzrdir = bzrdir
1436
 
        self._config = bzrdir._get_config()
 
1250
    def __init__(self, transport):
 
1251
        self._config = TransportConfig(transport, 'control.conf')
1437
1252
 
1438
1253
    def set_default_stack_on(self, value):
1439
1254
        """Set the default stacking location.
1443
1258
        This policy affects all branches contained by this bzrdir, except for
1444
1259
        those under repositories.
1445
1260
        """
1446
 
        if self._config is None:
1447
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1448
1261
        if value is None:
1449
1262
            self._config.set_option('', 'default_stack_on')
1450
1263
        else:
1458
1271
        This policy affects all branches contained by this bzrdir, except for
1459
1272
        those under repositories.
1460
1273
        """
1461
 
        if self._config is None:
1462
 
            return None
1463
1274
        value = self._config.get_option('default_stack_on')
1464
1275
        if value == '':
1465
1276
            value = None
1510
1321
            configobj.setdefault(section, {})[name] = value
1511
1322
        self._set_configobj(configobj)
1512
1323
 
1513
 
    def _get_config_file(self):
 
1324
    def _get_configobj(self):
1514
1325
        try:
1515
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1326
            return ConfigObj(self._transport.get(self._filename),
 
1327
                             encoding='utf-8')
1516
1328
        except errors.NoSuchFile:
1517
 
            return StringIO()
1518
 
 
1519
 
    def _get_configobj(self):
1520
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1329
            return ConfigObj(encoding='utf-8')
1521
1330
 
1522
1331
    def _set_configobj(self, configobj):
1523
1332
        out_file = StringIO()