~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: John Arbash Meinel
  • Date: 2009-03-27 22:29:55 UTC
  • mto: (3735.39.2 clean)
  • mto: This revision was merged to the branch mainline in revision 4280.
  • Revision ID: john@arbash-meinel.com-20090327222955-utifmfm888zerixt
Implement apply_delta_to_source which doesn't have to malloc another string.

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')
519
464
        self._write_config_file()
520
465
 
521
466
    def _write_config_file(self):
522
 
        path = self._get_filename()
523
 
        f = open(path, 'wb')
524
 
        osutils.copy_ownership_from_path(path)
 
467
        f = open(self._get_filename(), 'wb')
525
468
        self._get_parser().write(f)
526
469
        f.close()
527
470
 
727
670
 
728
671
        return self._get_best_value('_get_user_id')
729
672
 
730
 
    def _get_change_editor(self):
731
 
        return self._get_best_value('_get_change_editor')
732
 
 
733
673
    def _get_signature_checking(self):
734
674
        """See Config._get_signature_checking."""
735
675
        return self._get_best_value('_get_signature_checking')
769
709
                        trace.warning('Value "%s" is masked by "%s" from'
770
710
                                      ' branch.conf', value, mask_value)
771
711
 
 
712
 
772
713
    def _gpg_signing_command(self):
773
714
        """See Config.gpg_signing_command."""
774
715
        return self._get_safe_value('_gpg_signing_command')
821
762
            os.mkdir(parent_dir)
822
763
        trace.mutter('creating config directory: %r', path)
823
764
        os.mkdir(path)
824
 
        osutils.copy_ownership_from_path(path)
825
765
 
826
766
 
827
767
def config_dir():
873
813
    return osutils.pathjoin(config_dir(), 'ignore')
874
814
 
875
815
 
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.
 
816
def _auto_user_id():
 
817
    """Calculate automatic user identification.
 
818
 
 
819
    Returns (realname, email).
 
820
 
 
821
    Only used when none is set in the environment or the id file.
 
822
 
 
823
    This previously used the FQDN as the default domain, but that can
 
824
    be very slow on machines where DNS is broken.  So now we simply
 
825
    use the hostname.
884
826
    """
 
827
    import socket
 
828
 
885
829
    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')
 
830
        name = win32utils.get_user_name_unicode()
 
831
        if name is None:
 
832
            raise errors.BzrError("Cannot autodetect user name.\n"
 
833
                                  "Please, set your name with command like:\n"
 
834
                                  'bzr whoami "Your Name <name@domain.com>"')
 
835
        host = win32utils.get_host_name_unicode()
 
836
        if host is None:
 
837
            host = socket.gethostname()
 
838
        return name, (name + '@' + host)
 
839
 
 
840
    try:
 
841
        import pwd
 
842
        uid = os.getuid()
 
843
        try:
 
844
            w = pwd.getpwuid(uid)
 
845
        except KeyError:
 
846
            raise errors.BzrCommandError('Unable to determine your name.  '
 
847
                'Please use "bzr whoami" to set it.')
 
848
 
 
849
        # we try utf-8 first, because on many variants (like Linux),
 
850
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
 
851
        # false positives.  (many users will have their user encoding set to
 
852
        # latin-1, which cannot raise UnicodeError.)
 
853
        try:
 
854
            gecos = w.pw_gecos.decode('utf-8')
 
855
            encoding = 'utf-8'
 
856
        except UnicodeError:
 
857
            try:
 
858
                encoding = osutils.get_user_encoding()
 
859
                gecos = w.pw_gecos.decode(encoding)
 
860
            except UnicodeError:
 
861
                raise errors.BzrCommandError('Unable to determine your name.  '
 
862
                   'Use "bzr whoami" to set it.')
 
863
        try:
 
864
            username = w.pw_name.decode(encoding)
 
865
        except UnicodeError:
 
866
            raise errors.BzrCommandError('Unable to determine your name.  '
 
867
                'Use "bzr whoami" to set it.')
 
868
 
 
869
        comma = gecos.find(',')
 
870
        if comma == -1:
 
871
            realname = gecos
 
872
        else:
 
873
            realname = gecos[:comma]
 
874
        if not realname:
 
875
            realname = username
 
876
 
 
877
    except ImportError:
 
878
        import getpass
 
879
        try:
 
880
            user_encoding = osutils.get_user_encoding()
 
881
            realname = username = getpass.getuser().decode(user_encoding)
 
882
        except UnicodeDecodeError:
 
