Untitled

import time
import unittest
from selenium import webdriver

class BaseTest(unittest.TestCase):
    def setUp(self):
        """Set up the WebDriver instance and configure test environment."""
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(5)
        self.driver.maximize_window()

    def tearDown(self):
        """Tear down the WebDriver instance and capture screenshots on failure or success."""
        # Access the test outcome
        result = self._outcome.result
        
        # Generate timestamp for screenshot filenames
        timestamp = time.strftime("%Y%m%d-%H%M%S")

        # Capture a screenshot if there are any failures
        if any(error[1] for error in result.failures):
            screenshot_filename = f"screenshot_{timestamp}_failure.png"
            self.driver.save_screenshot(screenshot_filename)
            print(f"Failed screenshot {screenshot_filename} saved.")
        elif any(error[1] for error in result.errors):
            screenshot_filename = f"screenshot_{timestamp}_error.png"
            self.driver.save_screenshot(screenshot_filename)
            print(f"Error screenshot {screenshot_filename} saved.")
        else:
            screenshot_filename = f"screenshot_{timestamp}_success.png"
            self.driver.save_screenshot(screenshot_filename)
            print(f"Success screenshot {screenshot_filename} saved.")

        # Quit the WebDriver
        self.driver.quit()