Browse Source

Simplify save method

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

20
biscd/biscd/models/yaml_serializable.py

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

Loading…
Cancel
Save