Python au sélénium (5). Comment gérer les cookies avec Selenium et Python : importation et exportation de cookies

La gestion des cookies est un élément crucial de l'automatisation Web, en particulier lorsque vous travaillez avec des sessions en Selenium et Python. Cet article vous guidera tout au long du processus d'importation et d'exportation de cookies à l'aide de Selenium. Entrons dans les détails et explorons comment rendre vos tâches d'automatisation Web plus efficaces.

Que sont les cookies et pourquoi sont-ils importants ?

Les cookies sont de petits éléments de données stockés sur l'ordinateur de l'utilisateur par le navigateur Internet lors de sa navigation sur un site Internet. Ils sont essentiels au maintien des sessions et peuvent stocker les préférences de l'utilisateur, les informations de connexion et d'autres données nécessaires au fonctionnement du site Web.

Configuration de votre environnement Selenium

Avant de commencer à travailler avec les cookies, assurez-vous que Selenium est installé dans votre environnement Python. Vous pouvez l'installer en utilisant pip :

pip install selenium

Vous aurez également besoin d'un pilote Web pour votre navigateur (par exemple, ChromeDriver pour Google Chrome).

Importation des modules nécessaires

Commençons par importer les modules nécessaires :

from selenium import webdriver
import pickle
import time

Création d'une session de navigateur et connexion

Commençons par créer une session de navigateur et nous connecter à un site Web. À des fins de démonstration, nous utiliserons un forum aléatoire :

# Create a new browser session
browser = webdriver.Chrome()

# Navigate to the forum
browser.get('https://randomforum.com')

# Add a delay to ensure the page loads and user can log in
time.sleep(100)  # Adjust the delay as needed

Enregistrer des cookies dans un fichier

Une fois connecté, nous pouvons enregistrer les cookies de session dans un fichier à l'aide du module pickle :

# Save cookies to a file
with open('cookies.pkl', 'wb') as file:
    pickle.dump(browser.get_cookies(), file)

Chargement de cookies dans une nouvelle session de navigateur

Maintenant, chargeons ces cookies dans une nouvelle session de navigateur pour conserver l'état de connexion :

# Create a new browser session
browser = webdriver.Chrome()

# Navigate to the same website
browser.get('https://randomforum.com')

# Load cookies from the file
with open('cookies.pkl', 'rb') as file:
    cookies = pickle.load(file)

for cookie in cookies:
    browser.add_cookie(cookie)

# Refresh the page to apply cookies
browser.refresh()

En suivant ces étapes, la nouvelle session du navigateur devrait reconnaître les cookies stockés et vous garder connecté.

Considérations importantes

• Même domaine : Assurez-vous que les cookies sont ajoutés au même domaine pour lequel ils ont été créés. Toute tentative d'ajout de cookies à un domaine différent entraînera des erreurs.
• Expiration de la session: Sachez que les cookies de session peuvent expirer. La gestion du renouvellement de session peut être nécessaire pour les scripts de longue durée.

Exemple de code

Voici un exemple complet combinant toutes les étapes :

from selenium import webdriver
import pickle
import time

# Step 1: Create a browser session and log in
browser = webdriver.Chrome()
browser.get('https://randomforum.com')
time.sleep(100)  # Adjust the delay for login

# Step 2: Save cookies to a file
with open('cookies.pkl', 'wb') as file:
    pickle.dump(browser.get_cookies(), file)

# Step 3: Load cookies into a new browser session
browser = webdriver.Chrome()
browser.get('https://randomforum.com')

with open('cookies.pkl', 'rb') as file:
    cookies = pickle.load(file)

for cookie in cookies:
    browser.add_cookie(cookie)

browser.refresh()

Conclusion

La gestion des cookies dans Selenium avec Python peut rationaliser considérablement vos tâches d'automatisation Web en conservant les états de session sur différentes instances de navigateur. Cette méthode est particulièrement utile pour les tâches nécessitant des connexions et une gestion de session répétées.

Assurez-vous de suivre les meilleures pratiques et de gérer les cookies en toute sécurité, en tenant compte de l'expiration de la session et des contraintes de domaine. De cette façon, vous pouvez améliorer l'efficacité et la fiabilité de vos scripts d'automatisation Web.