python链接数据库

```python
import pymysql

class MysqlObject:
    def __init__(self,host,user,password,port,database):
        self.host = host
        self.user = user
        self.password = password
        self.port = port
        self.database = database
        self.db = self.connect_db()
        self.cursor = self.db.cursor()

    def connect_db(self):
        """
        链接数据库
        """
        try:
            db = pymysql.connect(
                                host=self.host,
                                user=self.user,
                                password=self.password,
                                port=self.port,
                                database=self.database)
        except Exception as e:
            print(e)
        return db

    def create(self,sql):
        """
        创建表/增加数据/改动数据/删除数据
        """
        try:
            self.cursor.execute(sql)
            self.db.commit() # 事件提交
        except Exception as e:
            print(e)
            self.db.rollback() # 事件回滚 (不保存操作前的数据)


    def query(self,sql):
        """
        查询
        """
        try:
            self.cursor.execute(sql)
            data = self.cursor.fetchall()
            self.db.commit()  # 事件提交
        except Exception as e:
            print(e)
            # self.db.rollback()  # 事件回滚
        return data

    def close(self,sql):
        """
        关闭链接
        """
        self.cursor.close()
        self.db.close()

if __name__ == "__main__":
    sql = "select num,scores from student;"
    mysql = MysqlObject(host="localhost",
                        user="root",
                        password=None,
                        port=3306,
                        database="ai_data")
    result = mysql.query(sql)
    print(result)