Yelp API запросы

from __future__ import print_function
import argparse
import pprint
import requests
import sys
from urllib.error import HTTPError
from urllib.parse import quote
 
 
API_KEY='KEY'
API_HOST = 'https://api.yelp.com'
SEARCH_PATH = '/v3/businesses/search'
BUSINESS_PATH = '/v3/businesses/'  # Business ID will come after slash.
 
 
# Defaults for our simple example.
DEFAULT_TERM = 'Vegan Cafe'
DEFAULT_LOCATION = 'San Francisco, CA'
SEARCH_LIMIT = 50
 
 
def request(host, path, api_key, url_params=None):
 
    url_params = url_params or {}
    url = '{0}{1}'.format(host, quote(path.encode('utf8')))
    headers = {
        'Authorization': 'Bearer %s' % api_key,
    }
 
    print(u'Querying {0} ...'.format(url))
 
    response = requests.request('GET', url, headers=headers, params=url_params)
 
    return response.json()
 
 
def search(api_key, term, location):
    """Query the Search API by a search term and location.
    Args:
        term (str): The search term passed to the API.
        location (str): The search location passed to the API.
    Returns:
        dict: The JSON response from the request.
    """
 
    url_params = {
        'term': term.replace(' ', '+'),
        'location': location.replace(' ', '+'),
        'limit': SEARCH_LIMIT
    }
    return request(API_HOST, SEARCH_PATH, api_key, url_params=url_params)
 
 
def get_business(api_key, business_id):
    """Query the Business API by a business ID.
    Args:
        business_id (str): The ID of the business to query.
    Returns:
        dict: The JSON response from the request.
    """
    business_path = BUSINESS_PATH + business_id
 
    return request(API_HOST, business_path, api_key)
 
 
def query_api(term, location):
    """Queries the API by the input values from the user.
    Args:
        term (str): The search term to query.
        location (str): The location of the business to query.
    """
    response = search(API_KEY, term, location)
 
    businesses = response.get('businesses')
 
    if not businesses:
        print(u'No businesses for {0} in {1} found.'.format(term, location))
        return
 
    business_id = businesses[0]['id']
 
    print(u'{0} businesses found, querying business info ' \
        'for the top result "{1}" ...'.format(
            len(businesses), business_id))
    response = get_business(API_KEY, business_id)
 
    print(u'Result for business "{0}" found:'.format(business_id))
    pprint.pprint(response, indent=2)
 
 
def main():
    parser = argparse.ArgumentParser()
 
    parser.add_argument('-q', '--term', dest='term', default=DEFAULT_TERM,
                        type=str, help='Search term (default: %(default)s)')
    parser.add_argument('-l', '--location', dest='location',
                        default=DEFAULT_LOCATION, type=str,
                        help='Search location (default: %(default)s)')
 
    input_values = parser.parse_args()
 
    try:
        query_api(input_values.term, input_values.location)
    except HTTPError as error:
        sys.exit(
            'Encountered HTTP error {0} on {1}:\n {2}\nAbort program.'.format(
                error.code,
                error.url,
                error.read(),
            )
        )
 
 
if __name__ == '__main__':
    main()