Create auth.py

auth.py

@@ -0,0 +1,34 @@
+from hashlib import sha256
+
+# 密码加密函数
+def hash_password(password):
+    return sha256(password.encode()).hexdigest()
+
+# 模拟数据库操作
+class Database:
+    def __init__(self):
+        self.users = {}
+
+    def save_user(self, username, hashed_password):
+        self.users[username] = {"password": hashed_password, "is_active": True}
+
+    def find_user(self, username):
+        return self.users.get(username)
+
+db = Database()
+
+# 注册函数
+def register(username, password):
+    hashed_password = hash_password(password)
+    db.save_user(username, hashed_password)
+    return "注册成功"
+
+# 登录函数
+def login(username, password):
+    user = db.find_user(username)
+    if not user:
+        return "用户名或密码错误"
+    if user["password"] == hash_password(password):
+        return "登录成功"
+    else:
+        return "用户名或密码错误"