/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: John Arbash Meinel
  • Date: 2009-04-08 16:33:19 UTC
  • mfrom: (3735.2.187 brisbane-core)
  • mto: This revision was merged to the branch mainline in revision 4280.
  • Revision ID: john@arbash-meinel.com-20090408163319-jzin98xnzklj8vun
Merge the a couple rev older brisbane-core into bzr.dev, most things are resolve in favor of bzr.dev

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
 
257
219
 
258
220
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
259
221
 
260
 
        $BZR_EMAIL can be set to override this, then
 
222
        $BZR_EMAIL can be set to override this (as well as the
 
223
        deprecated $BZREMAIL), then
261
224
        the concrete policy type is checked, and finally
262
225
        $EMAIL is examined.
263
 
        If no username can be found, errors.NoWhoami exception is raised.
 
226
        If none is found, a reasonable default is (hopefully)
 
227
        created.
264
228
 
265
229
        TODO: Check it's reasonably well-formed.
266
230
        """
276
240
        if v:
277
241
            return v.decode(osutils.get_user_encoding())
278
242
 
279
 
        raise errors.NoWhoami()
280
 
 
281
 
    def ensure_username(self):
282
 
        """Raise errors.NoWhoami if username is not set.
283
 
 
284
 
        This method relies on the username() function raising the error.
285
 
        """
286
 
        self.username()
 
243
        name, email = _auto_user_id()
 
244
        if name:
 
245
            return '%s <%s>' % (name, email)
 
246
        else:
 
247
            return email
287
248
 
288
249
    def signature_checking(self):
289
250
        """What is the current policy for signature checking?."""
334
295
                path = 'bzr'
335
296
            return path
336
297
 
337
 
    def suppress_warning(self, warning):
338
 
        """Should the warning be suppressed or emitted.
339
 
 
340
 
        :param warning: The name of the warning being tested.
341
 
 
342
 
        :returns: True if the warning should be suppressed, False otherwise.
343
 
        """
344
 
        warnings = self.get_user_option_as_list('suppress_warnings')
345
 
        if warnings is None or warning not in warnings:
346
 
            return False
347
 
        else:
348
 
            return True
349
 
 
350
298
 
351
299
class IniBasedConfig(Config):
352
300
    """A configuration policy that draws from ini files."""
353
301
 
354
 
    def __init__(self, get_filename):
355
 
        super(IniBasedConfig, self).__init__()
356
 
        self._get_filename = get_filename
357
 
        self._parser = None
358
 
 
359
302
    def _get_parser(self, file=None):
360
303
        if self._parser is not None:
361
304
            return self._parser
389
332
        """Return the policy for the given (section, option_name) pair."""
390
333
        return POLICY_NONE
391
334
 
392
 
    def _get_change_editor(self):
393
 
        return self.get_user_option('change_editor')
394
 
 
395
335
    def _get_signature_checking(self):
396
336
        """See Config._get_signature_checking."""
397
337
        policy = self._get_user_option('check_signatures')
441
381
        """See Config.log_format."""
442
382
        return self._get_user_option('log_format')
443
383
 
 
384
    def __init__(self, get_filename):
 
385
        super(IniBasedConfig, self).__init__()
 
386
        self._get_filename = get_filename
 
387
        self._parser = None
 
388
 
444
389
    def _post_commit(self):
445
390
        """See Config.post_commit."""
446
391
        return self._get_user_option('post_commit')
477
422
    def _get_nickname(self):
478
423
        return self.get_user_option('nickname')
479
424
 
480
 
    def _write_config_file(self):
481
 
        f = file(self._get_filename(), "wb")
482
 
        try:
483
 
            osutils.copy_ownership_from_path(f.name)
484
 
            self._get_parser().write(f)
485
 
        finally:
486
 
            f.close()
487
 
 
488
425
 
489
426
class GlobalConfig(IniBasedConfig):
490
427
    """The configuration that should be used for a specific location."""
526
463
        self._get_parser().setdefault(section, {})[option] = value
527
464
        self._write_config_file()
528
465
 
 
466
    def _write_config_file(self):
 
467
        f = open(self._get_filename(), 'wb')
 
468
        self._get_parser().write(f)
 
469
        f.close()
 
470
 
529
471
 
530
472
class LocationConfig(IniBasedConfig):
531
473
    """A configuration object that gives the policy for a location."""
665
607
        self._get_parser()[location][option]=value
666
608
        # the allowed values of store match the config policies
667
609
        self._set_option_policy(location, option, store)
668
 
        self._write_config_file()
 
610
        self._get_parser().write(file(self._get_filename(), 'wb'))
669
611
 
670
612
 
671
613
class BranchConfig(Config):
728
670
 
729
671
        return self._get_best_value('_get_user_id')
730
672
 
731
 
    def _get_change_editor(self):
732
 
        return self._get_best_value('_get_change_editor')
733
 
 
734
673
    def _get_signature_checking(self):
735
674
        """See Config._get_signature_checking."""
736
675
        return self._get_best_value('_get_signature_checking')
822
761
            os.mkdir(parent_dir)
823
762
        trace.mutter('creating config directory: %r', path)
824
763
        os.mkdir(path)
825
 
        osutils.copy_ownership_from_path(path)
826
764
 
827
765
 
828
766
def config_dir():
843
781
                                  ' or HOME set')
844
782
        return osutils.pathjoin(base, 'bazaar', '2.0')
845
783
    else:
 
784
        # cygwin, linux, and darwin all have a $HOME directory
846
785
        if base is None:
847
786
            base = os.path.expanduser("~")
848
787
        return osutils.pathjoin(base, ".bazaar")
873
812
    return osutils.pathjoin(config_dir(), 'ignore')
874
813
 
875
814
 
876
 
def crash_dir():
877
 
    """Return the directory name to store crash files.
