def test_calculate_real_lineup(self): league = League.objects.create(espn_id='930248', year='2014') file_browser = FileBrowser() rogues = Team.objects.create(league=league, espn_id='6') team2 = Team.objects.create(league=league, espn_id='1') game_html = file_browser.get_game(league, '1', 1) load_scores_from_game(league, 1, game_html) sc = Scorecard.objects.filter(week=1, team=rogues, actual=True) entries = ScorecardEntry.objects.filter(scorecard=sc) optimal_lineup = calculate_optimal_lineup(entries)
def test_load_settings(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') filebrowser = FileBrowser() html = filebrowser.get_settings(league.espn_id, league.year) public = get_public_on_from_settings(html) self.assertFalse(public)
def test_load_translog_source(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') team = Team.objects.create(espn_id='7', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '7') load_transactions_from_translog(html, '2014', team)
def test_load_translog_error2(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') team = Team.objects.create(espn_id='6', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '8') load_transactions_from_translog(html, '2014', team) foles = Player.objects.get(name='Nick Foles') DraftClaim.objects.get(player_added=foles)
def get_scraper(espn_user): if settings.LOCAL != True: scraper = EspnScraper() scraper.login(espn_user.username, espn_user.password) logger.info("returning espn scraper") return scraper else: if settings.DB_SCRAPE: logger.info("returning file scraper") fb = FileBrowser() fb.sleep = 0.1 return fb else: scraper = EspnScraper() scraper.login(espn_user.username, espn_user.password) logger.info("returning espn scraper") return scraper
def test_load_translog_error_allen_robinson(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team = Team.objects.create(espn_id='5', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '5') load_transactions_from_translog(html, '2014', team) allen_robinson = AddDrop.objects.filter(player__name='Allen Robinson') self.assertGreater(allen_robinson.count(), 0)
def handle(self, *args, **options): scraper = EspnScraper() scraper.login('gothamcityrogues', 'sincere1') file_browser = FileBrowser() league_scraper = LeagueScraper(scraper, file_browser) leagues = League.objects.filter(espn_id='930248', year='2014') print "length of leagues is %d" % len(leagues) for league in leagues: league_scraper.scrape_league(league)
def test_load_translog_error_failed_load(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='1880759', year='2014') team2 = Team.objects.create(espn_id='1', league=league, abbreviation='MM') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='Air') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('1880759', '2014', '1') load_transactions_from_translog(html, '2014', team2)
def test_load_translog_error_mohammed(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='1044126', year='2014') team2 = Team.objects.create(espn_id='11', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('1044126', '2014', '7') load_transactions_from_translog(html, '2014', team2) mo_sanu = AddDrop.objects.filter(player__name='Mohamed Sanu') self.assertGreater(mo_sanu.count(), 0)
def test_load_translog_error_alex_smith(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team2 = Team.objects.create(espn_id='11', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '11') load_transactions_from_translog(html, '2014', team2) alex_smith = AddDrop.objects.filter(player__name='Alex Smith') self.assertGreater(alex_smith.count(), 0)
def test_load_mside(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='451385', year='2014') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('451385', '2014', '9') team = Team.objects.get(league=league, espn_id='9') load_transactions_from_translog(html, '2014', team) maclin = Player.objects.get(name='Jeremy Maclin') trades = TradeEntry.objects.filter(players_added=maclin) self.assertGreater(trades.count(), 0) bad_trades = TradeEntry.objects.filter(players_removed=maclin) self.assertEqual(bad_trades.count(), 0)
def test_load_translog_error_ahmad_bradshaw(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='451385', year='2014') team2 = Team.objects.create(espn_id='12', league=league, abbreviation='MM') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='Air') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('451385', '2014', '12') load_transactions_from_translog(html, '2014', team2) ahmad_bradshaw = TradeEntry.objects.filter( players_added__name='Ahmad Bradshaw') self.assertGreater(ahmad_bradshaw.count(), 0)
def test_load_translog_error_toby_gerhart_2(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='BART') team2 = Team.objects.create(espn_id='10', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '10') load_transactions_from_translog(html, '2014', team2) toby_gerhart = TradeEntry.objects.filter( players_removed__name='Toby Gerhart') self.assertGreater(toby_gerhart.count(), 0)
def setUp(self): clearDb() self.browser = FileBrowser() self.sqlstore = SqlStore()
class ScraperTest(unittest.TestCase): def setUp(self): clearDb() self.browser = FileBrowser() self.sqlstore = SqlStore() def test_scrape_entrance(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') self.assertTrue(self.browser.has_entrance(espn_user)) html = self.browser.get_entrance(espn_user) leagues = get_leagues_from_entrance(html) self.assertEquals(len(leagues), 2) #self.assertIn(('Inglorious Basterds', '930248', '2013', '6'), leagues) self.assertIn(('Inglorious Basterds', '930248', '2014', '6'), leagues) self.assertIn(('Bizarro League III', '1880759', '2014', '9'), leagues) @unittest.skip("we dont scrape lineups in 2014 leagues currently") def test_scrape_players_from_lineup(self): league = League.objects.create(name='ib', espn_id='930248', year='2014') html = self.browser.get_roster(league, '1', 1) players = get_player_ids_from_lineup(html) self.assertIn('14874', players) self.assertEquals(len(players), 16) def test_scrape_teams_from_standings(self): league = League.objects.create(name='ib', espn_id='930248', year='2013') self.assertTrue(self.browser.has_standings(league)) teams = get_teams_from_standings(self.browser.get_standings(league)) self.assertEquals(len(teams), 12) self.assertIn(('5', 'Not a Prinhead ', 'Matthew Prin'), teams) self.assertIn(('4', "Geno's Cheesesteaks ", 'Matt Secko'), teams) def test_get_num_weeks_from_matchups(self): league = League.objects.create(name='ib', espn_id='930248', year='2013') self.assertTrue(self.browser.has_matchups(league, 1)) num_weeks = get_num_weeks_from_matchups( self.browser.get_matchups(league, 1)) self.assertEquals(num_weeks, 15) league = League.objects.create(name='ib', espn_id='930248', year='2014') self.assertTrue(self.browser.has_matchups(league, 1)) num_weeks = get_num_weeks_from_matchups( self.browser.get_matchups(league, 1)) self.assertEquals(num_weeks, 15) def test_get_matchups_from_matchups_page(self): league = League.objects.create(name='ib', espn_id='930248', year='2014') self.assertTrue(self.browser.has_matchups(league, 1)) matchups = get_teams_from_matchups( self.browser.get_matchups(league, 1), 1) self.assertEquals(len(matchups[0]), 6) for i in range(1, 13): self.assertTrue( str(i) in [item for sublist in matchups[0] for item in sublist]) self.assertEquals(len(matchups[1]), 1) self.assertEquals(matchups[1][0], 1) def test_get_matchups_during_playoffs(self): league = League.objects.create(name='ib', espn_id='930248', year='2014') self.assertTrue(self.browser.has_matchups(league, 15)) matchups = get_teams_from_matchups( self.browser.get_matchups(league, 15), 15) logger.info("matchup during playoffs %s" % str(matchups)) self.assertEquals(len(matchups[0]), 6) for i in range(1, 13): self.assertTrue( str(i) in [item for sublist in matchups[0] for item in sublist]) self.assertEquals(len(matchups[1]), 2) self.assertEquals(matchups[1][0], 16) self.assertEquals(matchups[1][1], 17)
class LeagueCreatorTest(unittest.TestCase): def setUp(self): clearDb() self.browser = FileBrowser() self.sqlstore = SqlStore() self.league_scraper = LeagueScraper(self.browser, self.sqlstore) def test_create_espn_user_leagues(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') self.assertFalse(self.sqlstore.has_entrance(espn_user)) self.league_scraper.scrape_espn_user_leagues(espn_user) self.assertTrue(self.sqlstore.has_entrance(espn_user)) def test_load_espn_user_leagues(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') self.assertFalse(self.sqlstore.has_entrance(espn_user)) self.league_scraper.scrape_espn_user_leagues(espn_user) self.league_scraper.load_espn_user_leagues(espn_user) self.assertGreater(len(Team.objects.filter(espn_user=espn_user)), 0) def test_load_legacy_entire_lineup(self): league = League.objects.create(espn_id='930248', year='2013') team_id = '1' week = 1 team = Team.objects.create(league=league, espn_id=team_id) roster_html = self.browser.get_roster(league, team_id, week) player_ids = get_player_ids_from_lineup(roster_html) for player_id in player_ids: player_html = self.browser.get_player(player_id) load_scores_from_playersheet(player_html, player_id, league.year) load_week_from_lineup(roster_html, week, team) scorecard = Scorecard.objects.get(team=team, week=1) self.assertEquals(scorecard.points, 96) @unittest.skip( "currently dont worry about loading players, legacy leagues only") def test_load_player(self): html = self.browser.get_player('2580', year='2014') brees = Player.objects.create(name='Drew Brees', espn_id='2580', position='QB') load_scores_from_playersheet(html, '2580', '2014') self.assertEquals(17, PlayerScoreStats.objects.all().count()) sc = ScoreEntry.objects.get(week=1, year='2014', player=brees) ps1 = sc.player_score_stats self.assertEquals(ps1.pass_yards, 333) self.assertEquals(ps1.pass_td, 1) self.assertEquals(ps1.default_points, 15) sc2 = ScoreEntry.objects.get(week=2, year='2014', player=brees) ps2 = sc2.player_score_stats self.assertEquals(ps2.pass_yards, 237) self.assertEquals(ps2.pass_td, 2) self.assertEquals(ps2.default_points, 15) def test_load_teams(self): league = League.objects.create(espn_id='930248', year='2014') user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') team = Team.objects.create(league=league, espn_user=espn_user, team_name='Gotham City Rogues', owner_name='William Prin', espn_id='6') html = self.browser.get_standings(league) logger.debug("test_load_teams(): ") load_teams_from_standings(html, league) teams = Team.objects.filter(league=league) self.assertEqual(len(teams), 12) rogues = Team.objects.get(league=league, espn_id='6') self.assertEqual(espn_user, rogues.espn_user) self.assertEquals(rogues.abbreviation, 'GCR') def test_load_game(self): league = League.objects.create(espn_id='930248', year='2014') team = Team.objects.create(espn_id='1', league=league) team2 = Team.objects.create(espn_id='6', league=league) game_html = self.browser.get_game(league, '1', 1) load_scores_from_game(league, 1, game_html) scorecard = Scorecard.objects.get(team=team, week=1) scorecard2 = Scorecard.objects.get(team=team2, week=1) luck = Player.objects.get(name="Andrew Luck") self.assertEquals(scorecard.points, 104) self.assertEquals(scorecard2.points, 69) self.assertEquals(luck.position, 'QB') def test_load_translog_error(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team = Team.objects.create(espn_id='7', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '7') load_transactions_from_translog(html, '2014', team) def test_load_translog_error2(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') team = Team.objects.create(espn_id='6', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '8') load_transactions_from_translog(html, '2014', team) foles = Player.objects.get(name='Nick Foles') DraftClaim.objects.get(player_added=foles) def test_load_translog_source(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') team = Team.objects.create(espn_id='7', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '7') load_transactions_from_translog(html, '2014', team) def test_load_settings(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='930248', year='2014') filebrowser = FileBrowser() html = filebrowser.get_settings(league.espn_id, league.year) public = get_public_on_from_settings(html) self.assertFalse(public) def test_load_mside(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='451385', year='2014') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('451385', '2014', '9') team = Team.objects.get(league=league, espn_id='9') load_transactions_from_translog(html, '2014', team) maclin = Player.objects.get(name='Jeremy Maclin') trades = TradeEntry.objects.filter(players_added=maclin) self.assertGreater(trades.count(), 0) bad_trades = TradeEntry.objects.filter(players_removed=maclin) self.assertEqual(bad_trades.count(), 0) def test_load_translog_error_eli_manning(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team = Team.objects.create(espn_id='7', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '8') load_transactions_from_translog(html, '2014', team) eli_manning = AddDrop.objects.filter(player__name='Eli Manning') self.assertGreater(eli_manning.count(), 0) def test_load_translog_error_allen_robinson(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team = Team.objects.create(espn_id='5', league=league) filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '5') load_transactions_from_translog(html, '2014', team) allen_robinson = AddDrop.objects.filter(player__name='Allen Robinson') self.assertGreater(allen_robinson.count(), 0) def test_load_translog_error_toby_gerhart(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team = Team.objects.create(espn_id='2', league=league, abbreviation='WAGN') team = Team.objects.create(espn_id='5', league=league, abbreviation='BART') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '2') load_transactions_from_translog(html, '2014', team) toby_gerhart = TradeEntry.objects.filter( players_added__name='Toby Gerhart') self.assertGreater(toby_gerhart.count(), 0) def test_load_translog_error_toby_gerhart_2(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='BART') team2 = Team.objects.create(espn_id='10', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('976898', '2014', '10') load_transactions_from_translog(html, '2014', team2) toby_gerhart = TradeEntry.objects.filter( players_removed__name='Toby Gerhart') self.assertGreater(toby_gerhart.count(), 0) def test_load_translog_error_alex_smith(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='976898', year='2014') team2 = Team.objects.create(espn_id='11', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('930248', '2014', '11') load_transactions_from_translog(html, '2014', team2) alex_smith = AddDrop.objects.filter(player__name='Alex Smith') self.assertGreater(alex_smith.count(), 0) def test_load_translog_error_mohammed(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='1044126', year='2014') team2 = Team.objects.create(espn_id='11', league=league, abbreviation='WAGN') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('1044126', '2014', '7') load_transactions_from_translog(html, '2014', team2) mo_sanu = AddDrop.objects.filter(player__name='Mohamed Sanu') self.assertGreater(mo_sanu.count(), 0) """ need to figure out cases where we write errors for other leagues def test_load_translog_error_aaron_rodgers(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='451385', year='2014') team2 = Team.objects.create(espn_id='11', league=league, abbreviation='RB') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='SMD') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('451385', '2014', '11') load_transactions_from_translog(html, '2014', team2) aaron_rodgers = DraftClaim.objects.filter(player_added__name='Aaron Rodgers') self.assertGreater(aaron_rodgers.count(), 0) """ def test_load_translog_error_ahmad_bradshaw(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='451385', year='2014') team2 = Team.objects.create(espn_id='12', league=league, abbreviation='MM') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='Air') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('451385', '2014', '12') load_transactions_from_translog(html, '2014', team2) ahmad_bradshaw = TradeEntry.objects.filter( players_added__name='Ahmad Bradshaw') self.assertGreater(ahmad_bradshaw.count(), 0) def test_load_translog_error_failed_load(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') league = League.objects.create(espn_id='1880759', year='2014') team2 = Team.objects.create(espn_id='1', league=league, abbreviation='MM') team1 = Team.objects.create(espn_id='2', league=league, abbreviation='Air') filebrowser = FileBrowser() html = filebrowser.get_standings(league) load_teams_from_standings(html, league) html = filebrowser.get_translog('1880759', '2014', '1') load_transactions_from_translog(html, '2014', team2) # ahmad_bradshaw = TradeEntry.objects.filter(players_added__name='Ahmad Bradshaw') #self.assertGreater(ahmad_bradshaw.count(), 0) def test_get_real_num_weeks(self): user = User.objects.create_user('*****@*****.**', '*****@*****.**', 'sincere1') espn_user = EspnUser.objects.create(pk=1, user=user, username='******', password='******') self.assertFalse(self.sqlstore.has_entrance(espn_user)) self.league_scraper.create_leagues(espn_user) current_league = League.objects.filter(year=2014)[0] current_num_weeks = get_real_num_weeks(17, league=current_league) self.assertLessEqual(current_num_weeks, 17) def test_load_translog(self): league = League.objects.create(espn_id='930248', year='2014') self.league_scraper.scrape_league(league) self.league_scraper.load_league(league) team = Team.objects.get(espn_id='6', league=league) self.assertEquals(16, len(DraftClaim.objects.filter(team=team))) team2 = Team.objects.get(espn_id='3', league=league) self.assertEquals(16, len(DraftClaim.objects.filter(team=team2))) team3 = Team.objects.get(espn_id='2', league=league) trade_count = TradeEntry.objects.filter(team=team3).count() trade_count_other = TradeEntry.objects.filter(other_team=team3).count() self.assertEquals(1, trade_count) self.assertEquals(1, trade_count_other) team6 = Team.objects.get(espn_id=6, league=league) hoyer = Player.objects.get(name="Brian Hoyer") waiver_add = AddDrop.objects.get(team=team6, player=hoyer) self.assertEquals(waiver_add.added, True) """ all = AddDrop.objects.filter(team=team3) self.assertGreater(len(all), 0) before_week2 = AddDrop.objects.get_before_week(team3, 2) self.assertEquals(len(before_week2), 10) before_week3 = AddDrop.objects.get_before_week(team3, 3) self.assertEquals(len(before_week3), 12) logger.debug("test_translog: get waiver points") waiver_score = team3.get_waiver_points(4) self.assertEquals(waiver_score, 3) """ davis = Player.objects.get(name='Austin Davis') davis_entry = ScorecardEntry.objects.filter(player=davis, week=6)[0] self.assertEquals(davis_entry.source, 'W')
def setUp(self): clearDb() self.browser = FileBrowser() self.sqlstore = SqlStore() self.league_scraper = LeagueScraper(self.browser, self.sqlstore)
def handle(self, *args, **options): store = SqlStore() filesystem = FileBrowser() # # entrance for espn_user in EspnUser.objects.all(): if store.has_entrance(espn_user): filesystem.write_entrance(espn_user, store.get_entrance(espn_user)) for league in League.objects.all(): ## standings if store.has_standings(league): filesystem.write_standings(league, store.get_standings(league)) ## settings if store.has_settings(league.espn_id, league.year): filesystem.write_settings( league.espn_id, league.year, store.get_settings(league.espn_id, league.year)) for team in Team.objects.filter(league=league): ## translog if store.has_translog(league.espn_id, league.year, team.espn_id): filesystem.write_translog( league.espn_id, league.year, team.espn_id, store.get_translog(league.espn_id, league.year, team.espn_id)) ## matchups for week in range(0, real_num_weeks() + 1): if store.has_matchups(league, week): logger.info("loading week %d" % week) html = store.get_matchups(league, week) filesystem.write_matchups(league, week, html) games = get_teams_from_matchups(html) ## games for game in games: if store.has_game(league, game[0], week): html = store.get_game(league, game[0], week) filesystem.write_game(league, game[0], week, html)