1 """Common utility functions
4 import errno
, os
, os
.path
, re
, sys
5 from stgit
.exception
import *
6 from stgit
.config
import config
7 from stgit
.out
import *
10 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
12 This program is free software; you can redistribute it and/or modify
13 it under the terms of the GNU General Public License version 2 as
14 published by the Free Software Foundation.
16 This program is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 GNU General Public License for more details.
21 You should have received a copy of the GNU General Public License
22 along with this program; if not, write to the Free Software
23 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 def mkdir_file(filename
, mode
):
27 """Opens filename with the given mode, creating the directory it's
28 in if it doesn't already exist."""
29 create_dirs(os
.path
.dirname(filename
))
30 return file(filename
, mode
)
32 def read_strings(filename
):
33 """Reads the lines from a file
35 f
= file(filename
, 'r')
36 lines
= [line
.strip() for line
in f
.readlines()]
40 def read_string(filename
, multiline
= False):
41 """Reads the first line from a file
43 f
= file(filename
, 'r')
47 result
= f
.readline().strip()
51 def write_strings(filename
, lines
):
52 """Write 'lines' sequence to file
54 f
= file(filename
, 'w+')
55 f
.writelines([line
+ '\n' for line
in lines
])
58 def write_string(filename
, line
, multiline
= False):
59 """Writes 'line' to file and truncates it
61 f
= mkdir_file(filename
, 'w+')
68 def append_strings(filename
, lines
):
69 """Appends 'lines' sequence to file
71 f
= mkdir_file(filename
, 'a+')
76 def append_string(filename
, line
):
77 """Appends 'line' to file
79 f
= mkdir_file(filename
, 'a+')
83 def insert_string(filename
, line
):
84 """Inserts 'line' at the beginning of the file
86 f
= mkdir_file(filename
, 'r+')
88 f
.seek(0); f
.truncate()
93 def create_empty_file(name
):
94 """Creates an empty file
96 mkdir_file(name
, 'w+').close()
98 def list_files_and_dirs(path
):
99 """Return the sets of filenames and directory names in a
102 for fd
in os
.listdir(path
):
103 full_fd
= os
.path
.join(path
, fd
)
104 if os
.path
.isfile(full_fd
):
106 elif os
.path
.isdir(full_fd
):
110 def walk_tree(basedir
):
111 """Starting in the given directory, iterate through all its
112 subdirectories. For each subdirectory, yield the name of the
113 subdirectory (relative to the base directory), the list of
114 filenames in the subdirectory, and the list of directory names in
118 subdir
= subdirs
.pop()
119 files
, dirs
= list_files_and_dirs(os
.path
.join(basedir
, subdir
))
121 subdirs
.append(os
.path
.join(subdir
, d
))
122 yield subdir
, files
, dirs
124 def strip_prefix(prefix
, string
):
125 """Return string, without the prefix. Blow up if string doesn't
126 start with prefix."""
127 assert string
.startswith(prefix
)
128 return string
[len(prefix
):]
130 def strip_suffix(suffix
, string
):
131 """Return string, without the suffix. Blow up if string doesn't
133 assert string
.endswith(suffix
)
134 return string
[:-len(suffix
)]
136 def remove_file_and_dirs(basedir
, file):
137 """Remove join(basedir, file), and then remove the directory it
138 was in if empty, and try the same with its parent, until we find a
139 nonempty directory or reach basedir."""
140 os
.remove(os
.path
.join(basedir
, file))
142 os
.removedirs(os
.path
.join(basedir
, os
.path
.dirname(file)))
144 # file's parent dir may not be empty after removal
147 def create_dirs(directory
):
148 """Create the given directory, if the path doesn't already exist."""
149 if directory
and not os
.path
.isdir(directory
):
150 create_dirs(os
.path
.dirname(directory
))
154 if e
.errno
!= errno
.EEXIST
:
157 def rename(basedir
, file1
, file2
):
158 """Rename join(basedir, file1) to join(basedir, file2), not
159 leaving any empty directories behind and creating any directories
161 full_file2
= os
.path
.join(basedir
, file2
)
162 create_dirs(os
.path
.dirname(full_file2
))
163 os
.rename(os
.path
.join(basedir
, file1
), full_file2
)
165 os
.removedirs(os
.path
.join(basedir
, os
.path
.dirname(file1
)))
167 # file1's parent dir may not be empty after move
170 class EditorException(StgException
):
174 for editor
in [os
.environ
.get('GIT_EDITOR'),
175 config
.get('stgit.editor'), # legacy
176 config
.get('core.editor'),
177 os
.environ
.get('VISUAL'),
178 os
.environ
.get('EDITOR'),
183 def call_editor(filename
):
184 """Run the editor on the specified filename."""
185 cmd
= '%s %s' % (get_editor(), filename
)
186 out
.start('Invoking the editor: "%s"' % cmd
)
189 raise EditorException
, 'editor failed, exit code: %d' % err
192 def edit_string(s
, filename
):
193 f
= file(filename
, 'w')
196 call_editor(filename
)
203 def append_comment(s
, comment
, separator
= '---'):
204 return ('%s\n\n%s\nEverything following the line with "%s" will be'
205 ' ignored\n\n%s' % (s
, separator
, separator
, comment
))
207 def strip_comment(s
, separator
= '---'):
209 return s
[:s
.index('\n%s\n' % separator
)]
213 def find_patch_name(patchname
, unacceptable
):
214 """Find a patch name which is acceptable."""
215 if unacceptable(patchname
):
217 while unacceptable('%s-%d' % (patchname
, suffix
)):
219 patchname
= '%s-%d' % (patchname
, suffix
)
222 def patch_name_from_msg(msg
):
223 """Return a string to be used as a patch name. This is generated
224 from the top line of the string passed as argument."""
228 name_len
= config
.getint('stgit.namelength')
232 subject_line
= msg
.split('\n', 1)[0].lstrip().lower()
233 words
= re
.sub('[\W]+', ' ', subject_line
).split()
235 # use loop to avoid truncating the last name
236 name
= words
and words
[0] or 'unknown'
237 for word
in words
[1:]:
238 new
= name
+ '-' + word
239 if len(new
) > name_len
:
245 def make_patch_name(msg
, unacceptable
, default_name
= 'patch'):
246 """Return a patch name generated from the given commit message,
247 guaranteed to make unacceptable(name) be false. If the commit
248 message is empty, base the name on default_name instead."""
249 patchname
= patch_name_from_msg(msg
)
251 patchname
= default_name
252 return find_patch_name(patchname
, unacceptable
)
254 # any and all functions are builtin in Python 2.5 and higher, but not
256 if not 'any' in dir(__builtins__
):
262 if not 'all' in dir(__builtins__
):
269 def add_sign_line(desc
, sign_str
, name
, email
):
272 sign_str
= '%s: %s <%s>' % (sign_str
, name
, email
)
276 if not any(s
in desc
for s
in ['\nSigned-off-by:', '\nAcked-by:']):
278 return '%s\n%s\n' % (desc
, sign_str
)
280 def parse_name_email(address
):
281 """Return a tuple consisting of the name and email parsed from a
282 standard 'name <email>' or 'email (name)' string."""
283 address
= re
.sub(r
'[\\"]', r
'\\\g<0>', address
)
284 str_list
= re
.findall(r
'^(.*)\s*<(.*)>\s*$', address
)
286 str_list
= re
.findall(r
'^(.*)\s*\((.*)\)\s*$', address
)
289 return (str_list
[0][1], str_list
[0][0])
292 def parse_name_email_date(address
):
293 """Return a tuple consisting of the name, email and date parsed
294 from a 'name <email> date' string."""
295 address
= re
.sub(r
'[\\"]', r
'\\\g<0>', address
)
296 str_list
= re
.findall('^(.*)\s*<(.*)>\s*(.*)\s*$', address
)
302 STGIT_SUCCESS
= 0 # everything's OK
303 STGIT_GENERAL_ERROR
= 1 # seems to be non-command-specific error
304 STGIT_COMMAND_ERROR
= 2 # seems to be a command that failed
305 STGIT_CONFLICT
= 3 # merge conflict, otherwise OK
306 STGIT_BUG_ERROR
= 4 # a bug in StGit
308 def add_dict(d1
, d2
):
309 """Return a new dict with the contents of both d1 and d2. In case of
310 conflicting mappings, d2 takes precedence."""