878
 
 
879
 
    This doesn't implicitly create it.
880
 
 
881
 
    On Windows it's in the config directory; elsewhere it's /var/crash
882
 
    which may be monitored by apport.  It can be overridden by
883
 
    $APPORT_CRASH_DIR.
 
815
def _auto_user_id():
 
816
    """Calculate automatic user identification.
 
817
 
 
818
    Returns (realname, email).
 
819
 
 
820
    Only used when none is set in the environment or the id file.
 
821
 
 
822
    This previously used the FQDN as the default domain, but that can
 
823
    be very slow on machines where DNS is broken.  So now we simply
 
824
    use the hostname.
884
825
    """
 
826
    import socket
 
827
 
885
828
    if sys.platform == 'win32':
886
 
        return osutils.pathjoin(config_dir(), 'Crash')
887
 
    else:
888
 
        # XXX: hardcoded in apport_python_hook.py; therefore here too -- mbp
889
 
        # 2010-01-31
890
 
        return os.environ.get('APPORT_CRASH_DIR', '/var/crash')
891
 
 
892
 
 
893
 
def xdg_cache_dir():
894
 
    # See http://standards.freedesktop.org/basedir-spec/latest/ar01s03.html
895
 
    # Possibly this should be different on Windows?
896
 
    e = os.environ.get('XDG_CACHE_DIR', None)
897
 
    if e:
898
 
        return e
899
 
    else:
900
 
        return os.path.expanduser('~/.cache')
 
829
        name = win32utils.get_user_name_unicode()
 
830
        if name is None:
 
831
            raise errors.BzrError("Cannot autodetect user name.\n"
 
832
                                  "Please, set your name with command like:\n"
 
833
                                  'bzr whoami "Your Name <name@domain.com>"')
 
834
        host = win32utils.get_host_name_unicode()
 
835
        if host is None:
 
836
            host = socket.gethostname()
 
837
        return name, (name + '@' + host)
 
838
 
 
839
    try:
 
840
        import pwd
 
841
        uid = os.getuid()
 
842
        try:
 
843
            w = pwd.getpwuid(uid)
 
844
        except KeyError:
 
845
            raise errors.BzrCommandError('Unable to determine your name.  '
 
846
                'Please use "bzr whoami" to set it.')
 
847
 
 
848
        # we try utf-8 first, because on many variants (like Linux),
 
849
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
850
        # false positives.  (many users will have their user encoding set to
 
851
        # latin-1, which cannot raise UnicodeError.)
 
852
        try:
 
853
            gecos = w.pw_gecos.decode('utf-8')
 
854
            encoding = 'utf-8'
 
855
        except UnicodeError:
 
856
            try:
 
857
                encoding = osutils.get_user_encoding()
 
858
                gecos = w.pw_gecos.decode(encoding)
 
859
            except UnicodeError:
 
860
                raise errors.BzrCommandError('Unable to determine your name.  '
 
861
                   'Use "bzr whoami" to set it.')
 
862
        try:
 
863
            username = w.pw_name.decode(encoding)
 
864
        except UnicodeError:
 
865
            raise errors.BzrCommandError('Unable to determine your name.  '
 
866
                'Use "bzr whoami" to set it.')
 
867
 
 
868
        comma = gecos.find(',')
 
869
        if comma == -1:
 
870
            realname = gecos
 
871
        else:
 
872
            realname = gecos[:comma]
 
873
        if not realname:
 
874
            realname = username
 
875
 
 
876
    except ImportError:
 
877
        import getpass
 
878
        try:
 
879
            user_encoding = osutils.get_user_encoding()
 
880
            realname = username = getpass.getuser().decode(user_encoding)
 
881
        except UnicodeDecodeError:
 
