Buyable ships: Switched from starcitizen.tools to UEX for more reliable information
This commit is contained in:
84
llm_tools/get_buyable_ships.py
Normal file
84
llm_tools/get_buyable_ships.py
Normal file
@@ -0,0 +1,84 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import requests
|
||||
import sqlite3
|
||||
|
||||
# --- Configuration ---
|
||||
API_URL = "https://api.uexcorp.space/2.0/vehicles_purchases_prices_all"
|
||||
with open("uex_api_key", "r") as f:
|
||||
BEARER_TOKEN = f.read().strip()
|
||||
|
||||
DB_NAME = "buyable_ships.db"
|
||||
TABLE_NAME = "buyable_ships"
|
||||
|
||||
def setup_database():
|
||||
"""
|
||||
Sets up the SQLite database and creates the table if it doesn't exist.
|
||||
The table uses a composite primary key (id_vehicle, id_terminal)
|
||||
to ensure each vehicle at each terminal has only one latest entry.
|
||||
"""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# Using "IF NOT EXISTS" prevents errors on subsequent runs.
|
||||
# The schema is derived from your provided image.
|
||||
# We use INSERT OR REPLACE later, so a primary key is important.
|
||||
# (id_vehicle, id_terminal) is a good candidate for a unique key.
|
||||
cursor.execute(f"""
|
||||
CREATE TABLE IF NOT EXISTS {TABLE_NAME} (
|
||||
id_vehicle TEXT,
|
||||
id_terminal TEXT,
|
||||
price_buy REAL,
|
||||
vehicle_name TEXT,
|
||||
terminal_name TEXT,
|
||||
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (id_vehicle, id_terminal)
|
||||
)
|
||||
""")
|
||||
conn.commit()
|
||||
conn.close()
|
||||
print("Database setup complete.")
|
||||
|
||||
def fetch_data_from_api():
|
||||
"""
|
||||
Fetches the latest vehicle purchase price data from the UAX Corp API.
|
||||
Returns the data as a list of dictionaries or None if an error occurs.
|
||||
"""
|
||||
headers = {"Authorization": f"Bearer {BEARER_TOKEN}"}
|
||||
try:
|
||||
response = requests.get(API_URL, headers=headers)
|
||||
response.raise_for_status() # Raise an error for HTTP errors
|
||||
data = response.json()
|
||||
return data.get("data", [])
|
||||
except requests.RequestException as e:
|
||||
print(f"Error fetching API data: {e}")
|
||||
return None
|
||||
|
||||
def save_data_to_db(data):
|
||||
"""
|
||||
Saves the fetched vehicle purchase price data to the SQLite database.
|
||||
"""
|
||||
conn = sqlite3.connect(DB_NAME)
|
||||
cursor = conn.cursor()
|
||||
|
||||
for item in data:
|
||||
cursor.execute(f"""
|
||||
INSERT OR REPLACE INTO {TABLE_NAME} (id_vehicle, id_terminal, price_buy, vehicle_name, terminal_name)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
""", (
|
||||
item.get("id_vehicle"),
|
||||
item.get("id_terminal"),
|
||||
item.get("price_buy"),
|
||||
item.get("vehicle_name"),
|
||||
item.get("terminal_name")
|
||||
))
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
print("Data saved to database.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
setup_database()
|
||||
data = fetch_data_from_api()
|
||||
if data:
|
||||
save_data_to_db(data)
|
||||
Reference in New Issue
Block a user