From 1de9198f0c69152fe6f474185ac351e9500a4ce8 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Thu, 24 Nov 2016 21:38:10 +0000 Subject: Implement document matcher parser --- mammoth/styles/parser/token_iterator.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 mammoth/styles/parser/token_iterator.py (limited to 'mammoth/styles/parser/token_iterator.py') diff --git a/mammoth/styles/parser/token_iterator.py b/mammoth/styles/parser/token_iterator.py new file mode 100644 index 0000000..810e50b --- /dev/null +++ b/mammoth/styles/parser/token_iterator.py @@ -0,0 +1,31 @@ +class TokenIterator(object): + def __init__(self, tokens): + self._tokens = tokens + self._index = 0 + + def next_value(self, token_type): + return self._next(token_type).value + + def _next(self, token_type): + token = self._tokens[self._index] + if token.type == token_type: + self._index += 1 + return token + else: + raise self._unexpected_token_type(token_type, token) + + def skip(self, token_type, token_value): + token = self._tokens[self._index] + if token.type == token_type and token.value == token_value: + self._index += 1 + return True + else: + raise self._unexpected_token_type(token_type, token) + + def try_skip(self, token_type, token_value): + token = self._tokens[self._index] + if token.type == token_type and token.value == token_value: + self._index += 1 + return True + else: + return False -- cgit v1.2.3