/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: Vincent Ladeuil
  • Date: 2008-10-06 10:06:53 UTC
  • mto: (3926.1.1 bzr.integration)
  • mto: This revision was merged to the branch mainline in revision 3928.
  • Revision ID: v.ladeuil+lp@free.fr-20081006100653-m74ulbg7rlfcuqox
Add a credential store for '.netrc'.

* plugins/netrc_credential_store/tests/test_netrc.py: 
Associated tests.

* plugins/netrc_credential_store/tests/__init__.py: 
Associated tests.

* plugins/netrc_credential_store/__init__.py: 
Provides a credential store for .netrc. This also documents how to
plug a credential store for authentication.conf.

* tests/__init__.py:
(TestCaseInTempDir.build_tree): Drive-by fix, use assertIsInstance
instead of assert_ to get a meaningful error message.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2007 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
 
37
37
[/home/robertc/source]
38
38
recurse=False|True(default)
39
39
email= as above
40
 
check_signatures= as above
 
40
check_signatures= as above 
41
41
create_signatures= as above.
42
42
 
43
43
explanation of options
45
45
editor - this option sets the pop up editor to use during commits.
46
46
email - this option sets the user id bzr will use when committing.
47
47
check_signatures - this option controls whether bzr will require good gpg
48
 
                   signatures, ignore them, or check them if they are
 
48
                   signatures, ignore them, or check them if they are 
49
49
                   present.
50
 
create_signatures - this option controls whether bzr will always create
 
50
create_signatures - this option controls whether bzr will always create 
51
51
                    gpg signatures, never create them, or create them if the
52
52
                    branch is configured to require them.
53
53
log_format - this option sets the default log format.  Possible values are
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
 
254
216
 
255
217
    def username(self):
256
218
        """Return email-style username.
257
 
 
 
219
    
258
220
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
259
 
 
 
221
        
260
222
        $BZR_EMAIL can be set to override this (as well as the
261
223
        deprecated $BZREMAIL), then
262
224
        the concrete policy type is checked, and finally
263
225
        $EMAIL is examined.
264
226
        If none is found, a reasonable default is (hopefully)
265
227
        created.
266
 
 
 
228
    
267
229
        TODO: Check it's reasonably well-formed.
268
230
        """
269
231
        v = os.environ.get('BZR_EMAIL')
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')
468
414
 
469
415
    def _get_alias(self, value):
470
416
        try:
471
 
            return self._get_parser().get_value("ALIASES",
 
417
            return self._get_parser().get_value("ALIASES", 
472
418
                                                value)
473
419
        except KeyError:
474
420
            pass
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
 
698
642
 
699
643
    def _get_safe_value(self, option_name):
700
644
        """This variant of get_best_value never returns untrusted values.
701
 
 
 
645
        
702
646
        It does not return values from the branch data, because the branch may
703
647
        not be controlled by the user.
704
648
 
713
657
 
714
658
    def _get_user_id(self):
715
659
        """Return the full user id for the branch.
716
 
 
 
660
    
717
661
        e.g. "John Hacker <jhacker@example.com>"
718
662
        This is looked up in the email controlfile for the branch.
719
663
        """
723
667
                    .rstrip("\r\n"))
724
668
        except errors.NoSuchFile, e:
725
669
            pass
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')
774
 
 
 
716
        
775
717
    def __init__(self, branch):
776
718
        super(BranchConfig, self).__init__()
777
719
        self._location_config = None
778
720
        self._branch_data_config = None
779
721
        self._global_config = None
780
722
        self.branch = branch
781
 
        self.option_sources = (self._get_location_config,
 
723
        self.option_sources = (self._get_location_config, 
782
724
                               self._get_branch_data_config,
783
725
                               self._get_global_config)
784
726
 
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():
827
768
    """Return per-user configuration directory.
828
769
 
829
770
    By default this is ~/.bazaar/
830
 
 
 
771
    
831
772
    TODO: Global option --config-dir to override this.
832
773
    """
833
774
    base = os.environ.get('BZR_HOME', None)
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
 
926
840
    try:
927
841
        import pwd
928
842
        uid = os.getuid()
929
 
        try:
930
 
            w = pwd.getpwuid(uid)
931
 
        except KeyError:
932
 
            raise errors.BzrCommandError('Unable to determine your name.  '
933
 
                'Please use "bzr whoami" to set it.')
 
843
        w = pwd.getpwuid(uid)
934
844
 
935
845
        # we try utf-8 first, because on many variants (like Linux),
936
846
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
984
894
def extract_email_address(e):
985
895
    """Return just the address part of an email string.
986
896
 
987
 
    That is just the user@domain part, nothing else.
 
897
    That is just the user@domain part, nothing else. 
988
898
    This part is required to contain only ascii characters.
989
899
    If it can't be extracted, raises an error.
990
900
 
1003
913
    # XXX: Really needs a better name, as this is not part of the tree! -- mbp 20080507
1004
914
 
1005
915
    def __init__(self, branch):
1006
 
        self._config = branch._get_config()
 
916
        # XXX: Really this should be asking the branch for its configuration
 
917
        # data, rather than relying on a Transport, so that it can work 
 
918
        # more cleanly with a RemoteBranch that has no transport.
 
919
        self._config = TransportConfig(branch._transport, 'branch.conf')
1007
920
        self.branch = branch
1008
921
 
1009
922
    def _get_parser(self, file=None):
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"""
1075
989
        section[option_name] = value
1076
990
        self._save()
1077
991
 
1078
 
    def get_credentials(self, scheme, host, port=None, user=None, path=None, 
1079
 
                        realm=None):
 
992
    def get_credentials(self, scheme, host, port=None, user=None, path=None):
1080
993
        """Returns the matching credentials from authentication.conf file.
1081
994
 
1082
995
        :param scheme: protocol
1088
1001
        :param user: login (optional)
1089
1002
 
1090
1003
        :param path: the absolute path on the server (optional)
1091
 
        
1092
 
        :param realm: the http authentication realm (optional)
1093
1004
 
1094
1005
        :return: A dict containing the matching credentials or None.
1095
1006
           This includes:
1096
1007
           - name: the section name of the credentials in the
1097
1008
             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),
 
