Fixed page size for ZoomEye

Xewdy444 · Apr 21, 2024 · 3b48c24 · 3b48c24
1 parent fc8c28c
commit 3b48c24
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/utils/utils.py b/utils/utils.py
@@ -104,8 +104,7 @@ def __len__(self) -> int:
         return len(self._tasks)
 
     def __iter__(self) -> Iterable[asyncio.Task[Any]]:
-        for task in self._tasks:
-            yield task
+        yield from self._tasks
 
     async def __aenter__(self) -> CoroutineExecutor:
         return self

diff --git a/utils/zoomeye.py b/utils/zoomeye.py
@@ -9,6 +9,8 @@
 
 import aiohttp
 
+USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36"
+
 
 @dataclass
 class ZoomEyeCredentials:
@@ -36,10 +38,7 @@ def __init__(self, credentials: ZoomEyeCredentials) -> None:
         self._session = aiohttp.ClientSession()
 
         self._session.headers.update(
-            {
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
-                "API-KEY": credentials.api_key,
-            }
+            {"User-Agent": USER_AGENT, "API-KEY": credentials.api_key}
         )
 
     def __repr__(self) -> str:
@@ -102,7 +101,7 @@ async def get_hosts(self, query: str, *, count: int = 500) -> List[Tuple[str, in
         """
         tasks = [
             asyncio.create_task(self.search(query, page=page))
-            for page in range(1, math.ceil(count / 20) + 1)
+            for page in range(1, math.ceil(count / 10) + 1)
         ]
 
         results = await asyncio.gather(*tasks)