Fix restyle

Author: raul-marquez-csa

src/python_testing/TC_SC_4_3.py

@@ -20,7 +20,7 @@
 
 import chip.clusters as Clusters
 from matter_testing_support import MatterBaseTest, async_test_body, default_matter_test_main
-from mdns_discovery.mdns_discovery import MdnsDiscovery, MdnsServiceType, DNSRecordType
+from mdns_discovery.mdns_discovery import DNSRecordType, MdnsDiscovery, MdnsServiceType
 from mobly import asserts
 
 '''
@@ -179,7 +179,6 @@ async def test_TC_SC_4_3(self):
             log_output=True,
             load_from_cache=False
         )
-        
 
         # # *** STEP 9 ***
         # self.print_step("9", "TH verifies ICD, SII, SAI, SAT, and T TXT record keys/vales of the returned record.")
@@ -261,11 +260,9 @@ async def test_TC_SC_4_3(self):
         # asserts.assert_in(op_sub_type, service_types, f"No PTR record with DNS-SD instance name '{op_sub_type}'")
 
         print("\n"*10)
-        
+
         # input()
 
 
 if __name__ == "__main__":
     default_matter_test_main()
-
-    

src/python_testing/mdns_discovery/mdns_async_service_info.py

@@ -21,34 +21,16 @@
 from ipaddress import IPv4Address, IPv6Address
 from random import randint
 from typing import TYPE_CHECKING, Dict, List, Optional, Set, Union
+
 from zeroconf import BadTypeInNameException, DNSQuestionType, ServiceInfo, Zeroconf, current_time_millis
-from zeroconf.const import (
-    _DNS_HOST_TTL,
-    _DNS_OTHER_TTL,
-    _LISTENER_TIME,
-    _MDNS_PORT,
-    _DUPLICATE_QUESTION_INTERVAL,
-    _FLAGS_QR_QUERY,
-    _CLASS_IN,
-    _TYPE_A,
-    _TYPE_AAAA,
-    _TYPE_SRV,
-    _TYPE_TXT
-    )
-from zeroconf._dns import (
-    DNSQuestion,
-    DNSAddress,
-    DNSPointer,
-    DNSQuestionType,
-    DNSRecord,
-    DNSService,
-    DNSText
-)
+from zeroconf._cache import DNSCache
+from zeroconf._dns import DNSQuestion, DNSAddress, DNSPointer, DNSQuestionType, DNSRecord, DNSService, DNSText
+from zeroconf._history import QuestionHistory
 from zeroconf._protocol.outgoing import DNSOutgoing
 from zeroconf._utils.name import service_type_name
 from zeroconf._utils.net import _encode_address
-from zeroconf._cache import DNSCache
-from zeroconf._history import QuestionHistory
+from zeroconf.const import (_DNS_HOST_TTL, _DNS_OTHER_TTL, _LISTENER_TIME,
+                            _MDNS_PORT, _DUPLICATE_QUESTION_INTERVAL, _FLAGS_QR_QUERY, _CLASS_IN, _TYPE_A, _TYPE_AAAA, _TYPE_SRV, _TYPE_TXT)
 
 
 int_ = int
@@ -60,6 +42,7 @@
 QM_QUESTION = DNSQuestionType.QM
 _AVOID_SYNC_DELAY_RANDOM_INTERVAL = (20, 120)
 
+
 @enum.unique
 class DNSRecordType(enum.Enum):
     """An MDNS record type.

src/python_testing/mdns_discovery/mdns_discovery.py

@@ -23,10 +23,10 @@
 from time import sleep
 from typing import Dict, List, Optional
 
+from mdns_discovery.mdns_async_service_info import DNSRecordType, MdnsAsyncServiceInfo
 from zeroconf import IPVersion, ServiceListener, ServiceStateChange, Zeroconf
 from zeroconf.asyncio import AsyncServiceBrowser, AsyncServiceInfo, AsyncZeroconfServiceTypes
 
-from mdns_discovery.mdns_async_service_info import DNSRecordType, MdnsAsyncServiceInfo
 
 @dataclass
 class MdnsServiceInfo:
@@ -226,12 +226,12 @@ async def get_service_types(self, log_output: bool = False) -> List[str]:
         return discovered_services
 
     async def get_service_by_record_type(self, service_name: str,
-                                      service_type: str,
-                                      record_type: DNSRecordType,
-                                      load_from_cache: bool = True,
-                                      discovery_timeout_sec: float = DISCOVERY_TIMEOUT_SEC,
-                                      log_output: bool = False
-                                      ) -> Optional[MdnsServiceInfo]:
+                                         service_type: str,
+                                         record_type: DNSRecordType,
+                                         load_from_cache: bool = True,
+                                         discovery_timeout_sec: float = DISCOVERY_TIMEOUT_SEC,
+                                         log_output: bool = False
+                                         ) -> Optional[MdnsServiceInfo]:
         """
         Asynchronously discovers an mDNS service within the network by service name, service type,
         and record type.