883
            raise errors.BzrError("Can't decode username as %s." % \
 
884
                    user_encoding)
 
885
 
 
886
    return realname, (username + '@' + socket.gethostname())
901
887
 
902
888
 
903
889
def parse_username(username):
931
917
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
932
918
 
933
919
    def __init__(self, branch):
934
 
        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')
935
924
        self.branch = branch
936
925
 
937
926
    def _get_parser(self, file=None):
945
934
            return self._config.get_option(name, section, default)
946
935
        finally:
947
936
            self.branch.unlock()
 
937
        return result
948
938
 
949
939
    def set_option(self, value, name, section=None):
950
940
        """Set a per-branch configuration option"""
1093
1083
                trace.mutter("Using authentication section: %r", auth_def_name)
1094
1084
            break
1095
1085
 
1096
 
        if credentials is None:
1097
 
            # No credentials were found in authentication.conf, try the fallback
1098
 
            # credentials stores.
1099
 
            credentials = credential_store_registry.get_fallback_credentials(
1100
 
                scheme, host, port, user, path, realm)
1101
 
 
1102
1086
        return credentials
1103
1087
 
1104
1088
    def set_credentials(self, name, host, user, scheme=None, password=None,
1146
1130
        config.update({name: values})
1147
1131
        self._save()
1148
1132
 
1149
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1150
 
                 prompt=None, ask=False, default=None):
 
1133
    def get_user(self, scheme, host, port=None,
 
1134
                 realm=None, path=None, prompt=None):
1151
1135
        """Get a user from authentication file.
1152
1136
 
1153
1137
        :param scheme: protocol
1160
1144
 
1161
1145
        :param path: the absolute path on the server (optional)
1162
1146
 
1163
 
        :param ask: Ask the user if there is no explicitly configured username 
1164
 
                    (optional)
1165
 
 
1166
 
        :param default: The username returned if none is defined (optional).
1167
 
 
1168
1147
        :return: The found user.
1169
1148
        """
