from selenium import webdriver
import pickle

class CookieManager:
    def __init__(self, driver=None):
        self.driver = driver

    def save_cookies(self, filename):
        if self.driver is None:
            raise ValueError("Driver instance not provided.")

        cookies = self.driver.get_cookies()
        with open(filename, 'wb') as f:
            pickle.dump(cookies, f)

    def load_cookies(self, filename):
        if self.driver is None:
            raise ValueError("Driver instance not provided.")

        with open(filename, 'rb') as f:
            cookies = pickle.load(f)
            for cookie in cookies:
                self.driver.add_cookie(cookie)

# Example usage
if __name__ == "__main__":
    # Initialize a WebDriver instance
    driver = webdriver.Chrome()  # You can use any browser driver you prefer

    # Create a CookieManager instance
    cookie_manager = CookieManager(driver)

    # Navigate to a website to get some cookies
    driver.get("https://example.com")
    # Perform some actions to get cookies if needed

    # Save cookies to a file
    cookie_manager.save_cookies("cookies.pkl")

    # Close the current browser session
    driver.quit()

    # Reinitialize WebDriver instance
    driver = webdriver.Chrome()

    # Create a new CookieManager instance
    cookie_manager = CookieManager(driver)

    # Load cookies from the file
    cookie_manager.load_cookies("cookies.pkl")

    # Navigate to a page that requires cookies
    driver.get("https://example.com/some-protected-page")

    # Now the browser should have the necessary cookies loaded

    # Close the browser session
    driver.quit()