python中pymysql的连接代码

# pip3 install pymysql 安装
import pymysql

user = input('姓名user>>: ').strip()
pwd = input('密码password>>: ').strip()

# 建立链接
conn = pymysql.connect(
    host='localhost',  # 我的IP地址
    port=3306,  # 不是字符串不需要加引号。
    user='root',
    password='123456',
    db='xing', # 数据库名
    charset='utf8'
)

# 拿到游标
cursor = conn.cursor()

class MyclassSQL:

    res = None

    def startSj(self):
        sql = 'select * from userinfo where user = "%s" and pwd="%s"' % (user, pwd)
        print(sql)
        self.res = cursor.execute(sql)
        print(self.res, '11111')
        # 进行判断
        if self.res:
            print('登录成功')
        else:
            print('登录失败')

    def addXr(self):
        if self.res:
            # 定义要执行的sql语句
            sqlAdd = 'insert into userinfo(user,pwd) values(%s,%s);'
            data = [
                ('july1', '1417'),
                ('june1', '2518'),
                ('marin1', '3619')
            ]
            # 拼接并执行sql语句
            dataIS = cursor.executemany(sqlAdd, data)
            # 涉及写 *** 作要注意提交
            conn.commit()
            if dataIS:
                print('写入成功')
            else:
                print('写入失败')
        else:
            print('写入失败')


if __name__ == '__main__':
    MyclassSQL().startSj()
    MyclassSQL().addXr()
    # 关闭连接
    cursor.close()
    conn.close()