本节课将学习如何在平台中安插一个数据库调用类:
在之前的进度最后,我们新建了这个数据库操作类,并设置了初始化init函数:
这个函数中,获取数据库id作为唯一参数,然后去数据库中拿到数据库真身。再把其四个字段设置成类变量:host,port,username,password。
第二个函数是连接函数:connect
注意,上述代码中,使用的第三方库是pymysql ,在成功连接后新建了对象引用:self.cur
第三个函数是:关闭数据库函数close
注意,这里用了俩个try,一个是关闭对象引用,一个是关闭连接。之所以都用try,就是因为在实际的使用中,可能出现各种各样的错误导致连接失败,或者对象引用创建失败。而你又恰好执行关闭数据库连接的时候,就可能因为压根没创建而去关闭导致报错,所以全用try :有则关,无则过。
本节课到此结束,内容比较简单,主要就是连接和关闭。
可复制版本代码为:
from Myapp.models import * import pymysql class SQLDB(object): def __init__(self,db_id): db_data = DB_db.objects.filter(id=int(db_id))[0] self.host = db_data.host self.port = db_data.port self.username = db_data.username self.password = db_data.password def connect(self, sql_db): try: self.conn = pymysql.connect(host=self.host,port=self.post,user=self.username,password=self.password) except: return False self.cur = self.conn.cursor() return True def close(self): try:self.cur.close() except:pass try:self.conn.close() except:pass