Create kroger.py
This commit is contained in:
parent
3a35889a0a
commit
6ca5668fdb
75
kroger.py
Normal file
75
kroger.py
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
import config
|
||||||
|
import requests
|
||||||
|
import schedule
|
||||||
|
import time
|
||||||
|
from datetime import datetime
|
||||||
|
import sqlite3
|
||||||
|
|
||||||
|
DATABASE = 'kroger_prices.db'
|
||||||
|
|
||||||
|
def get_token():
|
||||||
|
"""Authenticate with Kroger API and get access token."""
|
||||||
|
response = requests.post(config.TOKEN_URL, data={
|
||||||
|
'grant_type': 'client_credentials',
|
||||||
|
'scope': 'product.compact'
|
||||||
|
}, auth=(config.CLIENT_ID, config.CLIENT_SECRET))
|
||||||
|
|
||||||
|
response.raise_for_status()
|
||||||
|
return response.json()['access_token']
|
||||||
|
|
||||||
|
def get_prices(token, zip=config.ZIP_CODE):
|
||||||
|
"""Fetch egg prices from Kroger API."""
|
||||||
|
headers = {'Authorization': f'Bearer {token}'}
|
||||||
|
params = {
|
||||||
|
'filter.term': 'eggs',
|
||||||
|
'filter.locationId': zip,
|
||||||
|
'filter.limit': 10
|
||||||
|
}
|
||||||
|
|
||||||
|
response = requests.get('https://api.kroger.com/v1/products', headers=headers, params=params)
|
||||||
|
response.raise_for_status()
|
||||||
|
return response.json()['data']
|
||||||
|
|
||||||
|
def db_save(data):
|
||||||
|
"""Save product data to SQLite database."""
|
||||||
|
conn = sqlite3.connect(DATABASE)
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
# Create table if not exists
|
||||||
|
cursor.execute('''
|
||||||
|
CREATE TABLE IF NOT EXISTS prices (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
product_name TEXT,
|
||||||
|
price REAL,
|
||||||
|
currency TEXT,
|
||||||
|
date TEXT
|
||||||
|
)
|
||||||
|
''')
|
||||||
|
|
||||||
|
# Insert data
|
||||||
|
for item in data:
|
||||||
|
product_name = item['description']
|
||||||
|
price = item['items'][0]['price']['regular']
|
||||||
|
currency = item['items'][0]['price']['currency']
|
||||||
|
date = datetime.now().strftime('%Y-%m-%d')
|
||||||
|
|
||||||
|
cursor.execute('INSERT INTO prices (product_name, price, currency, date) VALUES (?, ?, ?, ?)',
|
||||||
|
(product_name, price, currency, date))
|
||||||
|
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
def daily_check():
|
||||||
|
"""Perform daily check for egg prices."""
|
||||||
|
token = get_token()
|
||||||
|
egg_price = get_prices(token)
|
||||||
|
db_save(egg_price)
|
||||||
|
|
||||||
|
# Schedule the script to run daily at the specified time
|
||||||
|
# schedule.every().day.at(config.CHECK_TIME).do(daily_check)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
while True:
|
||||||
|
daily_check()
|
||||||
|
# schedule.run_pending()
|
||||||
|
# time.sleep(60)
|
Loading…
Reference in New Issue
Block a user