1578
2252
configobj[name] = value
1580
2254
configobj.setdefault(section, {})[name] = value
2255
for hook in OldConfigHooks['set']:
2256
hook(self, name, value)
2257
self._set_configobj(configobj)
2259
def remove_option(self, option_name, section_name=None):
2260
configobj = self._get_configobj()
2261
if section_name is None:
2262
del configobj[option_name]
2264
del configobj[section_name][option_name]
2265
for hook in OldConfigHooks['remove']:
2266
hook(self, option_name)
1581
2267
self._set_configobj(configobj)
1583
2269
def _get_config_file(self):
1585
return StringIO(self._transport.get_bytes(self._filename))
2271
f = StringIO(self._transport.get_bytes(self._filename))
2272
for hook in OldConfigHooks['load']:
1586
2275
except errors.NoSuchFile:
1587
2276
return StringIO()
2277
except errors.PermissionDenied, e:
2278
trace.warning("Permission denied while trying to open "
2279
"configuration file %s.", urlutils.unescape_for_display(
2280
urlutils.join(self._transport.base, self._filename), "utf-8"))
2283
def _external_url(self):
2284
return urlutils.join(self._transport.external_url(), self._filename)
1589
2286
def _get_configobj(self):
1590
2287
f = self._get_config_file()
1592
return ConfigObj(f, encoding='utf-8')
2290
conf = ConfigObj(f, encoding='utf-8')
2291
except configobj.ConfigObjError, e:
2292
raise errors.ParseConfigError(e.errors, self._external_url())
2293
except UnicodeDecodeError:
2294
raise errors.ConfigContentError(self._external_url())
1596
2299
def _set_configobj(self, configobj):
1597
2300
out_file = StringIO()
1598
2301
configobj.write(out_file)
1599
2302
out_file.seek(0)
1600
2303
self._transport.put_file(self._filename, out_file)
2304
for hook in OldConfigHooks['save']:
2308
class Option(object):
2309
"""An option definition.
2311
The option *values* are stored in config files and found in sections.
2313
Here we define various properties about the option itself, its default
2314
value, how to convert it from stores, what to do when invalid values are
2315
encoutered, in which config files it can be stored.
2318
def __init__(self, name, default=None, default_from_env=None,
2320
from_unicode=None, invalid=None):
2321
"""Build an option definition.
2323
:param name: the name used to refer to the option.
2325
:param default: the default value to use when none exist in the config
2326
stores. This is either a string that ``from_unicode`` will convert
2327
into the proper type, a callable returning a unicode string so that
2328
``from_unicode`` can be used on the return value, or a python
2329
object that can be stringified (so only the empty list is supported
2332
:param default_from_env: A list of environment variables which can
2333
provide a default value. 'default' will be used only if none of the
2334
variables specified here are set in the environment.
2336
:param help: a doc string to explain the option to the user.
2338
:param from_unicode: a callable to convert the unicode string
2339
representing the option value in a store. This is not called for
2342
:param invalid: the action to be taken when an invalid value is
2343
encountered in a store. This is called only when from_unicode is
2344
invoked to convert a string and returns None or raise ValueError or
2345
TypeError. Accepted values are: None (ignore invalid values),
2346
'warning' (emit a warning), 'error' (emit an error message and
2349
if default_from_env is None:
2350
default_from_env = []
2352
# Convert the default value to a unicode string so all values are
2353
# strings internally before conversion (via from_unicode) is attempted.
2356
elif isinstance(default, list):
2357
# Only the empty list is supported
2359
raise AssertionError(
2360
'Only empty lists are supported as default values')
2362
elif isinstance(default, (str, unicode, bool, int, float)):
2363
# Rely on python to convert strings, booleans and integers
2364
self.default = u'%s' % (default,)
2365
elif callable(default):
2366
self.default = default
2368
# other python objects are not expected
2369
raise AssertionError('%r is not supported as a default value'
2371
self.default_from_env = default_from_env
2373
self.from_unicode = from_unicode
2374
if invalid and invalid not in ('warning', 'error'):
2375
raise AssertionError("%s not supported for 'invalid'" % (invalid,))
2376
self.invalid = invalid
2378
def convert_from_unicode(self, unicode_value):
2379
if self.from_unicode is None or unicode_value is None:
2380
# Don't convert or nothing to convert
2381
return unicode_value
2383
converted = self.from_unicode(unicode_value)
2384
except (ValueError, TypeError):
2385
# Invalid values are ignored
2387
if converted is None and self.invalid is not None:
2388
# The conversion failed
2389
if self.invalid == 'warning':
2390
trace.warning('Value "%s" is not valid for "%s"',
2391
unicode_value, self.name)
2392
elif self.invalid == 'error':
2393
raise errors.ConfigOptionValueError(self.name, unicode_value)
2396
def get_default(self):
2398
for var in self.default_from_env:
2400
# If the env variable is defined, its value is the default one
2401
value = os.environ[var]
2406
# Otherwise, fallback to the value defined at registration
2407
if callable(self.default):
2408
value = self.default()
2409
if not isinstance(value, unicode):
2410
raise AssertionError(
2411
'Callable default values should be unicode')
2413
value = self.default
2416
def get_help_text(self, additional_see_also=None, plain=True):
2418
from bzrlib import help_topics
2419
result += help_topics._format_see_also(additional_see_also)
2421
result = help_topics.help_as_plain_text(result)
2425
# Predefined converters to get proper values from store
2427
def bool_from_store(unicode_str):
2428
return ui.bool_from_string(unicode_str)
2431
def int_from_store(unicode_str):
2432
return int(unicode_str)
2435
def float_from_store(unicode_str):
2436
return float(unicode_str)
2440
# Use a an empty dict to initialize an empty configobj avoiding all
2441
# parsing and encoding checks
2442
_list_converter_config = configobj.ConfigObj(
2443
{}, encoding='utf-8', list_values=True, interpolation=False)
2446
def list_from_store(unicode_str):
2447
if not isinstance(unicode_str, basestring):
2449
# Now inject our string directly as unicode. All callers got their value
2450
# from configobj, so values that need to be quoted are already properly
2452
_list_converter_config.reset()
2453
_list_converter_config._parse([u"list=%s" % (unicode_str,)])
2454
maybe_list = _list_converter_config['list']
2455
# ConfigObj return '' instead of u''. Use 'str' below to catch all cases.
2456
if isinstance(maybe_list, basestring):
2458
# A single value, most probably the user forgot (or didn't care to
2459
# add) the final ','
2462
# The empty string, convert to empty list
2465
# We rely on ConfigObj providing us with a list already
2470
class OptionRegistry(registry.Registry):
2471
"""Register config options by their name.
2473
This overrides ``registry.Registry`` to simplify registration by acquiring
2474
some information from the option object itself.
2477
def register(self, option):
2478
"""Register a new option to its name.
2480
:param option: The option to register. Its name is used as the key.
2482
super(OptionRegistry, self).register(option.name, option,
2485
def register_lazy(self, key, module_name, member_name):
2486
"""Register a new option to be loaded on request.
2488
:param key: the key to request the option later. Since the registration
2489
is lazy, it should be provided and match the option name.
2491
:param module_name: the python path to the module. Such as 'os.path'.
2493
:param member_name: the member of the module to return. If empty or
2494
None, get() will return the module itself.
2496
super(OptionRegistry, self).register_lazy(key,
2497
module_name, member_name)
2499
def get_help(self, key=None):
2500
"""Get the help text associated with the given key"""
2501
option = self.get(key)
2502
the_help = option.help
2503
if callable(the_help):
2504
return the_help(self, key)
2508
option_registry = OptionRegistry()
2511
# Registered options in lexicographical order
2513
option_registry.register(
2514
Option('bzr.workingtree.worth_saving_limit', default=10,
2515
from_unicode=int_from_store, invalid='warning',
2517
How many changes before saving the dirstate.
2519
-1 means that we will never rewrite the dirstate file for only
2520
stat-cache changes. Regardless of this setting, we will always rewrite
2521
the dirstate file if a file is added/removed/renamed/etc. This flag only
2522
affects the behavior of updating the dirstate file after we notice that
2523
a file has been touched.
2525
option_registry.register(
2526
Option('dirstate.fdatasync', default=True,
2527
from_unicode=bool_from_store,
2529
Flush dirstate changes onto physical disk?
2531
If true (default), working tree metadata changes are flushed through the
2532
OS buffers to physical disk. This is somewhat slower, but means data
2533
should not be lost if the machine crashes. See also repository.fdatasync.
2535
option_registry.register(
2536
Option('debug_flags', default=[], from_unicode=list_from_store,
2537
help='Debug flags to activate.'))
2538
option_registry.register(
2539
Option('default_format', default='2a',
2540
help='Format used when creating branches.'))
2541
option_registry.register(
2542
Option('dpush_strict', default=None,
2543
from_unicode=bool_from_store,
2545
The default value for ``dpush --strict``.
2547
If present, defines the ``--strict`` option default value for checking
2548
uncommitted changes before pushing into a different VCS without any
2549
custom bzr metadata.
2551
option_registry.register(
2553
help='The command called to launch an editor to enter a message.'))
2554
option_registry.register(
2555
Option('ignore_missing_extensions', default=False,
2556
from_unicode=bool_from_store,
2558
Control the missing extensions warning display.
2560
The warning will not be emitted if set to True.
2562
option_registry.register(
2564
help='Language to translate messages into.'))
2565
option_registry.register(
2566
Option('locks.steal_dead', default=False, from_unicode=bool_from_store,
2568
Steal locks that appears to be dead.
2570
If set to True, bzr will check if a lock is supposed to be held by an
2571
active process from the same user on the same machine. If the user and
2572
machine match, but no process with the given PID is active, then bzr
2573
will automatically break the stale lock, and create a new lock for
2575
Otherwise, bzr will prompt as normal to break the lock.
2577
option_registry.register(
2578
Option('log_format', default='long',
2580
Log format to use when displaying revisions.
2582
Standard log formats are ``long``, ``short`` and ``line``. Additional formats
2583
may be provided by plugins.
2585
option_registry.register(
2586
Option('output_encoding',
2587
help= 'Unicode encoding for output'
2588
' (terminal encoding if not specified).'))
2589
option_registry.register(
2590
Option('push_strict', default=None,
2591
from_unicode=bool_from_store,
2593
The default value for ``push --strict``.
2595
If present, defines the ``--strict`` option default value for checking
2596
uncommitted changes before sending a merge directive.
2598
option_registry.register(
2599
Option('repository.fdatasync', default=True,
2600
from_unicode=bool_from_store,
2602
Flush repository changes onto physical disk?
2604
If true (default), repository changes are flushed through the OS buffers
2605
to physical disk. This is somewhat slower, but means data should not be
2606
lost if the machine crashes. See also dirstate.fdatasync.
2609
option_registry.register(
2610
Option('selftest.timeout',
2612
from_unicode=int_from_store,
2613
help='Abort selftest if one test takes longer than this many seconds',
2616
option_registry.register(
2617
Option('send_strict', default=None,
2618
from_unicode=bool_from_store,
2620
The default value for ``send --strict``.
2622
If present, defines the ``--strict`` option default value for checking
2623
uncommitted changes before pushing.
2626
option_registry.register(
2627
Option('serve.client_timeout',
2628
default=300.0, from_unicode=float_from_store,
2629
help="If we wait for a new request from a client for more than"
2630
" X seconds, consider the client idle, and hangup."))
2633
class Section(object):
2634
"""A section defines a dict of option name => value.
2636
This is merely a read-only dict which can add some knowledge about the
2637
options. It is *not* a python dict object though and doesn't try to mimic
2641
def __init__(self, section_id, options):
2642
self.id = section_id
2643
# We re-use the dict-like object received
2644
self.options = options
2646
def get(self, name, default=None, expand=True):
2647
return self.options.get(name, default)
2649
def iter_option_names(self):
2650
for k in self.options.iterkeys():
2654
# Mostly for debugging use
2655
return "<config.%s id=%s>" % (self.__class__.__name__, self.id)
2658
_NewlyCreatedOption = object()
2659
"""Was the option created during the MutableSection lifetime"""
2662
class MutableSection(Section):
2663
"""A section allowing changes and keeping track of the original values."""
2665
def __init__(self, section_id, options):
2666
super(MutableSection, self).__init__(section_id, options)
2669
def set(self, name, value):
2670
if name not in self.options:
2671
# This is a new option
2672
self.orig[name] = _NewlyCreatedOption
2673
elif name not in self.orig:
2674
self.orig[name] = self.get(name, None)
2675
self.options[name] = value
2677
def remove(self, name):
2678
if name not in self.orig:
2679
self.orig[name] = self.get(name, None)
2680
del self.options[name]
2683
class Store(object):
2684
"""Abstract interface to persistent storage for configuration options."""
2686
readonly_section_class = Section
2687
mutable_section_class = MutableSection
2689
def is_loaded(self):
2690
"""Returns True if the Store has been loaded.
2692
This is used to implement lazy loading and ensure the persistent
2693
storage is queried only when needed.
2695
raise NotImplementedError(self.is_loaded)
2698
"""Loads the Store from persistent storage."""
2699
raise NotImplementedError(self.load)
2701
def _load_from_string(self, bytes):
2702
"""Create a store from a string in configobj syntax.
2704
:param bytes: A string representing the file content.
2706
raise NotImplementedError(self._load_from_string)
2709
"""Unloads the Store.
2711
This should make is_loaded() return False. This is used when the caller
2712
knows that the persistent storage has changed or may have change since
2715
raise NotImplementedError(self.unload)
2718
"""Saves the Store to persistent storage."""
2719
raise NotImplementedError(self.save)
2721
def external_url(self):
2722
raise NotImplementedError(self.external_url)
2724
def get_sections(self):
2725
"""Returns an ordered iterable of existing sections.
2727
:returns: An iterable of (store, section).
2729
raise NotImplementedError(self.get_sections)
2731
def get_mutable_section(self, section_id=None):
2732
"""Returns the specified mutable section.
2734
:param section_id: The section identifier
2736
raise NotImplementedError(self.get_mutable_section)
2739
# Mostly for debugging use
2740
return "<config.%s(%s)>" % (self.__class__.__name__,
2741
self.external_url())
2744
class CommandLineStore(Store):
2745
"A store to carry command line overrides for the config options."""
2747
def __init__(self, opts=None):
2748
super(CommandLineStore, self).__init__()
2754
# The dict should be cleared but not replaced so it can be shared.
2755
self.options.clear()
2757
def _from_cmdline(self, overrides):
2758
# Reset before accepting new definitions
2760
for over in overrides:
2762
name, value = over.split('=', 1)
2764
raise errors.BzrCommandError(
2765
gettext("Invalid '%s', should be of the form 'name=value'")
2767
self.options[name] = value
2769
def external_url(self):
2770
# Not an url but it makes debugging easier and is never needed
2774
def get_sections(self):
2775
yield self, self.readonly_section_class('cmdline_overrides',
2779
class IniFileStore(Store):
2780
"""A config Store using ConfigObj for storage.
2782
:ivar transport: The transport object where the config file is located.
2784
:ivar file_name: The config file basename in the transport directory.
2786
:ivar _config_obj: Private member to hold the ConfigObj instance used to
2787
serialize/deserialize the config file.
2791
"""A config Store using ConfigObj for storage.
2793
super(IniFileStore, self).__init__()
2794
self._config_obj = None
2796
def is_loaded(self):
2797
return self._config_obj != None
2800
self._config_obj = None
2802
def _load_content(self):
2803
"""Load the config file bytes.
2805
This should be provided by subclasses
2807
:return: Byte string
2809
raise NotImplementedError(self._load_content)
2811
def _save_content(self, content):
2812
"""Save the config file bytes.
2814
This should be provided by subclasses
2816
:param content: Config file bytes to write
2818
raise NotImplementedError(self._save_content)
2821
"""Load the store from the associated file."""
2822
if self.is_loaded():
2824
content = self._load_content()
2825
self._load_from_string(content)
2826
for hook in ConfigHooks['load']:
2829
def _load_from_string(self, bytes):
2830
"""Create a config store from a string.
2832
:param bytes: A string representing the file content.
2834
if self.is_loaded():
2835
raise AssertionError('Already loaded: %r' % (self._config_obj,))
2836
co_input = StringIO(bytes)
2838
# The config files are always stored utf8-encoded
2839
self._config_obj = ConfigObj(co_input, encoding='utf-8',
2841
except configobj.ConfigObjError, e:
2842
self._config_obj = None
2843
raise errors.ParseConfigError(e.errors, self.external_url())
2844
except UnicodeDecodeError:
2845
raise errors.ConfigContentError(self.external_url())
2848
if not self.is_loaded():
2852
self._config_obj.write(out)
2853
self._save_content(out.getvalue())
2854
for hook in ConfigHooks['save']:
2857
def get_sections(self):
2858
"""Get the configobj section in the file order.
2860
:returns: An iterable of (store, section).
2862
# We need a loaded store
2865
except (errors.NoSuchFile, errors.PermissionDenied):
2866
# If the file can't be read, there is no sections
2868
cobj = self._config_obj
2870
yield self, self.readonly_section_class(None, cobj)
2871
for section_name in cobj.sections:
2873
self.readonly_section_class(section_name,
2874
cobj[section_name]))
2876
def get_mutable_section(self, section_id=None):
2877
# We need a loaded store
2880
except errors.NoSuchFile:
2881
# The file doesn't exist, let's pretend it was empty
2882
self._load_from_string('')
2883
if section_id is None:
2884
section = self._config_obj
2886
section = self._config_obj.setdefault(section_id, {})
2887
return self.mutable_section_class(section_id, section)
2890
class TransportIniFileStore(IniFileStore):
2891
"""IniFileStore that loads files from a transport.
2894
def __init__(self, transport, file_name):
2895
"""A Store using a ini file on a Transport
2897
:param transport: The transport object where the config file is located.
2898
:param file_name: The config file basename in the transport directory.
2900
super(TransportIniFileStore, self).__init__()
2901
self.transport = transport
2902
self.file_name = file_name
2904
def _load_content(self):
2906
return self.transport.get_bytes(self.file_name)
2907
except errors.PermissionDenied:
2908
trace.warning("Permission denied while trying to load "
2909
"configuration store %s.", self.external_url())
2912
def _save_content(self, content):
2913
self.transport.put_bytes(self.file_name, content)
2915
def external_url(self):
2916
# FIXME: external_url should really accepts an optional relpath
2917
# parameter (bug #750169) :-/ -- vila 2011-04-04
2918
# The following will do in the interim but maybe we don't want to
2919
# expose a path here but rather a config ID and its associated
2920
# object </hand wawe>.
2921
return urlutils.join(self.transport.external_url(), self.file_name)
2924
# Note that LockableConfigObjStore inherits from ConfigObjStore because we need
2925
# unlockable stores for use with objects that can already ensure the locking
2926
# (think branches). If different stores (not based on ConfigObj) are created,
2927
# they may face the same issue.
2930
class LockableIniFileStore(TransportIniFileStore):
2931
"""A ConfigObjStore using locks on save to ensure store integrity."""
2933
def __init__(self, transport, file_name, lock_dir_name=None):
2934
"""A config Store using ConfigObj for storage.
2936
:param transport: The transport object where the config file is located.
2938
:param file_name: The config file basename in the transport directory.
2940
if lock_dir_name is None:
2941
lock_dir_name = 'lock'
2942
self.lock_dir_name = lock_dir_name
2943
super(LockableIniFileStore, self).__init__(transport, file_name)
2944
self._lock = lockdir.LockDir(self.transport, self.lock_dir_name)
2946
def lock_write(self, token=None):
2947
"""Takes a write lock in the directory containing the config file.
2949
If the directory doesn't exist it is created.
2951
# FIXME: This doesn't check the ownership of the created directories as
2952
# ensure_config_dir_exists does. It should if the transport is local
2953
# -- vila 2011-04-06
2954
self.transport.create_prefix()
2955
return self._lock.lock_write(token)
2960
def break_lock(self):
2961
self._lock.break_lock()
2965
# We need to be able to override the undecorated implementation
2966
self.save_without_locking()
2968
def save_without_locking(self):
2969
super(LockableIniFileStore, self).save()
2972
# FIXME: global, bazaar, shouldn't that be 'user' instead or even
2973
# 'user_defaults' as opposed to 'user_overrides', 'system_defaults'
2974
# (/etc/bzr/bazaar.conf) and 'system_overrides' ? -- vila 2011-04-05
2976
# FIXME: Moreover, we shouldn't need classes for these stores either, factory
2977
# functions or a registry will make it easier and clearer for tests, focusing
2978
# on the relevant parts of the API that needs testing -- vila 20110503 (based
2979
# on a poolie's remark)
2980
class GlobalStore(LockableIniFileStore):
2982
def __init__(self, possible_transports=None):
2983
t = transport.get_transport_from_path(
2984
config_dir(), possible_transports=possible_transports)
2985
super(GlobalStore, self).__init__(t, 'bazaar.conf')
2989
class LocationStore(LockableIniFileStore):
2991
def __init__(self, possible_transports=None):
2992
t = transport.get_transport_from_path(
2993
config_dir(), possible_transports=possible_transports)
2994
super(LocationStore, self).__init__(t, 'locations.conf')
2995
self.id = 'locations'
2998
class BranchStore(TransportIniFileStore):
3000
def __init__(self, branch):
3001
super(BranchStore, self).__init__(branch.control_transport,
3003
self.branch = branch
3006
def lock_write(self, token=None):
3007
return self.branch.lock_write(token)
3010
return self.branch.unlock()
3014
# We need to be able to override the undecorated implementation
3015
self.save_without_locking()
3017
def save_without_locking(self):
3018
super(BranchStore, self).save()
3021
class ControlStore(LockableIniFileStore):
3023
def __init__(self, bzrdir):
3024
super(ControlStore, self).__init__(bzrdir.transport,
3026
lock_dir_name='branch_lock')
3029
class SectionMatcher(object):
3030
"""Select sections into a given Store.
3032
This is intended to be used to postpone getting an iterable of sections
3036
def __init__(self, store):
3039
def get_sections(self):
3040
# This is where we require loading the store so we can see all defined
3042
sections = self.store.get_sections()
3043
# Walk the revisions in the order provided
3044
for store, s in sections:
3048
def match(self, section):
3049
"""Does the proposed section match.
3051
:param section: A Section object.
3053
:returns: True if the section matches, False otherwise.
3055
raise NotImplementedError(self.match)
3058
class NameMatcher(SectionMatcher):
3060
def __init__(self, store, section_id):
3061
super(NameMatcher, self).__init__(store)
3062
self.section_id = section_id
3064
def match(self, section):
3065
return section.id == self.section_id
3068
class LocationSection(Section):
3070
def __init__(self, section, length, extra_path):
3071
super(LocationSection, self).__init__(section.id, section.options)
3072
self.length = length
3073
self.extra_path = extra_path
3074
self.locals = {'relpath': extra_path,
3075
'basename': urlutils.basename(extra_path)}
3077
def get(self, name, default=None, expand=True):
3078
value = super(LocationSection, self).get(name, default)
3079
if value is not None and expand:
3080
policy_name = self.get(name + ':policy', None)
3081
policy = _policy_value.get(policy_name, POLICY_NONE)
3082
if policy == POLICY_APPENDPATH:
3083
value = urlutils.join(value, self.extra_path)
3084
# expand section local options right now (since POLICY_APPENDPATH
3085
# will never add options references, it's ok to expand after it).
3087
for is_ref, chunk in iter_option_refs(value):
3089
chunks.append(chunk)
3092
if ref in self.locals:
3093
chunks.append(self.locals[ref])
3095
chunks.append(chunk)
3096
value = ''.join(chunks)
3100
class LocationMatcher(SectionMatcher):
3102
def __init__(self, store, location):
3103
super(LocationMatcher, self).__init__(store)
3104
if location.startswith('file://'):
3105
location = urlutils.local_path_from_url(location)
3106
self.location = location
3108
def _get_matching_sections(self):
3109
"""Get all sections matching ``location``."""
3110
# We slightly diverge from LocalConfig here by allowing the no-name
3111
# section as the most generic one and the lower priority.
3112
no_name_section = None
3114
# Filter out the no_name_section so _iter_for_location_by_parts can be
3115
# used (it assumes all sections have a name).
3116
for _, section in self.store.get_sections():
3117
if section.id is None:
3118
no_name_section = section
3120
all_sections.append(section)
3121
# Unfortunately _iter_for_location_by_parts deals with section names so
3122
# we have to resync.
3123
filtered_sections = _iter_for_location_by_parts(
3124
[s.id for s in all_sections], self.location)
3125
iter_all_sections = iter(all_sections)
3126
matching_sections = []
3127
if no_name_section is not None:
3128
matching_sections.append(
3129
LocationSection(no_name_section, 0, self.location))
3130
for section_id, extra_path, length in filtered_sections:
3131
# a section id is unique for a given store so it's safe to take the
3132
# first matching section while iterating. Also, all filtered
3133
# sections are part of 'all_sections' and will always be found
3136
section = iter_all_sections.next()
3137
if section_id == section.id:
3138
matching_sections.append(
3139
LocationSection(section, length, extra_path))
3141
return matching_sections
3143
def get_sections(self):
3144
# Override the default implementation as we want to change the order
3145
matching_sections = self._get_matching_sections()
3146
# We want the longest (aka more specific) locations first
3147
sections = sorted(matching_sections,
3148
key=lambda section: (section.length, section.id),
3150
# Sections mentioning 'ignore_parents' restrict the selection
3151
for section in sections:
3152
# FIXME: We really want to use as_bool below -- vila 2011-04-07
3153
ignore = section.get('ignore_parents', None)
3154
if ignore is not None:
3155
ignore = ui.bool_from_string(ignore)
3158
# Finally, we have a valid section
3159
yield self.store, section
3162
_option_ref_re = lazy_regex.lazy_compile('({[^{}\n]+})')
3163
"""Describes an expandable option reference.
3165
We want to match the most embedded reference first.
3167
I.e. for '{{foo}}' we will get '{foo}',
3168
for '{bar{baz}}' we will get '{baz}'
3171
def iter_option_refs(string):
3172
# Split isolate refs so every other chunk is a ref
3174
for chunk in _option_ref_re.split(string):
3179
class Stack(object):
3180
"""A stack of configurations where an option can be defined"""
3182
def __init__(self, sections_def, store=None, mutable_section_id=None):
3183
"""Creates a stack of sections with an optional store for changes.
3185
:param sections_def: A list of Section or callables that returns an
3186
iterable of Section. This defines the Sections for the Stack and
3187
can be called repeatedly if needed.
3189
:param store: The optional Store where modifications will be
3190
recorded. If none is specified, no modifications can be done.
3192
:param mutable_section_id: The id of the MutableSection where changes
3193
are recorded. This requires the ``store`` parameter to be
3196
self.sections_def = sections_def
3198
self.mutable_section_id = mutable_section_id
3200
def get(self, name, expand=None):
3201
"""Return the *first* option value found in the sections.
3203
This is where we guarantee that sections coming from Store are loaded
3204
lazily: the loading is delayed until we need to either check that an
3205
option exists or get its value, which in turn may require to discover
3206
in which sections it can be defined. Both of these (section and option
3207
existence) require loading the store (even partially).
3209
:param name: The queried option.
3211
:param expand: Whether options references should be expanded.
3213
:returns: The value of the option.
3215
# FIXME: No caching of options nor sections yet -- vila 20110503
3217
expand = _get_expand_default_value()
3219
# Ensuring lazy loading is achieved by delaying section matching (which
3220
# implies querying the persistent storage) until it can't be avoided
3221
# anymore by using callables to describe (possibly empty) section
3223
for sections in self.sections_def:
3224
for store, section in sections():
3225
value = section.get(name)
3226
if value is not None:
3228
if value is not None:
3230
# If the option is registered, it may provide additional info about
3233
opt = option_registry.get(name)
3237
def expand_and_convert(val):
3238
# This may need to be called twice if the value is None or ends up
3239
# being None during expansion or conversion.
3242
if isinstance(val, basestring):
3243
val = self._expand_options_in_string(val)
3245
trace.warning('Cannot expand "%s":'
3246
' %s does not support option expansion'
3247
% (name, type(val)))
3249
val = opt.convert_from_unicode(val)
3251
value = expand_and_convert(value)
3252
if opt is not None and value is None:
3253
# If the option is registered, it may provide a default value
3254
value = opt.get_default()
3255
value = expand_and_convert(value)
3256
for hook in ConfigHooks['get']:
3257
hook(self, name, value)
3260
def expand_options(self, string, env=None):
3261
"""Expand option references in the string in the configuration context.
3263
:param string: The string containing option(s) to expand.
3265
:param env: An option dict defining additional configuration options or
3266
overriding existing ones.
3268
:returns: The expanded string.
3270
return self._expand_options_in_string(string, env)
3272
def _expand_options_in_string(self, string, env=None, _refs=None):
3273
"""Expand options in the string in the configuration context.
3275
:param string: The string to be expanded.
3277
:param env: An option dict defining additional configuration options or
3278
overriding existing ones.
3280
:param _refs: Private list (FIFO) containing the options being expanded
3283
:returns: The expanded string.
3286
# Not much to expand there
3289
# What references are currently resolved (to detect loops)
3292
# We need to iterate until no more refs appear ({{foo}} will need two
3293
# iterations for example).
3298
for is_ref, chunk in iter_option_refs(result):
3300
chunks.append(chunk)
3305
raise errors.OptionExpansionLoop(string, _refs)
3307
value = self._expand_option(name, env, _refs)
3309
raise errors.ExpandingUnknownOption(name, string)
3310
chunks.append(value)
3312
result = ''.join(chunks)
3315
def _expand_option(self, name, env, _refs):
3316
if env is not None and name in env:
3317
# Special case, values provided in env takes precedence over
3321
value = self.get(name, expand=False)
3322
value = self._expand_options_in_string(value, env, _refs)
3325
def _get_mutable_section(self):
3326
"""Get the MutableSection for the Stack.
3328
This is where we guarantee that the mutable section is lazily loaded:
3329
this means we won't load the corresponding store before setting a value
3330
or deleting an option. In practice the store will often be loaded but
3331
this helps catching some programming errors.
3333
section = self.store.get_mutable_section(self.mutable_section_id)
3336
def set(self, name, value):
3337
"""Set a new value for the option."""
3338
section = self._get_mutable_section()
3339
section.set(name, value)
3340
for hook in ConfigHooks['set']:
3341
hook(self, name, value)
3343
def remove(self, name):
3344
"""Remove an existing option."""
3345
section = self._get_mutable_section()
3346
section.remove(name)
3347
for hook in ConfigHooks['remove']:
3351
# Mostly for debugging use
3352
return "<config.%s(%s)>" % (self.__class__.__name__, id(self))
3354
def _get_overrides(self):
3355
# Hack around library_state.initialize never called
3356
if bzrlib.global_state is not None:
3357
return bzrlib.global_state.cmdline_overrides.get_sections()
3361
class _CompatibleStack(Stack):
3362
"""Place holder for compatibility with previous design.
3364
This is intended to ease the transition from the Config-based design to the
3365
Stack-based design and should not be used nor relied upon by plugins.
3367
One assumption made here is that the daughter classes will all use Stores
3368
derived from LockableIniFileStore).
3370
It implements set() and remove () by re-loading the store before applying
3371
the modification and saving it.
3373
The long term plan being to implement a single write by store to save
3374
all modifications, this class should not be used in the interim.
3377
def set(self, name, value):
3380
super(_CompatibleStack, self).set(name, value)
3381
# Force a write to persistent storage
3384
def remove(self, name):
3387
super(_CompatibleStack, self).remove(name)
3388
# Force a write to persistent storage
3392
class GlobalStack(_CompatibleStack):
3393
"""Global options only stack."""
3397
gstore = GlobalStore()
3398
super(GlobalStack, self).__init__(
3399
[self._get_overrides, NameMatcher(gstore, 'DEFAULT').get_sections],
3400
gstore, mutable_section_id='DEFAULT')
3403
class LocationStack(_CompatibleStack):
3404
"""Per-location options falling back to global options stack."""
3406
def __init__(self, location):
3407
"""Make a new stack for a location and global configuration.
3409
:param location: A URL prefix to """
3410
lstore = LocationStore()
3411
if location.startswith('file://'):
3412
location = urlutils.local_path_from_url(location)
3413
matcher = LocationMatcher(lstore, location)
3414
gstore = GlobalStore()
3415
super(LocationStack, self).__init__(
3416
[self._get_overrides,
3417
matcher.get_sections, NameMatcher(gstore, 'DEFAULT').get_sections],
3418
lstore, mutable_section_id=location)
3421
class BranchStack(_CompatibleStack):
3422
"""Per-location options falling back to branch then global options stack."""
3424
def __init__(self, branch):
3425
bstore = branch._get_config_store()
3426
lstore = LocationStore()
3427
matcher = LocationMatcher(lstore, branch.base)
3428
gstore = GlobalStore()
3429
super(BranchStack, self).__init__(
3430
[self._get_overrides,
3431
matcher.get_sections, bstore.get_sections,
3432
NameMatcher(gstore, 'DEFAULT').get_sections],
3434
self.branch = branch
3437
class RemoteControlStack(_CompatibleStack):
3438
"""Remote control-only options stack."""
3440
# FIXME 2011-11-22 JRV This should probably be renamed to avoid confusion
3441
# with the stack used for remote bzr dirs. RemoteControlStack only uses
3442
# control.conf and is used only for stack options.
3444
def __init__(self, bzrdir):
3445
cstore = bzrdir._get_config_store()
3446
super(RemoteControlStack, self).__init__(
3447
[cstore.get_sections],
3449
self.bzrdir = bzrdir
3452
class RemoteBranchStack(_CompatibleStack):
3453
"""Remote branch-only options stack."""
3455
# FIXME 2011-11-22 JRV This should probably be renamed to avoid confusion
3456
# with the stack used for remote branches. RemoteBranchStack only uses
3457
# branch.conf and is used only for the stack options.
3459
def __init__(self, branch):
3460
bstore = branch._get_config_store()
3461
super(RemoteBranchStack, self).__init__(
3462
[bstore.get_sections],
3464
self.branch = branch
3466
# Use a an empty dict to initialize an empty configobj avoiding all
3467
# parsing and encoding checks
3468
_quoting_config = configobj.ConfigObj(
3469
{}, encoding='utf-8', interpolation=False)
3471
class cmd_config(commands.Command):
3472
__doc__ = """Display, set or remove a configuration option.
3474
Display the active value for a given option.
3476
If --all is specified, NAME is interpreted as a regular expression and all
3477
matching options are displayed mentioning their scope. The active value
3478
that bzr will take into account is the first one displayed for each option.
3480
If no NAME is given, --all .* is implied.
3482
Setting a value is achieved by using name=value without spaces. The value
3483
is set in the most relevant scope and can be checked by displaying the
3487
takes_args = ['name?']
3491
# FIXME: This should be a registry option so that plugins can register
3492
# their own config files (or not) and will also address
3493
# http://pad.lv/788991 -- vila 20101115
3494
commands.Option('scope', help='Reduce the scope to the specified'
3495
' configuration file.',
3497
commands.Option('all',
3498
help='Display all the defined values for the matching options.',
3500
commands.Option('remove', help='Remove the option from'
3501
' the configuration file.'),
3504
_see_also = ['configuration']
3506
@commands.display_command
3507
def run(self, name=None, all=False, directory=None, scope=None,
3509
if directory is None:
3511
directory = urlutils.normalize_url(directory)
3513
raise errors.BzrError(
3514
'--all and --remove are mutually exclusive.')
3516
# Delete the option in the given scope
3517
self._remove_config_option(name, directory, scope)
3519
# Defaults to all options
3520
self._show_matching_options('.*', directory, scope)
3523
name, value = name.split('=', 1)
3525
# Display the option(s) value(s)
3527
self._show_matching_options(name, directory, scope)
3529
self._show_value(name, directory, scope)
3532
raise errors.BzrError(
3533
'Only one option can be set.')
3534
# Set the option value
3535
self._set_config_option(name, value, directory, scope)
3537
def _get_stack(self, directory, scope=None):
3538
"""Get the configuration stack specified by ``directory`` and ``scope``.
3540
:param directory: Where the configurations are derived from.
3542
:param scope: A specific config to start from.
3544
# FIXME: scope should allow access to plugin-specific stacks (even
3545
# reduced to the plugin-specific store), related to
3546
# http://pad.lv/788991 -- vila 2011-11-15
3547
if scope is not None:
3548
if scope == 'bazaar':
3549
return GlobalStack()
3550
elif scope == 'locations':
3551
return LocationStack(directory)
3552
elif scope == 'branch':
3554
controldir.ControlDir.open_containing_tree_or_branch(
3556
return br.get_config_stack()
3557
raise errors.NoSuchConfig(scope)
3561
controldir.ControlDir.open_containing_tree_or_branch(
3563
return br.get_config_stack()
3564
except errors.NotBranchError:
3565
return LocationStack(directory)
3567
def _show_value(self, name, directory, scope):
3568
conf = self._get_stack(directory, scope)
3569
value = conf.get(name, expand=True)
3570
if value is not None:
3571
# Quote the value appropriately
3572
value = _quoting_config._quote(value)
3573
self.outf.write('%s\n' % (value,))
3575
raise errors.NoSuchConfigOption(name)
3577
def _show_matching_options(self, name, directory, scope):
3578
name = lazy_regex.lazy_compile(name)
3579
# We want any error in the regexp to be raised *now* so we need to
3580
# avoid the delay introduced by the lazy regexp. But, we still do
3581
# want the nicer errors raised by lazy_regex.
3582
name._compile_and_collapse()
3585
conf = self._get_stack(directory, scope)
3586
for sections in conf.sections_def:
3587
for store, section in sections():
3588
for oname in section.iter_option_names():
3589
if name.search(oname):
3590
if cur_store_id != store.id:
3591
# Explain where the options are defined
3592
self.outf.write('%s:\n' % (store.id,))
3593
cur_store_id = store.id
3595
if (section.id not in (None, 'DEFAULT')
3596
and cur_section != section.id):
3597
# Display the section if it's not the default (or
3599
self.outf.write(' [%s]\n' % (section.id,))
3600
cur_section = section.id
3601
value = section.get(oname, expand=False)
3602
value = _quoting_config._quote(value)
3603
self.outf.write(' %s = %s\n' % (oname, value))
3605
def _set_config_option(self, name, value, directory, scope):
3606
conf = self._get_stack(directory, scope)
3607
conf.set(name, value)
3609
def _remove_config_option(self, name, directory, scope):
3611
raise errors.BzrCommandError(
3612
'--remove expects an option to remove.')
3613
conf = self._get_stack(directory, scope)
3617
raise errors.NoSuchConfigOption(name)
3622
# We need adapters that can build a Store or a Stack in a test context. Test
3623
# classes, based on TestCaseWithTransport, can use the registry to parametrize
3624
# themselves. The builder will receive a test instance and should return a
3625
# ready-to-use store or stack. Plugins that define new store/stacks can also
3626
# register themselves here to be tested against the tests defined in
3627
# bzrlib.tests.test_config. Note that the builder can be called multiple times
3628
# for the same test.
3630
# The registered object should be a callable receiving a test instance
3631
# parameter (inheriting from tests.TestCaseWithTransport) and returning a Store
3633
test_store_builder_registry = registry.Registry()
3635
# The registered object should be a callable receiving a test instance
3636
# parameter (inheriting from tests.TestCaseWithTransport) and returning a Stack
3638
test_stack_builder_registry = registry.Registry()