16
16
# along with this program; if not, write to the Free Software
17
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
import os, types, re, time, errno, sys
19
import os, types, re, time, errno
20
20
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
22
from bzrlib.errors import BzrError
23
from bzrlib.trace import mutter
22
from errors import bailout, BzrError
23
from trace import mutter
26
26
def make_readonly(filename):
59
56
elif S_ISLNK(mode):
62
raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
65
def kind_marker(kind):
68
elif kind == 'directory':
70
elif kind == 'symlink':
73
raise BzrError('invalid file kind %r' % kind)
78
"""Copy a file to a backup.
80
Backups are named in GNU-style, with a ~ suffix.
82
If the file is already a backup, it's not copied.
95
outf = file(bfn, 'wb')
101
def rename(path_from, path_to):
102
"""Basically the same as os.rename() just special for win32"""
103
if sys.platform == 'win32':
107
if e.errno != e.ENOENT:
109
os.rename(path_from, path_to)
59
raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
132
def is_inside(dir, fname):
133
"""True if fname is inside dir.
135
return os.path.commonprefix([dir, fname]) == dir
138
def is_inside_any(dir_list, fname):
139
"""True if fname is inside any of given dirs."""
140
# quick scan for perfect match
141
if fname in dir_list:
144
for dirname in dir_list:
145
if is_inside(dirname, fname):
151
80
def pumpfile(fromfile, tofile):
152
81
"""Copy contents of one file to another."""
153
82
tofile.write(fromfile.read())
231
157
realname = gecos[:comma]
235
159
except ImportError:
237
realname = username = getpass.getuser().decode(bzrlib.user_encoding)
162
username = getpass.getuser().decode(bzrlib.user_encoding)
239
return realname, (username + '@' + socket.gethostname())
164
return realname, (username + '@' + os.gethostname())
242
167
def _get_user_id():
243
"""Return the full user id from a file or environment variable.
245
TODO: Allow taking this from a file in the branch directory too
246
for per-branch ids."""
247
168
v = os.environ.get('BZREMAIL')
249
170
return v.decode(bzrlib.user_encoding)
252
return (open(os.path.join(config_dir(), "email"))
173
return (open(os.path.expanduser("~/.bzr.email"))
254
175
.decode(bzrlib.user_encoding)
257
if e.errno != errno.ENOENT:
178
if e.errno != ENOENT:
260
181
v = os.environ.get('EMAIL')
357
282
if hasattr(os, 'urandom'): # python 2.4 and later
358
283
rand_bytes = os.urandom
359
elif sys.platform == 'linux2':
360
rand_bytes = file('/dev/urandom', 'rb').read
362
# not well seeded, but better than nothing
367
s += chr(random.randint(0, 255))
285
# FIXME: No good on non-Linux
286
_rand_file = file('/dev/urandom', 'rb')
287
rand_bytes = _rand_file.read
372
290
## TODO: We could later have path objects that remember their list
386
304
>>> splitpath('a/../b')
387
305
Traceback (most recent call last):
389
BzrError: sorry, '..' not allowed in path
307
BzrError: ("sorry, '..' not allowed in path", [])
391
309
assert isinstance(p, types.StringTypes)
393
# split on either delimiter because people might use either on
395
ps = re.split(r'[\\/]', p)
310
ps = [f for f in p.split('/') if (f != '.' and f != '')]
400
raise BzrError("sorry, %r not allowed in path" % f)
401
elif (f == '.') or (f == ''):
313
bailout("sorry, %r not allowed in path" % f)
408
317
assert isinstance(p, list)
410
319
if (f == '..') or (f == None) or (f == ''):
411
raise BzrError("sorry, %r not allowed in path" % f)
412
return os.path.join(*p)
320
bailout("sorry, %r not allowed in path" % f)
415
324
def appendpath(p1, p2):
419
return os.path.join(p1, p2)
422
331
def extern_command(cmd, ignore_errors = False):
423
332
mutter('external command: %s' % `cmd`)
424
333
if os.system(cmd):
425
334
if not ignore_errors:
426
raise BzrError('command failed')
429
def _read_config_value(name):
430
"""Read a config value from the file ~/.bzr.conf/<name>
431
Return None if the file does not exist"""
433
f = file(os.path.join(config_dir(), name), "r")
434
return f.read().decode(bzrlib.user_encoding).rstrip("\r\n")
436
if e.errno == errno.ENOENT:
442
"""Return a sequence of possible editor binaries for the current platform"""
443
e = _read_config_value("editor")
447
if os.name == "windows":
449
elif os.name == "posix":
451
yield os.environ["EDITOR"]
456
def _run_editor(filename):
457
"""Try to execute an editor to edit the commit message. Returns True on success,
459
for e in _get_editor():
460
x = os.spawnvp(os.P_WAIT, e, (e, filename))
467
raise BzrError("Could not start any editor. Please specify $EDITOR or use ~/.bzr.conf/editor")
471
def get_text_message(infotext, ignoreline = "default"):
474
if ignoreline == "default":
475
ignoreline = "-- This line and the following will be ignored --"
478
tmp_fileno, msgfilename = tempfile.mkstemp()
479
msgfile = os.close(tmp_fileno)
480
if infotext is not None and infotext != "":
482
msgfile = file(msgfilename, "w")
483
msgfile.write("\n\n%s\n\n%s" % (ignoreline, infotext))
488
if not _run_editor(msgfilename):
493
lastline, nlines = 0, 0
494
for line in file(msgfilename, "r"):
495
stripped_line = line.strip()
496
# strip empty line before the log message starts
498
if stripped_line != "":
502
# check for the ignore line only if there
503
# is additional information at the end
504
if hasinfo and stripped_line == ignoreline:
507
# keep track of the last line that had some content
508
if stripped_line != "":
514
# delete empty lines at the end
516
# add a newline at the end, if needed
517
if not msg[-1].endswith("\n"):
518
return "%s%s" % ("".join(msg), "\n")
522
# delete the msg file in any case
523
try: os.unlink(msgfilename)
335
bailout('command failed')
added
removed
bzrlib/osutils.py
