<s> from selenium . webdriver import Chrome <EOL> from selenium . webdriver . chrome . options import Options <EOL> from splinter . driver . webdriver import BaseWebDriver , WebDriverElement <EOL> from splinter . driver . webdriver . cookie_manager import CookieManager <EOL> class WebDriver ( BaseWebDriver ) : <EOL> driver_name = "<STR_LIT>" <EOL> def __init__ ( self , user_agent = None , wait_time = <NUM_LIT:2> , fullscreen = False , incognito = False , <EOL> ** kwargs ) : <EOL> options = Options ( ) <EOL> if user_agent is not None : <EOL> options . add_argument ( "<STR_LIT>" + user_agent ) <EOL> if incognito is not None : <EOL> options . add_argument ( "<STR_LIT>" ) <EOL> if fullscreen : <EOL> options . add_argument ( '<STR_LIT>' ) <EOL> self . driver = Chrome ( chrome_options = options , ** kwargs ) <EOL> self . element_class = WebDriverElement <EOL> self . _cookie_manager = CookieManager ( self . driver ) <EOL> super ( WebDriver , self ) . __init__ ( wait_time ) </s>
