#!/bin/bash/env python
# -*- coding:utf-8 -*-
def login(username,password):
"""
用于用户名密码的登录
:param username: 用户名
:param password: 密码
:return: True,用户验证成功;False,验证失败
"""
with open('cai.log','r',encoding="utf-8") as f:
for line in f: #一行一行的读取
line = line.strip() #去除两端的空格及换行符
line_list = line.split("$") #以$符号进行分割
if username == line_list[0] and password == line_list[1]:
return True
return False
def register(username,password):
"""
注册用户
:param username: 用户名
:param password: 密码
:return: True,注册成功
"""
with open('cai.log',"a",encoding="utf-8") as f:
temp = "\n" + username + "$" +password
f.write(temp)
return True
def user_exist(username):
"""
检查用户是否已经存在
:param username: 用户名
:return: True,则用户名已经存在;False,用户名不存在
"""
with open('cai.log',"r",encoding="utf-8") as f:
for line in f:
line = line.strip()
line_list = line.split("$")
if username == line_list[0]:
return True
return False
def main():
print("欢迎登陆瑞的系统")
inp = input("1: 登陆;2: 注册")
user = input("请输入用户名:")
pwd = input("请输入密码:")
if inp == "1":
is_login = login(user,pwd)
if is_login:
print("登录成功")
else:
print("登录失败")
elif inp == "2":
is_exist = user_exist(user)
if is_exist:
print("用户已经存在,无法注册")
else:
result = register(user,pwd)
if result:
print("注册成功")
else:
print("注册失败")
main()