import dbm

# 第一個參數是文件名
# 第二個參數是模式，有以下幾種
#     r：只讀，要求文件必須存在，默認就是這個模式
#     w：可讀可寫，要求文件必須存在
#     c：可讀可寫，文件不存在會創建，存在則追加
#     n：可讀可寫，文件不存在會創建，存在則清空
# 第三個參數是權限，用八進制數字表示，默認 0o666，即可讀可寫不可執行
db = dbm.open("store", "c")

# 打開文件就可以存儲值了，key 和 value 必須是字符串或 bytes 對象
db["name"] = "S せんせい"
db["age"] = "18"
db[b"corporation"] = "小摩".encode("utf-8")

# 關閉文件，將內容寫到磁盤上
db.close()

import dbm

db = dbm.open("store", "c")

# 獲取所有的 key，直接返回一個列表
print(db.keys())
"""
[b'corporation', b'name', b'age']
"""
# 判斷一個 key 是否存在，key 可以是字符串或 bytes 對象
print("name" in db, "NAME" in db)
"""
True False
"""
# 獲取一個 key 對應的 value，得到的是 bytes 對象
print(db["name"].decode("utf-8"))
print(db[b"corporation"].decode("utf-8"))
"""
S せんせい
小摩
"""
# key 如果不存在，會拋出 KeyError，我們可以使用 get 方法
print(db.get("NAME", b"unknown"))
"""
b'unknown'
"""
# 當然也可以使用 setdefault 方法，key 不存在時，自動寫進去
print(db.setdefault("gender", b"female"))
"""
b'female'
"""
print(db["gender"])
"""
b'female'
"""

import shelve

# 第二個參數表示模式，默認是 c
# 因此文件不存在會創建，存在則追加
sh = shelve.open("shelve")

sh["name"] = ["S 老師", "高老師", "電烤??架"]
sh["age"] = {18}
sh["job"] = {"tutu": "大學生", "xueer": "醫生"}

# 關閉文件，刷到磁盤中
sh.close()

import shelve

sh = shelve.open("shelve")

print(sh["name"])
print(sh["name"][2] == "電烤??架")
"""
['S 老師', '高老師', '電烤??架']
True
"""
print(sh["age"])
"""
{18}
"""
print(sh["job"])
"""
{'tutu': '大學生', 'xueer': '醫生'}
"""

sh.close()

import shelve

class People:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    @property
    def print_info(self):
        return f"name is {self.name}, age is {self.age}"

sh = shelve.open("shelve")

p = People("群主", 58)
# 將類、和該類的實例對象存儲進去
sh["People"] = People
sh["p"] = p
sh.close()

import shelve

sh = shelve.open("shelve")

# 需要注意的是，People 是我們自己定義的類
# 如果你想要將其還原出來，那么該類必須要出現在當前的命名空間中
try:
    sh["People"]
except AttributeError as e:
    print(e)
    """
    Can't get attribute 'People' on <module ...>
    """

class People:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    @property
    def print_info(self):
        return f"name is {self.name}, age is {self.age}"

print(sh["People"] is People)
"""
True
"""
print(sh["p"].print_info)
"""
name is 群主, age is 58
"""
print(sh["People"]("群主", 38).print_info)
"""
name is 群主, age is 38
"""

import shelve

# 打開文件，設置鍵值對
sh = shelve.open("shelve")
sh["name"] = "古明地覺"
sh["score"] = [80, 80, 80]
sh.close()

# 重新打開文件，修改鍵值對
sh = shelve.open("shelve")
sh["name"] = "芙蘭朵露"
sh["score"].append(90)
sh.close()

# 再次重新打開文件，查看鍵值對
sh = shelve.open("shelve")
print(sh["name"])
print(sh["score"])
"""
芙蘭朵露
[80, 80, 80]
"""
sh.close()

import shelve

# 打開文件，設置鍵值對
sh = shelve.open("shelve")
sh["name"] = "古明地覺"
sh["score"] = [80, 80, 80]
sh.close()

# 重新打開文件，修改鍵值對
sh = shelve.open("shelve", writeback=True)
sh["name"] = "芙蘭朵露"
sh["score"].append(90)
sh.close()

# 再次重新打開文件，查看鍵值對
sh = shelve.open("shelve")
print(sh["name"])
print(sh["score"])
"""
芙蘭朵露
[80, 80, 80, 90]
"""
sh.close()