gameday_api 0.5.0 → 0.5.1

data/lib/db_importer.rb

@@ -0,0 +1,205 @@
+require 'pitchfx_db_manager'
+require 'game'
+require 'date'
+
+
+
+class DbImporter
+  
+# player
+
+# team
+# game
+# atbat
+# pitch
+# pitch type
+# game type
+# umpire
+
+  def initialize(host, user, password, db_name)
+    @db = PitchfxDbManager.new(host, user, password, db_name)
+  end
+  
+  
+  def import_team_for_month(team_abbrev, year, month)
+    start_date = Date.new(year.to_i, month.to_i) # first day of month
+    end_date = (start_date >> 1)-1 # last day of month
+    ((start_date)..(end_date)).each do |dt| 
+      puts year.to_s + '/' + month.to_s + '/' + dt.day
+      team = Team.new('det')
+      games = team.games_for_date(year, month, dt.day.to_s)
+      games.each do |game|
+        import_for_game(game.gid)
+      end
+    end
+  end
+  
+  
+  def import_for_month(year, month)
+    start_date = Date.new(year.to_i, month.to_i) # first day of month
+    end_date = (start_date >> 1)-1 # last day of month
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      import_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def import_for_range(year, start_month, start_day, end_month, end_day)
+    start_date = Date.new(year.to_i, start_month.to_i, start_day.to_i)
+    end_date = Date.new(year.to_i, end_month.to_i, end_day.to_i)
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      import_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def import_for_date(year, month, day)
+    games = Game.find_by_date(year, month, day)
+    if games && games.length > 0
+      games.each do |game|
+        import_for_game(game.gid)
+      end
+    end
+  end
+  
+
+  def import_for_game(gid)
+    game = Game.new(gid)
+    visitor_id = @db.find_or_create_team(game.visiting_team)
+    home_id = @db.find_or_create_team(game.home_team)
+    import_players_for_game(gid, visitor_id, home_id)
+    game_id = @db.find_or_create_game(game, visitor_id, home_id)
+    abs = game.get_atbats
+    abs.each do |ab|
+      atbat_id = @db.find_or_create_atbat(ab, game_id)
+      pitches = ab.pitches
+      pitches.each do |pitch|
+        @db.find_or_create_pitch(pitch, atbat_id)
+      end
+    end
+    import_pitcher_lines_for_game(gid)
+  end
+  
+  
+  def import_players_for_game(gid, visitor_id, home_id)
+    players = Players.new
+    players.load_from_id(gid)
+    away = players.rosters[0]
+    home = players.rosters[1]    
+    away.players.each do |player|
+      @db.find_or_create_player(player, visitor_id) 
+    end
+    home.players.each do |player|
+      @db.find_or_create_player(player, home_id)
+    end
+  end
+  
+  
+  def import_pitcher_lines_for_month(year, month)
+    start_date = Date.new(year.to_i, month.to_i) # first day of month
+    end_date = (start_date >> 1)-1 # last day of month
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      import_pitcher_lines_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def import_pitcher_lines_for_date(year, month, day)
+    games = Game.find_by_date(year, month, day)
+    if games && games.length > 0
+      games.each do |game|
+        import_pitcher_lines_for_game(game.gid)
+      end
+    end
+  end
+  
+  
+  def import_pitcher_lines_for_range(year, start_month, start_day, end_month, end_day)
+    start_date = Date.new(year.to_i, start_month.to_i, start_day.to_i)
+    end_date = Date.new(year.to_i, end_month.to_i, end_day.to_i)
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      import_pitcher_lines_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def import_pitcher_lines_for_game(gid)
+    game = Game.new(gid)
+    pitchers = game.get_boxscore.pitchers
+    pitchers.each do |h_or_a_pitchers|
+      h_or_a_pitchers.each do |pitcher|
+        @db.find_or_create_pitcher_line(pitcher, game)
+      end
+    end
+  end
+  
+  
+  def set_status_for_range(year, start_month, start_day, end_month, end_day)
+    start_date = Date.new(year.to_i, start_month.to_i, start_day.to_i)
+    end_date = Date.new(year.to_i, end_month.to_i, end_day.to_i)
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      set_status_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def set_status_for_date(year, month, day)
+    games = Game.find_by_date(year, month, day)
+    if games && games.length > 0
+      games.each do |game|
+        set_status_for_game(game.gid)
+      end
+    end
+  end
+  
+  
+  def set_status_for_game(gid)
+    game = Game.new(gid)
+    @db.update_status_for_game(game)
+  end
+  
+  
+  def import_rosters_for_range(year, start_month, start_day, end_month, end_day)
+    start_date = Date.new(year.to_i, start_month.to_i, start_day.to_i)
+    end_date = Date.new(year.to_i, end_month.to_i, end_day.to_i)
+    ((start_date)..(end_date)).each do |dt| 
+      puts dt.year.to_s + '/' + dt.month.to_s + '/' + dt.day.to_s
+      import_rosters_for_date(dt.year.to_s, dt.month.to_s, dt.day.to_s)
+    end
+  end
+  
+  
+  def import_rosters_for_date(year, month, day)
+    games = Game.find_by_date(year, month, day)
+    if games && games.length > 0
+      games.each do |game|
+        import_rosters_for_game(game.gid)
+      end
+    end
+  end
+  
+  
+  def import_rosters_for_game(gid)
+    game = Game.new(gid)
+    away_id = @db.find_or_create_team(game.visiting_team)
+    home_id = @db.find_or_create_team(game.home_team)
+    players = Players.new
+    players.load_from_id(gid)
+    away = players.rosters[0]
+    home = players.rosters[1]     
+    roster_ids = @db.find_or_create_rosters(gid, away_id, home_id) 
+    away.players.each do |player|
+      @db.find_or_create_roster_player(player, roster_ids[0]) 
+    end
+    home.players.each do |player|
+      @db.find_or_create_roster_player(player, roster_ids[1])
+    end
+  end
+  
+  
+end

