def create(self, validated_data): user = None request = self.context.get("request") if request and hasattr(request, "user"): user = request.user image_instance = Image(**validated_data) image_instance.user = user image_instance.save() return image_instance
def setUp(self): self.user_data = { "username": "******", "first_name": "black", "last_name": "panther", } user_instance = User(**self.user_data) user_instance.set_password("gamora") user_instance.save() image_instance = Image() image_instance.user = user_instance image_instance.image = File(create_test_image()) image_instance.name = "test" image_instance.color = "BLACK" image_instance.save()