from contextlib import contextmanager
from selenium import webdriver@contextmanager
def get_browser():
driver = webdriver.Chrome()
driver.maximize_window()
try:
yield driver
finally:
driver.quit() # Guaranteed teardown!