<reponame>fuadahwal/optimum<filename>pydfs_lineup_optimizer/lineup_exporter.py<gh_stars>0
import csv


class LineupExporter(object):
    def __init__(self, lineup_generator):
        self.lineup_generator = lineup_generator

    @staticmethod
    def render_player(player):
        """
        :type player: Player
        """
        result = player.full_name
        if player.id:
            result += '(%s)' % player.id
        return result

    def export(self, filename):
        raise NotImplementedError


class CSVLineupExporter(LineupExporter):
    def export(self, filename, render_func=None):
        with open(filename, 'w') as csvfile:
            lineup_writer = csv.writer(csvfile, delimiter=',')
            for index, lineup in enumerate(self.lineup_generator):
                if index == 0:
                    header = [player.lineup_position for player in lineup.lineup]
                    header.extend(('Budget', 'FPPG'))
                    lineup_writer.writerow(header)
                row = [(render_func or self.render_player)(player) for player in lineup.lineup]
                row.append(lineup.salary_costs)
                row.append(lineup.fantasy_points_projection)
                lineup_writer.writerow(row)

import csv


class LineupExporter(object):
    def __init__(self, lineup_generator):
        self.lineup_generator = lineup_generator

    @staticmethod
    def render_player(player):
        """
        :type player: Player
        """
        result = player.full_name
        if player.id:
            result += '(%s)' % player.id
        return result

    def export(self, filename):
        raise NotImplementedError


class CSVLineupExporter(LineupExporter):
    def export(self, filename, render_func=None):
        with open(filename, 'w') as csvfile:
            lineup_writer = csv.writer(csvfile, delimiter=',')
            for index, lineup in enumerate(self.lineup_generator):
                if index == 0:
                    header = [player.lineup_position for player in lineup.lineup]
                    header.extend(('Budget', 'FPPG'))
                    lineup_writer.writerow(header)
                row = [(render_func or self.render_player)(player) for player in lineup.lineup]
                row.append(lineup.salary_costs)
                row.append(lineup.fantasy_points_projection)
                lineup_writer.writerow(row)
:type player: Player