Commit | Line | Data |
---|---|---|
41a6d859 CM |
1 | """Handles the Stacked GIT configuration files |
2 | """ | |
3 | ||
4 | __copyright__ = """ | |
5 | Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com> | |
6 | ||
7 | This program is free software; you can redistribute it and/or modify | |
8 | it under the terms of the GNU General Public License version 2 as | |
9 | published by the Free Software Foundation. | |
10 | ||
11 | This program is distributed in the hope that it will be useful, | |
12 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | GNU General Public License for more details. | |
15 | ||
16 | You should have received a copy of the GNU General Public License | |
17 | along with this program; if not, write to the Free Software | |
18 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
19 | """ | |
20 | ||
c73e63b7 | 21 | import os, re |
170f576b | 22 | from stgit import basedir |
87c93eab | 23 | from stgit.exception import * |
f0de3f92 | 24 | from stgit.run import * |
41a6d859 | 25 | |
87c93eab | 26 | class GitConfigException(StgException): |
c73e63b7 YD |
27 | pass |
28 | ||
29 | class GitConfig: | |
30 | __defaults={ | |
31 | 'stgit.autoresolved': 'no', | |
32 | 'stgit.smtpserver': 'localhost:25', | |
33 | 'stgit.smtpdelay': '5', | |
1576d681 CM |
34 | 'stgit.pullcmd': 'git pull', |
35 | 'stgit.fetchcmd': 'git fetch', | |
3b3c26fa | 36 | 'stgit.pull-policy': 'pull', |
c73e63b7 | 37 | 'stgit.autoimerge': 'no', |
c73e63b7 YD |
38 | 'stgit.keepoptimized': 'no', |
39 | 'stgit.extensions': '.ancestor .current .patched', | |
40 | 'stgit.shortnr': '5' | |
41 | } | |
42 | ||
a264e49b ST |
43 | __cache = None |
44 | ||
45 | def load(self): | |
46 | """Load the whole configuration in __cache unless it has been | |
47 | done already.""" | |
48 | if self.__cache is not None: | |
49 | return | |
50 | self.__cache = {} | |
51 | lines = Run('git', 'config', '--list', '--null').raw_output() | |
52 | for line in filter(None, lines.split('\0')): | |
53 | key, value = line.split('\n', 1) | |
54 | self.__cache.setdefault(key, []).append(value) | |
9a4bf454 | 55 | |
c73e63b7 | 56 | def get(self, name): |
a264e49b ST |
57 | self.load() |
58 | if name not in self.__cache: | |
59 | self.__cache[name] = [self.__defaults.get(name, None)] | |
60 | return self.__cache[name][0] | |
c73e63b7 YD |
61 | |
62 | def getall(self, name): | |
a264e49b ST |
63 | self.load() |
64 | try: | |
9a4bf454 | 65 | return self.__cache[name] |
a264e49b ST |
66 | except KeyError: |
67 | return [] | |
c73e63b7 YD |
68 | |
69 | def getint(self, name): | |
70 | value = self.get(name) | |
e928a3ec KH |
71 | if value == None: |
72 | return None | |
73 | elif value.isdigit(): | |
c73e63b7 YD |
74 | return int(value) |
75 | else: | |
76 | raise GitConfigException, 'Value for "%s" is not an integer: "%s"' % (name, value) | |
77 | ||
cb5be4c3 | 78 | def rename_section(self, from_name, to_name): |
f0de3f92 KH |
79 | """Rename a section in the config file. Silently do nothing if |
80 | the section doesn't exist.""" | |
cd885e08 | 81 | Run('git', 'config', '--rename-section', from_name, to_name |
a66b9072 | 82 | ).returns([0, 1, 128]).run() |
8591add9 | 83 | self.__cache.clear() |
cb5be4c3 | 84 | |
9a6bcbe2 KH |
85 | def remove_section(self, name): |
86 | """Remove a section in the config file. Silently do nothing if | |
87 | the section doesn't exist.""" | |
cd885e08 | 88 | Run('git', 'config', '--remove-section', name |
a66b9072 | 89 | ).returns([0, 1, 128]).discard_stderr().discard_output() |
9a6bcbe2 KH |
90 | self.__cache.clear() |
91 | ||
c73e63b7 | 92 | def set(self, name, value): |
cd885e08 | 93 | Run('git', 'config', name, value).run() |
8591add9 | 94 | self.__cache[name] = value |
c73e63b7 | 95 | |
0aee23c2 | 96 | def unset(self, name): |
cd885e08 | 97 | Run('git', 'config', '--unset', name) |
8591add9 | 98 | self.__cache[name] = None |
0aee23c2 | 99 | |
c73e63b7 YD |
100 | def sections_matching(self, regexp): |
101 | """Takes a regexp with a single group, matches it against all | |
102 | config variables, and returns a list whose members are the | |
103 | group contents, for all variable names matching the regexp. | |
104 | """ | |
105 | result = [] | |
cd885e08 | 106 | for line in Run('git', 'config', '--get-regexp', '"^%s$"' % regexp |
f0de3f92 | 107 | ).returns([0, 1]).output_lines(): |
c73e63b7 YD |
108 | m = re.match('^%s ' % regexp, line) |
109 | if m: | |
110 | result.append(m.group(1)) | |
c73e63b7 YD |
111 | return result |
112 | ||
113 | config=GitConfig() | |
abcc2620 | 114 | |
eee7283e CM |
115 | def config_setup(): |
116 | global config | |
117 | ||
eee7283e | 118 | # Set the PAGER environment to the config value (if any) |
c73e63b7 YD |
119 | pager = config.get('stgit.pager') |
120 | if pager: | |
121 | os.environ['PAGER'] = pager | |
122 | # FIXME: handle EDITOR the same way ? | |
eee7283e CM |
123 | |
124 | class ConfigOption: | |
125 | """Delayed cached reading of a configuration option. | |
126 | """ | |
127 | def __init__(self, section, option): | |
128 | self.__section = section | |
129 | self.__option = option | |
130 | self.__value = None | |
131 | ||
132 | def __str__(self): | |
133 | if not self.__value: | |
c73e63b7 | 134 | self.__value = config.get(self.__section + '.' + self.__option) |
eee7283e | 135 | return self.__value |
d7fade4b CM |
136 | |
137 | ||
138 | # cached extensions | |
139 | __extensions = None | |
140 | ||
141 | def file_extensions(): | |
142 | """Returns a dictionary with the conflict file extensions | |
143 | """ | |
144 | global __extensions | |
145 | ||
146 | if not __extensions: | |
c73e63b7 | 147 | cfg_ext = config.get('stgit.extensions').split() |
d7fade4b CM |
148 | if len(cfg_ext) != 3: |
149 | raise CmdException, '"extensions" configuration error' | |
150 | ||
151 | __extensions = { 'ancestor': cfg_ext[0], | |
152 | 'current': cfg_ext[1], | |
153 | 'patched': cfg_ext[2] } | |
154 | ||
155 | return __extensions |