(简易)测试数据构造平台:39 - 复杂工具(数据库组件)

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