8 lines
286 B
Python
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)
|