LineupGenerator.py

import sys
import csv
import pulp
import copy
import pandas as pd
from tqdm import tqdm
import os.path

class LineupGenerator:
	def __init__(self, sport, num_lineups, overlap, player_limit, teams_limit, stack, solver, correlation_file, players_file, defenses_goalies_file, output_file):
		self.sport = sport
		self.num_lineups = num_lineups
		self.overlap = overlap
		self.player_limit = player_limit
		self.teams_limit = teams_limit
		self.stack = stack
		self.solver = solver
		self.players = self.load(players_file)
		self.num_players = len(self.players.index)
		self.players_teams = []
		if sport == 'NFL':
			self.positions = {'QB':[], 'WR':[], 'RB':[], 'TE':[]}
			self.defenses = self.load(defenses_goalies_file)
			self.num_defenses = len(self.defenses.index)
		elif sport == 'NHL':
			self.positions = {'C':[], 'W':[], 'D':[]}
			self.goalies = self.load(defenses_goalies_file)
			self.num_goalies = len(self.goalies.index)
			self.goalies_teams = []
			self.goalies_opponents = []
			self.num_lines = None
			self.team_lines = []
		elif sport == 'NBA':
			self.positions = {'PG':[], 'SG':[], 'SF':[], 'PF':[], 'C':[]}
		if os.path.exists(correlation_file):
			self.correlation_matrix = self.load(correlation_file)
		self.output_file = output_file
		self.num_teams = None
		self.actuals = True if 'Actual FP' in self.players else False

	def load(self, filepath):
		try:
			data = pd.read_csv(filepath)
		except IOError:
			sys.exit('INVALID FILEPATH: {}'.format(filepath))
		return data

	def save_file(self, header, filled_lineups, show_proj=False):
		header_copy = copy.deepcopy(header)
		output_projection_path = self.output_file.split('.')[0] + '_proj.csv'
		if self.actuals:
			lineups_for_upload = [lineup[:-2] for lineup in filled_lineups]
			header_copy.extend(('PROJ', 'ACTUAL'))
		else:
			lineups_for_upload = [lineup[:-1] for lineup in filled_lineups]
			header_copy.extend(('PROJ'))
		if show_proj == False:
			with open(self.output_file, 'w') as f:
					writer = csv.writer(f)
					writer.writerow(header)
					writer.writerows(lineups_for_upload)
		elif show_proj == True:
			with open(output_projection_path, 'w') as f:
					writer = csv.writer(f)
					writer.writerow(header_copy)
					writer.writerows(filled_lineups)

	def create_indicators(self):
		teams = list(set(self.players['Team'].values))
		self.num_teams = len(teams)

		# positions used to show what position each player is
		for pos in self.players.loc[:, 'Pos']:
			for key in self.positions:
				self.positions[key].append(1 if key in pos else 0)

		# players_teams which teams each player is on
		for player_team in self.players.loc[:, 'Team']:
			self.players_teams.append([1 if player_team == team else 0 for team in teams])

		if self.sport == 'NHL':	
			# player_line indicates which line by their team they are on
			for i, line in enumerate(self.players.loc[:, 'Line']):
				player_line = []
				if int(line) == 1:
					player_line.extend((1, 0, 0, 0))
				elif int(line) == 2:
					player_line.extend((0, 1, 0, 0))
				elif int(line) == 3:
					player_line.extend((0, 0, 1, 0))
				elif int(line) == 4:
					player_line.extend((0, 0, 0, 1))
				else:
					player_line.extend((0, 0, 0, 0))
				player_lines = []
				for team in teams:
					if self.players.loc[i, 'Team'] == team:
						player_lines.extend(player_line)
					else:
						player_lines.extend((0, 0, 0, 0))
				self.team_lines.append(player_lines)
			self.num_lines = len(self.team_lines[0])
			# goalie_teams indicates which team the goalie is on
			for goalie_team in self.goalies.loc[:, 'Team']:
				self.goalies_teams.append([1 if goalie_team == team else 0 for team in teams])
			# goalies_opponents indicates which players are not on same team as goalie
			for player_opp in self.players.loc[:, 'Opp']:
				self.goalies_opponents.append([1 if player_opp == team else 0 for team in self.goalies.loc[:, 'Team']])

	def generate_lineups(self, formula):
		lineups = []
		for _ in tqdm(range(self.num_lineups)):
			lineup = formula(lineups)
			if lineup:
				lineups.append(lineup)
			else:
				break
		return lineups