diff --git a/{{ cookiecutter.name }}/src/app/models.py b/{{ cookiecutter.name }}/src/app/models.py index 6faf4908..27a21491 100644 --- a/{{ cookiecutter.name }}/src/app/models.py +++ b/{{ cookiecutter.name }}/src/app/models.py @@ -28,15 +28,13 @@ def __str__(self) -> str: def get_contenttype(cls) -> ContentType: return ContentType.objects.get_for_model(cls) - def update_from_kwargs(self, **kwargs: dict[str, Any]) -> None: - """A shortcut method to update model instance from the kwargs.""" + def update(self, **kwargs: "Any") -> "models.Model": for key, value in kwargs.items(): setattr(self, key, value) - def setattr_and_save(self, key: str, value: Any) -> None: - """Shortcut for testing -- set attribute of the model and save""" - setattr(self, key, value) - self.save() + self.save(update_fields=kwargs.keys()) + + return self @classmethod def get_label(cls) -> str: