laramies#1383 fix securitytrailssearch.py

theHarvester/discovery/securitytrailssearch.py

@@ -1,4 +1,5 @@
 import asyncio
+from typing import Sequence
 
 from theHarvester.discovery.constants import *
 from theHarvester.lib.core import *
@@ -14,7 +15,7 @@ def __init__(self, word) -> None:
         self.results = ""
         self.totalresults = ""
         self.api = "https://api.securitytrails.com/v1/"
-        self.info = ()
+        self.info: tuple[set, set] = (set(), set())
         self.proxy = False
 
     async def authenticate(self) -> None:

theHarvester/parsers/securitytrailsparser.py

@@ -1,23 +1,22 @@
-from typing import Union, Tuple, List, Set
+from typing import List, Set, Tuple, Union
 
 
 class Parser:
-
     def __init__(self, word, text) -> None:
         self.word = word
         self.text = text
         self.hostnames: Set = set()
         self.ips: Set = set()
 
-    async def parse_text(self) -> Union[List, Tuple]:
+    async def parse_text(self) -> tuple[set, set]:
         sub_domain_flag = 0
         self.text = str(self.text).splitlines()
         # Split lines to get a list of lines.
         for index in range(0, len(self.text)):
             line = self.text[index].strip()
             if '"ip":' in line:
                 # Extract IP.
-                ip = ''
+                ip = ""
                 for ch in line[7:]:
                     if ch == '"':
                         break
@@ -29,13 +28,17 @@ async def parse_text(self) -> Union[List, Tuple]:
                 sub_domain_flag = 1
                 continue
             elif sub_domain_flag > 0:
-                if ']' in line:
+                if "]" in line:
                     sub_domain_flag = 0
                 else:
-                    if 'www' in self.word:
-                        self.word = str(self.word).replace('www.', '').replace('www', '')
+                    if "www" in self.word:
+                        self.word = (
+                            str(self.word).replace("www.", "").replace("www", "")
+                        )
                     # Remove www from word if entered
-                    self.hostnames.add(str(line).replace('"', '').replace(',', '') + '.' + self.word)
+                    self.hostnames.add(
+                        str(line).replace('"', "").replace(",", "") + "." + self.word
+                    )
             else:
                 continue
-        return list(self.ips), list(self.hostnames)
+        return self.ips, self.hostnames