from urlparse import urljoin
import mysql
from db import PlayoffDB
import sql as p_sql
from dto import *

class PlayoffData(object):
    def __init__(self, settings):
        self.settings = settings
        self.database = PlayoffDB(self.settings.get('database'))
        self.phases = self.settings.get('phases')
        self.teams = self.settings.get('teams')
        self.match_info = {}
        self.leaderboard = []

    def generate_phases(self):
        grid = []
        for phase in self.phases:
            phase_count = len(phase['matches'])
            if 'dummies' in phase:
                phase_count += len(phase['dummies'])
            phase_grid = [None] * phase_count
            phase_pos = 0
            for match in phase['matches']:
                if 'dummies' in phase:
                    while phase_pos in phase['dummies']:
                        phase_pos += 1
                phase_grid[phase_pos] = match['id']
                phase_pos += 1
            grid.append(phase_grid)
        return grid

    def fill_match_info(self):
        self.match_info = {}
        for phase in self.phases:
            for match in phase['matches']:
                self.match_info[match['id']] = self.get_match_info(match)
                self.match_info[match['id']].link = phase['link'] if self.match_info[match['id']].link is None else urljoin(
                    phase['link'], self.match_info[match['id']].link)
        return self.match_info

    def get_match_link(self, match):
        try:
            row = self.database.fetch(match['database'], p_sql.PREFIX, ())
            if row is not None:
                if len(row) > 0:
                    return '%srunda%d.html' % (row[0], match['round'])
        except mysql.connector.Error:
            return None
        return None

    def get_db_match_teams(self, match):
        teams = [Team(), Team()]
        row = self.database.fetch(match['database'], p_sql.MATCH_RESULTS, (match['table'], match['round']))
        teams[0].name = row[0]
        teams[1].name = row[1]
        teams[0].score = row[3] + row[5]
        teams[1].score = row[4] + row[6]
        if row[2] > 0:
            teams[0].score += row[2]
        else:
            teams[1].score -= row[2]
        return teams

    def get_config_match_teams(self, match):
        teams = [Team(), Team()]
        for i in range(0, 2):
            if isinstance(match['teams'][i], basestring):
                teams[i].name = match['teams'][i]
            elif isinstance(match['teams'][i], list):
                teams[i].name = '<br />'.join(match['teams'][i])
            else:
                match_teams = []
                if 'winner' in match['teams'][i]:
                    match_teams += [
                        self.match_info[winner_match].winner
                        for winner_match in match['teams'][i]['winner']
                    ]
                if 'loser' in match['teams'][i]:
                    match_teams += [
                        self.match_info[loser_match].loser
                        for loser_match in match['teams'][i]['loser']
                    ]
                if 'place' in match['teams'][i]:
                    match_teams += [
                        self.teams[place-1][0]
                        for place in match['teams'][i]['place']
                    ]
            teams[i].name = '<br />'.join([
                team if team is not None else '??'
                for team in match_teams]
            ) if len([
                team for team in match_teams if team is not None
            ]) > 0 else ''
        return teams


    def get_match_info(self, match):
        info = Match()
        info.winner_matches = []
        info.loser_matches = []
        for i in range(0, 2):
            if 'winner' in match['teams'][i]:
                info.winner_matches += match['teams'][i]['winner']
            if 'loser' in match['teams'][i]:
                info.loser_matches += match['teams'][i]['loser']
        info.winner_matches = list(set(info.winner_matches))
        info.loser_matches = list(set(info.loser_matches))
        info.link = self.get_match_link(match)
        try:
            info.teams = self.get_db_match_teams(match)
        except (mysql.connector.Error, TypeError, IndexError):
            info.teams = self.get_config_match_teams(match)
        try:
            towels = self.database.fetch(match['database'], p_sql.TOWEL_COUNT, (match['table'], match['round']))
            row = [0 if r is None else r for r in self.database.fetch(match['database'], p_sql.BOARD_COUNT, (match['table'], match['round']))]
            if row[1] > 0:
                info.running = int(row[1])
            if row[1] >= row[0] - towels[0]:
                info.running = 0
                info.winner = info.teams[0].name if info.teams[0].score > info.teams[1].score else info.teams[1].name
                info.loser = info.teams[1].name if info.teams[0].score > info.teams[1].score else info.teams[0].name
        except (mysql.connector.Error, TypeError, KeyError):
            pass
        return info

    def prefill_leaderboard(self, teams):
        self.leaderboard = [None] * len(teams)
        for team in teams:
            if len(team) > 3:
                self.leaderboard[team[3]-1] = team[0]
        return self.leaderboard

    def fill_leaderboard(self):
        self.prefill_leaderboard(self.teams)
        leaderboard_teams = {}
        for phase in self.phases:
            for match in phase['matches']:
                if 'winner' in match:
                    winner_key = tuple(match['winner'])
                    if winner_key not in leaderboard_teams:
                        leaderboard_teams[winner_key] = []
                    leaderboard_teams[winner_key].append(self.match_info[match['id']].winner)
                if 'loser' in match:
                    loser_key = tuple(match['loser'])
                    if loser_key not in leaderboard_teams:
                        leaderboard_teams[loser_key] = []
                    leaderboard_teams[loser_key].append(self.match_info[match['id']].loser)
        for positions, position_teams in leaderboard_teams.iteritems():
            positions = list(positions)
            if len(positions) == len([team for team in position_teams if team is not None]):
                for table_team in self.teams:
                    if table_team[0] in position_teams:
                        position = positions.pop(0)
                        self.leaderboard[position-1] = table_team[0]
        return self.leaderboard

    def get_dimensions(self):
        return (
            len(self.phases),
            max([
                len(phase['matches']) + len(phase['dummies'])
                if 'dummies' in phase
                else len(phase['matches'])
                for phase in self.phases
            ])
        )

    def get_shortname(self, fullname):
        for team in self.teams:
            if team[0] == fullname:
                return team[1]
        return fullname

    def get_team_image(self, fullname):
        for team in self.teams:
            if team[0] == fullname and len(team) > 2:
                return team[2]
        return None