Python 自定义数据库连接类

创建一个配置文件：E:/mypy/mycnf.txt

[mssql]
host = HZC
user = kk
pwd = kk
dbname = DemoDB

数据库连接类：E:/mypy/MSSQL.py

# -*- coding: utf-8 -*-  
# python 3.5

import sys
import pymssql
import ConfigParser
sys.path.append(r'E:/mypy')

#从文件系统读取配置文件
cf = ConfigParser.ConfigParser()
cf.read("mycnf.txt")
host = cf.get("mssql", "host")
user = cf.get("mssql", "user")
pwd = cf.get("mssql", "pwd")
db = cf.get("mssql", "dbname")

class MSSQL:
    #def __init__(self,host=None,user=None,pwd=None,db=None):
    def __init__(self):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.db = db
        
        self._conn = self.GetConnect()
        if(self._conn):
            self._cur = self._conn.cursor()


    #连接数据库
    def GetConnect(self):
        conn = False
        try:
            conn = pymssql.connect(
                host=self.host,
                user=self.user,
                password=self.pwd,
                database =self.db
            )
        except Exception as err:
            print("连接数据库失败, %s" % err)
        else:
            return conn


    #执行查询
    def ExecQuery(self,sql):
        res = ""
        try:
            self._cur.execute(sql)
            res = self._cur.fetchall()
        except Exception as err:
            print("查询失败, %s" % err)
        else:
            return res


    #执行非查询类语句
    def ExecNonQuery(self, sql):
        flag = False
        try:
            self._cur.execute(sql)
            self._conn.commit()
            flag = True
        except Exception as err:
            flag = False
            self._conn.rollback()
            print("执行失败, %s" % err)
        else:
            return flag


    #获取连接信息
    def GetConnectInfo(self):
        print( "连接信息：" )
        print( "服务器:%s , 用户名:%s , 数据库:%s " % (self.host,self.user,self.db))



    #关闭数据库连接
    def Close(self):
        if(self._conn):
            try:
                if(type(self._cur)=='object'):
                    self._cur.close()
                if(type(self._conn)=='object'):
                    self._conn.close()
            except:
                raise("关闭异常, %s,%s" % (type(self._cur), type(self._conn)))

执行SQL：

# -*- coding: utf-8 -*-  
# python 3.5

import sys
sys.path.append(r'E:/mypy') 
from MSSQL import MSSQL;

sql = "select oid,name,craetedate from tab;"
ms = MSSQL()
resList = ms.ExecQuery(sql)
for row in resList:
    print("oid: %d\tname: %s\tcraetedate: %s" % (row[0],row[1],row[2]))

#ms.ExecNonQuery(sql)
#ms.GetConnectInfo()
#ms.Close()

​