exercise.py

"""
pymysql  数据库操作流程



"""
import pymysql


class Databases:
    def __init__(self):
        # 连接数据库
        self.db = pymysql.connect(host='localhost',
                                  port=3306,
                                  user='root',
                                  password='123456',
                                  database='stu',
                                  charset='utf8')

        # 创建游标对象(操作数据库语句，获取查询结果)
        self.cur = self.db.cursor()

    def close(self):
        # 关闭游标
        self.cur.close()
        # 断开数据库连接
        self.db.close()

    def register(self, name, passwd):
        sql = "select * from user where name='%s'" % name
        self.cur.execute(sql)
        resule = self.cur.fetchone()
        if resule:
            return False
        try:
            sql = "insert into user (name,passwd) values (%s,%s)"
            self.cur.execute(sql, [name, passwd])
            self.db.commit()
            return True
        except:
            self.db.rollback()
            return False

    def login(self, name, passwd):
        sql = "select * from user where name='%s' and passwd='%s'" % (name, passwd)
        self.cur.execute(sql)
        result = self.cur.fetchone()
        if result:
            return True
        else:
            return False


if __name__ == '__main__':
    db = Databases()
    while True:
        print("""
        ============
        1.注册  2.登录
        ============
        """)
        cmd = input("命令：")
        if cmd == '1':
            if db.register('张三', '123'):
                print("注册成功")
                break
            else:
                print("注册失败")

        elif cmd == '2':
            if db.login('张三', '123'):
                print("登录成功")
                break
            else:
                print("登录失败")

        else:
            print("请输入正确指令!")