def load_players_from_csv(self, filename: str):
"""
Load player list from CSV file with passed filename.
"""
csv_importer = self._csv_importer
if not csv_importer:
csv_importer = SitesRegistry.get_csv_importer(self._settings.site)
self._players = csv_importer(filename).import_players()
def load_players_from_csv(self, filename):
# type: (str) -> None
"""
Load player list from CSV file with passed filename.
"""
csv_importer = self._csv_importer
if not csv_importer:
csv_importer = SitesRegistry.get_csv_importer(self._settings.site)
self._players = csv_importer(filename).import_players()
self._set_available_teams()
def setUp(self):
self.players = load_players()
self.settings = SitesRegistry.get_settings(Site.DRAFTKINGS,
Sport.BASKETBALL)()
self.player_pool = PlayerPool(settings=self.settings)
self.player_pool.load_players(self.players)
self.test_player = Player(player_id=str(uuid4()),
first_name='Test',
last_name='Test',
team='Test',
fppg=10,
positions=['PG'],
salary=100)
def load_lineups_from_csv(self, filename: str) -> List[Lineup]:
csv_importer = self._csv_importer
if not csv_importer:
csv_importer = SitesRegistry.get_csv_importer(self._settings.site)
return csv_importer(filename).import_lineups(self.players)
def get_optimizer(site: str, sport: str, **kwargs) -> LineupOptimizer:
return LineupOptimizer(SitesRegistry.get_settings(site, sport), **kwargs)
def get_optimizer(site, sport, **kwargs):
# type: (str, str, **Any) -> LineupOptimizer
return LineupOptimizer(SitesRegistry.get_settings(site, sport), **kwargs)
def get_optimizer(site, sport):
# type: (str, str) -> LineupOptimizer
return LineupOptimizer(*SitesRegistry.get_optimizer_args(site, sport))