mirror of
https://github.com/guezoloic/millesima_projetS6.git
synced 2026-03-28 19:13:42 +00:00
ajout(main.py): meilleur systeme de cache
This commit is contained in:
61
main.py
61
main.py
@@ -2,8 +2,10 @@ from sys import stderr
|
|||||||
from typing import cast
|
from typing import cast
|
||||||
from requests import Response, Session
|
from requests import Response, Session
|
||||||
from bs4 import BeautifulSoup, Tag
|
from bs4 import BeautifulSoup, Tag
|
||||||
|
from collections import OrderedDict
|
||||||
from json import JSONDecodeError, loads
|
from json import JSONDecodeError, loads
|
||||||
|
|
||||||
|
|
||||||
class ScraperData:
|
class ScraperData:
|
||||||
def __init__(self, data: dict[str, object]) -> None:
|
def __init__(self, data: dict[str, object]) -> None:
|
||||||
if not data:
|
if not data:
|
||||||
@@ -33,7 +35,12 @@ class ScraperData:
|
|||||||
dict[str, object], current_value.get(name)
|
dict[str, object], current_value.get(name)
|
||||||
)
|
)
|
||||||
if app_dict:
|
if app_dict:
|
||||||
return cast(str, app_dict.get("valueId")).rstrip("+")
|
val: list[str] = (
|
||||||
|
cast(str, app_dict.get("valueId")).rstrip("+").split("-")
|
||||||
|
)
|
||||||
|
if len(val) > 1:
|
||||||
|
val[0] = str((int(val[0]) + int(val[1])) / 2)
|
||||||
|
return val[0]
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def parker(self) -> str | None:
|
def parker(self) -> str | None:
|
||||||
@@ -48,6 +55,7 @@ class ScraperData:
|
|||||||
def getdata(self) -> dict[str, object]:
|
def getdata(self) -> dict[str, object]:
|
||||||
return self._data
|
return self._data
|
||||||
|
|
||||||
|
|
||||||
class Scraper:
|
class Scraper:
|
||||||
"""
|
"""
|
||||||
Scraper est une classe qui permet de gerer
|
Scraper est une classe qui permet de gerer
|
||||||
@@ -64,8 +72,10 @@ class Scraper:
|
|||||||
# TCP et d'avoir toujours une connexion constante avec le server
|
# TCP et d'avoir toujours une connexion constante avec le server
|
||||||
self._session: Session = Session()
|
self._session: Session = Session()
|
||||||
# Système de cache pour éviter de solliciter le serveur inutilement
|
# Système de cache pour éviter de solliciter le serveur inutilement
|
||||||
self._latest_request: tuple[(str, Response | None)] = ("", None)
|
self._latest_request: tuple[(str, Response)] | None = None
|
||||||
self._latest_soup: tuple[(str, BeautifulSoup | None)] = ("", None)
|
self._latest_soups: OrderedDict[str, BeautifulSoup] = OrderedDict[
|
||||||
|
str, BeautifulSoup
|
||||||
|
]()
|
||||||
|
|
||||||
def _request(self, subdir: str) -> Response:
|
def _request(self, subdir: str) -> Response:
|
||||||
"""
|
"""
|
||||||
@@ -85,12 +95,14 @@ class Scraper:
|
|||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
return response
|
return response
|
||||||
|
|
||||||
def getresponse(self, subdir: str = "") -> Response:
|
def getresponse(self, subdir: str = "", use_cache: bool = True) -> Response:
|
||||||
"""
|
"""
|
||||||
Récupère la réponse d'une page, en utilisant le cache si possible.
|
Récupère la réponse d'une page, en utilisant le cache si possible.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
subdir (str, optional): Le chemin de la page.
|
subdir (str, optional): Le chemin de la page.
|
||||||
|
use_cache (bool, optional): Utilise la donnée deja sauvegarder ou
|
||||||
|
écrase la donnée utilisé avec la nouvelle
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
Response: L'objet réponse (cache ou nouvelle requête).
|
Response: L'objet réponse (cache ou nouvelle requête).
|
||||||
@@ -98,16 +110,24 @@ class Scraper:
|
|||||||
Raise:
|
Raise:
|
||||||
HTTPError: Si le serveur renvoie un code d'erreur (4xx, 5xx).
|
HTTPError: Si le serveur renvoie un code d'erreur (4xx, 5xx).
|
||||||
"""
|
"""
|
||||||
rq_subdir, rq_response = self._latest_request
|
|
||||||
|
|
||||||
if rq_response is not None and subdir == rq_subdir:
|
# si dans le cache, latest_request existe
|
||||||
return rq_response
|
if use_cache and self._latest_request is not None:
|
||||||
|
rq_subdir, rq_response = self._latest_request
|
||||||
|
|
||||||
|
# si c'est la meme requete et que use_cache est true,
|
||||||
|
# on renvoie celle enregistrer
|
||||||
|
if subdir == rq_subdir:
|
||||||
|
return rq_response
|
||||||
|
|
||||||
request: Response = self._request(subdir)
|
request: Response = self._request(subdir)
|
||||||
self._latest_request = (subdir, request)
|
# on recrée la structure pour le systeme de cache si activer
|
||||||
|
if use_cache:
|
||||||
|
self._latest_request = (subdir, request)
|
||||||
|
|
||||||
return request
|
return request
|
||||||
|
|
||||||
def getsoup(self, subdir: str = "") -> BeautifulSoup:
|
def getsoup(self, subdir: str = "", use_cache: bool = True) -> BeautifulSoup:
|
||||||
"""
|
"""
|
||||||
Récupère le contenu HTML d'une page et le transforme en objet BeautifulSoup.
|
Récupère le contenu HTML d'une page et le transforme en objet BeautifulSoup.
|
||||||
|
|
||||||
@@ -120,16 +140,19 @@ class Scraper:
|
|||||||
Raise:
|
Raise:
|
||||||
HTTPError: Si le serveur renvoie un code d'erreur (4xx, 5xx).
|
HTTPError: Si le serveur renvoie un code d'erreur (4xx, 5xx).
|
||||||
"""
|
"""
|
||||||
rq_subdir, rq_soup = self._latest_soup
|
|
||||||
|
|
||||||
if rq_soup is not None and subdir == rq_subdir:
|
if use_cache and subdir in self._latest_soups:
|
||||||
return rq_soup
|
return self._latest_soups[subdir]
|
||||||
|
|
||||||
soup: BeautifulSoup = BeautifulSoup(
|
markup: str = self.getresponse(subdir).text
|
||||||
markup=self.getresponse(subdir).text, features="html.parser"
|
soup: BeautifulSoup = BeautifulSoup(markup, features="html.parser")
|
||||||
)
|
|
||||||
|
if use_cache:
|
||||||
|
self._latest_soups[subdir] = soup
|
||||||
|
|
||||||
|
if len(self._latest_soups) > 10:
|
||||||
|
_ = self._latest_soups.popitem(last=False)
|
||||||
|
|
||||||
self._latest_soup = (subdir, soup)
|
|
||||||
return soup
|
return soup
|
||||||
|
|
||||||
def getjsondata(self, subdir: str = "", id: str = "__NEXT_DATA__") -> ScraperData:
|
def getjsondata(self, subdir: str = "", id: str = "__NEXT_DATA__") -> ScraperData:
|
||||||
@@ -182,7 +205,5 @@ class Scraper:
|
|||||||
print(f"Erreur lors de l'extraction JSON : {e}", file=stderr)
|
print(f"Erreur lors de l'extraction JSON : {e}", file=stderr)
|
||||||
return ScraperData({})
|
return ScraperData({})
|
||||||
|
|
||||||
# file = Scraper().getjsondata("/chateau-gloria-2016.html")
|
|
||||||
# print("parker: ", file.parker())
|
# print(Scraper().getjsondata("bordeaux.html?page=1").getdata())
|
||||||
# print("robinson: ", file.robinson())
|
|
||||||
# print("suckling: ", file.suckling())
|
|
||||||
|
|||||||
28
test_main.py
28
test_main.py
@@ -41,7 +41,7 @@ def mock_site():
|
|||||||
"name": "En promotion",
|
"name": "En promotion",
|
||||||
"value": "Non",
|
"value": "Non",
|
||||||
"sequence": 80,
|
"sequence": 80,
|
||||||
"displayable": "false",
|
"displayable": "False",
|
||||||
"type": "CHECKBOX",
|
"type": "CHECKBOX",
|
||||||
"isSpirit": False,
|
"isSpirit": False,
|
||||||
},
|
},
|
||||||
@@ -73,6 +73,24 @@ def mock_site():
|
|||||||
"isSpirit": False,
|
"isSpirit": False,
|
||||||
"groupIdentifier": "appellation_433",
|
"groupIdentifier": "appellation_433",
|
||||||
},
|
},
|
||||||
|
"note_rp": {
|
||||||
|
"valueId": "91",
|
||||||
|
"name": "Parker",
|
||||||
|
"value": "91",
|
||||||
|
"isSpirit": False,
|
||||||
|
},
|
||||||
|
"note_jr": {
|
||||||
|
"valueId": "17",
|
||||||
|
"name": "J. Robinson",
|
||||||
|
"value": "17",
|
||||||
|
"isSpirit": False,
|
||||||
|
},
|
||||||
|
"note_js": {
|
||||||
|
"valueId": "93-94",
|
||||||
|
"name": "J. Suckling",
|
||||||
|
"value": "93-94",
|
||||||
|
"isSpirit": False
|
||||||
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -123,3 +141,11 @@ def test_appellation(scraper: Scraper):
|
|||||||
"nino-negri-5-stelle-sfursat-2022.html"
|
"nino-negri-5-stelle-sfursat-2022.html"
|
||||||
)
|
)
|
||||||
assert appellation.appellation() == "Sforzato di Valtellina"
|
assert appellation.appellation() == "Sforzato di Valtellina"
|
||||||
|
|
||||||
|
def test_critiques(scraper: Scraper):
|
||||||
|
critiques: ScraperData = scraper.getjsondata(
|
||||||
|
"nino-negri-5-stelle-sfursat-2022.html"
|
||||||
|
)
|
||||||
|
assert critiques.parker() == "91"
|
||||||
|
assert critiques.robinson() == "17"
|
||||||
|
assert critiques.suckling() == "93.5"
|
||||||
Reference in New Issue
Block a user