1
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Rule-based definition of preferences for selected files in selected branches
19
See ``bzr help rules`` for details.
33
# Name of the file holding rules in a tree
34
RULES_TREE_FILENAME = ".bzrrules"
36
# Namespace prefix for per file preferences
37
FILE_PREFS_PREFIX = 'name '
38
FILE_PREFS_PREFIX_LEN = len(FILE_PREFS_PREFIX)
40
# The object providing default rules
41
_per_user_searcher = None
44
class UnknownRules(errors.BzrError):
46
_fmt = ('Unknown rules detected: %(unknowns_str)s.')
48
def __init__(self, unknowns):
49
errors.BzrError.__init__(self, unknowns_str=", ".join(unknowns))
52
class _RulesSearcher(object):
53
"""An object that provides rule-based preferences."""
55
def get_items(self, path):
56
"""Return the preferences for a path as name,value tuples.
58
:param path: tree relative path
59
:return: () if no rule matched, otherwise a sequence of name,value
62
raise NotImplementedError(self.get_items)
64
def get_selected_items(self, path, names):
65
"""Return selected preferences for a path as name,value tuples.
67
:param path: tree relative path
68
:param names: the list of preferences to lookup
69
:return: () if no rule matched, otherwise a sequence of name,value
70
tuples. The sequence is the same length as names,
71
tuple order matches the order in names, and
72
undefined preferences are given the value None.
74
raise NotImplementedError(self.get_selected_items)
76
def get_single_value(self, path, preference_name):
77
"""Get a single preference for a single file.
79
:returns: The string preference value, or None.
81
for key, value in self.get_selected_items(path, [preference_name]):
86
class _IniBasedRulesSearcher(_RulesSearcher):
88
def __init__(self, inifile):
89
"""Construct a _RulesSearcher based on an ini file.
91
The content will be decoded as utf-8.
93
:param inifile: the name of the file or a sequence of lines.
95
self._cfg = configobj.ConfigObj(inifile, encoding='utf-8')
96
sections = self._cfg.keys()
98
self.pattern_to_section = {}
100
if s.startswith(FILE_PREFS_PREFIX):
101
file_patterns = cmdline.split(s[FILE_PREFS_PREFIX_LEN:])
102
patterns.extend(file_patterns)
103
for fp in file_patterns:
104
self.pattern_to_section[fp] = s
105
if len(patterns) < len(sections):
106
unknowns = [s for s in sections
107
if not s.startswith(FILE_PREFS_PREFIX)]
108
raise UnknownRules(unknowns)
110
self._globster = globbing._OrderedGlobster(patterns)
112
self._globster = None
114
def get_items(self, path):
115
"""See _RulesSearcher.get_items."""
116
if self._globster is None:
118
pat = self._globster.match(path)
122
all = self._cfg[self.pattern_to_section[pat]]
123
return tuple(all.items())
125
def get_selected_items(self, path, names):
126
"""See _RulesSearcher.get_selected_items."""
127
if self._globster is None:
129
pat = self._globster.match(path)
133
all = self._cfg[self.pattern_to_section[pat]]
134
return tuple((k, all.get(k)) for k in names)
137
class _StackedRulesSearcher(_RulesSearcher):
139
def __init__(self, searchers):
140
"""Construct a _RulesSearcher based on a stack of other ones.
142
:param searchers: a sequence of searchers.
144
self.searchers = searchers
146
def get_items(self, path):
147
"""See _RulesSearcher.get_items."""
148
for searcher in self.searchers:
149
result = searcher.get_items(path)
154
def get_selected_items(self, path, names):
155
"""See _RulesSearcher.get_selected_items."""
156
for searcher in self.searchers:
157
result = searcher.get_selected_items(path, names)
164
"""Return the default rules file path."""
165
return osutils.pathjoin(bedding.config_dir(), 'rules')
169
global _per_user_searcher
170
_per_user_searcher = _IniBasedRulesSearcher(rules_path())