mirror of
https://github.com/guezoloic/millesima_projetS6.git
synced 2026-03-28 19:13:42 +00:00
Compare commits
30 Commits
0d96ff7714
...
jalon3
| Author | SHA1 | Date | |
|---|---|---|---|
| 68dffa6486 | |||
| c7d2077b23 | |||
| 106877a073 | |||
|
|
416cfcbf8b | ||
| 32c5310e37 | |||
| 9dfc7457a0 | |||
| f5d5703e49 | |||
| 888defb6b6 | |||
| 734e3898e9 | |||
| 4bb3112dd0 | |||
| 54e4b7860b | |||
| b865a59aba | |||
|
|
fde1f36148 | ||
| 6fbb36ea37 | |||
| 4b3c3c26e8 | |||
|
|
bcacd7a915 | ||
| de1d325fb7 | |||
| f4ded6d8b5 | |||
| acf4ddd881 | |||
| 69b8b4ce1f | |||
| 8047b06253 | |||
| 5303d36988 | |||
|
|
d182e08f9b | ||
|
|
cefdb94dd5 | ||
|
|
06097c257e | ||
|
|
b0eb5df07e | ||
|
|
5afb6e38fe | ||
|
|
f31de22693 | ||
|
|
73c6221080 | ||
|
|
99dd71989d |
18
.github/dependabot.yml
vendored
Normal file
18
.github/dependabot.yml
vendored
Normal file
@@ -0,0 +1,18 @@
|
||||
# To get started with Dependabot version updates, you'll need to specify which
|
||||
# package ecosystems to update and where the package manifests are located.
|
||||
# Please see the documentation for all configuration options:
|
||||
# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
|
||||
|
||||
version: 2
|
||||
updates:
|
||||
- package-ecosystem: "pip"
|
||||
directory: "/"
|
||||
schedule:
|
||||
interval: "weekly"
|
||||
day: "saturday"
|
||||
open-pull-requests-limit: 5
|
||||
groups:
|
||||
python-dependencies:
|
||||
patterns:
|
||||
- "*"
|
||||
|
||||
13
.github/workflows/python-app.yml
vendored
13
.github/workflows/python-app.yml
vendored
@@ -19,15 +19,15 @@ jobs:
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Python 3.10
|
||||
- name: Set up Python 3.x
|
||||
uses: actions/setup-python@v4
|
||||
with:
|
||||
python-version: "3.10"
|
||||
python-version: "3.x"
|
||||
|
||||
- name: install dependencies
|
||||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install ".[test,doc]"
|
||||
pip install ".[test]"
|
||||
|
||||
- name: Lint with flake8
|
||||
run: |
|
||||
@@ -36,10 +36,3 @@ jobs:
|
||||
|
||||
- name: Test with pytest
|
||||
run: pytest
|
||||
|
||||
- name: Deploy Doc
|
||||
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
|
||||
run: |
|
||||
git config user.name github-actions
|
||||
git config user.email github-actions@github.com
|
||||
mkdocs gh-deploy --force
|
||||
|
||||
57
.github/workflows/static.yml
vendored
Normal file
57
.github/workflows/static.yml
vendored
Normal file
@@ -0,0 +1,57 @@
|
||||
# Simple workflow for deploying static content to GitHub Pages
|
||||
name: Deploy static content to Pages
|
||||
|
||||
on:
|
||||
# Runs on pushes targeting the default branch
|
||||
push:
|
||||
branches: ["main"]
|
||||
|
||||
# Allows you to run this workflow manually from the Actions tab
|
||||
workflow_dispatch:
|
||||
|
||||
# Sets permissions of the GITHUB_TOKEN to allow deployment to GitHub Pages
|
||||
permissions:
|
||||
contents: read
|
||||
pages: write
|
||||
id-token: write
|
||||
|
||||
# Allow only one concurrent deployment, skipping runs queued between the run in-progress and latest queued.
|
||||
# However, do NOT cancel in-progress runs as we want to allow these production deployments to complete.
|
||||
concurrency:
|
||||
group: "pages"
|
||||
cancel-in-progress: false
|
||||
|
||||
jobs:
|
||||
# Single deploy job since we're just deploying
|
||||
deploy:
|
||||
environment:
|
||||
name: github-pages
|
||||
url: ${{ steps.deployment.outputs.page_url }}
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Python 3.x
|
||||
uses: actions/setup-python@v5
|
||||
with:
|
||||
python-version: '3.x'
|
||||
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -e ".[doc]"
|
||||
|
||||
- name: Setup Pages
|
||||
uses: actions/configure-pages@v5
|
||||
|
||||
- name: Build Documentation
|
||||
run: mkdocs build
|
||||
- name: Upload artifact
|
||||
uses: actions/upload-pages-artifact@v3
|
||||
with:
|
||||
# Upload entire repository
|
||||
path: './site'
|
||||
- name: Deploy to GitHub Pages
|
||||
id: deployment
|
||||
uses: actions/deploy-pages@v4
|
||||
38
README.md
38
README.md
@@ -1 +1,37 @@
|
||||
# millesima_projetS6
|
||||
# Millesima AI Engine 🍷
|
||||
|
||||
> A **University of Paris-Est Créteil (UPEC)** Semester 6 project.
|
||||
|
||||
## Documentation
|
||||
- 🇫🇷 [Version Française](https://guezoloic.github.io/millesima-ai-engine)
|
||||
> note: only french version enabled for now.
|
||||
---
|
||||
|
||||
## Installation
|
||||
> Make sure you have **Python 3.10+** installed.
|
||||
|
||||
1. **Clone the repository:**
|
||||
```bash
|
||||
git clone https://github.com/votre-pseudo/millesima-ai-engine.git
|
||||
cd millesima-ai-engine
|
||||
```
|
||||
|
||||
2. **Set up a virtual environment:**
|
||||
```bash
|
||||
python3 -m venv .venv
|
||||
source .venv/bin/activate # Windows: .venv\Scripts\activate
|
||||
```
|
||||
|
||||
3. **Install dependencies:**
|
||||
```bash
|
||||
pip install -e .
|
||||
```
|
||||
|
||||
## Usage
|
||||
|
||||
### 1. Data Extraction (Scraping)
|
||||
To fetch the latest wine data from Millesima:
|
||||
```bash
|
||||
python3 src/scraper.py
|
||||
```
|
||||
> Note: that will take some time to fetch all data depending on the catalog size.
|
||||
17
docs/cleaning.md
Normal file
17
docs/cleaning.md
Normal file
@@ -0,0 +1,17 @@
|
||||
# Cleaning
|
||||
|
||||
## Sommaire
|
||||
[TOC]
|
||||
|
||||
---
|
||||
|
||||
## Classe `Cleaning`
|
||||
::: src.cleaning.Cleaning
|
||||
options:
|
||||
heading_level: 3
|
||||
members:
|
||||
- __init__
|
||||
- getVins
|
||||
- drop_empty_appellation
|
||||
- fill_missing_scores
|
||||
- encode_appellation
|
||||
@@ -1 +1,16 @@
|
||||
# Millesima
|
||||
# Millesima
|
||||
|
||||
L’objectif de ce projet est d’étudier, en utilisant des méthodes d’apprentissage automatique, l’impact de différents critères (notes des critiques, appelation) sur le prix d’un vin. Pour ce faire, on s’appuiera sur le site Millesima (https://www.millesima.fr/), qui a l’avantage de ne pas posséder de protection contre les bots. Par respect pour l’hébergeur du site, on veillera à limiter au maximum le nombre de requêtes. En particulier, on s’assurera d’avoir un code fonctionnel avant de scraper l’intégralité du site, pour éviter les répétitions.
|
||||
|
||||
## projet
|
||||
<div style="text-align: center;">
|
||||
<object
|
||||
data="/millesima-ai-engine/projet.pdf"
|
||||
type="application/pdf"
|
||||
width="100%"
|
||||
height="1000px"
|
||||
>
|
||||
<p>Votre navigateur ne peut pas afficher ce PDF.
|
||||
<a href="/millesima-ai-engine/projet.pdf">Cliquez ici pour le télécharger.</a></p>
|
||||
</object>
|
||||
</div>
|
||||
Binary file not shown.
@@ -1,3 +1,31 @@
|
||||
# Scraper
|
||||
|
||||
::: scraper.Scraper
|
||||
## Sommaire
|
||||
[TOC]
|
||||
|
||||
---
|
||||
|
||||
## Classe `Scraper`
|
||||
::: scraper.Scraper
|
||||
options:
|
||||
members:
|
||||
- __init__
|
||||
- getvins
|
||||
- getjsondata
|
||||
- getresponse
|
||||
- getsoup
|
||||
heading_level: 4
|
||||
|
||||
## Classe `_ScraperData`
|
||||
::: scraper._ScraperData
|
||||
options:
|
||||
members:
|
||||
- __init__
|
||||
- getdata
|
||||
- appellation
|
||||
- parker
|
||||
- robinson
|
||||
- suckling
|
||||
- prix
|
||||
- informations
|
||||
heading_level: 4
|
||||
@@ -1,4 +0,0 @@
|
||||
|
||||
# _ScraperData
|
||||
|
||||
::: scraper._ScraperData
|
||||
387
learning.ipynb
Normal file
387
learning.ipynb
Normal file
File diff suppressed because one or more lines are too long
@@ -1,4 +1,5 @@
|
||||
site_name: "Projet Millesima S6"
|
||||
site_url: "https://github.guezoloic.com/millesima-ai-engine/"
|
||||
|
||||
theme:
|
||||
name: "material"
|
||||
@@ -7,6 +8,11 @@ plugins:
|
||||
- search
|
||||
- mkdocstrings
|
||||
|
||||
extra:
|
||||
generator: false
|
||||
|
||||
copyright: "Loïc GUEZO & Chahrazad DAHMANI – UPEC S6 – 2026"
|
||||
|
||||
markdown_extensions:
|
||||
- admonition
|
||||
- pymdownx.details
|
||||
|
||||
@@ -1,7 +1,18 @@
|
||||
[project]
|
||||
name = "projet-millesima-s6"
|
||||
version = "0.1.0"
|
||||
dependencies = ["requests==2.32.5", "beautifulsoup4==4.14.3", "pandas==2.3.3", "tqdm==4.67.3"]
|
||||
dependencies = [
|
||||
"requests==2.32.5",
|
||||
"beautifulsoup4==4.14.3",
|
||||
"pandas==2.3.3",
|
||||
"tqdm==4.67.3",
|
||||
"scikit-learn==1.7.2",
|
||||
"matplotlib==3.10.8"
|
||||
]
|
||||
|
||||
[tool.pytest.ini_options]
|
||||
pythonpath = "src"
|
||||
testpaths = ["tests"]
|
||||
|
||||
[project.optional-dependencies]
|
||||
test = ["pytest==8.4.2", "requests-mock==1.12.1", "flake8==7.3.0"]
|
||||
|
||||
119
src/cleaning.py
Executable file
119
src/cleaning.py
Executable file
@@ -0,0 +1,119 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
from os import getcwd
|
||||
from os.path import normpath, join
|
||||
from typing import cast
|
||||
from pandas import DataFrame, read_csv, to_numeric, get_dummies
|
||||
from sys import argv
|
||||
|
||||
|
||||
def path_filename(filename: str) -> str:
|
||||
return normpath(join(getcwd(), filename))
|
||||
|
||||
|
||||
class Cleaning:
|
||||
def __init__(self, filename) -> None:
|
||||
self._vins: DataFrame = read_csv(filename)
|
||||
# créer la liste de tout les scores
|
||||
self.SCORE_COLS: list[str] = [
|
||||
c for c in self._vins.columns if c not in ["Appellation", "Prix"]
|
||||
]
|
||||
# transforme tout les colonnes score en numérique
|
||||
for col in self.SCORE_COLS:
|
||||
self._vins[col] = to_numeric(self._vins[col], errors="coerce")
|
||||
|
||||
def getVins(self) -> DataFrame:
|
||||
return self._vins.copy(deep=True)
|
||||
|
||||
def __str__(self) -> str:
|
||||
"""
|
||||
Affiche un résumé du DataFrame
|
||||
- la taille
|
||||
- types des colonnes
|
||||
- valeurs manquantes
|
||||
- statistiques numériques
|
||||
"""
|
||||
return (
|
||||
f"Shape : {self._vins.shape[0]} lignes x {self._vins.shape[1]} colonnes\n\n"
|
||||
f"Types des colonnes :\n{self._vins.dtypes}\n\n"
|
||||
f"Valeurs manquantes :\n{self._vins.isna().sum()}\n\n"
|
||||
f"Statistiques numériques :\n{self._vins.describe().round(2)}\n\n"
|
||||
)
|
||||
|
||||
def drop_empty_appellation(self) -> "Cleaning":
|
||||
self._vins = self._vins.dropna(subset=["Appellation"])
|
||||
return self
|
||||
|
||||
def _mean_score(self, col: str) -> DataFrame:
|
||||
"""
|
||||
Calcule la moyenne d'une colonne de score par appellation.
|
||||
- Convertit les valeurs en numériques, en remplaçant les non-convertibles par NaN
|
||||
- Calcule la moyenne par appellation
|
||||
- Remplace les NaN résultants par 0
|
||||
|
||||
"""
|
||||
means = self._vins.groupby("Appellation", as_index=False)[col].mean()
|
||||
means = means.rename(
|
||||
columns={col: f"mean_{col}"}
|
||||
) # pyright: ignore[reportCallIssue]
|
||||
return cast(DataFrame, means.fillna(0))
|
||||
|
||||
def _mean_robert(self) -> DataFrame:
|
||||
return self._mean_score("Robert")
|
||||
|
||||
def _mean_robinson(self) -> DataFrame:
|
||||
return self._mean_score("Robinson")
|
||||
|
||||
def _mean_suckling(self) -> DataFrame:
|
||||
return self._mean_score("Suckling")
|
||||
|
||||
def fill_missing_scores(self) -> "Cleaning":
|
||||
"""
|
||||
Remplacer les notes manquantes par la moyenne
|
||||
des vins de la même appellation.
|
||||
"""
|
||||
for element in self.SCORE_COLS:
|
||||
means = self._mean_score(element)
|
||||
self._vins = self._vins.merge(means, on="Appellation", how="left")
|
||||
|
||||
mean_col = f"mean_{element}"
|
||||
self._vins[element] = self._vins[element].fillna(self._vins[mean_col])
|
||||
|
||||
self._vins = self._vins.drop(columns=["mean_" + element])
|
||||
return self
|
||||
|
||||
def encode_appellation(self, column: str = "Appellation") -> "Cleaning":
|
||||
"""
|
||||
Remplace la colonne 'Appellation' par des colonnes indicatrices
|
||||
"""
|
||||
appellations = self._vins[column].astype(str).str.strip()
|
||||
appellation_dummies = get_dummies(appellations, prefix="App")
|
||||
self._vins = self._vins.drop(columns=[column])
|
||||
self._vins = self._vins.join(appellation_dummies)
|
||||
return self
|
||||
|
||||
def drop_empty_price(self) -> "Cleaning":
|
||||
self._vins = self._vins.dropna(subset=["Prix"])
|
||||
return self
|
||||
|
||||
|
||||
def main() -> None:
|
||||
if len(argv) != 2:
|
||||
raise ValueError(f"Usage: {argv[0]} <filename.csv>")
|
||||
|
||||
filename = argv[1]
|
||||
cleaning: Cleaning = (
|
||||
Cleaning(filename)
|
||||
.drop_empty_appellation()
|
||||
.fill_missing_scores()
|
||||
.encode_appellation()
|
||||
.drop_empty_price()
|
||||
)
|
||||
cleaning.getVins().to_csv("clean.csv", index=False)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
main()
|
||||
except Exception as e:
|
||||
print(f"ERREUR: {e}")
|
||||
93
src/learning.py
Executable file
93
src/learning.py
Executable file
@@ -0,0 +1,93 @@
|
||||
|
||||
|
||||
from typing import Any, Callable
|
||||
from pandas import DataFrame
|
||||
from sklearn.linear_model import LinearRegression
|
||||
from sklearn.preprocessing import StandardScaler
|
||||
from sklearn.model_selection import train_test_split
|
||||
from sklearn.pipeline import make_pipeline
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
from cleaning import Cleaning
|
||||
|
||||
|
||||
class Learning:
|
||||
def __init__(self, vins: DataFrame, target: str) -> None:
|
||||
self.X = vins.drop(target, axis=1)
|
||||
self.y = vins[target]
|
||||
|
||||
self.X_train, self.X_test, self.y_train, self.y_test = train_test_split(
|
||||
self.X, self.y, test_size=0.25, random_state=49
|
||||
)
|
||||
|
||||
def evaluate(
|
||||
self,
|
||||
estimator,
|
||||
pretreatment=None,
|
||||
fn_score=lambda m, xt, yt: m.score(xt, yt),
|
||||
):
|
||||
|
||||
pipeline = make_pipeline(pretreatment, estimator) if pretreatment else estimator
|
||||
pipeline.fit(self.X_train, self.y_train)
|
||||
score = fn_score(pipeline, self.X_test, self.y_test)
|
||||
prediction = pipeline.predict(self.X_test)
|
||||
|
||||
return score, prediction
|
||||
|
||||
def draw(self, predictions, y_actual):
|
||||
plt.figure(figsize=(8, 6))
|
||||
|
||||
plt.scatter(
|
||||
predictions,
|
||||
y_actual,
|
||||
alpha=0.5,
|
||||
c="royalblue",
|
||||
edgecolors="k",
|
||||
label="Vins",
|
||||
)
|
||||
|
||||
mn = min(predictions.min(), y_actual.min())
|
||||
mx = max(predictions.max(), y_actual.max())
|
||||
plt.plot(
|
||||
[mn, mx],
|
||||
[mn, mx],
|
||||
color="red",
|
||||
linestyle="--",
|
||||
lw=2,
|
||||
label="Prédiction Parfaite",
|
||||
)
|
||||
|
||||
plt.xlabel("Prix estimés (estim_LR)")
|
||||
plt.ylabel("Prix réels (y_test)")
|
||||
plt.title("titre")
|
||||
plt.legend()
|
||||
plt.grid(True, linestyle=":", alpha=0.6)
|
||||
|
||||
plt.show()
|
||||
|
||||
|
||||
df_vins = (
|
||||
Cleaning("data.csv")
|
||||
.drop_empty_appellation()
|
||||
.fill_missing_scores()
|
||||
.encode_appellation()
|
||||
.drop_empty_price()
|
||||
.getVins()
|
||||
)
|
||||
|
||||
etude = Learning(df_vins, target="Prix")
|
||||
|
||||
print("--- Question 16 & 17 ---")
|
||||
score_simple, estim_simple = etude.evaluate(LinearRegression())
|
||||
print(f"Score R² (LR Simple) : {score_simple:.4f}")
|
||||
|
||||
etude.draw(estim_simple, etude.y_test)
|
||||
|
||||
|
||||
print("\n--- Question 18 ---")
|
||||
score_std, estim_std = etude.evaluate(
|
||||
estimator=LinearRegression(), pretreatment=StandardScaler()
|
||||
)
|
||||
print(f"Score R² (Standardisation + LR) : {score_std:.4f}")
|
||||
|
||||
etude.draw(estim_std, etude.y_test)
|
||||
@@ -1,7 +1,7 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
from collections import OrderedDict
|
||||
from io import SEEK_END, SEEK_SET, BufferedWriter
|
||||
from io import SEEK_END, SEEK_SET, BufferedWriter, TextIOWrapper
|
||||
from json import JSONDecodeError, loads
|
||||
from os import makedirs
|
||||
from os.path import dirname, exists, join, normpath, realpath
|
||||
@@ -215,6 +215,7 @@ class _ScraperData:
|
||||
robinson = self.robinson()
|
||||
suckling = self.suckling()
|
||||
prix = self.prix()
|
||||
prix = self.prix()
|
||||
|
||||
return f"{appellation},{parker},{robinson},{suckling},{prix}"
|
||||
|
||||
@@ -376,15 +377,11 @@ class Scraper:
|
||||
try:
|
||||
data: dict[str, object] = self.getjsondata(subdir).getdata()
|
||||
|
||||
for element in ["initialReduxState", "categ", "content"]:
|
||||
data = cast(dict[str, object], data.get(element))
|
||||
|
||||
products: list[dict[str, Any]] = cast(
|
||||
list[dict[str, Any]], data.get("products")
|
||||
)
|
||||
|
||||
if isinstance(products, list):
|
||||
return products
|
||||
return products
|
||||
|
||||
except (JSONDecodeError, HTTPError):
|
||||
return None
|
||||
@@ -407,6 +404,44 @@ class Scraper:
|
||||
except (JSONDecodeError, HTTPError) as e:
|
||||
print(f"Erreur sur le produit {link}: {e}")
|
||||
|
||||
def _initstate(self, reset: bool) -> tuple[int, set[str]]:
|
||||
"""
|
||||
appelle la fonction pour load le cache, si il existe
|
||||
pas, il utilise les variables de base sinon il override
|
||||
toute les variables pour continuer et pas recommencer le
|
||||
processus en entier.
|
||||
|
||||
Args:
|
||||
reset (bool): pouvoir le reset ou pas
|
||||
|
||||
Returns:
|
||||
tuple[int, set[str]]: le contenu de la page et du cache
|
||||
"""
|
||||
if not reset:
|
||||
#
|
||||
serializable: tuple[int, set[str]] | None = loadstate()
|
||||
if isinstance(serializable, tuple):
|
||||
return serializable
|
||||
return 1, set()
|
||||
|
||||
def _ensuretitle(self, f: TextIOWrapper, title: str) -> None:
|
||||
"""
|
||||
check si le titre est bien présent au début du buffer
|
||||
sinon il l'ecrit, petit bug potentiel, a+ ecrit tout le
|
||||
temps a la fin du buffer, si on a ecrit des choses avant
|
||||
le titre sera apres ces données mais on part du principe
|
||||
que personne va toucher le fichier.
|
||||
|
||||
Args:
|
||||
f (TextIOWrapper): buffer stream fichier
|
||||
title (str): titre du csv
|
||||
"""
|
||||
_ = f.seek(0, SEEK_SET)
|
||||
if not (f.read(len(title)) == title):
|
||||
_ = f.write(title)
|
||||
else:
|
||||
_ = f.seek(0, SEEK_END)
|
||||
|
||||
def getvins(self, subdir: str, filename: str, reset: bool = False) -> None:
|
||||
"""
|
||||
Scrape toutes les pages d'une catégorie et sauvegarde en CSV.
|
||||
@@ -420,35 +455,13 @@ class Scraper:
|
||||
mode: Literal["w", "a+"] = "w" if reset else "a+"
|
||||
# titre
|
||||
title: str = "Appellation,Robert,Robinson,Suckling,Prix\n"
|
||||
# page du début
|
||||
page: int = 1
|
||||
# le set qui sert de cache
|
||||
cache: set[str] = set[str]()
|
||||
# page: page où commence le scraper
|
||||
# cache: tout les pages déjà parcourir
|
||||
page, cache = self._initstate(reset)
|
||||
|
||||
custom_format = "{l_bar} {bar:20} {r_bar}"
|
||||
|
||||
if not reset:
|
||||
# appelle la fonction pour load le cache, si il existe
|
||||
# pas, il utilise les variables de base sinon il override
|
||||
# toute les variables pour continuer et pas recommencer le
|
||||
# processus en entier.
|
||||
serializable: tuple[int, set[str]] | None = loadstate()
|
||||
if isinstance(serializable, tuple):
|
||||
# override la page et le cache
|
||||
page, cache = serializable
|
||||
try:
|
||||
with open(filename, mode) as f:
|
||||
# check si le titre est bien présent au début du buffer
|
||||
# sinon il l'ecrit, petit bug potentiel, a+ ecrit tout le
|
||||
# temps a la fin du buffer, si on a ecrit des choses avant
|
||||
# le titre sera apres ces données mais on part du principe
|
||||
# que personne va toucher le fichier.
|
||||
_ = f.seek(0, SEEK_SET)
|
||||
if not (f.read(len(title)) == title):
|
||||
_ = f.write(title)
|
||||
else:
|
||||
_ = f.seek(0, SEEK_END)
|
||||
|
||||
self._ensuretitle(f, title)
|
||||
while True:
|
||||
products_list: list[dict[str, Any]] | None = (
|
||||
self._geturlproductslist(f"{subdir}?page={page}")
|
||||
@@ -457,15 +470,21 @@ class Scraper:
|
||||
break
|
||||
|
||||
pbar: tqdm[dict[str, Any]] = tqdm(
|
||||
products_list, bar_format=custom_format
|
||||
products_list, bar_format="{l_bar} {bar:20} {r_bar}"
|
||||
)
|
||||
for product in pbar:
|
||||
keyword = product.get("seoKeyword", "Inconnu")[:40]
|
||||
keyword: str = cast(
|
||||
str, product.get("seoKeyword", "Inconnu")[:40]
|
||||
)
|
||||
pbar.set_description(
|
||||
f"Page: {page:<3} | Product: {keyword:<40}"
|
||||
)
|
||||
self._writevins(cache, product, f)
|
||||
page += 1
|
||||
# va créer un fichier au début et l'override
|
||||
# tout les 5 pages au cas où SIGHUP ou autre
|
||||
if page % 5 == 0 and not reset:
|
||||
savestate((page, cache))
|
||||
except (Exception, HTTPError, KeyboardInterrupt, JSONDecodeError):
|
||||
if not reset:
|
||||
savestate((page, cache))
|
||||
|
||||
67
tests/test_cleaning.py
Executable file
67
tests/test_cleaning.py
Executable file
@@ -0,0 +1,67 @@
|
||||
import pytest
|
||||
from unittest.mock import patch, mock_open
|
||||
from cleaning import Cleaning
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def cleaning_raw() -> Cleaning:
|
||||
"""
|
||||
"Appellation": ["Pauillac", "Pauillac ", "Margaux", None , "Pomerol", "Pomerol"],
|
||||
"Robert": ["95" , None , "bad" , 90 , None , None ],
|
||||
"Robinson": [None , "93" , 18 , None , None , None ],
|
||||
"Suckling": [96 , None , None , None , 91 , None ],
|
||||
"Prix": ["10.0" , "11.0" , "20.0" , "30.0", "40.0" , "50.0" ],
|
||||
"""
|
||||
csv_content = """Appellation,Robert,Robinson,Suckling,Prix
|
||||
Pauillac,95,,96,10.0
|
||||
Pauillac ,,93,,11.0
|
||||
Margaux,bad,18,,20.0
|
||||
,90,,,30.0
|
||||
Pomerol,,,91,40.0
|
||||
Pomerol,,,,50.0
|
||||
"""
|
||||
m = mock_open(read_data=csv_content)
|
||||
with patch("builtins.open", m):
|
||||
return Cleaning("donnee.csv")
|
||||
|
||||
|
||||
def test_drop_empty_appellation(cleaning_raw: Cleaning) -> None:
|
||||
out = cleaning_raw.drop_empty_appellation().getVins()
|
||||
assert out["Appellation"].isna().sum() == 0
|
||||
assert len(out) == 5
|
||||
|
||||
|
||||
def test_mean_score_zero_when_no_scores(cleaning_raw: Cleaning) -> None:
|
||||
out = cleaning_raw.drop_empty_appellation()
|
||||
m = out._mean_score("Robert")
|
||||
assert list(m.columns) == ["Appellation", "mean_Robert"]
|
||||
pomerol_mean = m.loc[m["Appellation"].str.strip() == "Pomerol", "mean_Robert"].iloc[
|
||||
0
|
||||
]
|
||||
assert pomerol_mean == 0
|
||||
|
||||
|
||||
def test_fill_missing_scores(cleaning_raw: Cleaning):
|
||||
cleaning_raw._vins["Appellation"] = cleaning_raw._vins["Appellation"].str.strip()
|
||||
|
||||
cleaning_raw.drop_empty_appellation()
|
||||
filled = cleaning_raw.fill_missing_scores().getVins()
|
||||
for col in cleaning_raw.SCORE_COLS:
|
||||
assert filled[col].isna().sum() == 0
|
||||
|
||||
pauillac_robert = filled[filled["Appellation"] == "Pauillac"]["Robert"]
|
||||
assert (pauillac_robert == 95.0).all()
|
||||
|
||||
|
||||
def test_encode_appellation(cleaning_raw: Cleaning):
|
||||
cleaning_raw._vins["Appellation"] = cleaning_raw._vins["Appellation"].str.strip()
|
||||
|
||||
out = (
|
||||
cleaning_raw.drop_empty_appellation()
|
||||
.fill_missing_scores()
|
||||
.encode_appellation()
|
||||
.getVins()
|
||||
)
|
||||
assert "App_Appellation" not in out.columns
|
||||
assert "App_Pauillac" in out.columns
|
||||
assert int(out.loc[0, "App_Pauillac"]) == 1
|
||||
26
tests/test_scraper.py
Normal file → Executable file
26
tests/test_scraper.py
Normal file → Executable file
@@ -185,17 +185,11 @@ def mock_site():
|
||||
{dumps({
|
||||
"props": {
|
||||
"pageProps": {
|
||||
"initialReduxState": {
|
||||
"categ": {
|
||||
"content": {
|
||||
"products": [
|
||||
{"seoKeyword": "/nino-negri-5-stelle-sfursat-2022.html",},
|
||||
{"seoKeyword": "/poubelle",},
|
||||
{"seoKeyword": "/",}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
"products": [
|
||||
{"seoKeyword": "/nino-negri-5-stelle-sfursat-2022.html",},
|
||||
{"seoKeyword": "/poubelle",},
|
||||
{"seoKeyword": "/",}
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -213,14 +207,8 @@ def mock_site():
|
||||
{dumps({
|
||||
"props": {
|
||||
"pageProps": {
|
||||
"initialReduxState": {
|
||||
"categ": {
|
||||
"content": {
|
||||
"products": [
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
"products": [
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user