1009
           - user: can't de different from the provided user if any,
1104
1010
           - password: the decoded password, could be None if the credential
1105
1011
             defines only the user
1106
1012
           - verify_certificates: https specific, True if the server
1147
1053
            if a_user is None:
1148
1054
                # Can't find a user
1149
1055
                continue
1150
 
            # Prepare a credentials dictionary with additional keys
1151
 
            # for the credential providers
1152
1056
            credentials = dict(name=auth_def_name,
1153
1057
                               user=a_user,
1154
 
                               scheme=a_scheme,
1155
 
                               host=host,
1156
 
                               port=port,
1157
 
                               path=path,
1158
 
                               realm=realm,
1159
1058
                               password=auth_def.get('password', None),
1160
1059
                               verify_certificates=a_verify_certificates)
1161
 
            # Decode the password in the credentials (or get one)
1162
1060
            self.decode_password(credentials,
1163
1061
                                 auth_def.get('password_encoding', None))
1164
1062
            if 'auth' in debug.debug_flags:
1165
1063
                trace.mutter("Using authentication section: %r", auth_def_name)
1166
1064
            break
1167
1065
 
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
1066
        return credentials
1175
1067
 
1176
 
    def set_credentials(self, name, host, user, scheme=None, password=None,
1177
 
                        port=None, path=None, verify_certificates=None,
1178
 
                        realm=None):
1179
 
        """Set authentication credentials for a host.
1180
 
 
1181
 
        Any existing credentials with matching scheme, host, port and path
1182
 
        will be deleted, regardless of name.
1183
 
 
1184
 
        :param name: An arbitrary name to describe this set of credentials.
1185
 
        :param host: Name of the host that accepts these credentials.
1186
 
        :param user: The username portion of these credentials.
1187
 
        :param scheme: The URL scheme (e.g. ssh, http) the credentials apply
1188
 
            to.
1189
 
        :param password: Password portion of these credentials.
1190
 
        :param port: The IP port on the host that these credentials apply to.
1191
 
        :param path: A filesystem path on the host that these credentials
1192
 
            apply to.
1193
 
        :param verify_certificates: On https, verify server certificates if
1194
 
            True.
1195
 
        :param realm: The http authentication realm (optional).
1196
 
        """
1197
 
        values = {'host': host, 'user': user}
1198
 
        if password is not None:
1199
 
            values['password'] = password
1200
 
        if scheme is not None:
1201
 
            values['scheme'] = scheme
1202
 
        if port is not None:
1203
 
            values['port'] = '%d' % port
1204
 
        if path is not None:
1205
 
            values['path'] = path
1206
 
        if verify_certificates is not None:
1207
 
            values['verify_certificates'] = str(verify_certificates)
1208
 
        if realm is not None:
1209
 
            values['realm'] = realm
1210
 
        config = self._get_config()
1211
 
        for_deletion = []
1212
 
        for section, existing_values in config.items():
1213
 
            for key in ('scheme', 'host', 'port', 'path', 'realm'):
1214
 
                if existing_values.get(key) != values.get(key):
1215
 
                    break
1216
 
            else:
1217
 
                del config[section]
1218
 
        config.update({name: values})
