fix linting

Author: timgraham

django_mongodb/query.py

@@ -1,13 +1,13 @@
 from functools import reduce, wraps
 from operator import add as add_operator
 
-from django.core.exceptions import EmptyResultSet, FullResultSet, FieldDoesNotExist
+from django.core.exceptions import EmptyResultSet, FieldDoesNotExist, FullResultSet
 from django.db import DatabaseError, IntegrityError, NotSupportedError, connections
 from django.db.models import QuerySet
 from django.db.models.expressions import Case, Col, When
 from django.db.models.functions import Mod
 from django.db.models.lookups import Exact
-from django.db.models.query import RawQuerySet, RawModelIterable
+from django.db.models.query import RawModelIterable, RawQuerySet
 from django.db.models.sql.constants import INNER
 from django.db.models.sql.datastructures import Join
 from django.db.models.sql.query import RawQuery
@@ -308,7 +308,6 @@ def register_nodes():
 
 
 class MongoQuerySet(QuerySet):
-
     def raw_mql(self, raw_query, params=(), translations=None, using=None):
         return MongoRawQuerySet(
             raw_query,
@@ -320,7 +319,6 @@ def raw_mql(self, raw_query, params=(), translations=None, using=None):
 
 
 class MongoRawQuery(RawQuery):
-
     def __init__(self, sql, using, model, params=()):
         super().__init__(sql, using, params)
         self.model = model
@@ -352,7 +350,14 @@ def __init__(
         using=None,
         hints=None,
     ):
-        super().__init__(raw_query, model=model, query=query, using=using, hints=hints, translations=translations)
+        super().__init__(
+            raw_query,
+            model=model,
+            query=query,
+            using=using,
+            hints=hints,
+            translations=translations,
+        )
         self.query = query or MongoRawQuery(
             sql=raw_query, using=self.db, model=self.model, params=params
         )
@@ -362,17 +367,11 @@ def iterator(self):
 
     def resolve_model_init_order(self, columns):
         """Resolve the init field names and value positions."""
-        model_init_fields = [
-            f for f in self.model._meta.fields if f.column in columns
-        ]
+        model_init_fields = [f for f in self.model._meta.fields if f.column in columns]
         annotation_fields = [
-            (column, pos)
-            for pos, column in enumerate(columns)
-            if column not in self.model_fields
-        ]
-        model_init_order = [
-            columns.index(f.column) for f in model_init_fields
+            (column, pos) for pos, column in enumerate(columns) if column not in self.model_fields
         ]
+        model_init_order = [columns.index(f.column) for f in model_init_fields]
         model_init_names = [f.attname for f in model_init_fields]
         return model_init_names, model_init_order, annotation_fields
 
@@ -401,9 +400,7 @@ def __iter__(self):
             ) = self.queryset.resolve_model_init_order(columns)
             model_cls = self.queryset.model
             if model_cls._meta.pk.attname not in model_init_names:
-                raise FieldDoesNotExist(
-                    "Raw query must include the primary key"
-                )
+                raise FieldDoesNotExist("Raw query must include the primary key")
             fields = [self.queryset.model_fields.get(c) for c in self.queryset.columns]
             converters = compiler.get_converters(
                 [f.get_col(f.model._meta.db_table) if f else None for f in fields]