@@ -184,6 +184,9 @@ def _generate_xml(self, datasource_item: DatasourceItem, connection_credentials=
184184 project_element = ET .SubElement (datasource_element , "project" )
185185 project_element .attrib ["id" ] = datasource_item .project_id
186186
187+ if datasource_item .description is not None :
188+ datasource_element .attrib ["description" ] = datasource_item .description
189+
187190 if connection_credentials is not None and connections is not None :
188191 raise RuntimeError ("You cannot set both `connections` and `connection_credentials`" )
189192
@@ -196,7 +199,7 @@ def _generate_xml(self, datasource_item: DatasourceItem, connection_credentials=
196199 _add_connections_element (connections_element , connection )
197200 return ET .tostring (xml_request )
198201
199- def update_req (self , datasource_item ) :
202+ def update_req (self , datasource_item : DatasourceItem ) -> bytes :
200203 xml_request = ET .Element ("tsRequest" )
201204 datasource_element = ET .SubElement (xml_request , "datasource" )
202205 if datasource_item .name :
@@ -219,6 +222,8 @@ def update_req(self, datasource_item):
219222 datasource_element .attrib ["certificationNote" ] = str (datasource_item .certification_note )
220223 if datasource_item .encrypt_extracts is not None :
221224 datasource_element .attrib ["encryptExtracts" ] = str (datasource_item .encrypt_extracts ).lower ()
225+ if datasource_item .description is not None :
226+ datasource_element .attrib ["description" ] = datasource_item .description
222227
223228 return ET .tostring (xml_request )
224229
0 commit comments