from src.settings import Colors


def league_color(league: str) -> Colors:
    if league in [
        '1 HNL (Croatia)',
        'A-League (Australia)',
        'Bundesliga 2 (Germany)',
        'Coppa Italia (Italy)',
        'First Division (Cyprus)',
        'Jupiler League (Belgium)',
        'League Two (England)',
        'Liga 1 (Romania)',
        'Ligat ha\'Al (Israel)',
        'Ligue 2 (France)',
        'Major League Soccer (United States)',
        'Primera A (Colombia)',
        'Primera Division (Argentina)',
        'Primera Division (Peru)',
        'Prva Liga (Slovenia)',
        'Segunda Division (Spain)',
        'Segunda Liga (Portugal)',
        'Serie B (Brazil)',
        'Serie B (Italy)',
        'Super Liga (Slovakia)',
        'Superliga (Serbia)',
        'Tippeligaen (Norway)',
        'Vushaya Liga (Russia)',
    ]:
        return Colors.GREEN
    if league in [
        '1 CFL(Montenegro)',
        'A Grupa(Bulgaria)',
        'A Lyga(Lithuania)',
        'Erste Liga(Austria)',
        'J - League 2(Japan)',
        'K - League(South Korea)',
        'Kubok (Ukraine)',
        'OTP Bank Liga (Hungary)',
        'Primeira Liga (Portugal)',
        'Serie A (Italy)',
        'Superettan (Sweden)',
        'Veikkausliiga (Finland)',
        'Virsliga (Latvia)',
        'Vyscha Liga (Ukraine)',
    ]:
        return Colors.RED
    if league in [
        '1. Division (Russia)',
        'Adeccoligaen (Norway)',
        'Allsvenskan (Sweden)',
        'Bundesliga (Austria)',
        'Bundesliga 1 (Germany)',
        'Campeonato (Ecuador)',
        'Cempionat (Belarus)',
        'Challenge League (Switzerland)',
        'Championship (England)',
        'Championship (Scotland)',
        'Conference (England)',
        'Ekstraklasa (Poland)',
        'Eredivisie (Netherlands)',
        'Gambrinus League (Czech Republic)',
        'I Liga (Poland)',
        'J-League (Japan)',
        'Kubok (Belarus)',
        'League One (England)',
        'Ligue 1 (France)',
        'National (France)',
        'OFB Cup (Austria)',
        'Premier Division (Ireland)',
        'Premier League (Scotland)',
        'Premier League (Wales)',
        'Primera Division (Chile)',
        'Primera Division (Mexico)',
        'Primera Division (Spain)',
        'Primera Division (Uruguay)',
        'Proximus League (Belgium)',
        'Serie A (Brazil)',
        'Super Lig (Turkey)',
        'Superleague (Greece)',
        'Superligaen (Denmark)',
        'Taca da Liga (Portugal)',
        'TFF Lig A (Turkey)',
        'Viasat Sport Divisionen (Denmark)',
        'Ykkonen (Finland)',
    ]:
        return Colors.YELLOW
    return Colors.EMPTY
