软件测试精进之路

python连接MySQLdb的方法简单封装

2018-08-13  本文已影响0人  胆小的米老鼠

python连接MySQLdb方法简单封装:

from MySQLdb import *
class Helper():
    '''封装连接数据库,插入数据,查询数据的方法'''
    def __init__(self,host,port,db,user,userpwd,charset="utf-8"):
        self.host = host
        self.user = user
        self.userpwd= userpwd
        self.db = db
        self.charset = charset
        self.port = port

    def open(self):
        '''打开数据库'''
        self.coon =connect(host= self.host,port = self.port,db=self.db,user= self.user,userpwd = self.userpwd,cahrset = self.charset)
        #创建游标对象
        self.cursor = self.coon.cursor()
    def close(self):
        '''关闭数据库'''
        self.cursor.close()
        self.coon.close()

    def cub(self,sql,params):
        '''插入数据'''
        try:
            self.open()
            #创建sql语句
            sql = "insert into student(name,age) value (%s,%s)"
            #执行sql语句
            self.cursor.execute(sql,params)
            #提交事务
            self.coon.commit()
            self.close()
        except Exception as e:
            print(e)

    def all(self,pararm =""):
        '''查询数据'''
        try:
            self.open()
            #创建sql语句
            sql = "select * from student where id = 3"
            #执行sql语句
            self.cursor.execute(sql,pararm)
            #返回结果赋值给一个变量
            result = self.cursor.fetchall()

            self.close()
            #返回结果变量
            return  result

        except Exception as e:
            print(e)

日常工作中,函数封装后,增加代码重用性,让我们把重心更多的放到业务流程上,而尽量减少与系统的交互,增加工作效率,避免重复的代码占用自己的大部分时间。

上一篇下一篇

猜你喜欢

热点阅读