From f45eae62d055d2aa63bd8ebd73ee4e1f7625255d Mon Sep 17 00:00:00 2001 From: "duanchao.bill" Date: Tue, 26 Jul 2022 12:49:08 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=EF=BC=9Astrip=20sql?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- httprunner/database/engine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/httprunner/database/engine.py b/httprunner/database/engine.py index 9bddad1d..aa0c0cec 100644 --- a/httprunner/database/engine.py +++ b/httprunner/database/engine.py @@ -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))