requests
pandas
tqdm

# /usr/bin/python3
# The purpose of this script is to cache data from the OpenTable Directory API.

# Import Required Modules
import pickle
import requests
from tqdm import tqdm

# Define OpenTable Related Variables
opentable_token_uri = 'url provided by opentable'
opentable_offset = 0
opentable_directory_api = 'url provided by opentable'
opentable_directory_query = '?offset='
opentable_directory_uri = opentable_directory_api + opentable_directory_query + str(opentable_offset)
opentable_clientid = 'provided by opentable'
opentable_secret = 'provided by opentable'

# Send POST to OpenTable to retrieve authentication token.
r = requests.post(opentable_token_uri, auth=(opentable_clientid, opentable_secret))

# Create variable for managing authentication and populate it from initial request
jar = r.cookies

# Create variable for holding the active access token
access_token = r.json()['access_token']

# Create mandatory header variable for interaction with the OpenTable Director API
headers = {'Content-Type': 'application/json', 'Accept-Encoding': 'lz4', 'Authorization': 'bearer '+access_token}

# Query OpenTable Directory API
r2 = requests.get(opentable_directory_uri, headers=headers, cookies=jar)
raw = r2.json()
opentable_data_set = []
opentable_data_set.extend(raw['items'])
print("Loading Data From OpenTable Directory API...")

# Initialize Progress Bar
pbar = tqdm(total=raw['total_items'])

# Loop through API at 100 restaurants/per query until all items are stored in opentable_data_set
while raw['offset'] < raw['total_items']:
    opentable_offset = opentable_offset + 100
    opentable_directory_uri = opentable_directory_api + opentable_directory_query + str(opentable_offset)
    r2 = requests.get(opentable_directory_uri, headers=headers, cookies=jar)
    raw = r2.json()
    opentable_data_set.extend(raw['items'])
    pbar.update(100)  # Trigger Progress Bar Update

# Terminate Progress Bar
pbar.close()

# Refresh Cache
print('')
print('')
print('OpenTable Directory API Cache Refreshed')
with open('opentable_cache', 'wb') as handle:
    pickle.dump(opentable_data_set, handle)
handle.close()

# /usr/bin/python3
# The purpose of this script is to search cached data from the OpenTable Directory API.

# Import Required Modules
import pickle
import pandas as pd

# Define OpenTable Related Variables
opentable_refID = 'provided by opentable'

# OpenTable Directory API Cache Refresh Function


with open('opentable_cache', 'rb') as handle:
    opentable_data_set = pickle.load(handle)

# Load OpenTable Directory JSON Data into a Pandas Dataframe
df = pd.DataFrame(opentable_data_set)

# Request Name Restaurant's Name From User
restaurant = input("Search By Restaurant Name: ")

# Select Variables
results = df[df['name'].str.contains(restaurant)]
results_found = results['name'].count()
formatted_results = results[['name', 'metro_name', 'country']]

# Prevent Pandas from printing automatically formatted across multiple lines.
pd.set_option('expand_frame_repr', False)
# Print Total Search Results
print('')
print('Total Results Found: ' + str(results_found))

# Print Blank Line
print('')

# Print Formatted Results
print(formatted_results)

# Print Blank Line
print('')

# Request Row Number From User
row = int(input('Enter the leftmost line number to select a restaurant: '))

# Print Blank Line
print('')

# Set Selected Row
row_selected = df.loc[row]
reservation_link = str(row_selected['reservation_url'])

# Print Referral Link
print('Opentable Referral Link:')
print(reservation_link + '?ref=' + opentable_refID)
handle.close()