Browse Source

Simplify save method

master
Burathar 3 years ago
parent
commit
c68f3b6b57
  1. 10
      biscd/biscd/models/yaml_serializable.py

10
biscd/biscd/models/yaml_serializable.py

@ -40,18 +40,14 @@ class YamlSerializable(RecursiveProperty):
if self.name is None: if self.name is None:
raise TypeError("Name cannot be None") raise TypeError("Name cannot be None")
ymlsls = self._get_all_from_file() ymlsls = self._get_all_from_file()
found_match = False ymlsl = next((ymlsl for ymlsl in ymlsls if [*ymlsl][0] == self.name), None)
for ymlsl in ymlsls: if ymlsl:
if self.name == [*ymlsl][0]:
found_match = True
if overwrite: if overwrite:
ymlsl[self.name] = self.config_dict(values_only=True) ymlsl[self.name] = self.config_dict(values_only=True)
else: else:
raise ValueError( raise ValueError(
f"A {type(self).__name__} with name {self.name} already exists!") f"A {type(self).__name__} with name {self.name} already exists!")
break else:
if not found_match:
ymlsls.append(self.config_dict()) ymlsls.append(self.config_dict())
print(ymlsls) print(ymlsls)
self._save_all_to_file(ymlsls) self._save_all_to_file(ymlsls)

Loading…
Cancel
Save