data/lib/event.rb

@@ -0,0 +1,14 @@
+class Event
+  
+  attr_accessor :number, :inning, :description, :team
+  
+  def load(element, home_or_away)
+  	@xml_doc = element
+  	@team = home_or_away
+    @number = element.attributes["number"]
+    @inning = element.attributes["inning"]
+    @description = element.attributes["description"]
+  end
+  
+  
+end

data/lib/event_log.rb

@@ -0,0 +1,103 @@
+require 'event'
+require 'gameday_fetcher'
+
+
+# Parses the MLB Gameday eventLog.xml file
+class EventLog
+  
+  attr_accessor :home_team, :away_team, :gid
+  attr_accessor :home_events, :away_events
+  attr_accessor :max_inning
+  
+  
+  # Loads the eventLog XML from the MLB gameday server and parses it using REXML
+  def load_from_id(gid)
+    @gid = gid
+    @xml_data = GamedayFetcher.fetch_eventlog(gid)
+    @xml_doc = REXML::Document.new(@xml_data)
+    if @xml_doc.root
+      set_teams
+      set_events
+    end
+  end
+  
+  
+  # Sets the team names for the teams involved in this game
+  def set_teams
+  	@xml_doc.elements.each("game/team[@home_team='false']") do |element| 
+  		@away_team = element.attributes["name"]
+  	end
+  	@xml_doc.elements.each("game/team[@home_team='true']") do |element| 
+  		@home_team = element.attributes["name"]
+  	end
+  end
+  
+  
+  def set_events
+    @home_events, @away_events = [], []
+    @max_inning = 0
+    # set away team events
+    @xml_doc.elements.each("game/team[@home_team='false']/event") { |element| 
+      event = Event.new
+      event.load(element, 'away')
+      @max_inning = event.inning.to_i unless event.inning.to_i < @max_inning
+      @away_events.push event
+    }
+    # set home team events
+    @xml_doc.elements.each("game/team[@home_team='true']/event") { |element| 
+      event = Event.new
+      event.load(element, 'home')
+      @home_events.push event
+    }
+  end
+  
+  
+  def events_by_inning(inning)
+  	events = []
+  	@away_events.each do |event|
+  		if event.inning == inning
+  		  events << event
+  		end
+  	end
+  	@home_events.each do |event|
+  		if event.inning == inning
+  		  events << event
+  		end
+  	end
+  	events
+  end
+  
+  
+  def dump
+  	(1..@max_inning).each do |inning|
+  		dump_inning(inning.to_s)
+  	end
+  end
+  
+  
+  def dump_inning(inning)
+  	puts 'Inning - ' + inning.to_s
+  	events = events_by_inning(inning)
+  	events.each do |event|
+  		puts event.description
+  	end
+  end
+  
+  
+  # NOT IMPLEMENTED YET
+  def dump_to_file
+    GamedayUtil.save_file("eventlog.html", self.to_html('eventlog.html.erb'))
+  end
+  
+  
+  # NOT IMPLEMENTED YET
+  # Converts the eventlog into a formatted HTML representation.
+  # Relies on the eventlog.html.erb template for describing the layout
+  def to_html(template_filename)
+    gameday_info = GamedayUtil.parse_gameday_id('gid_' + gid)
+    template = ERB.new File.new(File.expand_path(File.dirname(__FILE__) + "/" + template_filename)).read, nil, "%"  
+    return template.result(binding)
+  end
+  
+  
+end