Merge pull request #4 from R-Brook/nflreadpy

Use Nflreadpy

Author: R-Brook

.github/workflows/nextjs.yml

@@ -1,5 +1,4 @@
 # Build and deploy a Next.js site to GitHub Pages
-#
 name: Deploy Next.js site to Pages
 
 on:
@@ -42,17 +41,19 @@ jobs:
           pip install -r requirements.txt
 
       - name: Run Python script
-        env:
-          BALLDONTLIEAPI_KEY: ${{ secrets.BALLDONTLIEAPI_KEY }}
-        run: python scripts/fetch-team-data.py
+        run: |
+          python scripts/fetch-team-data.py
+          python scripts/fetch-qb-data.py
 
       # Upload the file the script produced
       - name: Upload Python output
         uses: actions/upload-artifact@v4
         with:
           name: python-output
           # File the script produces
-          path: scripts/team-data.ts
+          path: |
+            scripts/team-data.ts
+            scripts/qb-data.ts
 
   # Build job
   build:
@@ -77,6 +78,7 @@ jobs:
       - name: Move Python output into project
         run: |
           cp ./python-output/team-data.ts scripts/team-data.ts
+          cp ./python-output/qb-data.ts scripts/qb-data.ts
 
       - name: Detect package manager
         id: detect-package-manager
@@ -120,6 +122,10 @@ jobs:
       - name: Install dependencies
         run: ${{ steps.detect-package-manager.outputs.manager }} ${{ steps.detect-package-manager.outputs.command }}
 
+      - name: Verify generated data files
+        run: |
+          ls -l scripts
+
       - name: Build with Next.js
         run: ${{ steps.detect-package-manager.outputs.runner }} next build
 

.gitignore

@@ -2,13 +2,6 @@
 
 # dependencies
 /node_modules
-/.pnp
-.pnp.*
-.yarn/*
-!.yarn/patches
-!.yarn/plugins
-!.yarn/releases
-!.yarn/versions
 
 # testing
 /coverage
@@ -26,9 +19,6 @@
 
 # debug
 npm-debug.log*
-yarn-debug.log*
-yarn-error.log*
-.pnpm-debug.log*
 
 # env files (can opt-in for committing if needed)
 .env*
@@ -39,3 +29,6 @@ yarn-error.log*
 # typescript
 *.tsbuildinfo
 next-env.d.ts
+
+# script outputs
+/scripts/*.ts

README.md

@@ -1,5 +1,7 @@
 This is a [Next.js](https://nextjs.org) project bootstrapped with [`create-next-app`](https://nextjs.org/docs/pages/api-reference/create-next-app).
 
+Uses [nflreadpy](https://nflreadpy.nflverse.com/) to fetch data.
+
 ## Getting Started
 
 First, run the development server:

pages/index.tsx

@@ -1,3 +1,4 @@
+import { qbData } from "@/scripts/qb-data"
 import { teamData } from "@/scripts/team-data"
 import Head from "next/head"
 
@@ -10,7 +11,12 @@ export default function Home() {
       <main>
         <div>
           <h1>NFL Stats</h1>
-          {teamData ? "has fetched data" : "has not fetched data"}
+          <span className="block">
+            {teamData ? "has fetched team data" : "has not fetched team data"}
+          </span>
+          <span className="block">
+            {qbData ? "has fetched QB data" : "has not fetched QB data"}
+          </span>
         </div>
       </main>
     </div>

requirements.txt

@@ -1,2 +1,2 @@
-balldontlie
-python-dotenv
+nflreadpy
+python-dotenv

scripts/fetch-qb-data.py

@@ -0,0 +1,34 @@
+import json
+import polars as pl
+import nflreadpy as nfl
+
+def parse_qbs():
+    """
+    Fetch current-season QB stats and write them to scripts/qb-data.ts
+    """
+    try:
+        # Fetch player stats (current season by default)
+        player_stats = nfl.load_player_stats(summary_level="reg")
+
+        # Filter to QBs
+        qb_data = (
+            player_stats
+            .filter(pl.col("position") == "QB")
+            .to_dicts()
+        )
+
+        # Write the data to a TypeScript file
+        output_path = "scripts/qb-data.ts"
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write("// file generated by fetch-qb-data.py\n\n")
+            f.write("export const qbData = ")
+            json.dump(qb_data, f, indent=2)
+            f.write(";\n")
+
+        print(f"Created file {output_path} with {len(qb_data)} QBs")
+
+    except Exception as e:
+        print(f"ERROR: {e}")
+        raise
+
+parse_qbs()

scripts/fetch-team-data.py

@@ -1,33 +1,27 @@
 import json
-import os
-from balldontlie import BalldontlieAPI
-from dotenv import load_dotenv
+import nflreadpy as nfl
 
-# Only load .env if it exists
-if os.path.exists(".env"):
-    load_dotenv()
+def parse_teams():
+    """
+    Fetch NFL teams via the API, convert them to dictionaries,
+    and write to a TypeScript file (scripts/team-data.ts).
+    """
+    try:
+        # Fetch data (Polars DataFrame)
+        team_data = nfl.load_teams().to_dicts()
 
-# Fetch the API key from environment
-api_key = os.getenv("BALLDONTLIEAPI_KEY")
-if not api_key:
-    raise ValueError(
-        "BALLDONTLIEAPI_KEY not set. "
-        "Set it in a local .env file or as a GitHub Actions secret."
-    )
+        # Write the data to a TypeScript file
+        output_path = "scripts/team-data.ts"
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write("// file generated by fetch-team-data.py\n\n")
+            f.write("export const teamData = ")
+            json.dump(team_data, f, indent=2)
+            f.write(";\n")
 
-# Initialise the API
-api = BalldontlieAPI(api_key=api_key)
+        print(f"Created file {output_path} with {len(team_data)} teams")
 
-# Fetch NFL teams data
-data = api.nfl.teams.list()
+    except Exception as e:
+        print(f"ERROR: {e}")
+        raise
 
-# Convert each team to a dictionary using Pydantic v2+ `.model_dump()`
-team_data = [team.model_dump() for team in data.data]
-
-# Write the data to a TypeScript file
-with open("scripts/team-data.ts", "w") as f:
-    f.write("// file generated by fetch-qb-data.py\n")
-    f.write("\n")
-    f.write("export const teamData = ")
-    json.dump(team_data, f, indent=2)
-    f.write("\n")
+parse_teams()