extract_tables() issue

[Eric L]

Hello, when I use the sample code of extract_tables() with the following SQL statement:

"select test from (select test2 from mytable where (a=b)) tt"

I get "tt", but I think I should get "tt" and "mytable"

I have looked at the code and it seems that the parenthesis are not properly handled in the code. Can you help with this?

[Ahmad Zareef]

You may replace the following function in the example:

extract_table_identifiers(token_stream)

with the below definition:

def extract_table_identifiers(token_stream):
    for item in token_stream:
        if isinstance(item, IdentifierList):
            for identifier in item.get_identifiers():
                yield identifier.value
        elif isinstance(item, Identifier):
            yield item.value
        # It's not necessary (in general cases) to check for Keyword here, but in case
        # some table names are identified as keywords...
        elif item.ttype is Keyword:
            yield item.value

Please let me know if it worked.

Thanks...