Cleaner file declaration

src/main/scala/Main.scala

@@ -5,9 +5,10 @@ import extract.Extract
 import transform.GamesAnalysis
 
 object Main extends App {
-  val teams_local = os.pwd / "teams.json"
-  val stats_local = os.pwd / "stats.json"
-  val games_local = os.pwd / "games.json"
+  val localDir = os.pwd / "local"
+  val localFiles = List("teams", "stats", "games")
+    .map(file => file -> localDir / s"$file.json")
+    .toMap
 
   val selectedTeams = List(
     "Phoenix Suns",
@@ -18,16 +19,17 @@ object Main extends App {
 
   val teamFilter = (team: Team) => selectedTeams.contains(team.full_name)
 
-  val teams = Extract.getTeams(teams_local, teamFilter)
+  val teams = Extract.getTeams(localFiles("teams"), teamFilter)
   println(teams)
 
-  val games = Extract.getGames(games_local, 2021, teams.map(_.id))
+  val games = Extract.getGames(localFiles("games"), 2021, teams.map(_.id))
   println(games.size)
 
-  val stats = Extract.getStats(stats_local, games.map(_.id))
+  val stats = Extract.getStats(localFiles("stats"), games.map(_.id))
   println(stats.size)
 
-  val analysis = GamesAnalysis(teams_local, games_local, stats_local)
+  val analysis =
+    GamesAnalysis(localFiles("teams"), localFiles("games"), localFiles("stats"))
 
   Map(
     "output/teams_games_df" -> analysis.teams_games.sort("game_id", "team_id"),