1219
 
        self._save()
1220
 
 
1221
 
    def get_user(self, scheme, host, port=None, realm=None, path=None,
1222
 
                 prompt=None, ask=False, default=None):
 
1068
    def get_user(self, scheme, host, port=None,
 
1069
                 realm=None, path=None, prompt=None):
1223
1070
        """Get a user from authentication file.
1224
1071
 
1225
1072
        :param scheme: protocol
1232
1079
 
1233
1080
        :param path: the absolute path on the server (optional)
1234
1081
 
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
1082
        :return: The found user.
1241
1083
        """
1242
1084
        credentials = self.get_credentials(scheme, host, port, user=None,
1243
 
                                           path=path, realm=realm)
 
1085
                                           path=path)
1244
1086
        if credentials is not None:
1245
1087
            user = credentials['user']
1246
1088
        else:
1247
1089
            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
1090
        return user
1262
1091
 
1263
1092
    def get_password(self, scheme, host, user, port=None,
1278
1107
 
1279
1108
        :return: The found password or the one entered by the user.
1280
1109
        """
1281
 
        credentials = self.get_credentials(scheme, host, port, user, path,
1282
 
                                           realm)
 
1110
        credentials = self.get_credentials(scheme, host, port, user, path)
1283
1111
        if credentials is not None:
1284
1112
            password = credentials['password']
1285
1113
            if password is not None and scheme is 'ssh':
1318
1146
    A credential store provides access to credentials via the password_encoding
1319
1147
    field in authentication.conf sections.
1320
1148
 
1321
 
    Except for stores provided by bzr itself, most stores are expected to be
 
1149
    Except for stores provided by bzr itself,most stores are expected to be
1322
1150
    provided by plugins that will therefore use
1323
1151
    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.
 
1152
    info=info) to install themselves.
1328
1153
    """
1329
1154
 
1330
1155
    def get_credential_store(self, encoding=None):
1333
1158
            cs = cs()
1334
1159
        return cs
1335
1160
 
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
1161
 
1399
1162
credential_store_registry = CredentialStoreRegistry()
1400
1163
 
1403
1166
    """An abstract class to implement storage for credentials"""
1404
1167
 
1405
1168
    def decode_password(self, credentials):
1406
 
        """Returns a clear text password for the provided credentials."""
 
1169
        """Returns a password for the provided credentials in clear text."""
1407
1170
        raise NotImplementedError(self.decode_password)
1408
1171
 
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
1172
 
1419
1173
class PlainTextCredentialStore(CredentialStore):
1420
 
    __doc__ = """Plain text credential store for the authentication.conf file"""
 
1174
    """Plain text credential store for the authentication.conf file."""
1421
1175
 
1422
1176
    def decode_password(self, credentials):
1423
1177
        """See CredentialStore.decode_password."""
1431
1185
 
1432
1186
class BzrDirConfig(object):
1433
1187
 
1434
 
    def __init__(self, bzrdir):
1435
 
        self._bzrdir = bzrdir
1436
 
        self._config = bzrdir._get_config()
 
1188
    def __init__(self, transport):
 
1189
        self._config = TransportConfig(transport, 'control.conf')
1437
1190
 
1438
1191
    def set_default_stack_on(self, value):
1439
1192
        """Set the default stacking location.
1443
1196
        This policy affects all branches contained by this bzrdir, except for
1444
1197
        those under repositories.
1445
1198
        """
1446
 
        if self._config is None:
1447
 
            raise errors.BzrError("Cannot set configuration in %s" % self._bzrdir)
1448
1199
        if value is None:
1449
1200
            self._config.set_option('', 'default_stack_on')
1450
1201
        else:
1458
1209
        This policy affects all branches contained by this bzrdir, except for
1459
1210
        those under repositories.
1460
1211
        """
1461
 
        if self._config is None:
1462
 
            return None
1463
1212
        value = self._config.get_option('default_stack_on')
1464
1213
        if value == '':
1465
1214
            value = None
1510
1259
            configobj.setdefault(section, {})[name] = value
1511
1260
        self._set_configobj(configobj)
1512
1261
 
1513
 
    def _get_config_file(self):
 
1262
    def _get_configobj(self):
1514
1263
        try:
1515
 
            return StringIO(self._transport.get_bytes(self._filename))
 
1264
            return ConfigObj(self._transport.get(self._filename),
 
1265
                             encoding='utf-8')
1516
1266
        except errors.NoSuchFile:
1517
 
            return StringIO()
1518
 
 
1519
 
    def _get_configobj(self):
1520
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1267
            return ConfigObj(encoding='utf-8')
1521
1268
 
1522
1269
    def _set_configobj(self, configobj):
1523
1270
        out_file = StringIO()