简单用户登录系统

view1='''
    ===================
        欢迎访问捡垃圾
        1.登录
        2.注册
        3.退出
    ===================
'''
islogin=False
users=[]
#加密
def code(pwd):
    li_pwd=list(pwd)
    li_pwd2=li_pwd[::-1]
    code_pwd=[]
    for i in range(len(pwd)):
        code_pwd.append(li_pwd[i])
        code_pwd.append(li_pwd2[i])
    return code_pwd

#登录
def login(id,pwd):
    global islogin
    print('请登录')
    user_login=get_user_by_id(id)
    if user_login:
        if user_login['user_pwd']==code(pwd):
            print('登录成功')
            islogin=True
        else:
            print('用户名或密码错误')
    else:
        print('该用户不存在')

#注册
def rigister(id,pwd,con_pwd):
    if pwd==con_pwd:
        user_rigister=get_user_by_id(id)
        if user_rigister:
            print('该用户已被注册')
        else:
            print('注册成功')
            user={}
            user['id']='00'+str(len(users)+1)
            user['user_id']=id
            user['user_pwd']=code(pwd)
            users.append(user)
            print(users)
    else:
        print('两次密码不一致')

#根据id取出用户信息
def get_user_by_id(id):
    for user in users:
        if user['user_id']==id:
            return user

#获取输入信息：用户名和密码
def get_dates():
    id_input=input('请输入用户名')
    pwd_input=input('请输入密码')
    return  id_input,pwd_input

#获取用户名和密码，注册用
def get_dates3():
    id_input=input('请输入用户名')
    pwd_input=input('请输入密码')
    con_input=input('请确认密码')
    return  id_input,pwd_input,con_input

#主函数
def main():
    print(view1)
    option=int(input('请选择'))
    if option==1:
        user_id,user_pwd=get_dates()
        login(user_id,user_pwd)

    if option==2:
        user_id,user_pwd,user_con=get_dates3()
        rigister(user_id,user_pwd,user_con)

    if option==3:
        print('再见')
main()