用装饰器去设置mysql的连接.
import pymysql
from pymysql.err import OperationalError
def cha(db):#这个db就是连接mysql的那个db
def decorator(func): #func是你自己的函数,本文就是sqlte
def wrapper(client,field): #你自己设置的参数,*args也可以用,但是我用args[0]或者args[1]的时候他会提示报错,暂时解决不了.
try:
client = pymysql.connect(host="localhost",user="root",password="admin",port=3306,db=db)
except OperationalError:
print('连接错误!')
raise OperationalError
return func(client,field) #将参数返回到你自己的函数中
return wrapper #返回wrapeer
return decorator #返回装饰器
db = 'spider'
@cha(db)
def sqlte(client,field):
db = client.cursor()
try:
db.execute('SELECT * FROM '+field)
results = db.fetchall()
for result in results:
print(result)
except :
db.rollback()
client = 'db'
field = 't'
sqlte(client,field)
我弄了一个下午的装饰器....
最新推荐文章于 2022-05-26 18:43:27 发布