GeezerVison is showing github.com — codeql/python/extractor/lark/indenter.py at codeql-cli/latest · github/codeql

Expand file tree

indenter.py

Copy path

More file actions

Latest commit

History

55 lines (42 loc) · 1.68 KB

indenter.py

Copy path

File metadata and controls

55 lines (42 loc) · 1.68 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

"Provides Indentation services for languages with indentation similar to Python"

from .lexer import Token

###{standalone

class Indenter:

def __init__(self):

self.paren_level = 0

self.indent_level = [0]

def handle_NL(self, token):

if self.paren_level > 0:

return

yield token

indent_str = token.rsplit('\n', 1)[1] # Tabs and spaces

indent = indent_str.count(' ') + indent_str.count('\t') * self.tab_len

if indent > self.indent_level[-1]:

self.indent_level.append(indent)

yield Token.new_borrow_pos(self.INDENT_type, indent_str, token)

else:

while indent < self.indent_level[-1]:

self.indent_level.pop()

yield Token.new_borrow_pos(self.DEDENT_type, indent_str, token)

assert indent == self.indent_level[-1], '%s != %s' % (indent, self.indent_level[-1])

def process(self, stream):

for token in stream:

if token.type == self.NL_type:

for t in self.handle_NL(token):

yield t

else:

yield token

if token.type in self.OPEN_PAREN_types:

self.paren_level += 1

elif token.type in self.CLOSE_PAREN_types:

self.paren_level -= 1

assert self.paren_level >= 0

while len(self.indent_level) > 1:

self.indent_level.pop()

yield Token(self.DEDENT_type, '')

assert self.indent_level == [0], self.indent_level

# XXX Hack for ContextualLexer. Maybe there's a more elegant solution?

@property

def always_accept(self):

return (self.NL_type,)

###}