882
            raise errors.BzrError("Can't decode username as %s." % \
 
883
                    user_encoding)
 
884
 
 
885
    return realname, (username + '@' + socket.gethostname())
901
886
 
902
887
 
903
888
def parse_username(username):
945
930
            return self._config.get_option(name, section, default)
946
931
        finally:
947
932
            self.branch.unlock()
 
933
        return result
948
934
 
949
935
    def set_option(self, value, name, section=None):
950
936
        """Set a per-branch configuration option"""
991
977
        """Save the config file, only tests should use it for now."""
992
978
        conf_dir = os.path.dirname(self._filename)
993
979
        ensure_config_dir_exists(conf_dir)
994
 
        f = file(self._filename, 'wb')
995
 
        try:
996
 
            self._get_config().write(f)
997
 
        finally:
998
 
            f.close()
 
980
        self._get_config().write(file(self._filename, 'wb'))
999
981
 
1000
982
    def _set_option(self, section_name, option_name, value):
1001
983
        """Set an authentication configuration option"""
1097
1079
                trace.mutter("Using authentication section: %r", auth_def_name)
1098
1080
            break
1099
1081
 
1100
 
        if credentials is None:
1101
 
            # No credentials were found in authentication.conf, try the fallback
1102
 
            # credentials stores.
1103
 
            credentials = credential_store_registry.get_fallback_credentials(
1104
 
                scheme, host, port, user, path, realm)
1105
 
 
1106
1082
        return credentials
1107
1083
 
1108
1084
    def set_credentials(self, name, host, user, scheme=None, password=None,
1150
1126
        config.update({name: values})
1151
1127
        self._save()
1152
1128
 
1153
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1154
 
                 prompt=None, ask=False, default=None):
 
1129
    def get_user(self, scheme, host, port=None,
 
1130
                 realm=None, path=None, prompt=None):
1155
1131
        """Get a user from authentication file.
1156
1132
 
1157
1133
        :param scheme: protocol
1164
1140
 
1165
1141
        :param path: the absolute path on the server (optional)
1166
1142
 
1167
 
        :param ask: Ask the user if there is no explicitly configured username 
1168
 
                    (optional)
1169
 
 
1170
 
        :param default: The username returned if none is defined (optional).
1171
 
 
1172
1143
        :return: The found user.
1173
1144
        """
