Fix alias placement: Move _InternalOptions and _Kwargs assignments after all imports to comply with PEP 8 guidelines

Co-authored-by: bambriz <[email protected]>

Author: Copilot

sdk/cosmos/azure-cosmos/azure/cosmos/_base.py

@@ -39,9 +39,6 @@
 from . import http_constants
 from . import _runtime_constants
 from ._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from .auth import _get_authorization_header
 from .offer import ThroughputProperties
 from .partition_key import _Empty, _Undefined
@@ -51,6 +48,10 @@
     from .aio._cosmos_client_connection_async import CosmosClientConnection as AsyncClientConnection
     from ._request_object import RequestObject
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 # InternalOptions alias for cleaner access to internal option constants
 InternalOptions = Constants.InternalOptions
 

sdk/cosmos/azure-cosmos/azure/cosmos/_cosmos_client_connection.py

@@ -64,9 +64,6 @@
 from ._change_feed.change_feed_state import ChangeFeedState
 from ._change_feed.feed_range_internal import FeedRangeInternalEpk
 from ._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from ._cosmos_http_logging_policy import CosmosHttpLoggingPolicy
 from ._cosmos_responses import CosmosDict, CosmosList
 from ._range_partition_resolver import RangePartitionResolver
@@ -84,6 +81,10 @@
     _return_undefined_or_empty_partition_key,
 )
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 class CredentialDict(TypedDict, total=False):
     masterKey: str
     resourceTokens: Mapping[str, Any]

sdk/cosmos/azure-cosmos/azure/cosmos/_global_partition_endpoint_manager_circuit_breaker.py

@@ -24,9 +24,6 @@
 from typing import TYPE_CHECKING, Optional
 
 from azure.cosmos._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from azure.cosmos.partition_key import _get_partition_key_from_partition_key_definition
 from azure.cosmos._global_partition_endpoint_manager_circuit_breaker_core import \
     _GlobalPartitionEndpointManagerForCircuitBreakerCore
@@ -39,6 +36,10 @@
 if TYPE_CHECKING:
     from azure.cosmos._cosmos_client_connection import CosmosClientConnection
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 class _GlobalPartitionEndpointManagerForCircuitBreaker(_GlobalEndpointManager):
     """
     This internal class implements the logic for partition endpoint management for

sdk/cosmos/azure-cosmos/azure/cosmos/_request_object.py

@@ -25,6 +25,7 @@
 from .documents import _OperationType
 from .http_constants import ResourceType
 from ._constants import _Constants as Constants
+
 # Shorter aliases for convenient usage
 _InternalOptions = Constants.InternalOptions
 _Kwargs = Constants.Kwargs

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_container.py

@@ -47,9 +47,6 @@
 from .._change_feed.feed_range_internal import FeedRangeInternalEpk
 from .._cosmos_responses import CosmosDict, CosmosList
 from .._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from .._routing.routing_range import Range
 from .._session_token_helpers import get_latest_session_token
 from ..offer import ThroughputProperties
@@ -58,6 +55,11 @@
     _return_undefined_or_empty_partition_key,
     _get_partition_key_from_partition_key_definition, NullPartitionKeyValue, _PartitionKeyType
 )
+
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 __all__ = ("ContainerProxy",)
 
 # pylint: disable=protected-access, too-many-lines

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_cosmos_client.py

@@ -35,16 +35,17 @@
 
 from ..cosmos_client import _parse_connection_str
 from .._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from ._cosmos_client_connection_async import CosmosClientConnection, CredentialDict
 from .._base import build_options as _build_options, _set_throughput_options
 from ._retry_utility_async import _ConnectionRetryPolicy
 from ._database import DatabaseProxy, _get_database_link
 from ..documents import ConnectionPolicy, DatabaseAccount
 from ..exceptions import CosmosResourceNotFoundError
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 # pylint: disable=docstring-keyword-should-match-keyword-only
 
 __all__ = ("CosmosClient",)

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_cosmos_client_connection_async.py

@@ -61,9 +61,6 @@
 from .._routing import routing_range
 from ..documents import ConnectionPolicy, DatabaseAccount
 from .._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from .._cosmos_responses import CosmosDict, CosmosList
 from .. import http_constants, exceptions
 from . import _query_iterable_async as query_iterable
@@ -86,6 +83,10 @@
 from .._cosmos_http_logging_policy import CosmosHttpLoggingPolicy
 from .._range_partition_resolver import RangePartitionResolver
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 
 
 

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_database.py

@@ -37,9 +37,6 @@
     _replace_throughput
 )
 from .._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from ._container import ContainerProxy
 from ..offer import ThroughputProperties
 from ..http_constants import StatusCodes
@@ -48,6 +45,10 @@
 from ..documents import IndexingMode
 from ..partition_key import PartitionKey
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 
 __all__ = ("DatabaseProxy",)
 

sdk/cosmos/azure-cosmos/azure/cosmos/container.py

@@ -44,9 +44,6 @@
 )
 from ._change_feed.feed_range_internal import FeedRangeInternalEpk
 from ._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from ._cosmos_client_connection import CosmosClientConnection
 from ._cosmos_responses import CosmosDict, CosmosList
 from ._routing.routing_range import Range
@@ -62,6 +59,10 @@
 )
 from .scripts import ScriptsProxy
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 __all__ = ("ContainerProxy",)
 
 # pylint: disable=too-many-lines,disable=protected-access

sdk/cosmos/azure-cosmos/azure/cosmos/cosmos_client.py

@@ -35,13 +35,14 @@
 from .offer import ThroughputProperties
 from ._retry_utility import ConnectionRetryPolicy
 from ._constants import _Constants as Constants
-# Shorter aliases for convenient usage
-_InternalOptions = Constants.InternalOptions
-_Kwargs = Constants.Kwargs
 from .database import DatabaseProxy, _get_database_link
 from .documents import ConnectionPolicy, DatabaseAccount
 from .exceptions import CosmosResourceNotFoundError
 
+# Shorter aliases for convenient usage
+_InternalOptions = Constants.InternalOptions
+_Kwargs = Constants.Kwargs
+
 
 __all__ = ("CosmosClient",)