python通过数据库实现简单的注册+登录

import pymysql
import time

db = pymysql.connect("localhost","root","subaobao666","newdatabase" ) #直接连入newdatabase库
cur = db.cursor()   #获取游标

pd =int(input('你好,登陆系统请回复:1 ,注册系统请回复:2\n'))

#登陆
if pd == 1:
    print('**************开始登陆')
    dlzh = str(input("请输入账号："))
    dlmm = str(input("请输入密码："))

    #数据库表中查询是否含有该账号和密码
    sql = """select * from user_1 where  (username = %s) and (userpassword = %s)"""
    aa = cur.execute(sql,[dlzh,dlmm])
    db.commit()

    if aa == 1:
        print("登陆成功")
        #登陆成功，更新lasttime字段
        sql = """update user_1 set lasttime = now() where username = %s"""
        cur.execute(sql,[dlzh])
        db.commit()
        #登陆成功，加入到登陆数据表
        sql = """insert into denglujilu(dl_time,dl_user) values (now(),%s)"""
        cur.execute(sql,[dlzh])
        db.commit()
        #登陆成功，显示用户信息

    else:
        print("账号或密码错误")
    db.close()

#注册
elif pd == 2:
    print('**************开始注册')
    user1 = str(input('请输入账号：'))
    password1 = str(input('请输入密码：'))
    key1 = str(input('请输入注册码：'))

    #判断注册码是否存在
    sql = """select * from key_1 where key_num = %s"""
    aa = cur.execute(sql, [key1])
    db.commit()

    if aa == 1:
        #判断注册码状态是否已使用
        sql = """select * from key_1 where (key_num = %s) and (syzt = 1)"""
        aa = cur.execute(sql, [key1])
        db.commit()
        if aa == 1:
            # 判断用户名是否已被注册
            sql = """select * from user_1 where username = %s"""
            aa = cur.execute(sql, [user1])
            db.commit()
            if aa == 1:
                print("用户名已存在，请重新注册")
            else:
                # 开始执行注册，加入用户数据到表中，且将该注册码标记为已使用状态：0
                sql = """insert into user_1(username,userpassword,zhucekey,firsttime) values (%s,%s,%s,now())"""
                cur.execute(sql, [user1, password1, key1])
                db.commit()

                sql = """update key_1 set syzt = 0 where key_num = %s"""
                cur.execute(sql, [key1])
                db.commit()
                print("注册成功")

        else:
            print("注册码已使用\n注册失败")

    else:
        print("注册码不存在");
    db.close()
#其他
else:
    print("输入的命令不正确")
    db.close()