snes kall)))))))

Reviewed-on: zovos/summer_practice#4

divisions.py

@@ -1,180 +0,0 @@
-import sys  # аргументы командной строки
-import os   # путь к main.py и имя csv-файла
-import pandas as pd
-import numpy as np
-
-'''
-Отдельный модуль: ДИВИЗИОНЫ и ПРАВИЛО ПОСЕВА победителей дивизионов.
-
-Зачем он нужен.
-Базовый main.py сортирует команды в конференции строго по проценту очков (%О)
-и тай-брейкам. Но в части сезонов КХЛ в таблице конференции действовало правило:
-ПОБЕДИТЕЛИ ДВУХ ДИВИЗИОНОВ получают места 1 и 2 в конференции, даже если по очкам
-они не входят в топ-2. Из-за этого, например, в сезоне 2020/21 в Западной
-конференции СКА (82 очка) стоит на сайте выше Динамо М (84) и Локомотива (83):
-СКА выиграл дивизион Боброва, а те двое - из дивизиона Тарасова, который выиграл ЦСКА.
-
-main.py этот посев не делает (в нём вообще нет деления на дивизионы, только
-Запад/Восток), поэтому правило вынесено сюда, в отдельный файл, и main.py не трогается.
-'''
-
-
-'''
-Расчёт таблицы берём из main.py, чтобы не дублировать логику (единственный источник
-правды - там). main.py в конце сам вызывает main(), поэтому импортировать его напрямую
-нельзя - он бы тут же попытался прочитать аргументы и упал. Поэтому читаем его исходник
-и выполняем без последней строки `main()`: так получаем готовые функции build_standings
-и словарь CONFERENCES, ничего при этом не запуская и не печатая.
-'''
-_main_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'main.py')
-_src = open(_main_path, encoding='utf-8').read()
-_src = _src.replace('\nmain()\n', '\n')  # убираем автозапуск
-_ns = {}
-exec(compile(_src, 'main.py', 'exec'), _ns)
-build_standings = _ns['build_standings']
-CONFERENCES = _ns['CONFERENCES']
-
-
-def make_div(bobrov, tarasov, kharlamov, chernyshev):
-    '''
-    Собрать словарь {команда -> дивизион} из четырёх списков команд.
-
-    Принимает: четыре списка - составы дивизионов Боброва и Тарасова (Запад),
-               Харламова и Чернышёва (Восток).
-    Возвращает: dict, где каждой команде сопоставлено название её дивизиона.
-
-    Аналог make_conf из main.py, только делений не два (Запад/Восток), а четыре.
-    '''
-    d = {}
-    d.update({t: 'Боброва' for t in bobrov})       # Запад
-    d.update({t: 'Тарасова' for t in tarasov})     # Запад
-    d.update({t: 'Харламова' for t in kharlamov})  # Восток
-    d.update({t: 'Чернышёва' for t in chernyshev}) # Восток
-    return d
-
-
-'''
-Состав дивизионов по сезонам (имена команд - ровно как в соответствующем csv).
-Источники:
-  2020/21 - официальное распределение КХЛ;
-  2021/22 - Википедия "КХЛ в сезоне 2021/2022";
-  2018/19, 2019/20 - стандартная сетка дивизионов тех сезонов.
-Для посева важен только победитель каждого дивизиона; выходные таблицы ниже
-сверены с официальным сайтом КХЛ по всем четырём сезонам и совпадают.
-'''
-DIVISIONS = {
-    '2018_19': make_div(
-        bobrov=['СКА', 'Йокерит', 'Динамо М', 'Динамо Мн', 'Динамо Р', 'Слован'],
-        tarasov=['ЦСКА', 'Локомотив', 'ХК Сочи', 'Спартак', 'Витязь', 'Северсталь'],
-        kharlamov=['Автомобилист', 'Металлург Мг', 'Ак Барс', 'Трактор',
-                   'Нефтехимик', 'Куньлунь РС', 'Адмирал'],
-        chernyshev=['Барыс', 'Авангард', 'Салават Юлаев', 'Сибирь',
-                    'Торпедо НН', 'Амур'],
-    ),
-    '2019_20': make_div(
-        bobrov=['СКА', 'Йокерит', 'Спартак', 'Динамо М', 'Динамо Мн', 'Динамо Р'],
-        tarasov=['ЦСКА', 'Локомотив', 'Витязь', 'Торпедо НН', 'ХК Сочи', 'Северсталь'],
-        kharlamov=['Ак Барс', 'Металлург Мг', 'Автомобилист', 'Нефтехимик',
-                   'Трактор', 'Адмирал'],
-        chernyshev=['Авангард', 'Барыс', 'Салават Юлаев', 'Сибирь',
-                    'Амур', 'Куньлунь РС'],
-    ),
-    '2020_21': make_div(
-        bobrov=['СКА', 'Йокерит', 'Северсталь', 'Спартак', 'Витязь', 'ХК Сочи'],
-        tarasov=['ЦСКА', 'Динамо М', 'Локомотив', 'Динамо Мн', 'Динамо Р'],
-        kharlamov=['Автомобилист', 'Ак Барс', 'Металлург Мг', 'Нефтехимик',
-                   'Трактор', 'Торпедо НН'],
-        chernyshev=['Авангард', 'Амур', 'Барыс', 'Куньлунь РС',
-                    'Салават Юлаев', 'Сибирь'],
-    ),
-    '2021_22': make_div(
-        bobrov=['Витязь', 'Йокерит', 'СКА', 'ХК Сочи', 'Торпедо', 'Спартак'],
-        tarasov=['Динамо М', 'Динамо Мн', 'Динамо Р', 'Локомотив', 'Северсталь', 'ЦСКА'],
-        kharlamov=['Автомобилист', 'Ак Барс', 'Куньлунь РС', 'Металлург Мг',
-                   'Нефтехимик', 'Трактор'],
-        chernyshev=['Авангард', 'Амур', 'Барыс', 'Сибирь', 'Салават Юлаев', 'Адмирал'],
-    ),
-}
-
-'''
-В каких сезонах правило посева победителей дивизионов отражалось в таблице конференции.
-2018/19, 2019/20, 2020/21 - да; в 2021/22 КХЛ перешла к чистому порядку по %О
-(на сайте за 2021/22 победители дивизионов наверх НЕ поднимаются).
-В сезонах 2018/19 и 2019/20 победители дивизионов и так были в топ-2 по очкам,
-поэтому правило там виден результат не меняет - но оставляем его включённым,
-чтобы логика была единой.
-'''
-SEED_BY_DIVISION = {
-    '2018_19': True,
-    '2019_20': True,
-    '2020_21': True,
-    '2021_22': False,
-}
-
-
-def seed_conference(conf_table):
-    '''
-    conf_table - часть итоговой таблицы по одной конференции, УЖЕ отсортированная
-    по критериям КХЛ (%О, затем В, ВО, ВБ, Р, ГЗ) - так её отдаёт build_standings.
-
-    Берём лучшего из каждого дивизиона (это и есть победитель дивизиона, т.к. таблица
-    отсортирована) и ставим этих двоих на места 1-2; между собой они идут в том же
-    порядке сортировки. Остальные команды - следом, тоже в порядке сортировки.
-    '''
-    # head(1) внутри каждого дивизиона = первая (а значит, лучшая) команда дивизиона.
-    winners = conf_table.groupby('Див', sort=False).head(1)
-    rest = conf_table.drop(winners.index)
-    return pd.concat([winners, rest])
-
-
-def show_table(table, title):
-    '''
-    Напечатать таблицу конференции в консоль.
-
-    То же, что show_table в main.py, но дополнительно выводит столбец 'Див'
-    (дивизион) - чтобы было видно, почему победители дивизионов оказались наверху.
-    Принимает: table - готовая (уже посеянная) таблица; title - заголовок.
-    Ничего не возвращает - только печатает.
-    '''
-    # Тот же набор колонок, что на сайте КХЛ, плюс столбец дивизиона.
-    cols = ['Див', 'И', 'В', 'ВО', 'ВБ', 'ПБ', 'ПО', 'П', 'ГЗ', 'ГП', 'Р', 'О', '%О']
-    view = table[cols].copy()
-    view.insert(0, 'Место', np.arange(1, len(view) + 1))
-    view.insert(1, 'Команда', view.index)
-
-    print('\n' + '=' * 86)
-    print(title)
-    print('=' * 86)
-    print(view.to_string(index=False))
-
-
-def main():
-    '''
-    Точка входа. Шаги:
-      1) определить сезон по имени CSV-файла;
-      2) построить таблицу команд через build_standings (логика из main.py);
-      3) приписать каждой команде дивизион (столбец 'Див');
-      4) для каждой конференции при необходимости применить посев победителей
-         дивизионов (seed_conference) и напечатать результат.
-    Пример:  python3 divisions.py khl_3/khl_2020_21.csv
-    '''
-    path = sys.argv[1]
-    name = os.path.basename(path)
-    season = name.replace('khl_', '').replace('.csv', '')
-
-    df = pd.read_csv(path)
-    table = build_standings(df, CONFERENCES[season])
-
-    # Подписываем каждой команде её дивизион.
-    table['Див'] = pd.Series(table.index, index=table.index).map(DIVISIONS[season])
-
-    use_seeding = SEED_BY_DIVISION[season]
-    for conf in ['Запад', 'Восток']:
-        part = table[table['Конф'] == conf]
-        if use_seeding:
-            part = seed_conference(part)
-        suffix = ' (посев по дивизионам)' if use_seeding else ' (по очкам, без посева)'
-        show_table(part, 'Конференция ' + conf + ' ' + season + suffix)
-
-
-main()