【练习】字典的循环遍历：实现多层级节点存取

关键点：字典、列表、集合等可变类型，若将一个可变类型赋给变量A和B，通过B改变了该可变类型内部的元素，指向该可变类型内存地址的变量A的值一样会跟着改变（因A、B同时指向了一个内存地址）

>>> a={"AB":"ab","CD":"cd"}
>>> b=a
>>> id(a)
33380536
>>> id(b)
33380536
>>>
>>> b["AB"]="efg"
>>> b
{'AB': 'efg', 'CD': 'cd'}
>>> a
{'AB': 'efg', 'CD': 'cd'}
>>> id(a)
33380536
>>> id(b)
33380536
>>>

 

题目：利用字典实现多层级节点的增加、查询功能，并且通过pickle序列化的方式保存为文件。

import pickle
import os
import time

if os.path.isfile('db.pk1'):
    print("载入数据库，请稍候…………")
    time.sleep(2)
    f = open("db.pk1", "rb")
    db = pickle.load(f)
    f.close()
    print("数据库加载完毕。")
    path = []
    while True:
        tmp = db
        if path:
            for item in path:
                if item in list(tmp.keys()):
                    tmp = tmp[item]
                else:
                    print("无此节点，请重新输入")
        print("当前节点下的子节点为：", list(tmp.keys()))

        choice = input("请选择：1、查看节点\t2、添加节点\t3、返回上级\t4、退出\n>>>>>")
        if choice == "1":
            n = input("请输入节点名称>>>")
            path.append(n)

        elif choice == "2":
            name = input("请输入节点名称：")
            tmp[name] = {}
        elif choice == "3":
            if path:
                path.pop()
            else:
                print("已经是根节点，无法返回。")
        elif choice == "4":
            print("保存数据库，请稍等……")
            time.sleep(2)
            f = open("db.pk1", "wb")
            pickle.dump(db, f, True)
            print("保存成功，程序退出。")
            break
        else:
            print("输入错误请重新输入")
elif not os.path.isfile('db.pk1'):
    choice1 = input("数据库不存在，是否初始化数据库：Y\\N\n>>>")
    if choice1.lower() == "y":
        db = {}
        time.sleep(2)
        f = open("db.pk1", "wb")
        pickle.dump(db, f, True)
        f.close()
        print("初始化成功。")
        
    elif choice1.lower() == "n":
        print("无法初始化数据库，程序退出。")