proxy-digest-auth-test.py.bak

import sys
import urllib2
import argparse


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--proxy", type=str, metavar="PROXY", required=True,
                        help="address of a proxy server")
    parser.add_argument("--user", type=str, metavar="USER", required=True,
                        help="proxy user")
    parser.add_argument("--password", type=str, metavar="PASSWORD", required=True,
                        help="proxt user's password")
    parser.add_argument("--url", type=str, metavar="URL", required=True,
                        help="URL to access")

    args = parser.parse_args()

    return args


def main():
    args = parse_args()

    password_manager = urllib2.HTTPPasswordMgrWithDefaultRealm()
    password_manager.add_password(None, args.url, args.user, args.password)
    auth_handler = urllib2.ProxyDigestAuthHandler(password_manager)
    proxy_support = urllib2.ProxyHandler({"http": args.proxy})
    opener = urllib2.build_opener(proxy_support, auth_handler)
    urllib2.install_opener(opener)
    handle = urllib2.urlopen(args.url)
    page = handle.read()

    print(page),


if __name__ == '__main__':
    try:
        main()
    except Exception as exc:
        print(exc)
        sys.exit(-1)