105
100
mod = os.lstat(filename).st_mode
106
101
if not stat.S_ISLNK(mod):
107
102
mod = mod & 0777555
108
os.chmod(filename, mod)
103
chmod_if_possible(filename, mod)
111
106
def make_writable(filename):
112
107
mod = os.lstat(filename).st_mode
113
108
if not stat.S_ISLNK(mod):
115
os.chmod(filename, mod)
110
chmod_if_possible(filename, mod)
113
def chmod_if_possible(filename, mode):
114
# Set file mode if that can be safely done.
115
# Sometimes even on unix the filesystem won't allow it - see
116
# https://bugs.launchpad.net/bzr/+bug/606537
118
# It is probably faster to just do the chmod, rather than
119
# doing a stat, and then trying to compare
120
os.chmod(filename, mode)
121
except (IOError, OSError),e:
122
# Permission/access denied seems to commonly happen on smbfs; there's
123
# probably no point warning about it.
124
# <https://bugs.launchpad.net/bzr/+bug/606537>
125
if getattr(e, 'errno') in (errno.EPERM, errno.EACCES):
126
trace.mutter("ignore error on chmod of %r: %r" % (
118
132
def minimum_path_selection(paths):
270
284
rename_func(tmp_name, new)
271
285
if failure_exc is not None:
272
raise failure_exc[0], failure_exc[1], failure_exc[2]
287
raise failure_exc[0], failure_exc[1], failure_exc[2]
275
292
# In Python 2.4.2 and older, os.path.abspath and os.path.realpath
276
293
# choke on a Unicode string containing a relative path if
277
294
# os.getcwd() returns a non-sys.getdefaultencoding()-encoded
279
_fs_enc = sys.getfilesystemencoding() or 'utf-8'
280
296
def _posix_abspath(path):
281
297
# jam 20060426 rather than encoding to fsencoding
282
298
# copy posixpath.abspath, but use os.getcwdu instead
283
299
if not posixpath.isabs(path):
284
300
path = posixpath.join(getcwd(), path)
285
return posixpath.normpath(path)
301
return _posix_normpath(path)
288
304
def _posix_realpath(path):
289
305
return posixpath.realpath(path.encode(_fs_enc)).decode(_fs_enc)
308
def _posix_normpath(path):
309
path = posixpath.normpath(path)
310
# Bug 861008: posixpath.normpath() returns a path normalized according to
311
# the POSIX standard, which stipulates (for compatibility reasons) that two
312
# leading slashes must not be simplified to one, and only if there are 3 or
313
# more should they be simplified as one. So we treat the leading 2 slashes
314
# as a special case here by simply removing the first slash, as we consider
315
# that breaking POSIX compatibility for this obscure feature is acceptable.
316
# This is not a paranoid precaution, as we notably get paths like this when
317
# the repo is hosted at the root of the filesystem, i.e. in "/".
318
if path.startswith('//'):
323
def _posix_path_from_environ(key):
324
"""Get unicode path from `key` in environment or None if not present
326
Note that posix systems use arbitrary byte strings for filesystem objects,
327
so a path that raises BadFilenameEncoding here may still be accessible.
329
val = os.environ.get(key, None)
333
return val.decode(_fs_enc)
334
except UnicodeDecodeError:
335
# GZ 2011-12-12:Ideally want to include `key` in the exception message
336
raise errors.BadFilenameEncoding(val, _fs_enc)
339
def _posix_getuser_unicode():
340
"""Get username from environment or password database as unicode"""
341
name = getpass.getuser()
342
user_encoding = get_user_encoding()
344
return name.decode(user_encoding)
345
except UnicodeDecodeError:
346
raise errors.BzrError("Encoding of username %r is unsupported by %s "
347
"application locale." % (name, user_encoding))
292
350
def _win32_fixdrive(path):
293
351
"""Force drive letters to be consistent.
1462
1536
# a similar effect.
1464
1538
# If BZR_COLUMNS is set, take it, user is always right
1539
# Except if they specified 0 in which case, impose no limit here
1466
return int(os.environ['BZR_COLUMNS'])
1541
width = int(os.environ['BZR_COLUMNS'])
1467
1542
except (KeyError, ValueError):
1544
if width is not None:
1470
1550
isatty = getattr(sys.stdout, 'isatty', None)
1471
1551
if isatty is None or not isatty():
1973
2054
return get_terminal_encoding()
2057
_message_encoding = None
2060
def get_message_encoding():
2061
"""Return the encoding used for messages
2063
While the message encoding is a general setting it should usually only be
2064
needed for decoding system error strings such as from OSError instances.
2066
global _message_encoding
2067
if _message_encoding is None:
2068
if os.name == "posix":
2070
# This is a process-global setting that can change, but should in
2071
# general just get set once at process startup then be constant.
2072
_message_encoding = locale.getlocale(locale.LC_MESSAGES)[1]
2074
# On windows want the result of GetACP() which this boils down to.
2075
_message_encoding = get_user_encoding()
2076
return _message_encoding or "ascii"
1976
2079
def get_host_name():
1977
2080
"""Return the current unicode host name.
1993
2096
# data at once.
1994
2097
MAX_SOCKET_CHUNK = 64 * 1024
2099
_end_of_stream_errors = [errno.ECONNRESET]
2100
for _eno in ['WSAECONNRESET', 'WSAECONNABORTED']:
2101
_eno = getattr(errno, _eno, None)
2102
if _eno is not None:
2103
_end_of_stream_errors.append(_eno)
1996
2107
def read_bytes_from_socket(sock, report_activity=None,
1997
2108
max_read_size=MAX_SOCKET_CHUNK):
1998
2109
"""Read up to max_read_size of bytes from sock and notify of progress.
2153
2264
return file_kind_from_stat_mode(mode)
2154
2265
file_kind_from_stat_mode = file_kind_from_stat_mode_thunk
2157
def file_kind(f, _lstat=os.lstat):
2267
def file_stat(f, _lstat=os.lstat):
2159
return file_kind_from_stat_mode(_lstat(f).st_mode)
2160
2271
except OSError, e:
2161
2272
if getattr(e, 'errno', None) in (errno.ENOENT, errno.ENOTDIR):
2162
2273
raise errors.NoSuchFile(f)
2276
def file_kind(f, _lstat=os.lstat):
2277
stat_value = file_stat(f, _lstat)
2278
return file_kind_from_stat_mode(stat_value.st_mode)
2166
2280
def until_no_eintr(f, *a, **kw):
2167
2281
"""Run f(*a, **kw), retrying if an EINTR error occurs.
2227
2341
termios.tcsetattr(fd, termios.TCSADRAIN, settings)
2231
if sys.platform == 'linux2':
2344
if sys.platform.startswith('linux'):
2232
2345
def _local_concurrency():
2234
prefix = 'processor'
2235
for line in file('/proc/cpuinfo', 'rb'):
2236
if line.startswith(prefix):
2237
concurrency = int(line[line.find(':')+1:]) + 1
2347
return os.sysconf('SC_NPROCESSORS_ONLN')
2348
except (ValueError, OSError, AttributeError):
2239
2350
elif sys.platform == 'darwin':
2240
2351
def _local_concurrency():
2241
2352
return subprocess.Popen(['sysctl', '-n', 'hw.availcpu'],
2242
2353
stdout=subprocess.PIPE).communicate()[0]
2243
elif sys.platform[0:7] == 'freebsd':
2354
elif "bsd" in sys.platform:
2244
2355
def _local_concurrency():
2245
2356
return subprocess.Popen(['sysctl', '-n', 'hw.ncpu'],
2246
2357
stdout=subprocess.PIPE).communicate()[0]
2274
2385
concurrency = os.environ.get('BZR_CONCURRENCY', None)
2275
2386
if concurrency is None:
2277
concurrency = _local_concurrency()
2278
except (OSError, IOError):
2388
import multiprocessing
2389
concurrency = multiprocessing.cpu_count()
2390
except (ImportError, NotImplementedError):
2391
# multiprocessing is only available on Python >= 2.6
2392
# and multiprocessing.cpu_count() isn't implemented on all
2395
concurrency = _local_concurrency()
2396
except (OSError, IOError):
2281
2399
concurrency = int(concurrency)
2282
2400
except (TypeError, ValueError):
2375
2481
name = "%s.~%d~" % (base, counter)
2485
def set_fd_cloexec(fd):
2486
"""Set a Unix file descriptor's FD_CLOEXEC flag. Do nothing if platform
2487
support for this is not available.
2491
old = fcntl.fcntl(fd, fcntl.F_GETFD)
2492
fcntl.fcntl(fd, fcntl.F_SETFD, old | fcntl.FD_CLOEXEC)
2493
except (ImportError, AttributeError):
2494
# Either the fcntl module or specific constants are not present
2498
def find_executable_on_path(name):
2499
"""Finds an executable on the PATH.
2501
On Windows, this will try to append each extension in the PATHEXT
2502
environment variable to the name, if it cannot be found with the name
2505
:param name: The base name of the executable.
2506
:return: The path to the executable found or None.
2508
path = os.environ.get('PATH')
2511
path = path.split(os.pathsep)
2512
if sys.platform == 'win32':
2513
exts = os.environ.get('PATHEXT', '').split(os.pathsep)
2514
exts = [ext.lower() for ext in exts]
2515
base, ext = os.path.splitext(name)
2517
if ext.lower() not in exts:
2525
f = os.path.join(d, name) + ext
2526
if os.access(f, os.X_OK):
2531
def _posix_is_local_pid_dead(pid):
2532
"""True if pid doesn't correspond to live process on this machine"""
2534
# Special meaning of unix kill: just check if it's there.
2537
if e.errno == errno.ESRCH:
2538
# On this machine, and really not found: as sure as we can be
2541
elif e.errno == errno.EPERM:
2542
# exists, though not ours
2545
mutter("os.kill(%d, 0) failed: %s" % (pid, e))
2546
# Don't really know.
2549
# Exists and our process: not dead.
2552
if sys.platform == "win32":
2553
is_local_pid_dead = win32utils.is_local_pid_dead
2555
is_local_pid_dead = _posix_is_local_pid_dead
2558
def fdatasync(fileno):
2559
"""Flush file contents to disk if possible.
2561
:param fileno: Integer OS file handle.
2562
:raises TransportNotPossible: If flushing to disk is not possible.
2564
fn = getattr(os, 'fdatasync', getattr(os, 'fsync', None))
2569
def ensure_empty_directory_exists(path, exception_class):
2570
"""Make sure a local directory exists and is empty.
2572
If it does not exist, it is created. If it exists and is not empty, an
2573
instance of exception_class is raised.
2578
if e.errno != errno.EEXIST:
2580
if os.listdir(path) != []:
2581
raise exception_class(path)
2584
def is_environment_error(evalue):
2585
"""True if exception instance is due to a process environment issue
2587
This includes OSError and IOError, but also other errors that come from
2588
the operating system or core libraries but are not subclasses of those.
2590
if isinstance(evalue, (EnvironmentError, select.error)):
2592
if sys.platform == "win32" and win32utils._is_pywintypes_error(evalue):
added
removed
bzrlib/osutils.py
