上代码:
# -*- 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()
如果对您有帮助,请赞助根棒棒糖~
如您对本文有疑问或者有任何想说的,请点击进行留言回复,万千网友为您解惑!