1174
1145
        credentials = self.get_credentials(scheme, host, port, user=None,
1177
1148
            user = credentials['user']
1178
1149
        else:
1179
1150
            user = None
1180
 
        if user is None:
1181
 
            if ask:
1182
 
                if prompt is None:
1183
 
                    # Create a default prompt suitable for most cases
1184
 
                    prompt = scheme.upper() + ' %(host)s username'
1185
 
                # Special handling for optional fields in the prompt
1186
 
                if port is not None:
1187
 
                    prompt_host = '%s:%d' % (host, port)
1188
 
                else:
1189
 
                    prompt_host = host
1190
 
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
1191
 
            else:
1192
 
                user = default
1193
1151
        return user
1194
1152
 
1195
1153
    def get_password(self, scheme, host, user, port=None,
1250
1208
    A credential store provides access to credentials via the password_encoding
1251
1209
    field in authentication.conf sections.
1252
1210
 
1253
 
    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
1254
1212
    provided by plugins that will therefore use
1255
1213
    register_lazy(password_encoding, module_name, member_name, help=help,
1256
 
    fallback=fallback) to install themselves.
1257
 
 
1258
 
    A fallback credential store is one that is queried if no credentials can be
1259
 
    found via authentication.conf.
 
1214
    info=info) to install themselves.
1260
1215
    """
1261
1216
 
1262
1217
    def get_credential_store(self, encoding=None):
1265
1220
            cs = cs()
1266
1221
        return cs
1267
1222
 
1268
 
    def is_fallback(self, name):
1269
 
        """Check if the named credentials store should be used as fallback."""
1270
 
        return self.get_info(name)
1271
 
 
1272
 
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
1273
 
                                 path=None, realm=None):
1274
 
        """Request credentials from all fallback credentials stores.
1275
 
 
1276
 
        The first credentials store that can provide credentials wins.
1277
 
        """
1278
 
        credentials = None
1279
 
        for name in self.keys():
1280
 
            if not self.is_fallback(name):
1281
 
                continue
1282
 
            cs = self.get_credential_store(name)
1283
 
            credentials = cs.get_credentials(scheme, host, port, user,
1284
 
                                             path, realm)
1285
 
            if credentials is not None:
1286
 
                # We found some credentials
1287
 
                break
1288
 
        return credentials
1289
 
 
1290
 
    def register(self, key, obj, help=None, override_existing=False,
1291
 
                 fallback=False):
1292
 
        """Register a new object to a name.
1293
 
 
1294
 
        :param key: This is the key to use to request the object later.
1295
 
        :param obj: The object to register.
1296
 
        :param help: Help text for this entry. This may be a string or
1297
 
                a callable. If it is a callable, it should take two
1298
 
                parameters (registry, key): this registry and the key that
1299
 
                the help was registered under.
1300
 
        :param override_existing: Raise KeyErorr if False and something has
1301
 
                already been registered for that key. If True, ignore if there
1302
 
                is an existing key (always register the new value).
1303
 
        :param fallback: Whether this credential store should be 
1304
 
                used as fallback.
1305
 
        """
1306
 
        return super(CredentialStoreRegistry,
1307
 
                     self).register(key, obj, help, info=fallback,
1308
 
                                    override_existing=override_existing)
1309
 
 
1310
 
    def register_lazy(self, key, module_name, member_name,
1311
 
                      help=None, override_existing=False,
1312
 
                      fallback=False):
1313
 
        """Register a new credential store to be loaded on request.
1314
 
 
1315
 
        :param module_name: The python path to the module. Such as 'os.path'.
1316
 
        :param member_name: The member of the module to return.  If empty or
1317
 
                None, get() will return the module itself.
1318
 
        :param help: Help text for this entry. This may be a string or
1319
 
                a callable.
1320
 
        :param override_existing: If True, replace the existing object
1321
 
                with the new one. If False, if there is already something
1322
 
                registered with the same key, raise a KeyError
1323
 
        :param fallback: Whether this credential store should be 
1324
 
                used as fallback.
1325
 
        """
1326
 
        return super(CredentialStoreRegistry, self).register_lazy(
1327
 
            key, module_name, member_name, help,
1328
 
            info=fallback, override_existing=override_existing)
1329
 
 
1330
1223
 
1331
1224
credential_store_registry = CredentialStoreRegistry()
1332
1225
 
1335
1228
    """An abstract class to implement storage for credentials"""
1336
1229
 
1337
1230
    def decode_password(self, credentials):
1338
 
        """Returns a clear text password for the provided credentials."""
 
1231
        """Returns a password for the provided credentials in clear text."""
1339
1232
        raise NotImplementedError(self.decode_password)
1340
1233
 
1341
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
1342
 
                        realm=None):
1343
 
        """Return the matching credentials from this credential store.
1344
 
 
1345
 
        This method is only called on fallback credential stores.
1346
 
        """
1347
 
        raise NotImplementedError(self.get_credentials)
1348
 
 
1349
 
 
1350
1234
 
1351
1235
class PlainTextCredentialStore(CredentialStore):
1352
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1236
    """Plain text credential store for the authentication.conf file."""
1353
1237
 
1354
1238
    def decode_password(self, credentials):
1355
1239
        """See CredentialStore.decode_password."""
1363
1247
 
1364
1248
class BzrDirConfig(object):
1365
1249
 
1366
 
    def __init__(self, bzrdir):
1367
 
        self._bzrdir = bzrdir
1368
 
        self._config = bzrdir._get_config()
 
1250
    def __init__(self, transport):
 
1251
        self._config = TransportConfig(transport, 'control.conf')
1369
1252
 
1370
1253
    def set_default_stack_on(self, value):
1371
1254
        """Set the default stacking location.
1375
1258
        This policy affects all branches contained by this bzrdir, except for
1376
1259
        those under repositories.
1377
1260
        """
1378
 
        if self._config is None:
1379
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1380
1261
        if value is None:
1381
1262
            self._config.set_option('', 'default_stack_on')
1382
1263
        else:
1390
1271
        This policy affects all branches contained by this bzrdir, except for
1391
1272
        those under repositories.
1392
1273
        """
1393
 
        if self._config is None:
1394
 
            return None
1395
1274
        value = self._config.get_option('default_stack_on')
1396
1275
        if value == '':
1397
1276
            value = None
1442
1321
            configobj.setdefault(section, {})[name] = value
1443
1322
        self._set_configobj(configobj)
1444
1323
 
1445
 
    def _get_config_file(self):
 
1324
    def _get_configobj(self):
1446
1325
        try:
1447
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1326
            return ConfigObj(self._transport.get(self._filename),
 
1327
                             encoding='utf-8')
1448
1328
        except errors.NoSuchFile:
1449
 
            return StringIO()
1450
 
 
1451
 
    def _get_configobj(self):
1452
 
        f = self._get_config_file()
1453
 
        try:
1454
 
            return ConfigObj(f, encoding='utf-8')
1455
 
        finally:
1456
 
            f.close()
 
1329
            return ConfigObj(encoding='utf-8')
1457
1330
 
1458
1331
    def _set_configobj(self, configobj):
1459
1332
        out_file = StringIO()