From 37f8d84a9c1fcadeee684f08ae8d3036f5a82213 Mon Sep 17 00:00:00 2001 From: I-Al-Istannen Date: Sun, 23 May 2021 19:00:01 +0200 Subject: [PATCH] Output total amount of http requests in HTTP Crawler --- PFERD/http_crawler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/PFERD/http_crawler.py b/PFERD/http_crawler.py index adbac5d..c6e679d 100644 --- a/PFERD/http_crawler.py +++ b/PFERD/http_crawler.py @@ -27,6 +27,7 @@ class HttpCrawler(Crawler): self._authentication_id = 0 self._authentication_lock = asyncio.Lock() self._current_cookie_jar: Optional[aiohttp.CookieJar] = None + self._request_count = 0 async def _current_auth_id(self) -> int: """ @@ -41,6 +42,7 @@ class HttpCrawler(Crawler): # This should reduce the amount of requests we make: If an authentication is in progress # all future requests wait for authentication to complete. async with self._authentication_lock: + self._request_count += 1 return self._authentication_id async def authenticate(self, caller_auth_id: int) -> None: @@ -85,6 +87,7 @@ class HttpCrawler(Crawler): async def run(self) -> None: self._current_cookie_jar = aiohttp.CookieJar() + self._request_count = 0 try: self._current_cookie_jar.load(self._cookie_jar_path) @@ -100,6 +103,7 @@ class HttpCrawler(Crawler): await super().run() finally: del self.session + log.explain_topic(f"Total amount of HTTP requests: {self._request_count}") # They are saved in authenticate, but a final save won't hurt await self._save_cookies()