1170
1149
        credentials = self.get_credentials(scheme, host, port, user=None,
1173
1152
            user = credentials['user']
1174
1153
        else:
1175
1154
            user = None
1176
 
        if user is None:
1177
 
            if ask:
1178
 
                if prompt is None:
1179
 
                    # Create a default prompt suitable for most cases
1180
 
                    prompt = scheme.upper() + ' %(host)s username'
1181
 
                # Special handling for optional fields in the prompt
1182
 
                if port is not None:
1183
 
                    prompt_host = '%s:%d' % (host, port)
1184
 
                else:
1185
 
                    prompt_host = host
1186
 
                user = ui.ui_factory.get_username(prompt, host=prompt_host)
1187
 
            else:
1188
 
                user = default
1189
1155
        return user
1190
1156
 
1191
1157
    def get_password(self, scheme, host, user, port=None,
1246
1212
    A credential store provides access to credentials via the password_encoding
1247
1213
    field in authentication.conf sections.
1248
1214
 
1249
 
    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
1250
1216
    provided by plugins that will therefore use
1251
1217
    register_lazy(password_encoding, module_name, member_name, help=help,
1252
 
    fallback=fallback) to install themselves.
1253
 
 
1254
 
    A fallback credential store is one that is queried if no credentials can be
1255
 
    found via authentication.conf.
 
1218
    info=info) to install themselves.
1256
1219
    """
1257
1220
 
1258
1221
    def get_credential_store(self, encoding=None):
1261
1224
            cs = cs()
1262
1225
        return cs
1263
1226
 
1264
 
    def is_fallback(self, name):
1265
 
        """Check if the named credentials store should be used as fallback."""
1266
 
        return self.get_info(name)
1267
 
 
1268
 
    def get_fallback_credentials(self, scheme, host, port=None, user=None,
1269
 
                                 path=None, realm=None):
1270
 
        """Request credentials from all fallback credentials stores.
1271
 
 
1272
 
        The first credentials store that can provide credentials wins.
1273
 
        """
1274
 
        credentials = None
1275
 
        for name in self.keys():
1276
 
            if not self.is_fallback(name):
1277
 
                continue
1278
 
            cs = self.get_credential_store(name)
1279
 
            credentials = cs.get_credentials(scheme, host, port, user,
1280
 
                                             path, realm)
1281
 
            if credentials is not None:
1282
 
                # We found some credentials
1283
 
                break
1284
 
        return credentials
1285
 
 
1286
 
    def register(self, key, obj, help=None, override_existing=False,
1287
 
                 fallback=False):
1288
 
        """Register a new object to a name.
1289
 
 
1290
 
        :param key: This is the key to use to request the object later.
1291
 
        :param obj: The object to register.
1292
 
        :param help: Help text for this entry. This may be a string or
1293
 
                a callable. If it is a callable, it should take two
1294
 
                parameters (registry, key): this registry and the key that
1295
 
                the help was registered under.
1296
 
        :param override_existing: Raise KeyErorr if False and something has
1297
 
                already been registered for that key. If True, ignore if there
1298
 
                is an existing key (always register the new value).
1299
 
        :param fallback: Whether this credential store should be 
1300
 
                used as fallback.
1301
 
        """
1302
 
        return super(CredentialStoreRegistry,
1303
 
                     self).register(key, obj, help, info=fallback,
1304
 
                                    override_existing=override_existing)
1305
 
 
1306
 
    def register_lazy(self, key, module_name, member_name,
1307
 
                      help=None, override_existing=False,
1308
 
                      fallback=False):
1309
 
        """Register a new credential store to be loaded on request.
1310
 
 
1311
 
        :param module_name: The python path to the module. Such as 'os.path'.
1312
 
        :param member_name: The member of the module to return.  If empty or
1313
 
                None, get() will return the module itself.
1314
 
        :param help: Help text for this entry. This may be a string or
1315
 
                a callable.
1316
 
        :param override_existing: If True, replace the existing object
1317
 
                with the new one. If False, if there is already something
1318
 
                registered with the same key, raise a KeyError
1319
 
        :param fallback: Whether this credential store should be 
1320
 
                used as fallback.
1321
 
        """
1322
 
        return super(CredentialStoreRegistry, self).register_lazy(
1323
 
            key, module_name, member_name, help,
1324
 
            info=fallback, override_existing=override_existing)
1325
 
 
1326
1227
 
1327
1228
credential_store_registry = CredentialStoreRegistry()
1328
1229
 
1331
1232
    """An abstract class to implement storage for credentials"""
1332
1233
 
1333
1234
    def decode_password(self, credentials):
1334
 
        """Returns a clear text password for the provided credentials."""
 
1235
        """Returns a password for the provided credentials in clear text."""
1335
1236
        raise NotImplementedError(self.decode_password)
1336
1237
 
1337
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None,
1338
 
                        realm=None):
1339
 
        """Return the matching credentials from this credential store.
1340
 
 
1341
 
        This method is only called on fallback credential stores.
1342
 
        """
1343
 
        raise NotImplementedError(self.get_credentials)
1344
 
 
1345
 
 
1346
1238
 
1347
1239
class PlainTextCredentialStore(CredentialStore):
1348
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1240
    """Plain text credential store for the authentication.conf file."""
1349
1241
 
1350
1242
    def decode_password(self, credentials):
1351
1243
        """See CredentialStore.decode_password."""
1359
1251
 
1360
1252
class BzrDirConfig(object):
1361
1253
 
1362
 
    def __init__(self, bzrdir):
1363
 
        self._bzrdir = bzrdir
1364
 
        self._config = bzrdir._get_config()
 
1254
    def __init__(self, transport):
 
1255
        self._config = TransportConfig(transport, 'control.conf')
1365
1256
 
1366
1257
    def set_default_stack_on(self, value):
1367
1258
        """Set the default stacking location.
1371
1262
        This policy affects all branches contained by this bzrdir, except for
1372
1263
        those under repositories.
1373
1264
        """
1374
 
        if self._config is None:
1375
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1376
1265
        if value is None:
1377
1266
            self._config.set_option('', 'default_stack_on')
1378
1267
        else:
1386
1275
        This policy affects all branches contained by this bzrdir, except for
1387
1276
        those under repositories.
1388
1277
        """
1389
 
        if self._config is None:
1390
 
            return None
1391
1278
        value = self._config.get_option('default_stack_on')
1392
1279
        if value == '':
1393
1280
            value = None
1438
1325
            configobj.setdefault(section, {})[name] = value
1439
1326
        self._set_configobj(configobj)
1440
1327
 
1441
 
    def _get_config_file(self):
 
1328
    def _get_configobj(self):
1442
1329
        try:
1443
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1330
            return ConfigObj(self._transport.get(self._filename),
 
1331
                             encoding='utf-8')
1444
1332
        except errors.NoSuchFile:
1445
 
            return StringIO()
1446
 
 
1447
 
    def _get_configobj(self):
1448
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1333
            return ConfigObj(encoding='utf-8')
1449
1334
 
1450
1335
    def _set_configobj(self, configobj):
1451
1336
        out_file = StringIO()