22# --------------------------------------------------------------------------
33# Copyright (c) Microsoft Corporation. All rights reserved.
44# Licensed under the MIT License. See License.txt in the project root for license information.
5- # Code generated by Microsoft (R) AutoRest Code Generator.
5+ # Code generated by Microsoft (R) Python Code Generator.
66# Changes may cause incorrect behavior and will be lost if the code is regenerated.
77# --------------------------------------------------------------------------
8+ from collections .abc import MutableMapping
89from io import IOBase
9- import sys
10+ import json
1011from typing import Any , Callable , Dict , IO , List , Optional , TypeVar , Union , overload
1112
13+ from azure .core import PipelineClient
1214from azure .core .exceptions import (
1315 ClientAuthenticationError ,
1416 HttpResponseError ,
2224from azure .core .tracing .decorator import distributed_trace
2325from azure .core .utils import case_insensitive_dict
2426
25- from .._serialization import Serializer
26- from .._vendor import LogsIngestionClientMixinABC
27+ from .._configuration import LogsIngestionClientConfiguration
28+ from .._utils .model_base import SdkJSONEncoder
29+ from .._utils .serialization import Serializer
30+ from .._utils .utils import ClientMixinABC
2731
28- if sys .version_info >= (3 , 9 ):
29- from collections .abc import MutableMapping
30- else :
31- from typing import MutableMapping # type: ignore
32- JSON = MutableMapping [str , Any ] # pylint: disable=unsubscriptable-object
3332T = TypeVar ("T" )
3433ClsType = Optional [Callable [[PipelineResponse [HttpRequest , HttpResponse ], T , Dict [str , Any ]], Any ]]
3534
3837
3938
4039def build_logs_ingestion_upload_request (
41- rule_id : str , stream : str , * , content_encoding : Optional [str ] = None , ** kwargs : Any
40+ rule_id : str , stream_name : str , * , content_encoding : Optional [str ] = None , ** kwargs : Any
4241) -> HttpRequest :
4342 _headers = case_insensitive_dict (kwargs .pop ("headers" , {}) or {})
4443 _params = case_insensitive_dict (kwargs .pop ("params" , {}) or {})
@@ -51,7 +50,7 @@ def build_logs_ingestion_upload_request(
5150 _url = "/dataCollectionRules/{ruleId}/streams/{stream}"
5251 path_format_arguments = {
5352 "ruleId" : _SERIALIZER .url ("rule_id" , rule_id , "str" ),
54- "stream" : _SERIALIZER .url ("stream " , stream , "str" ),
53+ "stream" : _SERIALIZER .url ("stream_name " , stream_name , "str" ),
5554 }
5655
5756 _url : str = _url .format (** path_format_arguments ) # type: ignore
@@ -69,14 +68,16 @@ def build_logs_ingestion_upload_request(
6968 return HttpRequest (method = "POST" , url = _url , params = _params , headers = _headers , ** kwargs )
7069
7170
72- class LogsIngestionClientOperationsMixin (LogsIngestionClientMixinABC ):
71+ class LogsIngestionClientOperationsMixin (
72+ ClientMixinABC [PipelineClient [HttpRequest , HttpResponse ], LogsIngestionClientConfiguration ]
73+ ):
7374
7475 @overload
7576 def _upload (
7677 self ,
7778 rule_id : str ,
78- stream : str ,
79- body : List [JSON ],
79+ stream_name : str ,
80+ body : List [Dict [ str , Any ] ],
8081 * ,
8182 content_encoding : Optional [str ] = None ,
8283 content_type : str = "application/json" ,
@@ -86,7 +87,7 @@ def _upload(
8687 def _upload (
8788 self ,
8889 rule_id : str ,
89- stream : str ,
90+ stream_name : str ,
9091 body : IO [bytes ],
9192 * ,
9293 content_encoding : Optional [str ] = None ,
@@ -98,24 +99,26 @@ def _upload(
9899 def _upload ( # pylint: disable=inconsistent-return-statements
99100 self ,
100101 rule_id : str ,
101- stream : str ,
102- body : Union [List [JSON ], IO [bytes ]],
102+ stream_name : str ,
103+ body : Union [List [Dict [ str , Any ] ], IO [bytes ]],
103104 * ,
104105 content_encoding : Optional [str ] = None ,
105106 ** kwargs : Any
106107 ) -> None :
107108 """Ingestion API used to directly ingest data using Data Collection Rules.
108109
109- See error response code and error response message for more detail .
110+ Ingestion API used to directly ingest data using Data Collection Rules .
110111
111- :param rule_id: The immutable Id of the Data Collection Rule resource. Required.
112+ :param rule_id: The immutable ID of the Data Collection Rule resource. Required.
112113 :type rule_id: str
113- :param stream: The streamDeclaration name as defined in the Data Collection Rule. Required.
114- :type stream: str
115- :param body: An array of objects matching the schema defined by the provided stream. Is either
116- a [JSON] type or a IO[bytes] type. Required.
117- :type body: list[JSON] or IO[bytes]
118- :keyword content_encoding: gzip. Default value is None.
114+ :param stream_name: The streamDeclaration name as defined in the Data Collection Rule.
115+ Required.
116+ :type stream_name: str
117+ :param body: The array of objects matching the schema defined by the provided stream. Is either
118+ a [{str: Any}] type or a IO[bytes] type. Required.
119+ :type body: list[dict[str, any]] or IO[bytes]
120+ :keyword content_encoding: The content encoding of the request body which is always 'gzip'.
121+ Default value is None.
119122 :paramtype content_encoding: str
120123 :return: None
121124 :rtype: None
@@ -136,20 +139,18 @@ def _upload( # pylint: disable=inconsistent-return-statements
136139 cls : ClsType [None ] = kwargs .pop ("cls" , None )
137140
138141 content_type = content_type or "application/json"
139- _json = None
140142 _content = None
141143 if isinstance (body , (IOBase , bytes )):
142144 _content = body
143145 else :
144- _json = body
146+ _content = json . dumps ( body , cls = SdkJSONEncoder , exclude_readonly = True ) # type: ignore
145147
146148 _request = build_logs_ingestion_upload_request (
147149 rule_id = rule_id ,
148- stream = stream ,
150+ stream_name = stream_name ,
149151 content_encoding = content_encoding ,
150152 content_type = content_type ,
151153 api_version = self ._config .api_version ,
152- json = _json ,
153154 content = _content ,
154155 headers = _headers ,
155156 params = _params ,
0 commit comments