python作业6

import json
import sys
import os
import hashlib

def login (login_ls):
    "登录过程"
    name=input("请输入用户名：")
    if name in login_ls:
        password = int(input('请你输入密码：'))
        if password==int(login_ls[name]):
            print('登录成功')
            return True
        else:
            print('密码错误')
            return False
    else:
        print('该用户不存在')
        return False

def get_md5(password):
    '密码加密'
    password=str(password)
    password1=hashlib.md5()
    password1.update(password.encode("utf-8"))
    password = password1.hexdigest()
    return password

def enroll():
    "注册用户"
    enroll_ls=dict()
    name=input('请输入注册的用户名字：')
    password=int(input('请输入密码：'))
    if input('是否注册') == 'yes' or input('是否注册') == 'Yes':
        #password=hashlib.md5().updata(password)
        change=input('是否注册')
        if change=='yes'or change=='Yes':
            password=get_md5(password)
        enroll_ls[name]=password
        print('注册成功')
        return enroll_ls
    else:
        print('注册失败')
        return enroll_ls

def write_article(article_new):
    '继续写文章'
    while True:
        with open(article_new,'w') as fle:
            w=fle.write(input('请输入：')+'\n')
            fle.flush()
        if w==0:
            break

def read_article(article_new):
    '读取文章'
    with open(article_new,'r') as fle:
        w=fle.readlines()
        for word in w:
            print(word)

def get_ls():
    path=' '    #路径为
    article=os.listdir(path)  #获取文章目录
    print('文章目录')

    for i in range(len(article)):
        i+=1
        print(f'{i}.{article[i]}')

if __name__=="__main__":
    print(''*10+'登录页面')
    print('~*'*10)
    print(''*12+'1.用户登录\n'+''*12+'2.新用户注册\n'+''*12)
    print('~*'*10)
    option1=int(input("请输入你的选项："))
    if option1==1:
        ls=open('','r') #读取文件位置
        login_ls=json.load(ls)
        print(login_ls)
        ls.close()

        if login(login_ls):
            get_ls()

    elif option1==2:
        '''
        ls=open(" ","a")
        enroll_ls=json.load(ls)
        print(enroll_ls)
        '''
        enroll_dict=enroll()
        #ls=open("",'w')
        #enroll_dict=enroll(enroll_ls)
        #ls.write(enroll_dict)
        json.dump(enroll_dict,open(""),'w')
        #ls.close()

    elif option1==3:
        sys.exit()
 '''
elif option1==4: 
'''