localstore.py
86 lines
| 2.9 KiB
| text/x-python
|
PythonLexer
|
r15168 | # Copyright 2009-2010 Gregory P. Ward | ||
# Copyright 2009-2010 Intelerad Medical Systems Incorporated | ||||
# Copyright 2010-2011 Fog Creek Software | ||||
# Copyright 2010-2011 Unity Technologies | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
|
r15252 | '''store class for local filesystem''' | ||
|
r15168 | |||
import os | ||||
from mercurial import util | ||||
from mercurial.i18n import _ | ||||
import lfutil | ||||
import basestore | ||||
class localstore(basestore.basestore): | ||||
|
r15317 | '''localstore first attempts to grab files out of the store in the remote | ||
Mercurial repository. Failling that, it attempts to grab the files from | ||||
the user cache.''' | ||||
|
r15168 | |||
def __init__(self, ui, repo, remote): | ||||
|
r17191 | url = os.path.join(remote.local().path, '.hg', lfutil.longname) | ||
|
r15168 | super(localstore, self).__init__(ui, repo, util.expandpath(url)) | ||
|
r17191 | self.remote = remote.local() | ||
|
r15168 | |||
|
r15317 | def put(self, source, hash): | ||
|
r15371 | util.makedirs(os.path.dirname(lfutil.storepath(self.remote, hash))) | ||
|
r15317 | if lfutil.instore(self.remote, hash): | ||
return | ||||
lfutil.link(lfutil.storepath(self.repo, hash), | ||||
lfutil.storepath(self.remote, hash)) | ||||
|
r15168 | |||
|
r17411 | def exists(self, hashes): | ||
retval = {} | ||||
for hash in hashes: | ||||
retval[hash] = lfutil.instore(self.remote, hash) | ||||
return retval | ||||
|
r15168 | |||
def _getfile(self, tmpfile, filename, hash): | ||||
|
r15317 | if lfutil.instore(self.remote, hash): | ||
path = lfutil.storepath(self.remote, hash) | ||||
elif lfutil.inusercache(self.ui, hash): | ||||
path = lfutil.usercachepath(self.ui, hash) | ||||
else: | ||||
raise basestore.StoreError(filename, hash, '', | ||||
|
r16928 | _("can't get file locally")) | ||
|
r15317 | fd = open(path, 'rb') | ||
try: | ||||
return lfutil.copyandhash(fd, tmpfile) | ||||
finally: | ||||
fd.close() | ||||
|
r15168 | |||
def _verifyfile(self, cctx, cset, contents, standin, verified): | ||||
filename = lfutil.splitstandin(standin) | ||||
if not filename: | ||||
return False | ||||
fctx = cctx[standin] | ||||
key = (filename, fctx.filenode()) | ||||
if key in verified: | ||||
return False | ||||
expecthash = fctx.data()[0:40] | ||||
verified.add(key) | ||||
|
r15317 | if not lfutil.instore(self.remote, expecthash): | ||
|
r15168 | self.ui.warn( | ||
_('changeset %s: %s missing\n' | ||||
' (looked for hash %s)\n') | ||||
% (cset, filename, expecthash)) | ||||
return True # failed | ||||
if contents: | ||||
|
r15317 | storepath = lfutil.storepath(self.remote, expecthash) | ||
|
r15168 | actualhash = lfutil.hashfile(storepath) | ||
if actualhash != expecthash: | ||||
self.ui.warn( | ||||
_('changeset %s: %s: contents differ\n' | ||||
' (%s:\n' | ||||
' expected hash %s,\n' | ||||
' but got %s)\n') | ||||
% (cset, filename, storepath, expecthash, actualhash)) | ||||
return True # failed | ||||
return False | ||||