第一步: pip install pymysql
第二步: 如下是連接數據庫代碼:
import pymysql
class ConnectDatabase:
# 封裝一個數據庫類
# 用於數據庫查詢/刪除/更新/插入
def __init__(self,
host="xxx.xx.xx.xx", # 數據庫地址
port=3306, # 端口
user='xxxx',
password='xxxx', # 登錄數據庫密碼
database='xxxx', # 數據庫名
charset='utf8'): # 編碼
# 第一步:連接數據庫
self.connect = pymysql.connect(host=host, user=user, password=password,
database=database, port=int(port), charset=charset)
# 第二步:創建遊標
self.cur = self.connect.cursor(cursor=pymysql.cursors.DictCursor)
# 執行sql語句,如果是查詢語句,則返回第一條
def run_sql(self, sql_param):
if sql_param.strip().upper().startswith('SELECT'): # 判斷是否是select查詢語句
self.cur.execute(sql_param)
# fetchone只返回一條數據 ;fetchmany(5) 只返回5條數據 ;fetchall() 返回所有數據
alldata = self.cur.fetchone()
# 此處並非是提交事務,主要是用於數據庫數據刷新,源於接口請求之後,數據庫可能會數據更新,如果此時不更新數據表,
# 那麼查詢到的數據依舊是老數據,非更新數據
self.connect.commit()
return alldata
else:
# 如果是insert,update,delete語句,那麼需要進行提交事務,否則操作失敗
self.cur.execute(sql_param)
self.connect.commit() #
return None
def colse(self):
self.cur.close()
self.connect.close()