def extract_table_identifiers()

in src/redash_stmo/handlers/query_results/parser.py [0:0]

• 12 lines of code
• 8 McCabe index (conditional complexity)

def extract_table_identifiers(token_stream):
    for item in token_stream:
        try:
            if isinstance(item, IdentifierList):
                for identifier in item.get_identifiers():
                    yield identifier.get_real_name()
            elif isinstance(item, Identifier):
                yield item.get_real_name()
            # It's a bug to check for Keyword here, but in the example
            # above some tables names are identified as keywords...
            elif item.ttype is Keyword:
                yield item.value
        except StopIteration:
            return