封裝一個資料庫模塊有三個功能:查詢,插入,關閉

封裝一個資料庫模塊有三個功能:查詢,插入,關閉

1.查看

2.提交

3.關閉

import pymysql
cur = None
conn = None

def getall(sql): #用來執行查詢
    # 連接資料庫
    conn = pymysql.connect(host='localhost', user='root', password='123', db='day300', charset='utf8')
    cur = conn.cursor()          #獲取cursor對象
    # 通過cursor的對象去執行SQL語句
    cur.execute(sql)
    return cur.fetchall()

def exceDML(sql): #用來執行插入
    conn = pymysql.connect(host='localhost', user='root', password='123', db='day300', charset='utf8')
    cur = conn.cursor()
# 通過cursor的對象去執行SQL語句
    cur.execute(sql)
    # 提交事物
    conn.commit()

def close():  #用來關閉連接
    if cur:
        cur.close()
    if conn:
        conn.close()


# 使用工具模塊：
#
# from day3 import mysqlHelper
#
# name = input("請輸入名字：")
# id = input("請輸入ID：")
# sql1 = 'insert into t_user values(%d,"%s")'%(int(id),name)
# sql2 = 'select * from t_user'
# mysqlHelper.exceDML(sql1)
# print(mysqlHelper.getall(sql2))
# mysqlHelper.close()