python连接sqlserver工具类

上代码：

# -*- coding:utf-8 -*-
 
import pymssql
import pandas as pd

class mssql(object):
    def __init__(self,host,user,pwd,db):
        self.host = host
        self.user = user
        self.pwd = pwd
        self.db = db
 
    def __getconnect(self):
        if not self.db:
            raise(nameerror,"没有设置数据库信息")
        self.conn = pymssql.connect(host=self.host,user=self.user,password=self.pwd,database=self.db,charset="utf8")
        cursor = self.conn.cursor()
        if not cursor:
            raise(nameerror,"连接数据库失败")
        else:
            return cursor
 
    def execquery(self,sql):
        cursor = self.__getconnect()
        cursor.execute(sql)
        # 调出数据
        reslist = cursor.fetchall()
 
        #查询完毕后必须关闭连接
        self.conn.close()
        return reslist

    def execquerytodataframe(self,sql):
        cursor = self.__getconnect()
        cursor.execute(sql)
        # 调出数据
        reslist = cursor.fetchall()
        # cols为字段信息 例如((''))
        cols = cursor.description 
        #查询完毕后必须关闭连接
        self.conn.close()

        # 将数据转换为dataframe
        col = []
        for i in cols:
            col.append(i[0])
        data = list(map(list, reslist))
        data = pd.dataframe(data,columns=col) 

        return data
 
    def execnonquery(self,sql):
        cursor = self.__getconnect()
        cursor.execute(sql)
        self.conn.commit()
        self.conn.close()

 

如果对您有帮助，请赞助根棒棒糖~