1
Fork 0
yamdl-playground/yamdl_playground/core/utils.py

8 lines
286 B
Python

from yamdl.loader import ModelLoader
class CustomYamdlLoader(ModelLoader):
def load_fixture(self, model_class, data, file_path):
if file_path.suffix not in self.EXT_YAML:
data["slug"] = file_path.stem
super().load_fixture(model_class, data, file_path)