python 封装一个mysql类

import pymysql


def get_list(sql, args=None):
    conn = pymysql.connect(host='127.0.0.1', port=3306, user='root',
                           passwd='root', db='classes',
                           charset='utf8')
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    cursor.execute(sql, args)
    reslt = cursor.fetchall()
    cursor.close()
    conn.close()
    return reslt

def get_one(sql, args=None):
    conn = pymysql.connect(host='127.0.0.1', port=3306, user='root',
                           passwd='root', db='classes',
                           charset='utf8')
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    cursor.execute(sql, args)
    reslt = cursor.fetchone()
    cursor.close()
    conn.close()
    return reslt

def modify(sql, args):
    conn = pymysql.connect(host='127.0.0.1', port=3306, user='root',
                           passwd='root', db='classes',
                           charset='utf8')
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    cursor.execute(sql, args)
    conn.commit()
    cursor.close()
    conn.close()

def create(sql, args):
    conn = pymysql.connect(host='127.0.0.1', port=3306, user='root',
                           passwd='root', db='classes',
                           charset='utf8')
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    cursor.execute(sql, args)
    conn.commit()
    last_row_id = cursor.lastrowid
    cursor.close()
    conn.close()
    return last_row_id




class SqlHeper(object): def __init__(self): # 读配置文件,后进行connect复制连接更好 self.connect() def connect(self): self.conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='classes', charset='utf8') self.cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor) def get_list(self,sql,args=None): self.cursor.execute(sql, args) result = self.cursor.fetchall() return result def get_one(self,sql,args=None ): self.cursor.execute(sql, args) result = self.cursor.fetchone() return result def modify(self,sql,args): self.cursor.execute(sql, args) self.conn.commit() #批量插入值, def multiple_modify(self,sql,args): #这里的args可以传入一个列表中包含多个元祖的一个序列 # self.cursor.executemany('insert into a(id,name) values(%s,%s)',[(1,'alex'),(2,'eric')]) self.cursor.executemany(sql,args) self.conn.commit() def create(self,sql, args): self.cursor.execute(sql, args) self.conn.commit() # last_row_id = self.cursor.lastrowid return self.cursor.lastrowid def close(self): self.cursor.close() self.conn.close()

 

posted @ 2021-04-19 23:31  茶叶蛋蛋  阅读(229)  评论(0)    收藏  举报