scmutil.py
122 lines
| 4.1 KiB
| text/x-python
|
PythonLexer
/ mercurial / scmutil.py
Adrian Buehlmann
|
r13962 | # scmutil.py - Mercurial core utility functions | ||
# | ||||
# Copyright Matt Mackall <mpm@selenic.com> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
from i18n import _ | ||||
import util, error | ||||
Adrian Buehlmann
|
r13970 | import os, errno | ||
Adrian Buehlmann
|
r13962 | |||
def checkportable(ui, f): | ||||
'''Check if filename f is portable and warn or abort depending on config''' | ||||
util.checkfilename(f) | ||||
val = ui.config('ui', 'portablefilenames', 'warn') | ||||
lval = val.lower() | ||||
abort = os.name == 'nt' or lval == 'abort' | ||||
bval = util.parsebool(val) | ||||
if abort or lval == 'warn' or bval: | ||||
msg = util.checkwinfilename(f) | ||||
if msg: | ||||
if abort: | ||||
raise util.Abort("%s: %r" % (msg, f)) | ||||
ui.warn(_("warning: %s: %r\n") % (msg, f)) | ||||
elif bval is None and lval != 'ignore': | ||||
raise error.ConfigError( | ||||
_("ui.portablefilenames value is invalid ('%s')") % val) | ||||
Adrian Buehlmann
|
r13970 | |||
class opener(object): | ||||
'''Open files relative to a base directory | ||||
This class is used to hide the details of COW semantics and | ||||
remote file access from higher level code. | ||||
''' | ||||
def __init__(self, base, audit=True): | ||||
self.base = base | ||||
if audit: | ||||
self.auditor = util.path_auditor(base) | ||||
else: | ||||
self.auditor = util.always | ||||
self.createmode = None | ||||
self._trustnlink = None | ||||
@util.propertycache | ||||
def _can_symlink(self): | ||||
return util.checklink(self.base) | ||||
def _fixfilemode(self, name): | ||||
if self.createmode is None: | ||||
return | ||||
os.chmod(name, self.createmode & 0666) | ||||
def __call__(self, path, mode="r", text=False, atomictemp=False): | ||||
r = util.checkosfilename(path) | ||||
if r: | ||||
raise Abort("%s: %r" % (r, path)) | ||||
self.auditor(path) | ||||
f = os.path.join(self.base, path) | ||||
if not text and "b" not in mode: | ||||
mode += "b" # for that other OS | ||||
nlink = -1 | ||||
dirname, basename = os.path.split(f) | ||||
# If basename is empty, then the path is malformed because it points | ||||
# to a directory. Let the posixfile() call below raise IOError. | ||||
if basename and mode not in ('r', 'rb'): | ||||
if atomictemp: | ||||
if not os.path.isdir(dirname): | ||||
util.makedirs(dirname, self.createmode) | ||||
return util.atomictempfile(f, mode, self.createmode) | ||||
try: | ||||
if 'w' in mode: | ||||
util.unlink(f) | ||||
nlink = 0 | ||||
else: | ||||
# nlinks() may behave differently for files on Windows | ||||
# shares if the file is open. | ||||
fd = util.posixfile(f) | ||||
nlink = util.nlinks(f) | ||||
if nlink < 1: | ||||
nlink = 2 # force mktempcopy (issue1922) | ||||
fd.close() | ||||
except (OSError, IOError), e: | ||||
if e.errno != errno.ENOENT: | ||||
raise | ||||
nlink = 0 | ||||
if not os.path.isdir(dirname): | ||||
util.makedirs(dirname, self.createmode) | ||||
if nlink > 0: | ||||
if self._trustnlink is None: | ||||
self._trustnlink = nlink > 1 or util.checknlink(f) | ||||
if nlink > 1 or not self._trustnlink: | ||||
util.rename(util.mktempcopy(f), f) | ||||
fp = util.posixfile(f, mode) | ||||
if nlink == 0: | ||||
self._fixfilemode(f) | ||||
return fp | ||||
def symlink(self, src, dst): | ||||
self.auditor(dst) | ||||
linkname = os.path.join(self.base, dst) | ||||
try: | ||||
os.unlink(linkname) | ||||
except OSError: | ||||
pass | ||||
dirname = os.path.dirname(linkname) | ||||
if not os.path.exists(dirname): | ||||
util.makedirs(dirname, self.createmode) | ||||
if self._can_symlink: | ||||
try: | ||||
os.symlink(src, linkname) | ||||
except OSError, err: | ||||
raise OSError(err.errno, _('could not symlink to %r: %s') % | ||||
(src, err.strerror), linkname) | ||||
else: | ||||
f = self(dst, "w") | ||||
f.write(src) | ||||
f.close() | ||||
self._fixfilemode(dst) | ||||