从事python进行数据挖掘、数据处理、数据分析等跟数据打交道的工作,不免会接触到MySQL和SqlServer数据库,下面是python连接MySQL和SqlServer常用的方法进行简单的封装,我从事的大部分工作是获取数据下来进行数据挖掘处理分析,然后对处理完的数据进行写入,所以我只对查询和写入进行封装。
终端直接执行pip命令:
pip install pymysql
pip install pymssql
pip install pandas
下面直接上代码
# -*- coding: utf-8 -*-
import pymssql
import pymysql
# 连接MySQL和SqlServer的基类
class DataBase(object):
def __init__(self, host=None, port=None, user=None, pwd=None, db=None, conn=None):
self.host = host
self.port = port
self.user = user
self.pwd = pwd
self.db = db
self.conn = conn
def get_cursor(self):
raise NotImplementedError
def query(self, sql):
with self.get_cursor() as cur:
cur.execute(sql)
# 获取提交sql语句后获取到字段等信息
# 先获取信息却没执行execute提交sql语句,将获取不到
description_field_info = cur.description
description_field = []
for i in description_field_info:
description_field.append(i[0])
# 字段,结果
return description_field, cur.fetchall()
def query_one(self, sql):
with self.get_cursor()