[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

api/environments/models.py

@@ -4,7 +4,6 @@
 from copy import deepcopy
 from typing import TYPE_CHECKING, Literal
 
-from flagsmith_schemas.api import V1EnvironmentDocumentResponse
 from common.core.utils import using_database_replica
 from django.conf import settings
 from django.contrib.contenttypes.fields import GenericRelation
@@ -21,6 +20,7 @@
     LifecycleModel,
     hook,
 )
+from flagsmith_schemas.api import V1EnvironmentDocumentResponse
 from rest_framework.request import Request
 from softdelete.models import SoftDeleteObject  # type: ignore[import-untyped]
 
@@ -363,16 +363,16 @@ def write_environment_documents(
             cache_payload = {}
             for e in environments:
                 fat_doc = map_environment_to_environment_document(e)
-                
+
                 #  Safely check for Pydantic V2's model_dump, with a fallback
                 if hasattr(fat_doc, "model_dump"):
                     doc_dict = fat_doc.model_dump()
-                elif hasattr(fat_doc, "dict"): 
+                elif hasattr(fat_doc, "dict"):
                     # Just in case Flagsmith has a legacy custom class here
                     doc_dict = fat_doc.dict()
                 else:
                     doc_dict = fat_doc
-                
+
                 # Manually extract ONLY the keys allowed by the V1 TypedDict
                 clean_doc: V1EnvironmentDocumentResponse = {
                     "api_key": doc_dict.get("api_key"),

api/tests/unit/environments/test_unit_environments_models.py

@@ -10,6 +10,7 @@
 from django.db.models import Count, Q
 from django.test import override_settings
 from django.utils import timezone
+from flagsmith_schemas.api import V1EnvironmentDocumentResponse
 from mypy_boto3_dynamodb.service_resource import Table
 from pytest_django import DjangoAssertNumQueries
 from pytest_django.asserts import assertQuerySetEqual as assert_queryset_equal
@@ -19,9 +20,8 @@
 from audit.related_object_type import RelatedObjectType
 from core.constants import STRING
 from core.request_origin import RequestOrigin
-from environments.identities.models import Identity
 from environments.enums import EnvironmentDocumentCacheMode
-from flagsmith_schemas.api import V1EnvironmentDocumentResponse
+from environments.identities.models import Identity
 from environments.metrics import CACHE_HIT, CACHE_MISS
 from environments.models import (
     Environment,
@@ -1269,7 +1269,7 @@ def test_environment_clone_from_non_versioned_environment_with_use_v2_feature_ve
     # When
     new_environment = environment.clone(name="new-environment")
 
- # Then
+    # Then
     assert new_environment.use_v2_feature_versioning
 
     # we only expect a single environment feature version as we are essentially
@@ -1283,15 +1283,14 @@ def test_environment_clone_from_non_versioned_environment_with_use_v2_feature_ve
     assert latest_feature_states.count() == 2
     assert {fs.environment_feature_version for fs in latest_feature_states} == {efv}
 
+
 @mock.patch("environments.models.environment_document_cache")
 def test_write_environment_documents_strips_internal_fields_from_cache(
-    mock_document_cache: MagicMock, 
-    environment: Environment, 
-    settings: typing.Any
+    mock_document_cache: MagicMock, environment: Environment, settings: typing.Any
 ) -> None:
     # 1. SETUP: Force the persistent caching mode
     settings.CACHE_ENVIRONMENT_DOCUMENT_MODE = EnvironmentDocumentCacheMode.PERSISTENT
-    
+
     # 2. ACTION: Trigger the document builder
     Environment.write_environment_documents(environment_id=environment.id)
 
@@ -1304,7 +1303,7 @@ def test_write_environment_documents_strips_internal_fields_from_cache(
     assert "compress_dynamo_documents" not in cached_document
     assert "use_v2_feature_versioning" not in cached_document
 
-    # 5. SCHEMA VALIDATION: Prove the dictionary perfectly matches the TypedDict 
+    # 5. SCHEMA VALIDATION: Prove the dictionary perfectly matches the TypedDict
     # required by the API. If this raises an error, the fix failed.
     clean_doc: V1EnvironmentDocumentResponse = {
         "api_key": cached_document["api_key"],