封裝DB础锐,后續(xù)對數(shù)據(jù)庫進行操作時盖彭,不需要再連一遍
import pymysql
def query_db(sql):
"""
封裝DB
:return:
"""
# 建立數(shù)據(jù)庫連接
db = pymysql.connect(host="xxx", port=xxx, user="xxx",
password="xxx", database="xxx", charset="utf8")
# 創(chuàng)建游標
cursor =db.cursor()
# 使用游標執(zhí)行sql
cursor.execute(sql)
print("行數(shù):", cursor.rowcount)
datas = cursor.fetchall()
print("查詢到的數(shù)據(jù)為:", datas)
cursor.close()
db.close()
return datas
以上需替換連接串賬號密碼等信息~
調用
# 導包
from interface.po.base_api import BaseApi
# 查一條數(shù)據(jù)測試一下
def test_search():
sql = f"SELECT * FROM xxx limit 1"
dat = BaseApi.query_db(sql)
print(dat)