python之MySQL學習——簡單的增刪改查封裝

1.增刪改查封裝類MysqlHelper.py

import pymysql as ps

class MysqlHelper:
    def __init__(self, host, user, password, database, charset):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.charset = charset
        self.db = None
        self.curs = None
    # 數據庫連接
    def open(self):
        self.db = ps.connect(host=self.host, user=self.user, password=self.password,database=self.database, charset=self.charset)
        self.curs = self.db.cursor()
    # 數據庫關閉
    def close(self):
        self.curs.close()
        self.db.close()
    # 數據增刪改
    def cud(self, sql, params):
        self.open()
        try:
            self.curs.execute(sql, params)
            self.db.commit()
            print("ok")
        except :
            print('cud出現錯誤')
            self.db.rollback()
        self.close()
    # 數據查詢
    def find(self, sql, params):
        self.open()
        try:
            result = self.curs.execute(sql, params)
            self.close()
            print("ok")
            return result
        except:
            print('find出現錯誤')

2.數據查詢（引入封裝類）

from MysqlHelper import MysqlHelper

mh = MysqlHelper('localhost', 'root', '123456', 'test', 'utf8')
sql = "select * from user where name=%s"
print(mh.find(sql, '小明'))

3.數據修改（引入封裝類）

from MysqlHelper import MysqlHelper

mh = MysqlHelper('localhost', 'root', '123456', 'test', 'utf8')
sql = "insert into user(name,password) values(%s,%s)"
mh.cud(sql, ('小光', '123456'))