[py] Add support for full page screenshots in Chrome and Edge

[cgoldberg]

User description

🔗 Related Issues

Similar to #16725 for Ruby
Python implementation of #14116

💥 What does this PR do?

This PR adds support for full page screenshots for Chrome and Edge via session/{session_id}/screenshot/full endpoint (like Firefox already has).

🔧 Implementation Notes

This essentially implements the same methods that the FireFox WebDriver class uses in the Chromium WebDriver class, so Chrome and Edge can use them.

🔄 Types of changes

• New feature (non-breaking change which adds functionality and tests!)

---

PR Type

Enhancement

---

Description

• Add full page screenshot methods to Chromium WebDriver

• Implement four screenshot methods with different output formats

• Add comprehensive tests for Chrome and Edge browsers

• Methods support base64, PNG binary, and file output formats

---

Diagram Walkthrough

flowchart LR
  A["Chromium WebDriver"] -->|"add methods"| B["get_full_page_screenshot_as_base64"]
  A -->|"add methods"| C["get_full_page_screenshot_as_png"]
  A -->|"add methods"| D["get_full_page_screenshot_as_file"]
  A -->|"add methods"| E["save_full_page_screenshot"]
  B -->|"decode"| C
  C -->|"write to file"| D
  D -->|"alias"| E
  F["Chrome Tests"] -->|"verify"| B
  F -->|"verify"| C
  G["Edge Tests"] -->|"verify"| B
  G -->|"verify"| C

Loading

File Walkthrough

	Relevant files
Enhancement	webdriver.py Add four full page screenshot methods to Chromium                py/selenium/webdriver/chromium/webdriver.py Add imports for base64 and warnings modules Implement get_full_page_screenshot_as_base64() method to retrieve full page screenshot as base64-encoded string Implement get_full_page_screenshot_as_png() method to decode base64 to binary PNG data Implement get_full_page_screenshot_as_file() method to save screenshot to file with validation Implement save_full_page_screenshot() as alias method for file saving +67/-0
Formatting	webdriver.py Minor formatting adjustment                                                            py/selenium/webdriver/firefox/webdriver.py Add blank line after license header for consistency +1/-0
Tests	chrome_takes_full_page_screenshots_tests.py Add Chrome full page screenshot tests                                        py/test/selenium/webdriver/chrome/chrome_takes_full_page_screenshots_tests.py Create new test file for Chrome full page screenshots Add test for get_full_page_screenshot_as_base64() method Add test for get_full_page_screenshot_as_png() method Verify returned data is valid PNG format using filetype library +34/-0    edge_takes_full_page_screenshots_tests.py Add Edge full page screenshot tests                                            py/test/selenium/webdriver/edge/edge_takes_full_page_screenshots_tests.py Create new test file for Edge full page screenshots Add test for get_full_page_screenshot_as_base64() method Add test for get_full_page_screenshot_as_png() method Verify returned data is valid PNG format using filetype library +34/-0

---

[qodo-code-review]

PR Compliance Guide 🔍

Below is a summary of compliance checks for this PR:

Security Compliance
🟢	No security concerns identified No security vulnerabilities detected by AI analysis. Human verification advised for critical code.
Ticket Compliance
⚪	🎫 No ticket provided Create ticket/issue
Codebase Duplication Compliance
⚪	Codebase context is not defined Follow the guide to enable codebase context checks.
Custom Compliance
🟢	Generic: Meaningful Naming and Self-Documenting Code Objective: Ensure all identifiers clearly express their purpose and intent, making code self-documenting Status: Passed Learn more about managing compliance generic rules or creating your own custom rules
	Generic: Secure Error Handling Objective: To prevent the leakage of sensitive system information through error messages while providing sufficient detail for internal debugging. Status: Passed Learn more about managing compliance generic rules or creating your own custom rules
	Generic: Secure Logging Practices Objective: To ensure logs are useful for debugging and auditing without exposing sensitive information like PII, PHI, or cardholder data. Status: Passed Learn more about managing compliance generic rules or creating your own custom rules
	Generic: Security-First Input Validation and Data Handling Objective: Ensure all data inputs are validated, sanitized, and handled securely to prevent vulnerabilities Status: Passed Learn more about managing compliance generic rules or creating your own custom rules
⚪	Generic: Comprehensive Audit Trails Objective: To create a detailed and reliable record of critical system actions for security analysis and compliance. Status: No auditing: The new screenshot methods perform actions without emitting any audit logs, but WebDriver libraries typically avoid logging at this level and may rely on higher-level frameworks, so applicability needs human review. Referred Code def get_full_page_screenshot_as_file(self, filename) -> bool: """Save a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Returns: False if there is any IOError, else returns True. Use full paths in your filename. Example: driver.get_full_page_screenshot_as_file("/Screenshots/foo.png") """ if not filename.lower().endswith(".png"): warnings.warn( "name used for saved screenshot does not match file type. It should end with a `.png` extension", UserWarning, ) png = self.get_full_page_screenshot_as_png() try: with open(filename, "wb") as f: ... (clipped 44 lines) Learn more about managing compliance generic rules or creating your own custom rules
	Generic: Robust Error Handling and Edge Case Management Objective: Ensure comprehensive error handling that provides meaningful context and graceful degradation Status: Narrow errors: File write errors only return False without conveying context, and base64 decode or command execution errors are not explicitly handled, which may be acceptable for parity with existing drivers but merits review. Referred Code if not filename.lower().endswith(".png"): warnings.warn( "name used for saved screenshot does not match file type. It should end with a `.png` extension", UserWarning, ) png = self.get_full_page_screenshot_as_png() try: with open(filename, "wb") as f: f.write(png) except OSError: return False finally: del png return True def save_full_page_screenshot(self, filename) -> bool: """Save a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. ... (clipped 20 lines) Learn more about managing compliance generic rules or creating your own custom rules
Update

Compliance status legend

🟢 - Fully Compliant
🟡 - Partial Compliant
🔴 - Not Compliant
⚪ - Requires Further Human Verification
🏷️ - Compliance label

[titusfortner]

Policy is to only implement w3c compliant endpoints and Google does not want to update this to be w3c compliant- https://issues.chromium.org/issues/42322971

Unfortunately, we need to wait for the BiDi implementation

[cgoldberg]

Thanks.. that's too bad