Thun/implementations/Python/joy/utils/snippets.py

34 lines
639 B
Python

from collections import namedtuple
from re import compile as RE
Snippet = namedtuple('Snippet', 'sha offset length')
_fmt = '{%s %i %i}'
pat = (
'{'
'\s*'
'(?P<sha>[a-f0-9]+)'
'\s+'
'(?P<offset>\d+)'
'\s+'
'(?P<length>\d+)'
'\s*'
'}'
)
_PAT = RE(pat)
def to_string(snip):
return _fmt % _ts(*snip)
def _ts(sha, offset, length):
return sha.decode('ascii'), offset, length
def from_string(text):
m = _PAT.match(text)
if not m:
raise ValueError
return _fs(**m.groupdict())
def _fs(sha, offset, length):
return Snippet(sha.encode('ascii'), int(offset), int(length))