71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
import config
|
|
import requests
|
|
from base64 import b64encode
|
|
from garfpy import logger
|
|
|
|
|
|
class Kroger:
|
|
def __init__(self):
|
|
self.client_id = config.CLIENT_ID
|
|
self.client_secret = config.CLIENT_SECRET
|
|
self.auth = b64encode(f"{self.client_id}:{self.client_secret}".encode()).decode()
|
|
|
|
def kroger_token(self):
|
|
headers = {
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
'Authorization': f'Basic {self.auth}'
|
|
}
|
|
|
|
response = requests.post('https://api.kroger.com/v1/connect/oauth2/token', headers=headers, data={
|
|
'grant_type': 'client_credentials',
|
|
'scope': 'product.compact'
|
|
})
|
|
|
|
response.raise_for_status()
|
|
return response.json()['access_token']
|
|
|
|
def find_store(self, zipcode, kroken):
|
|
headers = {
|
|
'Authorization': f'Bearer {kroken}',
|
|
}
|
|
params = {
|
|
'filter.zipCode.near': zipcode,
|
|
'filter.limit': 1,
|
|
}
|
|
response = requests.get('https://api.kroger.com/v1/locations', headers=headers, params=params)
|
|
return response.json()
|
|
|
|
def search_product(self, product, loc_id, kroken):
|
|
logger.info(f"Searching for {product}...")
|
|
headers = {
|
|
'Authorization': f'Bearer {kroken}',
|
|
}
|
|
params = {
|
|
'filter.term': product,
|
|
'filter.locationId': loc_id,
|
|
'filter.limit': 10
|
|
}
|
|
response = requests.get('https://api.kroger.com/v1/products', headers=headers, params=params)
|
|
return response.json()
|
|
|
|
def garfshop(self, query):
|
|
try:
|
|
query = query.split()
|
|
kroken = self.kroger_token()
|
|
product = query[-2]
|
|
zipcode = query[-1]
|
|
loc_data = self.find_store(zipcode, kroken)
|
|
loc_id = loc_data['data'][0]['locationId']
|
|
store_name = loc_data['data'][0]['name']
|
|
product_query = self.search_product(product, loc_id, kroken)
|
|
products = product_query['data']
|
|
sorted_products = sorted(products, key=lambda item: item['items'][0]['price']['regular'])
|
|
response = f"Prices for `{product}` at `{store_name}` near `{zipcode}`:\n"
|
|
for item in sorted_products:
|
|
product_name = item['description']
|
|
price = item['items'][0]['price']['regular']
|
|
response += f"- `${price}`: {product_name} \n"
|
|
return response
|
|
except Exception as e:
|
|
return e
|