GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-31 13:29:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化：strip sql

Browse Source
This commit is contained in:
duanchao.bill
2022-07-26 12:49:08 +08:00
parent 7340740170
commit f45eae62d0
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
httprunner/database/engine.py
View File

@@ -37,6 +37,7 @@ class DBEngine(object):
pass pass
def _fetch(self, query, size=-1, commit=True): def _fetch(self, query, size=-1, commit=True):
query = query.strip()
result = self.session.execute(query) result = self.session.execute(query)
if query.upper()[:6] == "SELECT": if query.upper()[:6] == "SELECT":
if size < 0: if size < 0:
@@ -80,5 +81,6 @@ class DBEngine(object):
if __name__ == "__main__": if __name__ == "__main__":
# db = DBEngine(f"mysql+pymysql://xxxxx:xxxxx@10.0.0.1:3306/dbname?charset=utf8mb4") # db = DBEngine(f"mysql+pymysql://xxxxx:xxxxx@10.0.0.1:3306/dbname?charset=utf8mb4")
db = DBEngine(f"sqlite:////Users/bytedance/HttpRunner/examples/data/sqlite.db") db = DBEngine(f"sqlite:////Users/bytedance/HttpRunner/examples/data/sqlite.db")
print(db.fetchmany("select* from student", 5)) print(db.fetchmany("""
select* from student""", 5))
print(db.fetchmany("select* from student", 5)) print(db.fetchmany("select* from student", 5))