Initial release of scripts and knowledge base

llm_tools/fleetyard.py

@@ -0,0 +1,108 @@
+#!/usr/bin/env python3
+
+import requests
+import sqlite3
+import configparser
+from collections import defaultdict
+import json
+
+# Testing out the fleetyard API
+
+class FleetyardAPI:
+    def __init__(self, base_url):
+        self.base_url = base_url
+        self.session = requests.Session()
+
+    def create_session(self, ini_path="fleetyard_login.ini"):
+        """Create a new session using credentials from an INI file."""
+        config = configparser.ConfigParser()
+        config.read(ini_path)
+        login = config.get('login', 'username', fallback=None)
+        password = config.get('login', 'password', fallback=None)
+        remember_me = config.getboolean('login', 'rememberMe', fallback=True)
+        if not login or not password:
+            raise ValueError("Missing login or password in fleetyard_login.ini")
+        payload = {
+            "login": login,
+            "password": password,
+            "rememberMe": remember_me,
+        }
+        response = self.session.post(f"{self.base_url}/sessions", json=payload)
+        response.raise_for_status()
+        return response.json()
+
+    def get_igns_fleet(self):
+        """Get the IGNs fleet."""
+        fleet_url = f"{self.base_url}/fleets/igns/vehicles/export"
+        response = self.session.get(fleet_url)
+        response.raise_for_status()
+        return response.json()
+
+def process_fleet_data(fleet_data):
+    """Groups fleet data by ship and aggregates owners."""
+    ship_owners = defaultdict(list)
+    for ship in fleet_data:
+        if "username" in ship:
+            key = (ship["manufacturerName"], ship["name"])
+            ship_owners[key].append(ship["username"])
+    return ship_owners
+
+def store_in_database(ship_owners):
+    """Stores the processed fleet data in a SQLite database."""
+    db_file = "fleet.db"
+    conn = sqlite3.connect(db_file)
+    cursor = conn.cursor()
+
+    # Create table
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS ship_owner_list (
+            manufacturerName TEXT,
+            name TEXT,
+            usernames TEXT,
+            PRIMARY KEY (manufacturerName, name)
+        )
+    ''')
+
+    # Clear existing data to prevent duplicates on re-runs
+    cursor.execute('DELETE FROM ship_owner_list')
+
+    # Insert new data
+    for (manufacturer, ship_name), owners in ship_owners.items():
+        usernames_str = ", ".join(sorted(owners))
+        cursor.execute(
+            "INSERT INTO ship_owner_list (manufacturerName, name, usernames) VALUES (?, ?, ?)",
+            (manufacturer, ship_name, usernames_str)
+        )
+
+    conn.commit()
+    conn.close()
+    print(f"Data successfully stored in {db_file}")
+
+if __name__ == "__main__":
+    base_url = "https://api.fleetyards.net/v1"
+    api = FleetyardAPI(base_url)
+
+    try:
+        # Create a session
+        session_response = api.create_session()
+        print("Session created.")
+
+        # Get the IGNs fleet
+        if session_response.get("code") == "success":
+            fleet_json = api.get_igns_fleet()
+            print("IGNs fleet data retrieved.")
+
+            # Process the data
+            processed_data = process_fleet_data(fleet_json)
+
+            # Store in database
+            store_in_database(processed_data)
+
+        else:
+            print("Failed to create session, cannot retrieve fleet.")
+
+    except requests.exceptions.HTTPError as e:
+        print(f"An HTTP error occurred: {e}")
+        print(f"Response body: {e.response.text}")
+    except requests.exceptions.RequestException as e:
+        print(f"A request error occurred: {e}")