crate/garfbot
0
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

format etc

Browse Source
This commit is contained in:
crate
2025-06-05 17:25:43 -05:00
parent 2415a8146d
commit 6b9c2b638a
9 changed files with 284 additions and 216 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
garfpy/kroger.py
View File

@ -8,44 +8,51 @@ 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()
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}'
"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 = 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']
return response.json()["access_token"]
def find_store(self, zipcode, kroken):
headers = {
'Authorization': f'Bearer {kroken}',
"Authorization": f"Bearer {kroken}",
}
params = {
'filter.zipCode.near': zipcode,
'filter.limit': 1,
"filter.zipCode.near": zipcode,
"filter.limit": 1,
}
response = requests.get('https://api.kroger.com/v1/locations', headers=headers, params=params)
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}',
"Authorization": f"Bearer {kroken}",
}
params = {
'filter.term': product,
'filter.locationId': loc_id,
'filter.limit': 10
"filter.term": product,
"filter.locationId": loc_id,
"filter.limit": 10,
}
response = requests.get('https://api.kroger.com/v1/products', headers=headers, params=params)
response = requests.get(
"https://api.kroger.com/v1/products", headers=headers, params=params
)
return response.json()
def garfshop(self, query):
@ -55,15 +62,17 @@ class Kroger:
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']
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'])
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']
product_name = item["description"]
price = item["items"][0]["price"]["regular"]
response += f"- `${price}`: {product_name} \n"
return response
except Exception as e: