homebrew-cask/developer/cgi/check_url_filename.py

#!/usr/bin/python
from __future__ import print_function
import sys
import cgi
try:  # Python 3
  from urllib.request import Request, urlopen
except NameError:  # Python 2
  from urllib2 import Request, urlopen


url = sys.argv[1]
headers = {'User-Agent': 'Mozilla'}
if len(sys.argv) > 2:
    ua = sys.argv[2]
    headers = {'User-Agent': ua}


try:
    req = Request(url, None, headers)
    response = urlopen(req)
    _, params = cgi.parse_header(
        response.headers.get('Content-Disposition', ''))
    filename = params['filename']
except Exception:
    import traceback
    filename = 'generic exception: ' + traceback.format_exc()

print(filename)