headless-driver 0.1.0

A simple Python package for managing Selenium Chrome drivers in headless mode.

headless-driver

A simple Python package for managing Selenium Chrome drivers in headless mode.

Features

• Headless Chrome driver management
• Temporary user data directory
• Custom window size and user agent
• Easy cleanup and context manager support
• Remote and local driver support

Installation

pip install headless-driver

Usage

from headless import Headless

hl = Headless()
driver = hl.get_driver()
driver.get("https://example.com")
print(driver.title)
hl.quit()

Or use as a context manager:

from headless import Headless

with Headless() as driver:
    driver.get("https://example.com")
    print(driver.title)

Search

from headless import SearchScraper, Headless

hl = Headless()
driver = hl.get_driver()

scraper = SearchScraper(driver=driver, max_results=5)

results = scraper.search("Nuhman PK github")
print(results)

API Documentation

Headless class

Headless(
    user_data_dir: Optional[str] = None,
    window_size: Tuple[int, int] = (1920, 1080),
    user_agent: Optional[str] = None,
    headless: bool = True,
    chrome_driver_path: Optional[str] = '/opt/homebrew/bin/chromedriver',
    additional_args: Optional[List[str]] = None,
    remote_url: Optional[str] = None,
)

• user_data_dir: Path for Chrome user data (temporary if not provided)
• window_size: Browser window size (default: 1920x1080)
• user_agent: Custom user agent string
• headless: Run Chrome in headless mode (default: True)
• chrome_driver_path: Path to chromedriver executable
• additional_args: List of extra Chrome arguments
• remote_url: Use remote Selenium server if provided

Methods

• get_driver(): Returns a Selenium WebDriver instance
• quit(): Quits the driver and cleans up user data

Running Tests

python -m unittest discover tests

License

MIT