How to deal with "Not supported query type!"?

[christopherpickering]

I'm running the sql-metadata parser on a bunch of sqlserver queries. Its pretty common to start a sql server query with the database name -

use cool_db;

In the parser, this throws a not supported error because the parser is only looked at the first query from sqlparse.

Or, if I have a messy statement -

;
;
;
select a from b;
;

The tool is currently only setup for us to pass in select a from b;, but it would be nice it if could handle messier stuff.

What do you think about iterating through the parsed parts instead of grabbing part one?

I'm using a simple wrapper which removes leading whitespace and punctuation -

parsed = sqlparse.parse(query)

for parse in parsed:
    count = 0

    # remove blank line tokens from front of query
    for x, token in enumerate(copy(parse.tokens)):
        if token.ttype == sqlparse.tokens.Text.Whitespace.Newline and count == 0:
            parse.tokens.pop(x)
        else:
            count += 1

    if parse.tokens[0].ttype != sqlparse.tokens.Punctuation:
        if parse.tokens[0].value == 'USE':
            continue

        a = Parser(str(parse))

I guess this leads to another question! What if we are querying cool_db and joining to another_cool_db - how can we determine which tables/column belong to which db?