unterklassiert Objekt Ergebnisse in NoneType

Question

Ich versuche, den Chrome WebDriver Unterklasse einig Initialisierung und Bereinigungscode enthält, aber dann klagt Python, dass das erstellte Objekt zu None gesetzt:

import glob 
import selenium 
import subprocess 
from selenium.webdriver.common.by import By 


class WebDriver(selenium.webdriver.Chrome): 
    def __init__(self, url, *args, **kwargs): 
     super().__init__(*args, **kwargs) 
     self.url = url 

    def __enter__(self): 
     self.get(self.url) 
     self.implicitly_wait(15) 

    def __exit__(self, type, value, traceback): 
     self.quit() 
     for path in glob.glob('/tmp/.org.chromium.Chromium.*'): 
      subprocess.run(['rm', '-rf', path], check=True) 


with WebDriver('https://google.com') as driver: 
    driver.find_element(By.ID, 'lst-ib').send_keys('Search') 

den Code mit Python Lauf 3:

$ python3 test.py 
Traceback (most recent call last): 
    File "test.py", line 43, in <module> 
    driver.find_element(By.ID, 'lst-ib').send_keys('Search') 
AttributeError: 'NoneType' object has no attribute 'find_element' 

Answer 1

Ihre __enter__() magische Methode self für die driver Variable werden wies auf die Instanz der WebDriver Klasse zurückgeben sollte:

def __enter__(self): 
    self.get(self.url) 
    self.implicitly_wait(15) 
    return self 

Um weitere Informationen darüber, warum und wie das funktioniert, bitte zu sehen bekommen:

• Understanding Python's "with" statement
• Explaining Python's '__enter__' and '__exit__'