view oembed/core.py @ 878:fb0d2871afa4

Use UNIX socket for memcached.
author Brian Neal <bgneal@gmail.com>
date Sat, 03 Jan 2015 14:24:52 -0600
parents 89b240fe9297
children 344f7914d421
line wrap: on
line source
"""
This module contains core functionality for the oembed application.

"""
import json
import urllib
import urllib2
import gzip
try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO



USER_AGENT = 'gremmies python oembed'


def get_oembed(api_endpoint, url, format='json', **opts):
    """
    Perform the GET request to retrieve the embedded media data from the given
    API endpoint for the given URL. Return the result as a Python dictionary.

    format specifies the response format, and should be 'json' or 'xml'.
    opts are any additional GET options that should be present in the GET
    request.

    """
    opts['url'] = url
    opts['format'] = format
    api_url = "%s?%s" % (api_endpoint, urllib.urlencode(opts))

    headers = {
        'User-Agent': USER_AGENT,
        'Accept-Encoding': 'gzip',
    }
    request = urllib2.Request(api_url, headers=headers)

    opener = urllib2.build_opener()
    f = opener.open(request)
    headers = f.info()
    result = f.read()
    f.close()

    if headers.get('content-encoding') == 'gzip':
        f = gzip.GzipFile(fileobj=StringIO(result))
        result = f.read()
        f.close()

    return json.loads(result)

if __name__ == "__main__":
    try:
        print get_oembed("http://www.youtube.com/oembed",
                "http://www.youtube.com/watch?v=7_IMzJldOf4")
    except urllib2.HTTPError, e:
        print e