GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +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
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))