CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-04 06:50:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update related to the #2769

Browse source
This commit is contained in:
Miroslav Stampar 2018-01-25 12:29:56 +01:00
parent c5730ee88d
commit 4bf20066ec
3 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
lib/utils/sqlalchemy.py
View file

@ -41,6 +41,8 @@ class SQLAlchemy(GenericConnector):
def connect(self):
if _sqlalchemy:
import pdb
pdb.set_trace()
self.initConnection()
try:
@ -54,7 +56,13 @@ class SQLAlchemy(GenericConnector):
if self.dialect:
conf.direct = conf.direct.replace(conf.dbms, self.dialect, 1)
engine = _sqlalchemy.create_engine(conf.direct, connect_args={"check_same_thread": False} if self.dialect == "sqlite" else {})
if self.dialect == "sqlite":
engine = _sqlalchemy.create_engine(conf.direct, connect_args={"check_same_thread": False})
elif self.dialect == "oracle":
engine = _sqlalchemy.create_engine(conf.direct, connect_args={"allow_twophase": False})
else:
engine = _sqlalchemy.create_engine(conf.direct, connect_args={})
self.connector = engine.connect()
except (TypeError, ValueError):
if "_get_server_version_info" in traceback.format_exc():