Merge pull request #94 from HackSoftware/update-service

POC: Update service

Author: wencakisa

styleguide_example/common/services.py

@@ -0,0 +1,48 @@
+from typing import List, Dict, Any, Tuple
+
+from styleguide_example.common.types import DjangoModelType
+
+
+def model_update(
+    *,
+    instance: DjangoModelType,
+    fields: List[str],
+    data: Dict[str, Any]
+) -> Tuple[DjangoModelType, bool]:
+    """
+    Generic update service meant to be reused in local update services
+
+    For example:
+
+    def user_update(*, user: User, data) -> User:
+        fields = ['first_name', 'last_name']
+        user, has_updated = model_update(instance=user, fields=fields, data=data)
+
+        // Do other actions with the user here
+
+        return user
+
+    Return value: Tuple with the following elements:
+        1. The instance we updated
+        2. A boolean value representing whether we performed an update or not.
+    """
+    has_updated = False
+
+    for field in fields:
+        # Skip if a field is not present in the actual data
+        if field not in data:
+            continue
+
+        if getattr(instance, field) != data[field]:
+            has_updated = True
+            setattr(instance, field, data[field])
+
+    # Perform an update only if any of the fields was actually changed
+    if has_updated:
+        instance.full_clean()
+        # Update only the fields that are meant to be updated.
+        # Django docs reference:
+        # https://docs.djangoproject.com/en/dev/ref/models/instances/#specifying-which-fields-to-save
+        instance.save(update_fields=fields)
+
+    return instance, has_updated

styleguide_example/common/tests/services/__init__.py

@@ -0,0 +1,58 @@
+import unittest
+from unittest.mock import Mock
+
+from styleguide_example.common.services import model_update
+
+
+class ModelUpdateTests(unittest.TestCase):
+    def setUp(self):
+        self.instance = Mock(
+            field_a=None,
+            field_b=None,
+            field_c=None
+        )
+
+    def test_model_update_does_not_update_if_none_of_the_fields_are_in_the_data(self):
+        update_fields = ['non_existing_field']
+        data = {'field_a': 'value_a'}
+
+        updated_instance, has_updated = model_update(
+            instance=self.instance,
+            fields=update_fields,
+            data=data
+        )
+
+        self.assertEqual(updated_instance, self.instance)
+        self.assertFalse(has_updated)
+
+        self.assertIsNone(updated_instance.field_a)
+        self.assertIsNone(updated_instance.field_b)
+        self.assertIsNone(updated_instance.field_c)
+
+        self.instance.full_clean.assert_not_called()
+        self.instance.save.assert_not_called()
+
+    def test_model_update_updates_only_passed_fields_from_data(self):
+        update_fields = ['field_a']
+        data = {
+            'field_a': 'value_a',
+            'field_b': 'value_b'
+        }
+
+        updated_instance, has_updated = model_update(
+            instance=self.instance,
+            fields=update_fields,
+            data=data
+        )
+
+        self.assertTrue(has_updated)
+
+        self.assertEqual(updated_instance.field_a, 'value_a')
+        # Even though `field_b` is passed in `data` - it does not get updated
+        # because it is not present in the `fields` list.
+        self.assertIsNone(updated_instance.field_b)
+        # `field_c` remains `None`, because it is not passed anywhere.
+        self.assertIsNone(updated_instance.field_c)
+
+        self.instance.full_clean.assert_called_once()
+        self.instance.save.assert_called_once_with(update_fields=update_fields)

styleguide_example/common/tests/services/test_model_update.py

@@ -0,0 +1,7 @@
+from typing import TypeVar
+
+from django.db import models
+
+# Generic type for a Django model
+# Reference: https://mypy.readthedocs.io/en/stable/kinds_of_types.html#the-type-of-class-objects
+DjangoModelType = TypeVar('DjangoModelType', bound=models.Model)

styleguide_example/common/types.py

@@ -1,5 +1,9 @@
 from typing import Optional
 
+from django.db import transaction
+
+from styleguide_example.common.services import model_update
+
 from styleguide_example.users.models import BaseUser
 
 
@@ -18,3 +22,20 @@ def user_create(
     )
 
     return user
+
+
+@transaction.atomic
+def user_update(*, user: BaseUser, data) -> BaseUser:
+    non_side_effect_fields = ['first_name', 'last_name']
+
+    user, has_updated = model_update(
+        instance=user,
+        fields=non_side_effect_fields,
+        data=data
+    )
+
+    # Side-effect fields update here (e.g. username is generated based on first & last name)
+
+    # ... some additional tasks with the user ...
+
+    return user