1. 程式人生 > >python寫使用者登入程式(繼續更新中)第四講(3)

python寫使用者登入程式(繼續更新中)第四講(3)

需求:使用者登入或者註冊,後繼續會更新刪除使用者和更改密碼,以及呼叫資料庫

def login(username,password):
    """
    login my system  ---elcome to my sys!
    :param username:
    :param password:
    :return:
    """
    with open("log","r") as f:
        for line in f:
            line = line.strip()
            line_l = line.split(":") #分割完會寫在一個列表裡面
            if line_l[0] == username and line_l[1]:
                return True
        return False
#前提檔名為log的檔案
def register(username,password):
    """
    註冊使用者
    :param username:使用者名稱
    :param password: 密碼
    :return: 註冊成功
    """
    with open("log","a") as f:
        temp = "\n" + username + ":" + password
        f.write(temp)
    return True


def user_exsit(username):
    """
    檢測使用者是否存在
    :param username:要檢測的使用者名稱
    :param password: 密碼
    :return: 使用者已經存在
    """
    with open("log","r") as f:
        for line in f:
            line= line.strip()
            line_li = line.split(":")
            if line_li[0] == username:
                return True
        return False
def main():
    print("歡迎登陸我係統")
    inp = input("1:登陸,2:註冊")
    user = input("please username:")
    passwd = input("please password:")
    if inp == "1":
        if_login = login(user,passwd)
        if if_login:
            print("登陸成功")
        else:
            print("登陸失敗")
    elif inp == "2":
        is_exsit = user_exsit(user)
        if is_exsit:
            print("使用者名稱已經註冊")
        else:
            result = register(user,passwd)
            if result:
                print("註冊成功")
            else:
                print("註冊失敗")
main()