python封装连接数据库_python工作日常：数据库链接 方法封装

拿来即用 下面是代码

import pymysql

import sys

class Config(object):

@staticmethod

def get_config(name):

config = {

'localhost': {

'host': '127.0.0.1',

'user': 'mysql',

'password': 'mysql',

'database': 'mysql',

'port': 3306

}

}

return config[name]

class UseDB(object):

def __init__(self, name):

self.__conn = self.build_conn(name)

self.__cursor = self.__conn.cursor()

@property

def conn(self):

return self.__conn

def build_conn(self, name):

try:

config = Config.get_config(name)

conn = pymysql.connect(host=config["host"], user=config["user"], passwd=config["password"],

db=config["database"], charset='utf8')

return conn

except Exception as e:

print('Something wrong: %s' % format(e))

def getData(self, sql, type="all"):

self.__cursor.execute(sql)

if type != "all":

return self.__cursor.fetchone()

return self.__cursor.fetchall()

def postData(self, sql):

try:

self.__cursor.execute(sql)

self.conn.commit()

except Exception as e:

self.conn.rollback()

def close(self):

self.conn.close()

class ToExecute(object):

def __init__(self, db1, db2):

self.db1 = UseDB(db1)

self.db2 = UseDB(db2)

def xxx(self):

pass

to = ToExecute(db1='',db2='')