sqlparse
sqlparse copied to clipboard
how to get tablename
hi ~ If provide an SQL parameter, how to get the SQL tablename? thanks ~
def getTableName(sql):
result = set()
tokens = re.split(r"[\s)(;`]+", sql.upper())
get_next = False
for tok in tokens:
if get_next:
if tok not in ["", "SELECT"]:
result.add(tok)
get_next = False
get_next = tok in ['INTO','FROM','JOIN']
return result