"""
寫一個登錄類，這個類里有一個屬性，存放著賬號密碼
實例化這個類會生成一段四位數隨機驗證碼
這個類有兩個方法:
·登錄方法，賬號密碼和屬性中的賬號密碼和驗證碼是否一致，一致就返回成功
·修改密碼，接收賬號密碼和新密碼，如果賬號密碼和驗證碼一致，則可以將屬性中的對應的密碼修改為新密碼
"""
import random

class LoginSystem:
    def __init__(self, username, password):
        self.username = username
        self.password = password
        print(self.username)
        print(self.password)
        self.verfication_code = random.randint(1000,9999)
    def login(self, username, password, verfication_code):
        if username == self.username and password == self.password and self.verfication_code == verfication_code:
            return "登錄成功"
        else:
            return "登錄失敗"
    def change_password(self, username, password, new_password, verfication_code):
        if username == self.username and password == self.password and verfication_code == self.verfication_code:
            self.password = new_password
            return "修改密碼成功"
        else:
            return "修改密碼失敗"

login_system = LoginSystem("root", "1234")
code = login_system.verfication_code
print("驗證碼：", code)

"""登錄測試"""
username = input("用戶名：")
password = input("密碼：")
verification_code = int(input("請輸入驗證碼："))
login_result = login_system.login(username,password,verification_code)
print("登錄結果", login_result)

"""修改密碼測試"""
# username = input("用戶名：")
# password = input("密碼：")
# new_password = input("請輸入新的密碼：")
# verification_code = int(input("請輸入驗證碼："))
# login_result = login_system.change_password(username,password,new_password,verification_code)
# print("修改密碼結果：", login_result)