mirrors/pferd
1
0
Fork 0
mirror of https://github.com/Garmelon/PFERD.git synced 2023-12-21 10:23:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

Output total amount of http requests in HTTP Crawler

Browse Source
This commit is contained in:
I-Al-Istannen 2021-05-23 19:00:01 +02:00
parent 5edd868d5b
commit 37f8d84a9c
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
PFERD/http_crawler.py
View File

@ -27,6 +27,7 @@ class HttpCrawler(Crawler):
self._authentication_id = 0 self._authentication_id = 0
self._authentication_lock = asyncio.Lock() self._authentication_lock = asyncio.Lock()
self._current_cookie_jar: Optional[aiohttp.CookieJar] = None self._current_cookie_jar: Optional[aiohttp.CookieJar] = None
self._request_count = 0
async def _current_auth_id(self) -> int: 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 # This should reduce the amount of requests we make: If an authentication is in progress
# all future requests wait for authentication to complete. # all future requests wait for authentication to complete.
async with self._authentication_lock: async with self._authentication_lock:
self._request_count += 1
return self._authentication_id return self._authentication_id
async def authenticate(self, caller_auth_id: int) -> None: async def authenticate(self, caller_auth_id: int) -> None:
@ -85,6 +87,7 @@ class HttpCrawler(Crawler):
async def run(self) -> None: async def run(self) -> None:
self._current_cookie_jar = aiohttp.CookieJar() self._current_cookie_jar = aiohttp.CookieJar()
self._request_count = 0
try: try:
self._current_cookie_jar.load(self._cookie_jar_path) self._current_cookie_jar.load(self._cookie_jar_path)
@ -100,6 +103,7 @@ class HttpCrawler(Crawler):
await super().run() await super().run()
finally: finally:
del self.session 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 # They are saved in authenticate, but a final save won't hurt
await self._save_cookies() await self._save_cookies()