how to get tablename

[lmy1223]

hi ~ If provide an SQL parameter, how to get the SQL tablename？
thanks ~

[yzjkzq]

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