Phantom JS custom headers + proxy

from selenium import webdriver


PHANTOMJS_DRIVER = '/home/dwn/phantomjs-2.1.1-linux-x86_64/bin/phantomjs'


def main(headers=None, proxy=False):
    if headers is None:
        headers = {
            'User-Agent': 'Test Chrome Headless (KHTML like Gecko)',
            'Referer': 'https://youtube.com/testvideo',
        }
    assert isinstance(headers, dict)

    capabilities = webdriver.DesiredCapabilities.PHANTOMJS.copy()
    capabilities['platform'] = "WINDOWS"
    capabilities['version'] = "10"
    capabilities['browserName'] = 'Test browserName'

    for key in headers:
        capabilities['phantomjs.page.customHeaders.{}'.format(key)] = headers[key]

    if proxy:
        webdriver.DesiredCapabilities.PHANTOMJS['proxy'] = {
            "httpProxy": proxy,
            "ftpProxy": proxy,
            "sslProxy": proxy,
            "noProxy": None,
            "proxyType": "MANUAL",
            "class": "org.openqa.selenium.Proxy",
            "autodetect": False
        }

    driver = webdriver.PhantomJS(PHANTOMJS_DRIVER, desired_capabilities=capabilities)
    driver.get('http://httpbin.org/get')

    print(driver.page_source)

    driver.close()


if __name__ == '__main__':
    main()