Forráskód Böngészése

Change "output" to "local"

theenglishway (time) 2 éve
szülő
commit
0f67373aa0
2 módosított fájl, 21 hozzáadás és 21 törlés
  1. 14 14
      src/main/scala/Extract.scala
  2. 7 7
      src/main/scala/Main.scala

+ 14 - 14
src/main/scala/Extract.scala

@@ -6,47 +6,47 @@ import upickle.default._
 
 object Extract {
   def writeOrReadBack[T: ReadWriter](
-      file: os.Path,
+      localCopy: os.Path,
       getRemoteData: () => T
   ): T = {
-    if (!os.exists(file)) {
+    if (!os.exists(localCopy)) {
       val data = getRemoteData()
-      Utils.writeToFile[T](data, file)
+      Utils.writeToFile[T](data, localCopy)
 
       data
     } else {
-      println(s"file $file already exists, reading back from it")
-      Utils.readFromFile[T](file)
+      println(s"file $localCopy already exists, reading back from it")
+      Utils.readFromFile[T](localCopy)
     }
   }
 
-  def getTeams(output: os.Path, filterTeams: Team => Boolean) = {
+  def getTeams(local: os.Path, filterTeams: Team => Boolean) = {
     def getData() = {
       EndpointResponse
         .getTeams()
         .filter(filterTeams)
     }
 
-    writeOrReadBack(output, getData)
+    writeOrReadBack(local, getData)
   }
 
-  def getGames(output: os.Path, season: Int, teamIds: List[Int]) = {
+  def getGames(local: os.Path, season: Int, teamIds: List[Int]) = {
     def getData() = {
       EndpointResponse.getGames(2021, teamIds)
     }
 
-    writeOrReadBack(output, getData)
+    writeOrReadBack(local, getData)
   }
 
-  def getStats(file: os.Path, gameIds: List[Int]) = {
-    if (!os.exists(file)) {
+  def getStats(local: os.Path, gameIds: List[Int]) = {
+    if (!os.exists(local)) {
       val data = EndpointResponse.getStats(gameIds)
-      Utils.writeToFile(PlayerGameStats.as_json_value(data), file)
+      Utils.writeToFile(PlayerGameStats.as_json_value(data), local)
 
       data
     } else {
-      println(s"file $file already exists, reading back from it")
-      Utils.readFromFile[List[PlayerGameStats]](file)
+      println(s"file $local already exists, reading back from it")
+      Utils.readFromFile[List[PlayerGameStats]](local)
     }
   }
 }

+ 7 - 7
src/main/scala/Main.scala

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