Snippets support.

Parse, print.
This commit is contained in:
Simon Forman
2022-03-01 19:13:44 -08:00
parent 142d6e53b0
commit 87fe1d0b3c
3 changed files with 20 additions and 8 deletions
+4 -5
View File
@@ -2,7 +2,7 @@ from collections import namedtuple
from re import compile as RE
Snippet = namedtuple('Snippet', 'sha offset length')
fmt = '{%s %i %i}'
_fmt = '{%s %i %i}'
pat = (
'{'
'\s*'
@@ -14,21 +14,20 @@ pat = (
'\s*'
'}'
)
PAT = RE(pat)
_PAT = RE(pat)
def to_string(snip):
return fmt % _ts(*snip)
return _fmt % _ts(*snip)
def _ts(sha, offset, length):
return sha.decode('ascii'), offset, length
def from_string(text):
m = PAT.match(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))
+8 -3
View File
@@ -71,6 +71,7 @@ printed left-to-right. These functions are written to support :doc:`../pretty`.
'''
from .errors import NotAListError
from .snippets import Snippet, to_string as snip_to_string
def list_to_stack(el, stack=()):
@@ -133,20 +134,24 @@ _JOY_BOOL_LITS = 'false', 'true'
def _joy_repr(thing):
if isinstance(thing, bool):
return _JOY_BOOL_LITS[thing]
return repr(thing)
if isinstance(thing, bool): return _JOY_BOOL_LITS[thing]
if isinstance(thing, Snippet): return snip_to_string(thing)
return repr(thing)
def _to_string(stack, f):
if not isinstance(stack, tuple): return _joy_repr(stack)
if not stack: return '' # shortcut
if isinstance(stack, Snippet): return snip_to_string(stack)
return ' '.join(map(_s, f(stack)))
_s = lambda s: (
'[%s]' % expression_to_string(s)
if isinstance(s, tuple)
and not isinstance(s, Snippet)
# Is it worth making a non-tuple class for Snippet?
# Doing this check on each tuple seems a bit much.
else _joy_repr(s)
)