python對象轉字典

1.基礎實現

 1 class TestDict:
 2     name = "wyb"
 3     age = "21"
 4 
 5     def __init__(self):
 6         self.gender = 'male'
 7 
 8     def keys(self):                         # 獲取字典的鍵
 9         s = ('name', 'age', 'gender')
10         return s
11 
12     def __getitem__(self, item):            # 獲取鍵對應的值
13         return getattr(self, item)          # getattr獲取對象下某個屬性的值
14 
15 
16 o = TestDict()
17 print(dict(o))          # 建立字典 -> 先調用對象下的keys方法再用o["xxx"]獲取值([]本質上是調用對象下的__getitem__方法)

 

 

2.進階實現

 1 class Base(object):
 2     def __init__(self):
 3         self.fields = []
 4 
 5     def __getitem__(self, item):
 6         return getattr(self, item)
 7 
 8     def keys(self):
 9         return self.fields
10 
11     def hide(self, *keys):                  # 隱藏域
12         for key in keys:
13             self.fields.remove(key)
14         return self
15 
16     def append(self, *keys):                # 添加域
17         for key in keys:
18             self.fields.append(key)
19         return self
20 
21 
22 class Sx(Base):
23     name = "wyb"
24     age = 21
25 
26     def __init__(self):
27         super(Base, self).__init__()
28         self.gender = 'male'
29         # self.fields定義默認輸出字段
30         self.fields = ["name", "age"]
31 
32 
33 o1 = Sx()
34 print(dict(o1))
35 o2 = Sx()
36 print(dict(o2.hide("age")))
37 o3 = Sx()
38 print(dict(o3.append("gender")))
39 
40 # 輸出結果:
41 # {'name': 'wyb', 'age': 21}
42 # {'name': 'wyb'}
43 # {'name': 'wyb', 'age': 21, 'gender': 'male'}