Mercurial > hg
annotate hgext/largefiles/basestore.py @ 18155:5206af8894a3
largefiles: cleanup of warnings on errors getting largefiles
Especially the "no default or default-push path set in hgrc" was often very
misleading and didn't give any hint where it actually was looking.
A long error messages is better than several multi-line messages.
author | Mads Kiilerich <madski@unity3d.com> |
---|---|
date | Thu, 13 Dec 2012 19:19:06 +0100 |
parents | e7cfe3587ea4 |
children | abfbb04fab8e |
rev | line source |
---|---|
15168 | 1 # Copyright 2009-2010 Gregory P. Ward |
2 # Copyright 2009-2010 Intelerad Medical Systems Incorporated | |
3 # Copyright 2010-2011 Fog Creek Software | |
4 # Copyright 2010-2011 Unity Technologies | |
5 # | |
6 # This software may be used and distributed according to the terms of the | |
7 # GNU General Public License version 2 or any later version. | |
8 | |
15252
6e809bb4f969
largefiles: improve comments, internal docstrings
Greg Ward <greg@gerg.ca>
parents:
15169
diff
changeset
|
9 '''base class for store implementations and store-related utility code''' |
15168 | 10 |
11 import binascii | |
12 import re | |
13 | |
14 from mercurial import util, node, hg | |
15 from mercurial.i18n import _ | |
16 | |
17 import lfutil | |
18 | |
19 class StoreError(Exception): | |
20 '''Raised when there is a problem getting files from or putting | |
21 files to a central store.''' | |
22 def __init__(self, filename, hash, url, detail): | |
23 self.filename = filename | |
24 self.hash = hash | |
25 self.url = url | |
26 self.detail = detail | |
27 | |
28 def longmessage(self): | |
18155
5206af8894a3
largefiles: cleanup of warnings on errors getting largefiles
Mads Kiilerich <madski@unity3d.com>
parents:
17424
diff
changeset
|
29 return (_("error getting %s from %s for %s: %s\n") % |
5206af8894a3
largefiles: cleanup of warnings on errors getting largefiles
Mads Kiilerich <madski@unity3d.com>
parents:
17424
diff
changeset
|
30 (self.hash, self.url, self.filename, self.detail)) |
15168 | 31 |
32 def __str__(self): | |
33 return "%s: %s" % (self.url, self.detail) | |
34 | |
35 class basestore(object): | |
36 def __init__(self, ui, repo, url): | |
37 self.ui = ui | |
38 self.repo = repo | |
39 self.url = url | |
40 | |
41 def put(self, source, hash): | |
42 '''Put source file into the store under <filename>/<hash>.''' | |
43 raise NotImplementedError('abstract method') | |
44 | |
17127
9e1616307c4c
largefiles: batch statlfile requests when pushing a largefiles repo (issue3386)
Na'Tosha Bard <natosha@unity3d.com>
parents:
16247
diff
changeset
|
45 def exists(self, hashes): |
9e1616307c4c
largefiles: batch statlfile requests when pushing a largefiles repo (issue3386)
Na'Tosha Bard <natosha@unity3d.com>
parents:
16247
diff
changeset
|
46 '''Check to see if the store contains the given hashes.''' |
15168 | 47 raise NotImplementedError('abstract method') |
48 | |
49 def get(self, files): | |
50 '''Get the specified largefiles from the store and write to local | |
51 files under repo.root. files is a list of (filename, hash) | |
17424
e7cfe3587ea4
fix trivial spelling errors
Mads Kiilerich <mads@kiilerich.com>
parents:
17127
diff
changeset
|
52 tuples. Return (success, missing), lists of files successfully |
15168 | 53 downloaded and those not found in the store. success is a list |
54 of (filename, hash) tuples; missing is a list of filenames that | |
55 we could not get. (The detailed error message will already have | |
56 been presented to the user, so missing is just supplied as a | |
57 summary.)''' | |
58 success = [] | |
59 missing = [] | |
60 ui = self.ui | |
61 | |
62 at = 0 | |
63 for filename, hash in files: | |
64 ui.progress(_('getting largefiles'), at, unit='lfile', | |
65 total=len(files)) | |
66 at += 1 | |
67 ui.note(_('getting %s:%s\n') % (filename, hash)) | |
68 | |
15316
c65f5b6e26d4
largefiles: rename functions and methods to match desired behavior
Benjamin Pollack <benjamin@bitquabit.com>
parents:
15302
diff
changeset
|
69 storefilename = lfutil.storepath(self.repo, hash) |
16154
9b072a5f8f92
largefiles: respect store.createmode in basestore.get
Martin Geisler <mg@aragost.com>
parents:
15943
diff
changeset
|
70 tmpfile = util.atomictempfile(storefilename, |
9b072a5f8f92
largefiles: respect store.createmode in basestore.get
Martin Geisler <mg@aragost.com>
parents:
15943
diff
changeset
|
71 createmode=self.repo.store.createmode) |
15168 | 72 |
73 try: | |
74 hhash = binascii.hexlify(self._getfile(tmpfile, filename, hash)) | |
75 except StoreError, err: | |
76 ui.warn(err.longmessage()) | |
77 hhash = "" | |
78 | |
79 if hhash != hash: | |
80 if hhash != "": | |
81 ui.warn(_('%s: data corruption (expected %s, got %s)\n') | |
82 % (filename, hash, hhash)) | |
16154
9b072a5f8f92
largefiles: respect store.createmode in basestore.get
Martin Geisler <mg@aragost.com>
parents:
15943
diff
changeset
|
83 tmpfile.discard() # no-op if it's already closed |
15168 | 84 missing.append(filename) |
85 continue | |
86 | |
16154
9b072a5f8f92
largefiles: respect store.createmode in basestore.get
Martin Geisler <mg@aragost.com>
parents:
15943
diff
changeset
|
87 tmpfile.close() |
15316
c65f5b6e26d4
largefiles: rename functions and methods to match desired behavior
Benjamin Pollack <benjamin@bitquabit.com>
parents:
15302
diff
changeset
|
88 lfutil.linktousercache(self.repo, hash) |
15168 | 89 success.append((filename, hhash)) |
90 | |
91 ui.progress(_('getting largefiles'), None) | |
92 return (success, missing) | |
93 | |
94 def verify(self, revs, contents=False): | |
95 '''Verify the existence (and, optionally, contents) of every big | |
96 file revision referenced by every changeset in revs. | |
97 Return 0 if all is well, non-zero on any errors.''' | |
98 write = self.ui.write | |
99 failed = False | |
100 | |
101 write(_('searching %d changesets for largefiles\n') % len(revs)) | |
102 verified = set() # set of (filename, filenode) tuples | |
103 | |
104 for rev in revs: | |
105 cctx = self.repo[rev] | |
106 cset = "%d:%s" % (cctx.rev(), node.short(cctx.node())) | |
107 | |
15319
9da7e96cd5c2
largefiles: remove redundant any_ function
Benjamin Pollack <benjamin@bitquabit.com>
parents:
15316
diff
changeset
|
108 failed = util.any(self._verifyfile( |
15168 | 109 cctx, cset, contents, standin, verified) for standin in cctx) |
110 | |
16247
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
111 numrevs = len(verified) |
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
112 numlfiles = len(set([fname for (fname, fnode) in verified])) |
15168 | 113 if contents: |
114 write(_('verified contents of %d revisions of %d largefiles\n') | |
16247
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
115 % (numrevs, numlfiles)) |
15168 | 116 else: |
117 write(_('verified existence of %d revisions of %d largefiles\n') | |
16247
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
118 % (numrevs, numlfiles)) |
15168 | 119 |
120 return int(failed) | |
121 | |
122 def _getfile(self, tmpfile, filename, hash): | |
123 '''Fetch one revision of one file from the store and write it | |
124 to tmpfile. Compute the hash of the file on-the-fly as it | |
125 downloads and return the binary hash. Close tmpfile. Raise | |
126 StoreError if unable to download the file (e.g. it does not | |
127 exist in the store).''' | |
128 raise NotImplementedError('abstract method') | |
129 | |
130 def _verifyfile(self, cctx, cset, contents, standin, verified): | |
131 '''Perform the actual verification of a file in the store. | |
132 ''' | |
133 raise NotImplementedError('abstract method') | |
134 | |
135 import localstore, wirestore | |
136 | |
137 _storeprovider = { | |
138 'file': [localstore.localstore], | |
139 'http': [wirestore.wirestore], | |
140 'https': [wirestore.wirestore], | |
141 'ssh': [wirestore.wirestore], | |
142 } | |
143 | |
144 _scheme_re = re.compile(r'^([a-zA-Z0-9+-.]+)://') | |
145 | |
146 # During clone this function is passed the src's ui object | |
147 # but it needs the dest's ui object so it can read out of | |
148 # the config file. Use repo.ui instead. | |
149 def _openstore(repo, remote=None, put=False): | |
150 ui = repo.ui | |
151 | |
152 if not remote: | |
15943
f9efb325ea32
largefiles: fix caching largefiles from an aliased repo (issue3212)
Na'Tosha Bard <natosha@unity3d.com>
parents:
15319
diff
changeset
|
153 lfpullsource = getattr(repo, 'lfpullsource', None) |
f9efb325ea32
largefiles: fix caching largefiles from an aliased repo (issue3212)
Na'Tosha Bard <natosha@unity3d.com>
parents:
15319
diff
changeset
|
154 if lfpullsource: |
f9efb325ea32
largefiles: fix caching largefiles from an aliased repo (issue3212)
Na'Tosha Bard <natosha@unity3d.com>
parents:
15319
diff
changeset
|
155 path = ui.expandpath(lfpullsource) |
f9efb325ea32
largefiles: fix caching largefiles from an aliased repo (issue3212)
Na'Tosha Bard <natosha@unity3d.com>
parents:
15319
diff
changeset
|
156 else: |
f9efb325ea32
largefiles: fix caching largefiles from an aliased repo (issue3212)
Na'Tosha Bard <natosha@unity3d.com>
parents:
15319
diff
changeset
|
157 path = ui.expandpath('default-push', 'default') |
15252
6e809bb4f969
largefiles: improve comments, internal docstrings
Greg Ward <greg@gerg.ca>
parents:
15169
diff
changeset
|
158 |
6e809bb4f969
largefiles: improve comments, internal docstrings
Greg Ward <greg@gerg.ca>
parents:
15169
diff
changeset
|
159 # ui.expandpath() leaves 'default-push' and 'default' alone if |
6e809bb4f969
largefiles: improve comments, internal docstrings
Greg Ward <greg@gerg.ca>
parents:
15169
diff
changeset
|
160 # they cannot be expanded: fallback to the empty string, |
6e809bb4f969
largefiles: improve comments, internal docstrings
Greg Ward <greg@gerg.ca>
parents:
15169
diff
changeset
|
161 # meaning the current directory. |
15168 | 162 if path == 'default-push' or path == 'default': |
163 path = '' | |
164 remote = repo | |
165 else: | |
166 remote = hg.peer(repo, {}, path) | |
167 | |
168 # The path could be a scheme so use Mercurial's normal functionality | |
169 # to resolve the scheme to a repository and use its path | |
15169
aa262fff87ac
largefile: fix up hasattr usage
Matt Mackall <mpm@selenic.com>
parents:
15168
diff
changeset
|
170 path = util.safehasattr(remote, 'url') and remote.url() or remote.path |
15168 | 171 |
172 match = _scheme_re.match(path) | |
173 if not match: # regular filesystem path | |
174 scheme = 'file' | |
175 else: | |
176 scheme = match.group(1) | |
177 | |
178 try: | |
179 storeproviders = _storeprovider[scheme] | |
180 except KeyError: | |
181 raise util.Abort(_('unsupported URL scheme %r') % scheme) | |
182 | |
16247
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
183 for classobj in storeproviders: |
15168 | 184 try: |
16247
d87d9d8a8e03
largefiles: remove use of underscores that breaks coding convention
Na'Tosha Bard <natosha@unity3d.com>
parents:
16154
diff
changeset
|
185 return classobj(ui, repo, remote) |
15168 | 186 except lfutil.storeprotonotcapable: |
187 pass | |
188 | |
15302
225d30bacabd
largefiles: string formatting typo in basestore._openstore where comma is used instead of modulo
Hao Lian <hao@fogcreek.com>
parents:
15255
diff
changeset
|
189 raise util.Abort(_('%s does not appear to be a largefile store') % path) |