[utils][all] Pre-commit cleanup

beat/web/utils/activity_stream_serializer.py

@@ -25,22 +25,22 @@
 #                                                                             #
 ###############################################################################
 
+import json
+import sys
+
+from actstream.models import Action
+from django.contrib.contenttypes.models import ContentType
 from django.core.serializers.base import DeserializationError
-from django.core.serializers.python import Deserializer as PythonDeserializer, _get_model
 from django.core.serializers.json import Serializer as JSONSerializer
+from django.core.serializers.python import Deserializer as PythonDeserializer
+from django.core.serializers.python import _get_model
 from django.utils.encoding import force_text
-from django.contrib.contenttypes.models import ContentType
-
 
-
-from actstream.models import Action
-
-import sys
-import json
-
-elements_to_convert = [('actor_object_id', 'actor_content_type'),
-                       ('action_object_object_id', 'action_object_content_type'),
-                       ('target_object_id', 'target_content_type')]
+elements_to_convert = [
+    ("actor_object_id", "actor_content_type"),
+    ("action_object_object_id", "action_object_content_type"),
+    ("target_object_id", "target_content_type"),
+]
 
 
 class Serializer(JSONSerializer):
@@ -48,13 +48,14 @@ class Serializer(JSONSerializer):
     Custom JSON serializer that replaces the actstream Action model ids by
     their natural key counterpart.
     """
+
     def get_dump_object(self, obj):
         data = {
             "model": force_text(obj._meta),
             "fields": self._current,
         }
 
-        if not self.use_natural_primary_keys or not hasattr(obj, 'natural_key'):
+        if not self.use_natural_primary_keys or not hasattr(obj, "natural_key"):
             data["pk"] = force_text(obj._get_pk_val(), strings_only=True)
 
         if self.use_natural_primary_keys and isinstance(obj, Action):
@@ -62,15 +63,19 @@ class Serializer(JSONSerializer):
                 element_object_id = element[0]
                 element_content_type = element[1]
 
-                element_info = data['fields'][element_content_type]
+                element_info = data["fields"][element_content_type]
 
                 if element_info:
                     element_app = element_info[0]
                     element_model = element_info[1]
-                    element_type = ContentType.objects.get(app_label=element_app, model=element_model)
-                    element_id = data['fields'][element_object_id]
-                    element_object = element_type.get_object_for_this_type(id=int(element_id))
-                    data['fields'][element_object_id] = element_object.natural_key()
+                    element_type = ContentType.objects.get(
+                        app_label=element_app, model=element_model
+                    )
+                    element_id = data["fields"][element_object_id]
+                    element_object = element_type.get_object_for_this_type(
+                        id=int(element_id)
+                    )
+                    data["fields"][element_object_id] = element_object.natural_key()
         return data
 
 
@@ -82,7 +87,7 @@ def Deserializer(stream_or_string, **options):
     if not isinstance(stream_or_string, (bytes, str)):
         stream_or_string = stream_or_string.read()
     if isinstance(stream_or_string, bytes):
-        stream_or_string = stream_or_string.decode('utf-8')
+        stream_or_string = stream_or_string.decode("utf-8")
     try:
         objects = json.loads(stream_or_string)
 
@@ -94,16 +99,20 @@ def Deserializer(stream_or_string, **options):
                 for element in elements_to_convert:
                     element_object_id = element[0]
                     element_content_type = element[1]
-                    element_info = d['fields'][element_content_type]
+                    element_info = d["fields"][element_content_type]
 
                     if element_info:
                         element_app = element_info[0]
                         element_model = element_info[1]
-                        element_type = ContentType.objects.get(app_label=element_app, model=element_model)
-                        element_natural_key = d['fields'][element_object_id]
-                        element_object = element_type.model_class().objects.get_by_natural_key(*element_natural_key)
+                        element_type = ContentType.objects.get(
+                            app_label=element_app, model=element_model
+                        )
+                        element_natural_key = d["fields"][element_object_id]
+                        element_object = element_type.model_class().objects.get_by_natural_key(
+                            *element_natural_key
+                        )
 
-                        d['fields'][element_object_id] = str(element_object.id)
+                        d["fields"][element_object_id] = str(element_object.id)
 
         for obj in PythonDeserializer(objects, **options):
             yield obj
@@ -111,4 +120,4 @@ def Deserializer(stream_or_string, **options):
         raise
     except Exception as e:
         # Map to deserializer error
-        raise(DeserializationError, DeserializationError(e), sys.exc_info()[2])
+        raise (DeserializationError, DeserializationError(e), sys.exc_info()[2])

beat/web/utils/drf.py

@@ -27,12 +27,11 @@
 """
 Django REST framework helpers
 """
-
 import logging
 
-from rest_framework.views import exception_handler
 from rest_framework.exceptions import APIException
 from rest_framework.status import is_client_error
+from rest_framework.views import exception_handler
 
 logger = logging.getLogger("beat.drf_exceptions")