80 lines
2.6 KiB
Python
80 lines
2.6 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
|