← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/rules.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-08-17 18:13:57 UTC
  • mfrom: (5268.7.29 transport-segments)
  • Revision ID: pqm@pqm.ubuntu.com-20110817181357-y5q5eth1hk8bl3om
(jelmer) Allow specifying the colocated branch to use in the branch URL,
 and retrieving the branch name using ControlDir._get_selected_branch.
 (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/rules.py
1
 
# Copyright (C) 2008 Canonical Ltd
 
1
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Rule-based definition of preferences for selected files in selected branches.
18
18
 
21
21
 
22
22
from bzrlib import (
23
23
    config,
 
24
    cmdline,
24
25
    errors,
25
26
    globbing,
26
27
    osutils,
35
36
FILE_PREFS_PREFIX = 'name '
36
37
FILE_PREFS_PREFIX_LEN = len(FILE_PREFS_PREFIX)
37
38
 
 
39
# The object providing default rules
 
40
_per_user_searcher = None
 
41
 
38
42
 
39
43
class _RulesSearcher(object):
40
44
    """An object that provides rule-based preferences."""
60
64
        """
61
65
        raise NotImplementedError(self.get_selected_items)
62
66
 
 
67
    def get_single_value(self, path, preference_name):
 
68
        """Get a single preference for a single file.
 
69
        
 
70
        :returns: The string preference value, or None.
 
71
        """
 
72
        for key, value in self.get_selected_items(path, [preference_name]):
 
73
            return value
 
74
        return None
 
75
 
63
76
 
64
77
class _IniBasedRulesSearcher(_RulesSearcher):
65
78
 
70
83
 
71
84
        :param inifile: the name of the file or a sequence of lines.
72
85
        """
73
 
        options = {'encoding': 'utf-8'}
74
 
        self._cfg = configobj.ConfigObj(inifile, options=options)
 
86
        self._cfg = configobj.ConfigObj(inifile, encoding='utf-8')
75
87
        sections = self._cfg.keys()
76
 
        patterns = [s[FILE_PREFS_PREFIX_LEN:] for s in sections
77
 
            if s.startswith(FILE_PREFS_PREFIX)]
 
88
        patterns = []
 
89
        self.pattern_to_section = {}
 
90
        for s in sections:
 
91
            if s.startswith(FILE_PREFS_PREFIX):
 
92
                file_patterns = cmdline.split(s[FILE_PREFS_PREFIX_LEN:])
 
93
                patterns.extend(file_patterns)
 
94
                for fp in file_patterns:
 
95
                    self.pattern_to_section[fp] = s
78
96
        if len(patterns) < len(sections):
79
97
            unknowns = [s for s in sections
80
98
                if not s.startswith(FILE_PREFS_PREFIX)]
92
110
        if pat is None:
93
111
            return ()
94
112
        else:
95
 
            all = self._cfg[FILE_PREFS_PREFIX + pat]
 
113
            all = self._cfg[self.pattern_to_section[pat]]
96
114
            return tuple(all.items())
97
115
 
98
116
    def get_selected_items(self, path, names):
103
121
        if pat is None:
104
122
            return ()
105
123
        else:
106
 
            all = self._cfg[FILE_PREFS_PREFIX + pat]
 
124
            all = self._cfg[self.pattern_to_section[pat]]
107
125
            return tuple((k, all.get(k)) for k in names)
108
126
 
109
127
 
138
156
    return osutils.pathjoin(config.config_dir(), 'rules')
139
157
 
140
158
 
141
 
# The object providing default rules
142
 
_per_user_searcher = _IniBasedRulesSearcher(rules_filename())
 
159
def reset_rules():
 
160
    global _per_user_searcher
 
161
    _per_user_searcher = _IniBasedRulesSearcher(rules_filename())
 
162
 
 
163
reset_rules()