Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def query_type(cls, query: str) -> QueryTypes:
check = query.lstrip()[:8].upper()
if any(
[
check.startswith("SELECT"),
check.startswith("SHOW"),
check.startswith("DESCRIBE"),
check.startswith("EXISTS"),
]
):
return QueryTypes.FETCH
if check.startswith("INSERT"):
return QueryTypes.INSERT
return QueryTypes.OTHER