UPC WiFi Router Enable/Disable

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from datetime import datetime
from pyvirtualdisplay import Display
from selenium import webdriver

display = Display(visible=0, size=(800, 600))
display.start()

wifi_address = "http://192.168.1.1"
wifi_username = "admin"
wifi_password = "admin"

driver  = webdriver.Firefox()
driver.get(wifi_address + '/login.asp')
driver.find_element_by_id("loginUsername").clear()
driver.find_element_by_id("loginUsername").send_keys(wifi_username)
driver.find_element_by_id("loginPassword").clear()
driver.find_element_by_id("loginPassword").send_keys(wifi_password)
driver.find_element_by_css_selector("button.upc_button1").click()
driver.get(wifi_address +'/wireless/radio.asp')

if str(sys.argv[1]).lower() == "enable":
    driver.find_element_by_xpath("//select[@name='WirelessEnable']/option[text()='Enabled']").click()
    print str(datetime.now()) + " Enabling WiFi on router"
else:
    driver.find_element_by_xpath("//select[@name='WirelessEnable']/option[text()='Disabled']").click()
    print str(datetime.now()) + " Disabling WiFi on router"

driver.find_element_by_id("CommitRadioSubmit").click()
driver.find_element_by_css_selector("a.logout").click()
driver.quit()
display.stop()
sys.exit()