python中的class库_Python--操作数据库class

import pymysql

class OpMySql1: # 经典类

pass

class OpMySql(object): # 新式类

def __init__(self, host, user, password, db, port=3306, charset='utf8'):

schema = {

'user': user,

'host': host,

'password': password,

'db': db,

'port': port,

'charset': charset

}

try:

self.coon = pymysql.connect(**schema)

except Exception as e:

print('数据库连接异常！%s' % e)

quit()

else: # 没有异常的情况下，建立游标

self.cur = self.coon.cursor(cursor=pymysql.cursors.DictCursor)

def execute(self, sql):

try:

self.cur.execute(sql)

except Exception as e:

print('sql语句有错误！%s' % e)

return e

if sql[:6].upper() == 'SELECT':

return self.cur.fetchall()

else: # 其他sql语句的话

self.coon.commit()

return 'ok'

def __del__(self):

self.cur.close()

self.coon.close()

ybq = OpMySql('211.149.218.16', 'jxz', '123456', db='jxz') # 实例化

print(ybq.execute('select * from stu;'))