对象变量的try/except设置的python方式



我正在编写一段代码,它接受字典并创建对象的实例。在学习tryexcept模块时,我将其合并到我的小函数中。我已经准备好很多次了,在做某事之前检查不是python的,所以我想知道我该如何修改我的代码(下面)来满足这个建议。在它的立场,代码只工作,因为我使用getattr设置之前。否则,setattr将创建该属性,而不是因为该属性不存在而失败。

class Gem:
def __init__(self):
self.name = None
self.val = None

class Rock:
def __init__(self):
self.name = None
self.weight = None

def create_objects(data):
obj_list = []
for i in data:
new_obj = Gem()
try:
for k, v in i.items():
getattr(new_obj, k)
setattr(new_obj, k, v)
obj_list.append(new_obj)
except:
print("Fail")
return obj_list
data_list = [
{"name": "Star", "val": 5},
{"name": "Square", "val": 1},
{"name": "Granite", "weight": 50}
]
obj_list = create_objects(data_list)
for o in obj_list:
for k, v in vars(o).items():
print(k, v)

在创建实例时设置属性,将它们作为参数传递给__init__。这样,您可以确定它们还不存在,并且您不需要try/except:

class Gem:
def __init__(self, name, val):
self.name = name
self.val = val
obj_list = []
for gem_attributes in data_list:
gem = Gem(gem_attributes['name'], gem_attributes['val'])
obj_list.append(gem)

将字典值作为与键名相同的参数传递,可以使用**参数解包语法缩短:

gem = Gem(**gem_attributes)

整个列表创建可以缩短为列表推导:

obj_list = [
Gem(**gem_attributes)
for gem_attributes in data_list
]

作为一个额外的复杂性,你有不同的对象类型。

为了在同一个循环中创建GemRock对象,我建议将对象类型显式地添加到输入数据中,而不是依靠属性名称来区分它们:

data_list = [
('Gem', {'name': 'Star', 'val': 5}),
...
('Rock', {'name': 'Granite', 'weight': 50})
]
obj_list = []
for obj_type, obj_attributes in data_list:
obj_class = {'Gem': Gem, 'Rock': Rock}[obj_type]
obj_list.append(obj_class(**obj_attributes))

最新更新