mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 14:08:08 +01:00
Spike of football fixture lookup.
This commit is contained in:
59
tools/find_fixtures.py
Normal file
59
tools/find_fixtures.py
Normal file
@@ -0,0 +1,59 @@
|
||||
import os
|
||||
import requests
|
||||
from datetime import datetime
|
||||
from dotenv import load_dotenv
|
||||
|
||||
BASE_URL = 'https://api.football-data.org/v4'
|
||||
|
||||
def find_fixtures(args: dict) -> dict:
|
||||
load_dotenv(override=True)
|
||||
api_key = os.getenv("FOOTBALL_DATA_API_KEY", "YOUR_DEFAULT_KEY")
|
||||
|
||||
team_name = args.get("team")
|
||||
month = args.get("month")
|
||||
headers = {'X-Auth-Token': api_key}
|
||||
team_name = team_name.lower()
|
||||
month = month.capitalize()
|
||||
|
||||
try:
|
||||
month_number = datetime.strptime(month, "%B").month
|
||||
except ValueError:
|
||||
return {"error": "Invalid month provided."}
|
||||
|
||||
# Fetch team ID
|
||||
teams_response = requests.get(f'{BASE_URL}/competitions/PL/teams', headers=headers)
|
||||
if teams_response.status_code != 200:
|
||||
return {"error": "Failed to fetch teams data."}
|
||||
|
||||
teams_data = teams_response.json()
|
||||
team_id = None
|
||||
for team in teams_data['teams']:
|
||||
if team_name in team['name'].lower():
|
||||
team_id = team['id']
|
||||
break
|
||||
|
||||
if not team_id:
|
||||
return {"error": "Team not found."}
|
||||
|
||||
# Fetch fixtures
|
||||
match_date_from = f'2025-{month_number:02}-01'
|
||||
match_date_to = f'2025-{month_number:02}-31'
|
||||
print(f'{BASE_URL}/teams/{team_id}/matches?dateFrom={match_date_from}&dateTo={match_date_to}')
|
||||
fixtures_response = requests.get(f'{BASE_URL}/teams/{team_id}/matches?dateFrom={match_date_from}&dateTo={match_date_to}', headers=headers)
|
||||
if fixtures_response.status_code != 200:
|
||||
return {"error": "Failed to fetch fixtures data."}
|
||||
|
||||
fixtures_data = fixtures_response.json()
|
||||
matching_fixtures = []
|
||||
|
||||
for match in fixtures_data['matches']:
|
||||
print(match)
|
||||
match_date = datetime.strptime(match['utcDate'], "%Y-%m-%dT%H:%M:%SZ")
|
||||
if match['competition']['code'] == 'PL':
|
||||
matching_fixtures.append({
|
||||
"date": match_date.strftime("%Y-%m-%d"),
|
||||
"homeTeam": match['homeTeam']['name'],
|
||||
"awayTeam": match['awayTeam']['name'],
|
||||
})
|
||||
|
||||
return {"fixtures": matching_fixtures}
|
||||
Reference in New Issue
Block a user