1
# Bazaar-NG -- distributed version control
3
# Copyright (C) 2005 by Canonical Ltd
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
# GNU General Public License for more details.
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19
from shutil import copyfile
20
from stat import (S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE,
21
S_ISCHR, S_ISBLK, S_ISFIFO, S_ISSOCK)
22
from cStringIO import StringIO
34
from bzrlib.errors import (BzrError,
35
BzrBadParameterNotUnicode,
39
from bzrlib.trace import mutter
42
def make_readonly(filename):
43
"""Make a filename read-only."""
44
mod = os.stat(filename).st_mode
46
os.chmod(filename, mod)
49
def make_writable(filename):
50
mod = os.stat(filename).st_mode
52
os.chmod(filename, mod)
59
"""Return a quoted filename filename
61
This previously used backslash quoting, but that works poorly on
63
# TODO: I'm not really sure this is the best format either.x
66
_QUOTE_RE = re.compile(r'([^a-zA-Z0-9.,:/\\_~-])')
68
if _QUOTE_RE.search(f):
75
mode = os.lstat(f)[ST_MODE]
94
def kind_marker(kind):
97
elif kind == 'directory':
99
elif kind == 'symlink':
102
raise BzrError('invalid file kind %r' % kind)
105
if hasattr(os.path, 'lexists'):
106
return os.path.lexists(f)
108
if hasattr(os, 'lstat'):
114
if e.errno == errno.ENOENT:
117
raise BzrError("lstat/stat of (%r): %r" % (f, e))
119
def fancy_rename(old, new, rename_func, unlink_func):
120
"""A fancy rename, when you don't have atomic rename.
122
:param old: The old path, to rename from
123
:param new: The new path, to rename to
124
:param rename_func: The potentially non-atomic rename function
125
:param unlink_func: A way to delete the target file if the full rename succeeds
128
# sftp rename doesn't allow overwriting, so play tricks:
130
base = os.path.basename(new)
131
dirname = os.path.dirname(new)
132
tmp_name = u'tmp.%s.%.9f.%d.%d' % (base, time.time(), os.getpid(), random.randint(0, 0x7FFFFFFF))
133
tmp_name = pathjoin(dirname, tmp_name)
135
# Rename the file out of the way, but keep track if it didn't exist
136
# We don't want to grab just any exception
137
# something like EACCES should prevent us from continuing
138
# The downside is that the rename_func has to throw an exception
139
# with an errno = ENOENT, or NoSuchFile
142
rename_func(new, tmp_name)
143
except (NoSuchFile,), e:
146
# RBC 20060103 abstraction leakage: the paramiko SFTP clients rename
147
# function raises an IOError with errno == None when a rename fails.
148
# This then gets caught here.
149
if e.errno not in (None, errno.ENOENT, errno.ENOTDIR):
152
if (not hasattr(e, 'errno')
153
or e.errno not in (errno.ENOENT, errno.ENOTDIR)):
160
# This may throw an exception, in which case success will
162
rename_func(old, new)
166
# If the file used to exist, rename it back into place
167
# otherwise just delete it from the tmp location
169
unlink_func(tmp_name)
171
rename_func(tmp_name, new)
173
# Default is to just use the python builtins
174
abspath = os.path.abspath
175
realpath = os.path.realpath
176
pathjoin = os.path.join
177
normpath = os.path.normpath
179
mkdtemp = tempfile.mkdtemp
181
dirname = os.path.dirname
182
basename = os.path.basename
184
if os.name == "posix":
185
# In Python 2.4.2 and older, os.path.abspath and os.path.realpath
186
# choke on a Unicode string containing a relative path if
187
# os.getcwd() returns a non-sys.getdefaultencoding()-encoded
189
_fs_enc = sys.getfilesystemencoding()
191
return os.path.abspath(path.encode(_fs_enc)).decode(_fs_enc)
194
return os.path.realpath(path.encode(_fs_enc)).decode(_fs_enc)
196
if sys.platform == 'win32':
197
# We need to use the Unicode-aware os.path.abspath and
198
# os.path.realpath on Windows systems.
200
return os.path.abspath(path).replace('\\', '/')
203
return os.path.realpath(path).replace('\\', '/')
206
return os.path.join(*args).replace('\\', '/')
209
return os.path.normpath(path).replace('\\', '/')
212
return os.getcwdu().replace('\\', '/')
214
def mkdtemp(*args, **kwargs):
215
return tempfile.mkdtemp(*args, **kwargs).replace('\\', '/')
217
def rename(old, new):
218
fancy_rename(old, new, rename_func=os.rename, unlink_func=os.unlink)
221
def normalizepath(f):
222
if hasattr(os.path, 'realpath'):
226
[p,e] = os.path.split(f)
227
if e == "" or e == "." or e == "..":
230
return pathjoin(F(p), e)
234
"""Copy a file to a backup.
236
Backups are named in GNU-style, with a ~ suffix.
238
If the file is already a backup, it's not copied.
244
if has_symlinks() and os.path.islink(fn):
245
target = os.readlink(fn)
246
os.symlink(target, bfn)
254
outf = file(bfn, 'wb')
262
"""True if f is an accessible directory."""
264
return S_ISDIR(os.lstat(f)[ST_MODE])
270
"""True if f is a regular file."""
272
return S_ISREG(os.lstat(f)[ST_MODE])
277
"""True if f is a symlink."""
279
return S_ISLNK(os.lstat(f)[ST_MODE])
283
def is_inside(dir, fname):
284
"""True if fname is inside dir.
286
The parameters should typically be passed to osutils.normpath first, so
287
that . and .. and repeated slashes are eliminated, and the separators
288
are canonical for the platform.
290
The empty string as a dir name is taken as top-of-tree and matches
293
>>> is_inside('src', pathjoin('src', 'foo.c'))
295
>>> is_inside('src', 'srccontrol')
297
>>> is_inside('src', pathjoin('src', 'a', 'a', 'a', 'foo.c'))
299
>>> is_inside('foo.c', 'foo.c')
301
>>> is_inside('foo.c', '')
303
>>> is_inside('', 'foo.c')
306
# XXX: Most callers of this can actually do something smarter by
307
# looking at the inventory
317
return fname.startswith(dir)
320
def is_inside_any(dir_list, fname):
321
"""True if fname is inside any of given dirs."""
322
for dirname in dir_list:
323
if is_inside(dirname, fname):
329
def pumpfile(fromfile, tofile):
330
"""Copy contents of one file to another."""
333
b = fromfile.read(BUFSIZE)
339
def file_iterator(input_file, readsize=32768):
341
b = input_file.read(readsize)
348
if hasattr(f, 'tell'):
361
def sha_strings(strings):
362
"""Return the sha-1 of concatenation of strings"""
364
map(s.update, strings)
374
def fingerprint_file(f):
379
return {'size': size,
380
'sha1': s.hexdigest()}
383
def compare_files(a, b):
384
"""Returns true if equal in contents"""
395
def local_time_offset(t=None):
396
"""Return offset of local zone from GMT, either at present or at time t."""
397
# python2.3 localtime() can't take None
401
if time.localtime(t).tm_isdst and time.daylight:
404
return -time.timezone
407
def format_date(t, offset=0, timezone='original', date_fmt=None,
409
## TODO: Perhaps a global option to use either universal or local time?
410
## Or perhaps just let people set $TZ?
411
assert isinstance(t, float)
413
if timezone == 'utc':
416
elif timezone == 'original':
419
tt = time.gmtime(t + offset)
420
elif timezone == 'local':
421
tt = time.localtime(t)
422
offset = local_time_offset(t)
424
raise BzrError("unsupported timezone format %r" % timezone,
425
['options are "utc", "original", "local"'])
427
date_fmt = "%a %Y-%m-%d %H:%M:%S"
429
offset_str = ' %+03d%02d' % (offset / 3600, (offset / 60) % 60)
432
return (time.strftime(date_fmt, tt) + offset_str)
435
def compact_date(when):
436
return time.strftime('%Y%m%d%H%M%S', time.gmtime(when))
441
"""Return size of given open file."""
442
return os.fstat(f.fileno())[ST_SIZE]
444
# Define rand_bytes based on platform.
446
# Python 2.4 and later have os.urandom,
447
# but it doesn't work on some arches
449
rand_bytes = os.urandom
450
except (NotImplementedError, AttributeError):
451
# If python doesn't have os.urandom, or it doesn't work,
452
# then try to first pull random data from /dev/urandom
453
if os.path.exists("/dev/urandom"):
454
rand_bytes = file('/dev/urandom', 'rb').read
455
# Otherwise, use this hack as a last resort
457
# not well seeded, but better than nothing
462
s += chr(random.randint(0, 255))
466
## TODO: We could later have path objects that remember their list
467
## decomposition (might be too tricksy though.)
470
"""Turn string into list of parts.
476
>>> splitpath('a/./b')
478
>>> splitpath('a/.b')
480
>>> splitpath('a/../b')
481
Traceback (most recent call last):
483
BzrError: sorry, '..' not allowed in path
485
assert isinstance(p, types.StringTypes)
487
# split on either delimiter because people might use either on
489
ps = re.split(r'[\\/]', p)
494
raise BzrError("sorry, %r not allowed in path" % f)
495
elif (f == '.') or (f == ''):
502
assert isinstance(p, list)
504
if (f == '..') or (f == None) or (f == ''):
505
raise BzrError("sorry, %r not allowed in path" % f)
509
def appendpath(p1, p2):
513
return pathjoin(p1, p2)
517
"""Split s into lines, but without removing the newline characters."""
518
return StringIO(s).readlines()
521
def hardlinks_good():
522
return sys.platform not in ('win32', 'cygwin', 'darwin')
525
def link_or_copy(src, dest):
526
"""Hardlink a file, or copy it if it can't be hardlinked."""
527
if not hardlinks_good():
532
except (OSError, IOError), e:
533
if e.errno != errno.EXDEV:
539
if hasattr(os, 'symlink'):
545
def contains_whitespace(s):
546
"""True if there are any whitespace characters in s."""
547
for ch in string.whitespace:
554
def contains_linebreaks(s):
555
"""True if there is any vertical whitespace in s."""
563
def relpath(base, path):
564
"""Return path relative to base, or raise exception.
566
The path may be either an absolute path or a path relative to the
567
current working directory.
569
os.path.commonprefix (python2.4) has a bad bug that it works just
570
on string prefixes, assuming that '/u' is a prefix of '/u2'. This
571
avoids that problem."""
576
while len(head) >= len(base):
579
head, tail = os.path.split(head)
583
# XXX This should raise a NotChildPath exception, as its not tied
585
raise PathNotChild(rp, base)
593
def safe_unicode(unicode_or_utf8_string):
594
"""Coerce unicode_or_utf8_string into unicode.
596
If it is unicode, it is returned.
597
Otherwise it is decoded from utf-8. If a decoding error
598
occurs, it is wrapped as a If the decoding fails, the exception is wrapped
599
as a BzrBadParameter exception.
601
if isinstance(unicode_or_utf8_string, unicode):
602
return unicode_or_utf8_string
604
return unicode_or_utf8_string.decode('utf8')
605
except UnicodeDecodeError:
606
raise BzrBadParameterNotUnicode(unicode_or_utf8_string)
609
def terminal_width():
610
"""Return estimated terminal width."""
612
# TODO: Do something smart on Windows?
614
# TODO: Is there anything that gets a better update when the window
615
# is resized while the program is running? We could use the Python termcap
618
return int(os.environ['COLUMNS'])
619
except (IndexError, KeyError, ValueError):
added
removed
bzrlib/osutils.py
