python利用数据库实现注册登录系统

用数据库实现注册登录系统

import null
from pymysql import connect


class JD(object):
    def __init__(self):
        # 打开数据库连接
        self.conn = connect(host='localhost', port=3306, user='xxxx', password='xxxx', database='xxxxx',
                            charset='utf8')

        # 使用 cursor() 方法创建一个游标对象 cursor
        self.cursor = self.conn.cursor()

    def __del__(self):
        """关闭cursor对象"""
        self.cursor.close()
        self.conn.close()

    def add_brands(self):
        item_name = input("请输入新用户名（用户名由字母，数字，下划线组成）：")
        item_key=input("请输入登录密码")
        # sql1 = ("{},{}".format(item_name, item_key))

        sql = 'insert into register_user values(null,"%s","%s");' % (item_name, item_key)
        self.cursor.execute(sql)
        self.conn.commit()  # 确定插入

    def get_info_by_name(self):
        find_name = input("请输入用户名：")
        key=input("请输入登录密码")
        sql="select * from register_user where user=%s"
        self.cursor.execute(sql,[find_name])
        value = self.cursor.fetchall()

        try:
            user=value[0][1]
            keyword=value[0][2]

            if user==find_name and key==keyword:
                print("登录成功")
            else:
                choice=input("登录失败，用户名或密码错误，继续输入1，退出输入666")
                if choice=="1":
                    self.get_info_by_name()
                elif choice=="666":
                    self.bye()
                else:
                    print("输入有误，请重新输入")
                    self.get_info_by_name()
        except:
            choice = input("登录失败，用户名或密码错误，继续输入1，退出输入666")
            if choice == "1":
                self.get_info_by_name()
            elif choice == "666":
                self.bye()


            else:
                print("输入有误，请重新输入")
                self.get_info_by_name()


        print("\n")

    def change(self):
        try:
            find_name = input("请输入要修改的用户名：")
            key = input("请输入要修改的密码")
            sql="update register_user set kyes=%s where user=%s;"%(key,find_name) #修改id = 6的多个
            self.cursor.execute(sql)
            self.conn.commit()
        except:
            # 如果发生错误则回滚
            self.conn.rollback()




    def bye(self):
        print("谢谢使用，再见!!!")



    def print_menu(self):
        print("-------注册主页-------")
        print("4:添加（注册）用户")
        print("5:根据名字查找")
        print("0:结束注册")
        return input("请输入对应功能的序号：")

    def run(self):
        while True:
            num = self.print_menu()
            if num == "4":
                # 添加一个商品分类
                self.add_brands()

            elif num == "5":
                # 根据名字查商品
                self.get_info_by_name()

            elif num == "0":
                return False
            else:
                print("您的输入有误，请重新输入...")


def start():
    # 创建一个京东商城对象
    jd = JD()
    # 调用这个对象的run方法，让其运行
    jd.run()


if __name__ == '__main__':
    start()