##// END OF EJS Templates
move opener from util to scmutil
Adrian Buehlmann -
r13970:d1391335 default
parent child Browse files
Show More
@@ -4,12 +4,12
4 4 # $ undumprevlog < repo.dump
5 5
6 6 import sys
7 from mercurial import revlog, node, util, transaction
7 from mercurial import revlog, node, scmutil, util, transaction
8 8
9 9 for fp in (sys.stdin, sys.stdout, sys.stderr):
10 10 util.set_binary(fp)
11 11
12 opener = util.opener('.', False)
12 opener = scmutil.opener('.', False)
13 13 tr = transaction.transaction(sys.stderr.write, opener, "undump.journal")
14 14 while 1:
15 15 l = sys.stdin.readline()
@@ -10,7 +10,7 import tempfile
10 10 import urllib
11 11 import urllib2
12 12
13 from mercurial import strutil, util, encoding
13 from mercurial import strutil, scmutil, util, encoding
14 14 from mercurial.i18n import _
15 15
16 16 # Subversion stuff. Works best with very recent Python SVN bindings
@@ -998,8 +998,8 class svn_sink(converter_sink, commandli
998 998 self.run0('checkout', path, wcpath)
999 999
1000 1000 self.wc = wcpath
1001 self.opener = util.opener(self.wc)
1002 self.wopener = util.opener(self.wc)
1001 self.opener = scmutil.opener(self.wc)
1002 self.wopener = scmutil.opener(self.wc)
1003 1003 self.childmap = mapfile(ui, self.join('hg-childmap'))
1004 1004 self.is_exec = util.checkexec(self.wc) and util.is_exec or None
1005 1005
@@ -61,7 +61,7 pretty fast (at least faster than having
61 61
62 62 from mercurial.i18n import _
63 63 from mercurial.node import short, nullid
64 from mercurial import cmdutil, util, commands, encoding
64 from mercurial import cmdutil, scmutil, util, commands, encoding
65 65 import os, shlex, shutil, tempfile, re
66 66
67 67 def snapshot(ui, repo, files, node, tmproot):
@@ -81,7 +81,7 def snapshot(ui, repo, files, node, tmpr
81 81 else:
82 82 ui.note(_('making snapshot of %d files from working directory\n') %
83 83 (len(files)))
84 wopener = util.opener(base)
84 wopener = scmutil.opener(base)
85 85 fns_and_mtime = []
86 86 ctx = repo[node]
87 87 for fn in files:
@@ -45,7 +45,7 create other, independent patch queues w
45 45 from mercurial.i18n import _
46 46 from mercurial.node import bin, hex, short, nullid, nullrev
47 47 from mercurial.lock import release
48 from mercurial import commands, cmdutil, hg, patch, util
48 from mercurial import commands, cmdutil, hg, patch, scmutil, util
49 49 from mercurial import repair, extensions, url, error
50 50 import os, sys, re, errno, shutil
51 51
@@ -259,7 +259,7 class queue(object):
259 259 except IOError:
260 260 curpath = os.path.join(path, 'patches')
261 261 self.path = patchdir or curpath
262 self.opener = util.opener(self.path)
262 self.opener = scmutil.opener(self.path)
263 263 self.ui = ui
264 264 self.applied_dirty = 0
265 265 self.series_dirty = 0
@@ -16,7 +16,7 map from a changeset hash to its hash in
16 16 from mercurial.i18n import _
17 17 import os, tempfile
18 18 from mercurial import bundlerepo, cmdutil, hg, merge, match
19 from mercurial import patch, revlog, util, error
19 from mercurial import patch, revlog, scmutil, util, error
20 20 from mercurial import revset, templatekw
21 21
22 22 class transplantentry(object):
@@ -31,7 +31,7 class transplants(object):
31 31 self.opener = opener
32 32
33 33 if not opener:
34 self.opener = util.opener(self.path)
34 self.opener = scmutil.opener(self.path)
35 35 self.transplants = {}
36 36 self.dirty = False
37 37 self.read()
@@ -74,7 +74,7 class transplanter(object):
74 74 def __init__(self, ui, repo):
75 75 self.ui = ui
76 76 self.path = repo.join('transplant')
77 self.opener = util.opener(self.path)
77 self.opener = scmutil.opener(self.path)
78 78 self.transplants = transplants(self.path, 'transplants',
79 79 opener=self.opener)
80 80
@@ -8,7 +8,7
8 8 from i18n import _
9 9 from node import hex
10 10 import cmdutil
11 import util, encoding
11 import scmutil, util, encoding
12 12 import cStringIO, os, tarfile, time, zipfile
13 13 import zlib, gzip
14 14
@@ -187,7 +187,7 class fileit(object):
187 187
188 188 def __init__(self, name, mtime):
189 189 self.basedir = name
190 self.opener = util.opener(self.basedir)
190 self.opener = scmutil.opener(self.basedir)
191 191
192 192 def addfile(self, name, mode, islink, data):
193 193 if islink:
@@ -9,7 +9,7 from node import hex, bin, nullid, nullr
9 9 from lock import release
10 10 from i18n import _, gettext
11 11 import os, re, sys, difflib, time, tempfile
12 import hg, util, revlog, extensions, copies, error, bookmarks
12 import hg, scmutil, util, revlog, extensions, copies, error, bookmarks
13 13 import patch, help, mdiff, url, encoding, templatekw, discovery
14 14 import archival, changegroup, cmdutil, sshserver, hbisect, hgweb, hgweb.server
15 15 import merge as mergemod
@@ -975,7 +975,7 def debugancestor(ui, repo, *args):
975 975 """find the ancestor revision of two revisions in a given index"""
976 976 if len(args) == 3:
977 977 index, rev1, rev2 = args
978 r = revlog.revlog(util.opener(os.getcwd(), audit=False), index)
978 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False), index)
979 979 lookup = r.lookup
980 980 elif len(args) == 2:
981 981 if not repo:
@@ -1393,7 +1393,7 def debugdag(ui, repo, file_=None, *revs
1393 1393 spaces = opts.get('spaces')
1394 1394 dots = opts.get('dots')
1395 1395 if file_:
1396 rlog = revlog.revlog(util.opener(os.getcwd(), audit=False), file_)
1396 rlog = revlog.revlog(scmutil.opener(os.getcwd(), audit=False), file_)
1397 1397 revs = set((int(r) for r in revs))
1398 1398 def events():
1399 1399 for r in rlog:
@@ -1443,7 +1443,8 def debugdata(ui, repo, file_, rev):
1443 1443 if len(filelog):
1444 1444 r = filelog
1445 1445 if not r:
1446 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_[:-2] + ".i")
1446 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False),
1447 file_[:-2] + ".i")
1447 1448 try:
1448 1449 ui.write(r.revision(r.lookup(rev)))
1449 1450 except KeyError:
@@ -1482,7 +1483,7 def debugindex(ui, repo, file_, **opts):
1482 1483 raise util.Abort(_("unknown format %d") % format)
1483 1484
1484 1485 if not r:
1485 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_)
1486 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False), file_)
1486 1487
1487 1488 if format == 0:
1488 1489 ui.write(" rev offset length base linkrev"
@@ -1515,7 +1516,7 def debugindexdot(ui, repo, file_):
1515 1516 if len(filelog):
1516 1517 r = filelog
1517 1518 if not r:
1518 r = revlog.revlog(util.opener(os.getcwd(), audit=False), file_)
1519 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False), file_)
1519 1520 ui.write("digraph G {\n")
1520 1521 for i in r:
1521 1522 node = r.node(i)
@@ -10,7 +10,7 from i18n import _
10 10 import repo, changegroup, subrepo, discovery, pushkey
11 11 import changelog, dirstate, filelog, manifest, context, bookmarks
12 12 import lock, transaction, store, encoding
13 import util, extensions, hook, error
13 import scmutil, util, extensions, hook, error
14 14 import match as matchmod
15 15 import merge as mergemod
16 16 import tags as tagsmod
@@ -32,8 +32,8 class localrepository(repo.repository):
32 32 self.path = os.path.join(self.root, ".hg")
33 33 self.origroot = path
34 34 self.auditor = util.path_auditor(self.root, self._checknested)
35 self.opener = util.opener(self.path)
36 self.wopener = util.opener(self.root)
35 self.opener = scmutil.opener(self.path)
36 self.wopener = scmutil.opener(self.root)
37 37 self.baseui = baseui
38 38 self.ui = baseui.copy()
39 39
@@ -90,7 +90,7 class localrepository(repo.repository):
90 90 if inst.errno != errno.ENOENT:
91 91 raise
92 92
93 self.store = store.store(requirements, self.sharedpath, util.opener)
93 self.store = store.store(requirements, self.sharedpath, scmutil.opener)
94 94 self.spath = self.store.path
95 95 self.sopener = self.store.opener
96 96 self.sjoin = self.store.join
@@ -11,7 +11,7 import tempfile, zlib
11 11
12 12 from i18n import _
13 13 from node import hex, nullid, short
14 import base85, mdiff, util, diffhelpers, copies, encoding
14 import base85, mdiff, scmutil, util, diffhelpers, copies, encoding
15 15
16 16 gitre = re.compile('diff --git a/(.*) b/(.*)')
17 17
@@ -1111,7 +1111,7 def _applydiff(ui, fp, patcher, copyfn,
1111 1111 err = 0
1112 1112 current_file = None
1113 1113 cwd = os.getcwd()
1114 opener = util.opener(cwd)
1114 opener = scmutil.opener(cwd)
1115 1115
1116 1116 for state, values in iterhunks(ui, fp):
1117 1117 if state == 'hunk':
@@ -7,7 +7,7
7 7
8 8 from i18n import _
9 9 import util, error
10 import os
10 import os, errno
11 11
12 12 def checkportable(ui, f):
13 13 '''Check if filename f is portable and warn or abort depending on config'''
@@ -25,3 +25,98 def checkportable(ui, f):
25 25 elif bval is None and lval != 'ignore':
26 26 raise error.ConfigError(
27 27 _("ui.portablefilenames value is invalid ('%s')") % val)
28
29 class opener(object):
30 '''Open files relative to a base directory
31
32 This class is used to hide the details of COW semantics and
33 remote file access from higher level code.
34 '''
35 def __init__(self, base, audit=True):
36 self.base = base
37 if audit:
38 self.auditor = util.path_auditor(base)
39 else:
40 self.auditor = util.always
41 self.createmode = None
42 self._trustnlink = None
43
44 @util.propertycache
45 def _can_symlink(self):
46 return util.checklink(self.base)
47
48 def _fixfilemode(self, name):
49 if self.createmode is None:
50 return
51 os.chmod(name, self.createmode & 0666)
52
53 def __call__(self, path, mode="r", text=False, atomictemp=False):
54 r = util.checkosfilename(path)
55 if r:
56 raise Abort("%s: %r" % (r, path))
57 self.auditor(path)
58 f = os.path.join(self.base, path)
59
60 if not text and "b" not in mode:
61 mode += "b" # for that other OS
62
63 nlink = -1
64 dirname, basename = os.path.split(f)
65 # If basename is empty, then the path is malformed because it points
66 # to a directory. Let the posixfile() call below raise IOError.
67 if basename and mode not in ('r', 'rb'):
68 if atomictemp:
69 if not os.path.isdir(dirname):
70 util.makedirs(dirname, self.createmode)
71 return util.atomictempfile(f, mode, self.createmode)
72 try:
73 if 'w' in mode:
74 util.unlink(f)
75 nlink = 0
76 else:
77 # nlinks() may behave differently for files on Windows
78 # shares if the file is open.
79 fd = util.posixfile(f)
80 nlink = util.nlinks(f)
81 if nlink < 1:
82 nlink = 2 # force mktempcopy (issue1922)
83 fd.close()
84 except (OSError, IOError), e:
85 if e.errno != errno.ENOENT:
86 raise
87 nlink = 0
88 if not os.path.isdir(dirname):
89 util.makedirs(dirname, self.createmode)
90 if nlink > 0:
91 if self._trustnlink is None:
92 self._trustnlink = nlink > 1 or util.checknlink(f)
93 if nlink > 1 or not self._trustnlink:
94 util.rename(util.mktempcopy(f), f)
95 fp = util.posixfile(f, mode)
96 if nlink == 0:
97 self._fixfilemode(f)
98 return fp
99
100 def symlink(self, src, dst):
101 self.auditor(dst)
102 linkname = os.path.join(self.base, dst)
103 try:
104 os.unlink(linkname)
105 except OSError:
106 pass
107
108 dirname = os.path.dirname(linkname)
109 if not os.path.exists(dirname):
110 util.makedirs(dirname, self.createmode)
111
112 if self._can_symlink:
113 try:
114 os.symlink(src, linkname)
115 except OSError, err:
116 raise OSError(err.errno, _('could not symlink to %r: %s') %
117 (src, err.strerror), linkname)
118 else:
119 f = self(dst, "w")
120 f.write(src)
121 f.close()
122 self._fixfilemode(dst)
@@ -18,7 +18,7
18 18 # s: "i hate that."
19 19
20 20 from i18n import _
21 import util, mdiff
21 import scmutil, util, mdiff
22 22 import sys, os
23 23
24 24 class CantReprocessAndShowBase(Exception):
@@ -429,7 +429,7 def simplemerge(ui, local, base, other,
429 429
430 430 local = os.path.realpath(local)
431 431 if not opts.get('print'):
432 opener = util.opener(os.path.dirname(local))
432 opener = scmutil.opener(os.path.dirname(local))
433 433 out = opener(os.path.basename(local), "w", atomictemp=True)
434 434 else:
435 435 out = sys.stdout
@@ -895,101 +895,6 def makedirs(name, mode=None):
895 895 makedirs(parent, mode)
896 896 makedirs(name, mode)
897 897
898 class opener(object):
899 """Open files relative to a base directory
900
901 This class is used to hide the details of COW semantics and
902 remote file access from higher level code.
903 """
904 def __init__(self, base, audit=True):
905 self.base = base
906 if audit:
907 self.auditor = path_auditor(base)
908 else:
909 self.auditor = always
910 self.createmode = None
911 self._trustnlink = None
912
913 @propertycache
914 def _can_symlink(self):
915 return checklink(self.base)
916
917 def _fixfilemode(self, name):
918 if self.createmode is None:
919 return
920 os.chmod(name, self.createmode & 0666)
921
922 def __call__(self, path, mode="r", text=False, atomictemp=False):
923 r = checkosfilename(path)
924 if r:
925 raise Abort("%s: %r" % (r, path))
926 self.auditor(path)
927 f = os.path.join(self.base, path)
928
929 if not text and "b" not in mode:
930 mode += "b" # for that other OS
931
932 nlink = -1
933 dirname, basename = os.path.split(f)
934 # If basename is empty, then the path is malformed because it points
935 # to a directory. Let the posixfile() call below raise IOError.
936 if basename and mode not in ('r', 'rb'):
937 if atomictemp:
938 if not os.path.isdir(dirname):
939 makedirs(dirname, self.createmode)
940 return atomictempfile(f, mode, self.createmode)
941 try:
942 if 'w' in mode:
943 unlink(f)
944 nlink = 0
945 else:
946 # nlinks() may behave differently for files on Windows
947 # shares if the file is open.
948 fd = posixfile(f)
949 nlink = nlinks(f)
950 if nlink < 1:
951 nlink = 2 # force mktempcopy (issue1922)
952 fd.close()
953 except (OSError, IOError), e:
954 if e.errno != errno.ENOENT:
955 raise
956 nlink = 0
957 if not os.path.isdir(dirname):
958 makedirs(dirname, self.createmode)
959 if nlink > 0:
960 if self._trustnlink is None:
961 self._trustnlink = nlink > 1 or checknlink(f)
962 if nlink > 1 or not self._trustnlink:
963 rename(mktempcopy(f), f)
964 fp = posixfile(f, mode)
965 if nlink == 0:
966 self._fixfilemode(f)
967 return fp
968
969 def symlink(self, src, dst):
970 self.auditor(dst)
971 linkname = os.path.join(self.base, dst)
972 try:
973 os.unlink(linkname)
974 except OSError:
975 pass
976
977 dirname = os.path.dirname(linkname)
978 if not os.path.exists(dirname):
979 makedirs(dirname, self.createmode)
980
981 if self._can_symlink:
982 try:
983 os.symlink(src, linkname)
984 except OSError, err:
985 raise OSError(err.errno, _('could not symlink to %r: %s') %
986 (src, err.strerror), linkname)
987 else:
988 f = self(dst, "w")
989 f.write(src)
990 f.close()
991 self._fixfilemode(dst)
992
993 898 class chunkbuffer(object):
994 899 """Allow arbitrary sized chunks of data to be efficiently read from an
995 900 iterator over chunks of arbitrary size."""
@@ -26,7 +26,7 We approximate that by reducing the read
26 26 summary: change foo
27 27
28 28 $ cat >> test.py << EOF
29 > from mercurial import changelog, util
29 > from mercurial import changelog, scmutil
30 30 > from mercurial.node import *
31 31 >
32 32 > class singlebyteread(object):
@@ -42,7 +42,7 We approximate that by reducing the read
42 42 > return getattr(self.real, key)
43 43 >
44 44 > def opener(*args):
45 > o = util.opener(*args)
45 > o = scmutil.opener(*args)
46 46 > def wrapper(*a):
47 47 > f = o(*a)
48 48 > return singlebyteread(f)
General Comments 0
You need to be logged in to leave comments. Login now