feat: implement usage of the requests cacher

This commit is contained in:
Benex254
2024-09-29 20:40:14 +03:00
parent 1427a3193c
commit ba4c41d888

View File

@@ -1,3 +1,5 @@
import os
import requests
from yt_dlp.utils.networking import random_user_agent
@@ -8,6 +10,17 @@ class AnimeProvider:
USER_AGENT = random_user_agent()
HEADERS = {}
def __init__(self) -> None:
self.session = requests.session()
def __init__(
self, cache_requests=os.environ.get("FASTANIME_CACHE_REQUESTS", "true")
) -> None:
if cache_requests.lower() == "true":
from ...constants import APP_CACHE_DIR
from ..common.requests_cacher import CachedRequestsSession
self.session = CachedRequestsSession(
os.path.join(APP_CACHE_DIR, "cached_requests.db"),
os.path.join(APP_CACHE_DIR, "cached_requests.lock"),
)
else:
self.session = requests.session()
self.session.headers.update({"User-Agent": self.USER_AGENT, **self.HEADERS})