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):
40
40
_QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/_~-])')
42
"""Return a quoted filename filename
44
This previously used backslash quoting, but that works poorly on
46
# TODO: I'm not really sure this is the best format either.x
47
if _QUOTE_RE.search(f):
42
"""Return shell-quoted filename"""
43
## We could be a bit more terse by using double-quotes etc
44
f = _QUOTE_RE.sub(r'\\\1', f)
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')
59
raise BzrError("can't handle file kind with mode %o of %r" % (mode, f))
121
def is_inside(dir, fname):
122
"""True if fname is inside dir.
124
return os.path.commonprefix([dir, fname]) == dir
127
def is_inside_any(dir_list, fname):
128
"""True if fname is inside any of given dirs."""
129
# quick scan for perfect match
130
if fname in dir_list:
133
for dirname in dir_list:
134
if is_inside(dirname, fname):
140
80
def pumpfile(fromfile, tofile):
141
81
"""Copy contents of one file to another."""
142
82
tofile.write(fromfile.read())
146
86
"""Return a new UUID"""
88
## XXX: Could alternatively read /proc/sys/kernel/random/uuid on
89
## Linux, but we need something portable for other systems;
90
## preferably an implementation in Python.
148
return file('/proc/sys/kernel/random/uuid').readline().rstrip('\n')
92
return chomp(file('/proc/sys/kernel/random/uuid').readline())
150
94
return chomp(os.popen('uuidgen').readline())
98
if s and (s[-1] == '\n'):
106
## TODO: Maybe read in chunks to handle big files
155
107
if hasattr(f, 'tell'):
156
108
assert f.tell() == 0
164
111
return s.hexdigest()
326
274
tt = time.localtime(t)
327
275
offset = local_time_offset(t)
329
raise BzrError("unsupported timezone format %r",
277
bailout("unsupported timezone format %r",
330
278
['options are "utc", "original", "local"'])
332
280
return (time.strftime("%a %Y-%m-%d %H:%M:%S", tt)
346
294
if hasattr(os, 'urandom'): # python 2.4 and later
347
295
rand_bytes = os.urandom
348
elif sys.platform == 'linux2':
349
rand_bytes = file('/dev/urandom', 'rb').read
351
# not well seeded, but better than nothing
356
s += chr(random.randint(0, 255))
297
# FIXME: No good on non-Linux
298
_rand_file = file('/dev/urandom', 'rb')
299
rand_bytes = _rand_file.read
361
302
## TODO: We could later have path objects that remember their list
375
316
>>> splitpath('a/../b')
376
317
Traceback (most recent call last):
378
BzrError: sorry, '..' not allowed in path
319
BzrError: ("sorry, '..' not allowed in path", [])
380
321
assert isinstance(p, types.StringTypes)
412
353
mutter('external command: %s' % `cmd`)
413
354
if os.system(cmd):
414
355
if not ignore_errors:
415
raise BzrError('command failed')
418
def _read_config_value(name):
419
"""Read a config value from the file ~/.bzr.conf/<name>
420
Return None if the file does not exist"""
422
f = file(os.path.join(config_dir(), name), "r")
423
return f.read().decode(bzrlib.user_encoding).rstrip("\r\n")
425
if e.errno == errno.ENOENT:
431
"""Return a sequence of possible editor binaries for the current platform"""
432
e = _read_config_value("editor")
436
if os.name == "windows":
438
elif os.name == "posix":
440
yield os.environ["EDITOR"]
445
def _run_editor(filename):
446
"""Try to execute an editor to edit the commit message. Returns True on success,
448
for e in _get_editor():
449
x = os.spawnvp(os.P_WAIT, e, (e, filename))
456
raise BzrError("Could not start any editor. Please specify $EDITOR or use ~/.bzr.conf/editor")
460
def get_text_message(infotext, ignoreline = "default"):
463
if ignoreline == "default":
464
ignoreline = "-- This line and the following will be ignored --"
467
tmp_fileno, msgfilename = tempfile.mkstemp()
468
msgfile = os.close(tmp_fileno)
469
if infotext is not None and infotext != "":
471
msgfile = file(msgfilename, "w")
472
msgfile.write("\n\n%s\n\n%s" % (ignoreline, infotext))
477
if not _run_editor(msgfilename):
482
lastline, nlines = 0, 0
483
for line in file(msgfilename, "r"):
484
stripped_line = line.strip()
485
# strip empty line before the log message starts
487
if stripped_line != "":
491
# check for the ignore line only if there
492
# is additional information at the end
493
if hasinfo and stripped_line == ignoreline:
496
# keep track of the last line that had some content
497
if stripped_line != "":
503
# delete empty lines at the end
505
# add a newline at the end, if needed
506
if not msg[-1].endswith("\n"):
507
return "%s%s" % ("".join(msg), "\n")
511
# delete the msg file in any case
512
try: os.unlink(msgfilename)
356
bailout('command failed')
added
removed
bzrlib/osutils.py
