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.
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.
2065
2176
report_activity(sent, 'write')
2179
def connect_socket(address):
2180
# Slight variation of the socket.create_connection() function (provided by
2181
# python-2.6) that can fail if getaddrinfo returns an empty list. We also
2182
# provide it for previous python versions. Also, we don't use the timeout
2183
# parameter (provided by the python implementation) so we don't implement
2185
err = socket.error('getaddrinfo returns an empty list')
2186
host, port = address
2187
for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
2188
af, socktype, proto, canonname, sa = res
2191
sock = socket.socket(af, socktype, proto)
2195
except socket.error, err:
2196
# 'err' is now the most recent error
2197
if sock is not None:
2068
2202
def dereference_path(path):
2069
2203
"""Determine the real path to a file.
2130
2264
return file_kind_from_stat_mode(mode)
2131
2265
file_kind_from_stat_mode = file_kind_from_stat_mode_thunk
2134
def file_kind(f, _lstat=os.lstat):
2267
def file_stat(f, _lstat=os.lstat):
2136
return file_kind_from_stat_mode(_lstat(f).st_mode)
2137
2271
except OSError, e:
2138
2272
if getattr(e, 'errno', None) in (errno.ENOENT, errno.ENOTDIR):
2139
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)
2143
2280
def until_no_eintr(f, *a, **kw):
2144
2281
"""Run f(*a, **kw), retrying if an EINTR error occurs.
2204
2341
termios.tcsetattr(fd, termios.TCSADRAIN, settings)
2208
if sys.platform == 'linux2':
2344
if sys.platform.startswith('linux'):
2209
2345
def _local_concurrency():
2211
prefix = 'processor'
2212
for line in file('/proc/cpuinfo', 'rb'):
2213
if line.startswith(prefix):
2214
concurrency = int(line[line.find(':')+1:]) + 1
2347
return os.sysconf('SC_NPROCESSORS_ONLN')
2348
except (ValueError, OSError, AttributeError):
2216
2350
elif sys.platform == 'darwin':
2217
2351
def _local_concurrency():
2218
2352
return subprocess.Popen(['sysctl', '-n', 'hw.availcpu'],
2219
2353
stdout=subprocess.PIPE).communicate()[0]
2220
elif sys.platform[0:7] == 'freebsd':
2354
elif "bsd" in sys.platform:
2221
2355
def _local_concurrency():
2222
2356
return subprocess.Popen(['sysctl', '-n', 'hw.ncpu'],
2223
2357
stdout=subprocess.PIPE).communicate()[0]
2251
2385
concurrency = os.environ.get('BZR_CONCURRENCY', None)
2252
2386
if concurrency is None:
2254
concurrency = _local_concurrency()
2255
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):
2258
2399
concurrency = int(concurrency)
2259
2400
except (TypeError, ValueError):
2321
2462
open_file = open
2324
def getuser_unicode():
2325
"""Return the username as unicode.
2328
user_encoding = get_user_encoding()
2329
username = getpass.getuser().decode(user_encoding)
2330
except UnicodeDecodeError:
2331
raise errors.BzrError("Can't decode username as %s." % \
2465
def available_backup_name(base, exists):
2466
"""Find a non-existing backup file name.
2468
This will *not* create anything, this only return a 'free' entry. This
2469
should be used for checking names in a directory below a locked
2470
tree/branch/repo to avoid race conditions. This is LBYL (Look Before You
2471
Leap) and generally discouraged.
2473
:param base: The base name.
2475
:param exists: A callable returning True if the path parameter exists.
2478
name = "%s.~%d~" % (base, counter)
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
