import os
import pickle


def main():  # 主函數(shù)
    while True:
        print('-' * 30)
        print('------菜單------')
        print('1.線路查詢(xún)')
        print('2.站點(diǎn)查詢(xún)')
        print('3.換乘信息')
        print('4.后臺(tái)管理員界面')
        print('5.管理員的賬號(hào)管理界面')
        print('6.退出程序')
        print('-' * 30)
        num = int(input('請(qǐng)選擇你的操作：\n'))
        if num == 1:
            line_find()
            continue
        if num == 2:
            site_find()
            continue
        if num == 3:
            change_line()
            continue
        if num == 5:
            manager_account()
            continue
        if num == 6:
            break
        if num == 4:
            manager()
            continue
        else:
            print('請(qǐng)重新做出你的選擇??！')


def manager_account():  # 管理員賬號(hào)管理界面的函數(shù)
    def new_():
        file_path = 'D:\公交查詢(xún)系統(tǒng)\info.pkl'
        a = input('請(qǐng)?jiān)O(shè)置你的賬號(hào)：')
        b = input('請(qǐng)?jiān)O(shè)置入你的密碼：')
        w = input('請(qǐng)?jiān)俅屋斎肽愕牡拿艽a：')
        d = {
            'user': a,
            'password': b
        }
        if b == w:
            if os.path.exists(file_path) and os.path.getsize(file_path):
                o = open('info.pkl', 'rb')
                m = pickle.load(o)
                with open('info.pkl', 'ab') as fp:
                    pickle.dump(d, fp)
                print('您已經(jīng)注冊(cè)成功?。?)
            else:
                with open('info.pkl', 'wb') as fp:
                    pickle.dump(d, fp)
                print('您已經(jīng)注冊(cè)成功??！')
        else:
            print('對(duì)不起，你兩次輸入的密碼不一致，請(qǐng)您重新輸入?。?)

    def xiaohui():
        h = open('info.pkl', 'w+')  # 清空文件里的內(nèi)容。。。
        h.truncate()
        print('你已經(jīng)銷(xiāo)毀成功。。。')

    while True:
        print('-----管理員的賬號(hào)管理界面------\n'
              '1.注冊(cè)\n'
              '2.注銷(xiāo)原有的所有賬戶(hù)\n'
              '3.退出管理員的賬號(hào)管理界面')
        choice = input('請(qǐng)做出你的選擇：\n')
        if choice == '1':
            new_()
        if choice == '2':
            xiaohui()
        elif choice == '3':
            print('-' * 35)
            print('------菜單------')
            print('1.線路查詢(xún)')
            print('2.站點(diǎn)查詢(xún)')
            print('3.換乘信息')
            print('4.后臺(tái)管理員界面')
            print('5.管理員的賬號(hào)管理界面')
            print('6.退出程序')
            print('-' * 35)
            break
        else:
            print('請(qǐng)重新做出你的選擇')


def login():  # 用戶(hù)登錄的函數(shù)
    with open('info.pkl', 'rb') as fp:
        q = input('請(qǐng)輸入你的的賬號(hào)：')
        w = input('請(qǐng)輸入你的的密碼：')
        while True:
            try:
                data = pickle.load(fp)
                if q == data['user'] and w == data['password']:
                    print('您已經(jīng)登錄成功?。?！')
                    return True

            except:
                break


def manager():  # 后臺(tái)管理員界面的函數(shù)
    if login():
        while True:
            print('-' * 30)
            print('-----后臺(tái)管理員界面-----')
            print('1.添加線路')
            print('2.刪除線路')
            print('3.修改線路')
            print('4.查看所有線路')
            print('5.刪除所有線路')
            print('6.退出')
            print('-' * 30)
            num = input('請(qǐng)做出你的選擇:\n')
            if num == '3':
                revise_line()
                continue
            if num == '1':
                add_line()
                continue
            if num == '6':
                print('-' * 25)
                print('------菜單------')
                print('1.線路查詢(xún)')
                print('2.站點(diǎn)查詢(xún)')
                print('3.換乘信息')
                print('4.后臺(tái)管理員界面')
                print('5.管理員的賬號(hào)管理界面')
                print('6.退出程序')
                print('-' * 25)
                break
            if num == '5':
                clear_line_all()
                continue
            if num == '4':
                show_line_all()
                continue
            if num == '2':
                num = input('請(qǐng)輸入你要?jiǎng)h除的線路：\n').strip()
                clear_line(num)
                continue
            else:
                print('請(qǐng)重新做出你的選擇...')
    else:
        print('對(duì)不起，你登錄失敗，不可以讓您進(jìn)入后臺(tái)管理員界面')


def revise_line():  # 定義修改線路的函數(shù)
    num = input('請(qǐng)輸入您要修改的線路：').strip()
    if check_add_line(num):
        show_line_all()
        tag = input('請(qǐng)確認(rèn)是否繼續(xù)修改線路：(1為是，0為否)\n')
        if tag == '1':
            file_path = 'D:\公交查詢(xún)系統(tǒng)\line_info.pkl'
            a = input('請(qǐng)?jiān)俅屋斎胍薷牡木€路：').strip()
            clear_line(a)
            b = input('請(qǐng)輸入經(jīng)過(guò)的站點(diǎn)：')
            d = {
                a: 
            }
            if os.path.exists(file_path) and os.path.getsize(file_path):
                with open('line_info.pkl', 'ab') as f:
                    pickle.dump(d, f)
                print('您已經(jīng)修改線路成功??！')
            else:
                with open('line_info.pkl', 'wb') as fp:
                    pickle.dump(d, fp)
                print("您已經(jīng)修改線路成功了！?。?！")
    else:
        print('您要修改的線路不存在，需要?jiǎng)?chuàng)建一個(gè)新的線路。。。')
        add_line()


def line_check(num):  # 判斷是否存在線路的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                if num in data:
                    print(f'{num}號(hào)線的路線經(jīng)過(guò)的站點(diǎn)為{data[num]}')
                    return True
            except:
                break


def line_find():  # 線路查詢(xún)的函數(shù)
    num = input('請(qǐng)輸入你要查詢(xún)的線路：\n').strip()
    if line_check(num):
        pass
    else:
        print('您要查找的線路不存在....')


def site_check():  # 判斷站點(diǎn)是否存在的函數(shù)
    res = []
    with open('line_info.pkl', 'rb') as fp:
        num = input('請(qǐng)輸入你要查詢(xún)的站點(diǎn)名：\n').strip()
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    ls = str(value)
                    if num in ls:
                        res.append(f'經(jīng)過(guò){num}站點(diǎn)的線路為{key}號(hào)線')
                        continue
            except:
                break
    if res:
        print(res)
        return True
    else:
        return False


def site_find():  # 站點(diǎn)查詢(xún)的函數(shù)
    if site_check():
        pass
    else:
        print('您要查詢(xún)的站點(diǎn)不存在.....')


def check_add_line(num):  # 檢查添加線路是否存在的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    if num in key:
                        return True
            except:
                break


def add_line():  # 添加線路的函數(shù)
    file_path = 'D:\公交查詢(xún)系統(tǒng)\line_info.pkl'
    a = input('請(qǐng)輸入添加的線路：').strip()
    b = input('請(qǐng)輸入經(jīng)過(guò)的站點(diǎn)：')
    d = {
        a: 
    }
    if os.path.exists(file_path) and os.path.getsize(file_path):
        if check_add_line(a):
            print('您要添加的線路已經(jīng)存在了......')
        else:
            with open('line_info.pkl', 'ab') as f:
                pickle.dump(d, f)
            print('您已經(jīng)添加線路成功?。?)
    else:
        with open('line_info.pkl', 'wb') as fp:
            pickle.dump(d, fp)
        print('您已經(jīng)添加線路成功了??！')


def clear_line_all():  # 清除所有線路的函數(shù)
    h = open('line_info.pkl', 'w+')  # 清空文件里的內(nèi)容。。。
    h.truncate()
    print('你已經(jīng)銷(xiāo)毀成功。。。')


def show_line_all():  # 展示所有線路的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    print(f'{key}線路經(jīng)過(guò)的站點(diǎn)是{value}')
            except:
                break


def clear_line(num):  # 刪除線路的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        ls = []
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    if num in key:
                        print(f'{num}線路已經(jīng)刪除成功！！')
                        pass
                    else:
                        ls.append(data)
            except:
                break
        save_data(ls)


def line_check(num):  # 判斷是否存在線路的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                if num in data:
                    print(f'{num}號(hào)線的路線經(jīng)過(guò)的站點(diǎn)為{data[num]}')
                    return True
            except:
                break


def show_site(num):  # 展示特定線路中的站點(diǎn)的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                if num in data:
                    return data[num]
            except:
                break


def change_line():  # 換乘信息的函數(shù)
    begin = input('請(qǐng)輸入你的起點(diǎn)站點(diǎn)：\n').strip()
    end = input('請(qǐng)輸入你的終點(diǎn)站點(diǎn)：\n').strip()
    if direct_line(begin, end):
        pass
    else:
        ls = []  # 存儲(chǔ)經(jīng)過(guò)起點(diǎn)的線路的全部站點(diǎn)
        ls_ = []  # 存儲(chǔ)經(jīng)過(guò)終點(diǎn)的線路的全部站點(diǎn)
        tag = True
        data = check_site(begin, end)
        # print(type(data[0]),type(data[1]))
        print(f'經(jīng)過(guò){begin}站點(diǎn)的線路為：{data[0]}，經(jīng)過(guò){end}站點(diǎn)的線路為{data[1]}')  # 判斷經(jīng)過(guò)起始點(diǎn)和終點(diǎn)的對(duì)應(yīng)線路
        for i in range(len(list(data)[0])):
            ls.append(show_site(data[0][i]))
            print(f"{line_check(str(data[0][i]))}")
        for j in range(len(list(data)[1])):
            ls_.append(show_site(data[1][j]))
            print(f"{line_check(str(data[1][j]))}")
        for i in ls:  # i為集合的形式
            for a in list(i):
                for b in str(a).split():  # 切割字符串
                    if b in str(ls_):  # 判斷是否存在相同的站點(diǎn)
                        tag = False
                        print(f"您可以通過(guò)從{data[0]}號(hào)線的站轉(zhuǎn)到{data[1]}號(hào)線來(lái)到達(dá)目的地?。?)
            #             break
            #         break
            #     break
            # break
        if tag:
            print('對(duì)不起，無(wú)法通過(guò)換乘來(lái)到達(dá)目的地')


def direct_line(begin, end):  # 判斷線路直達(dá)的函數(shù)
    with open('line_info.pkl', 'rb') as fp:
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    ls = str(value)
                    if begin in ls and end in ls:
                        print(f'您可以通過(guò){key}號(hào)線路直接到達(dá)')
                        return True
            except:
                break


def save_data(ls):  # 存儲(chǔ)數(shù)據(jù)的函數(shù)
    with open('line_info.pkl', 'wb') as fp:
        for data in ls:
            pickle.dump(data, fp)


def check_site(begin, end):  # 判斷站點(diǎn)所在的線路的函數(shù)。
    with open('line_info.pkl', 'rb') as fp:
        ls = []
        ls_ = []
        while True:
            try:
                data = pickle.load(fp)
                # 遍歷字典
                for key, value in data.items():
                    ls1 = str(value)
                    if begin in ls1:
                        ls.append(key)
                    if end in ls1:
                        ls_.append(key)
            except:
                break
        return ls, ls_


# def help_change_line_info(i, j):
#     with open('line_info.pkl', 'rb') as fp:
#         ls = []
#         ls_ = []
#         while True:
#             try:
#                 data = pickle.load(fp)
#                 # 遍歷字典
#                 for key, value in data.items():
#                     ls1 = str(key)
#                     if i in ls1:
#                         ls.append(value)
#                     if j in ls1:
#                         ls_.append(value)
#             except:
#                 break
#         return ls, ls_


if __name__ == 'main':
    main()

main()