import os
import sys

import matplotlib.pyplot as plt
import pandas as pd

import main as mn

def calculate_team_points(filename):
    df = pd.read_csv(filename)
    # Считаем очки команд за периоды, овертаймы и булиты
    h1, a1 = mn.parse_score(df, 'Период_1')
    h2, a2 = mn.parse_score(df, 'Период_2')
    h3, a3 = mn.parse_score(df, 'Период_3')
    oth, ota = mn.parse_score(df, 'Овертайм')
    soh, soa = mn.parse_score(df, 'Буллиты') # Счет по буллитам

    # Складываем очки, получая итоговый счет
    home_goals = h1 + h2 + h3
    away_goals = a1 + a2 + a3
    tie = home_goals == away_goals

    # Проверка кто проиграл/выйграл по итогу матча или овертайма, или буллита
    home_win = (home_goals > away_goals) | (tie & ((oth > ota) | ((oth == ota) & (soh > soa))))
    away_win = (away_goals > home_goals) | (tie & ((ota > oth) | ((oth == ota) & (soa > soh))))

    home_overtime_loss = tie & ((ota > oth) | ((oth == ota) & (soa > soh)))
    away_overtime_loss = tie & ((oth > ota) | ((oth == ota) & (soh > soa)))

    # Подсчитываем итоговое кол-во очков
    home_points = 2 * home_win.astype(int) + home_overtime_loss.astype(int)
    away_points = 2 * away_win.astype(int) + away_overtime_loss.astype(int)

    # Соединяем 2 таблицы
    points = pd.concat([
        pd.DataFrame({'Команда': df['Команда_1'], 'О': home_points}),
        pd.DataFrame({'Команда': df['Команда_2'], 'О': away_points}),
    ])

    # Собираем таблицу с помощью groupby,
    return points.groupby('Команда')['О'].sum().sort_values(ascending=False)


def show_points_histogram(filename, output=None):
    points = calculate_team_points(filename) # Получаем очки

    plt.figure(figsize=(12, 7)) # Создаем таблицу 12 на 7 дюймов
    plt.bar(points.index, points.values, color='steelblue') # Строим столбчатую диаграмму, по очкам по y
    plt.title('Гистограмма набранных командами очков') 
    plt.xlabel('Команда')
    plt.ylabel('Очки')
    plt.xticks(rotation=75, ha='right') # Чтобы не накладывались подписи, мы выравниваем и вращаем их
    plt.grid(axis='y', alpha=0.3) # Отображение координатной сетки
    plt.tight_layout() # Автоматическая настройка отступов
    
    # Сохранение если указан нужный аргумент
    if output:
        plt.savefig(output, dpi=200)
        print(f'Гистограмма сохранена в файл: {output}')
    else:
        plt.show()


def main():
    if len(sys.argv) not in (2, 3):
        print('Использование: python3 individual_1.py <csv-файл> [файл_для_сохранения]')
        sys.exit(1)

    filename = sys.argv[1]
    output = sys.argv[2] if len(sys.argv) == 3 else None

    if not os.path.isfile(filename):
        raise FileNotFoundError(f'Файл не найден: {filename}')

    show_points_histogram(filename, output)


if __name__ == '__main__':
    main()