This commit is contained in:
Miroslav Stampar
2021-03-25 17:29:14 +01:00
parent 8d6125f067
commit bce3abc829
2 changed files with 12 additions and 3 deletions

View File

@@ -101,13 +101,22 @@ class SQLAlchemy(GenericConnector):
return None
def execute(self, query):
retVal = False
try:
self.cursor = self.connector.execute(query)
retVal = True
except (_sqlalchemy.exc.OperationalError, _sqlalchemy.exc.ProgrammingError) as ex:
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % getSafeExString(ex))
except _sqlalchemy.exc.InternalError as ex:
raise SqlmapConnectionException(getSafeExString(ex))
return retVal
def select(self, query):
self.execute(query)
return self.fetchall()
retVal = None
if self.execute(query):
retVal = self.fetchall()
return retVal