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

Merge pull request #1428 from billduan/optimize_strip_sql

Browse Source 
优化：strip sql
This commit is contained in:
debugtalk
2022-08-12 15:07:12 +08:00
committed by GitHub
parent 88c25d26c4 f45eae62d0
commit 4132c324a4
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
def _fetch(self, query, size=-1, commit=True):
query = query.strip()
result = self.session.execute(query)
if query.upper()[:6] == "SELECT":
if size < 0:
@@ -80,5 +81,6 @@ class DBEngine(object):
if __name__ == "__main__":
# 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")
print(db.fetchmany("select* from student", 5))
print(db.fetchmany("""
select* from student""", 5))
print(db.fetchmany("select* from student", 5))