import pymysql

#连接数据库
class Mysql(object):
    #连接数据库
    def __init__(self,port_test,database_test):  #传参输入数据库的接口和需要操作的库
        self.db = pymysql.connect(host='127.0.0.1',              #数据库地址
                         port=port_test,                         #链接接口
                         user='admin',                           #连接数据库的账户
                         password='password',        #连接数据库的密码
                         database=database_test)                 #需要操作的库
        self.cursor = self.db.cursor()

    #输入sql进行查询操作
    def select_mysql(self,sql):
        try:
            self.cursor.execute(sql)
            results = self.cursor.fetchall()
            print(results)
        except:
            print ("Error: unable to fetch data")
        self.db.close()

    #输入sql进行执行操作
    def  execute_mysql(self,sql):
        try:
            self.cursor.execute(sql)
            self.db.commit()
            print("执行成功")
        except:
            self.db.rollback()
            print ("Error: unable to fetch data")
        self.db.close()