pointer.py
81 lines
| 2.7 KiB
| text/x-python
|
PythonLexer
|
r35097 | # pointer.py - Git-LFS pointer serialization | ||
# | ||||
# Copyright 2017 Facebook, Inc. | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
from __future__ import absolute_import | ||||
import re | ||||
|
r35098 | from mercurial.i18n import _ | ||
|
r35097 | from mercurial import ( | ||
error, | ||||
|
r36620 | pycompat, | ||
|
r35097 | ) | ||
|
r37945 | from mercurial.utils import ( | ||
stringutil, | ||||
) | ||||
|
r35097 | |||
class InvalidPointer(error.RevlogError): | ||||
pass | ||||
class gitlfspointer(dict): | ||||
VERSION = 'https://git-lfs.github.com/spec/v1' | ||||
def __init__(self, *args, **kwargs): | ||||
self['version'] = self.VERSION | ||||
|
r36620 | super(gitlfspointer, self).__init__(*args) | ||
self.update(pycompat.byteskwargs(kwargs)) | ||||
|
r35097 | |||
@classmethod | ||||
def deserialize(cls, text): | ||||
try: | ||||
return cls(l.split(' ', 1) for l in text.splitlines()).validate() | ||||
except ValueError: # l.split returns 1 item instead of 2 | ||||
|
r37961 | raise InvalidPointer(_('cannot parse git-lfs text: %s') | ||
% stringutil.pprint(text)) | ||||
|
r35097 | |||
def serialize(self): | ||||
sortkeyfunc = lambda x: (x[0] != 'version', x) | ||||
items = sorted(self.validate().iteritems(), key=sortkeyfunc) | ||||
return ''.join('%s %s\n' % (k, v) for k, v in items) | ||||
def oid(self): | ||||
return self['oid'].split(':')[-1] | ||||
def size(self): | ||||
return int(self['size']) | ||||
# regular expressions used by _validate | ||||
# see https://github.com/git-lfs/git-lfs/blob/master/docs/spec.md | ||||
|
r36618 | _keyre = re.compile(br'\A[a-z0-9.-]+\Z') | ||
_valuere = re.compile(br'\A[^\n]*\Z') | ||||
|
r35097 | _requiredre = { | ||
|
r36618 | 'size': re.compile(br'\A[0-9]+\Z'), | ||
'oid': re.compile(br'\Asha256:[0-9a-f]{64}\Z'), | ||||
'version': re.compile(br'\A%s\Z' % re.escape(VERSION)), | ||||
|
r35097 | } | ||
def validate(self): | ||||
"""raise InvalidPointer on error. return self if there is no error""" | ||||
requiredcount = 0 | ||||
for k, v in self.iteritems(): | ||||
if k in self._requiredre: | ||||
if not self._requiredre[k].match(v): | ||||
|
r37961 | raise InvalidPointer(_('unexpected value: %s=%s') | ||
% (k, stringutil.pprint(v))) | ||||
|
r35097 | requiredcount += 1 | ||
elif not self._keyre.match(k): | ||||
raise InvalidPointer(_('unexpected key: %s') % k) | ||||
if not self._valuere.match(v): | ||||
|
r37961 | raise InvalidPointer(_('unexpected value: %s=%s') | ||
% (k, stringutil.pprint(v))) | ||||
|
r35097 | if len(self._requiredre) != requiredcount: | ||
miss = sorted(set(self._requiredre.keys()).difference(self.keys())) | ||||
raise InvalidPointer(_('missed keys: %s') % ', '.join(miss)) | ||||
return self | ||||
deserialize = gitlfspointer.deserialize | ||||