前言&思维导图

用户登录模块是最基本的模块之一，主要设计的有当前用户存在判断，用户注册，用户登录名和密码的保存，用户信息输出等等；

1.判断首次启动

通过判断flag.txt是否存在，判断是不是首次启动；

这是一种简单而常见的手法，常在程序初始化的时候会有这样的需求。

判断一个特定文件存在与否，或者判断配置文件的值是否改变；

就能够判断出来当前是不是首次启动，是的话，初始化程序，不是则不变等等

(这是后续了，根据需求来，记住这个小方法即可！)

这里是判断flag.txt是否存在

2.用户注册

用户信息单独文件保存

# 用户注册 用户信息单独文件保存
def user_register():user_id = input('输入账户id：')user_pwd = input('输入用户密码：')user_name = input('输入用户昵称：')user = {'u_id': user_id, 'u_pwd': user_pwd, 'u_name': user_name}user_path = "./users/" + user_id # 新建文件夹保存信息file_user = open(user_path, 'w')  file_user.write(str(user))  file_user.close()  

在文件中保存的信息如下：

3、管理员信息&登录

管理员身份是写死的 名称 root 密码 123456

身份信息同样是新建一个文件，保存

# 初始化管理员
def init():if os.path.exists('users') == False:file = open('u_root.txt', 'w') root = {'rnum': 'root', 'rpwd': "123456"}file.write(str(root)) file.close()  os.mkdir('users')  

管理员登录；

# 管理员登陆
def root_login():while True:print('****管理员登陆****')root_number = input('请输入管理员账户名：')root_password = input('请输入管理员密码：')file_root = open('u_root.txt', 'r')  # 只读打开文件root = eval(file_root.read())  # 读取账户信息if root_number == root['rnum'] and root_password == root['rpwd']:print('管理员登陆成功！')breakelse:print('很遗憾！验证失败')

4.用户登录

选择是注册还是登录；

注册和 上述2 一样

登录和管理员登录类似，匹配信息就好了

# 普通用户登录
def user_login():while True:print('****普通用户登录****')user_id = input('请输入账户名')user_pwd = input('请输入密码：')user_list = os.listdir('./users')  # 遍历元组，判断user_id是否在元组中flag = 0for user in user_list:if user == user_id:flag = 1print('登录中····')# 打开文件file_name = './users/' + user_idfile_user = open(file_name)# 获取文件内容user_info = eval(file_user.read())if user_pwd == user_info['u_pwd']:print('登录成功！')# 此处根据需求，可以跳转到登陆成功之后的操作breakif flag == 1:breakelif flag == 0:print('您并没有注册！请您先注册用户')break

users文件夹下，就会新增 qqqqqq 用户的信息了！

5.完整代码

# 文件存储管理信息
import os# 判断是否首次使用系统
def is_first_use():if os.path.exists('flag.txt') == False:print('首次启动')flag = open('flag.txt', 'w+')flag.write('1')flag.close()  # 关闭文件init()  # 初始化资源print_login_menu()  # 打印登录菜单user_select()  # 选择用户else:flag = open('flag.txt', 'r')word = flag.read()if len(word) == 1:init()  # 初始化资源print_login_menu()  # 打印登录菜单user_select()  # 选择用户# 初始化管理员
def init():if os.path.exists('users') == False:file = open('u_root.txt', 'w')  # 创建并打开管理员账户文件root = {'rnum': 'root', 'rpwd': "123456"}file.write(str(root))  # 写入管理员信息file.close()  # 关闭管理员账户文件os.mkdir('users')  # 创建普通用户文件夹# 打印登录菜单
def print_login_menu():print('----用户登录----')print('1-管理员登陆')print('2-普通用户登陆')print('--------------')print('请选择您的登陆身份')# 用户选择
def user_select():while True:user_type_select = input('请选择用户类型')if user_type_select == '1':  # 管理员登陆验证root_login()breakelif user_type_select == '2':  # 普通用户while True:select = input('是否需要注册？（y/n）：')if select == 'y' or select == 'Y':print('----用户注册----')user_register()  # 用户注册breakelif select == 'n' or select == 'N':print('----用户登录----')breakelse:print('输入有误，请重新选择')user_login()  # 用户登录breakelse:print('输入有误，请重新选择')# 管理员登陆
def root_login():while True:print('****管理员登陆****')root_number = input('请输入管理员账户名：')root_password = input('请输入管理员密码：')file_root = open('u_root.txt', 'r')  # 只读打开文件root = eval(file_root.read())  # 读取账户信息# 信息匹配if root_number == root['rnum'] and root_password == root['rpwd']:print('管理员登陆成功！')breakelse:print('很遗憾！验证失败')# 用户注册 用户信息单独文件保存
def user_register():user_id = input('输入账户id：')user_pwd = input('输入用户密码：')user_name = input('输入用户昵称：')user = {'u_id': user_id, 'u_pwd': user_pwd, 'u_name': user_name}user_path = "./users/" + user_id # 新建文件夹保存信息file_user = open(user_path, 'w')file_user.write(str(user))file_user.close()# 普通用户登录
def user_login():while True:print('****普通用户登录****')user_id = input('请输入账户名')user_pwd = input('请输入密码：')user_list = os.listdir('./users')  # 遍历元组，判断user_id是否在元组中flag = 0for user in user_list:if user == user_id:flag = 1print('登录中····')# 打开文件file_name = './users/' + user_idfile_user = open(file_name)# 获取文件内容user_info = eval(file_user.read())if user_pwd == user_info['u_pwd']:print('登录成功！')# 此处根据需求，可以跳转到登陆成功之后的操作breakif flag == 1:breakelif flag == 0:print('您并没有注册！请您先注册用户')breakif __name__ == '__main__':is_first_use()