65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import csv
|
||
|
||
|
||
def register(username, password):
|
||
# 读取现有的用户信息
|
||
existing_users = read_users()
|
||
|
||
# 检查用户名是否已存在
|
||
if username in existing_users:
|
||
return False, "Username already exists."
|
||
|
||
# 添加新用户信息
|
||
new_user = {"username": username, "password": password}
|
||
existing_users[username] = new_user
|
||
|
||
# 将用户信息保存到CSV文件
|
||
save_users(existing_users)
|
||
|
||
return True, "Registration successful."
|
||
|
||
|
||
def login(username, password):
|
||
# 读取用户信息
|
||
existing_users = read_users()
|
||
|
||
# 验证用户名和密码
|
||
user = existing_users.get(username)
|
||
if user and user["password"] == password:
|
||
return True, "Login successful."
|
||
return False, "Invalid credentials."
|
||
|
||
|
||
def read_users():
|
||
# 从CSV文件中读取用户信息
|
||
users = {}
|
||
try:
|
||
with open("data/users.csv", "r", newline="") as csvfile:
|
||
reader = csv.DictReader(csvfile)
|
||
for row in reader:
|
||
users[row["username"]] = {"username": row["username"], "password": row["password"]}
|
||
except FileNotFoundError:
|
||
pass # 如果文件不存在，说明还没有注册用户，返回一个空字典
|
||
return users
|
||
|
||
|
||
def save_users(users):
|
||
# 将用户信息保存到CSV文件
|
||
with open("data/users.csv", "w", newline="") as csvfile:
|
||
fieldnames = ["username", "password"]
|
||
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
|
||
writer.writeheader()
|
||
for user in users.values():
|
||
writer.writerow(user)
|
||
|
||
# 测试注册功能
|
||
# register("user1", "password1")
|
||
# register("user2", "password2")
|
||
|
||
# 测试登录功能
|
||
# login_success, message = login("user1", "password1")
|
||
# print(message) # 输出：Login successful.
|
||
#
|
||
# login_success, message = login("user3", "password3")
|
||
# print(message) # 输出：Invalid credentials.
|