---

from collections import Counter, defaultdict
import difflib
import os
import re
import openpyxl

# ================= ФАЙЛЫ =================
FILE_MAIN = "result_perfect.xlsx"
FILE_NAGRUZKA = "Svodnaya_uchebnaya_nagruzka_12_09_2026_-_krasnym_izmenenia.xlsx"
FILE_PODRAZDEL = "Подразделения.xlsx"
FILE_OUTPUT = "result_ready.xlsx"

# Точные номера столбцов основной таблицы:
COL_GROUP = 2    # Столбец B — Группа
COL_SUBJECT = 5  # Столбец E — Наименование дисциплины
COL_FIO = 6      # Столбец F — ФИО преподавателя
COL_KAFEDRA = 7  # Столбец G — Кафедра
# =========================================


def resolve_file(filename, keyword=None):
  if os.path.exists(filename):
    return filename
  if os.path.exists(filename + ".xlsx"):
    return filename + ".xlsx"
  if filename.endswith(".xlsx.xlsx") and os.path.exists(filename[:-5]):
    return filename[:-5]
  if keyword:
    for f in os.listdir("."):
      if keyword.lower() in f.lower() and f.endswith(".xlsx") and not f.startswith("~$"):
        return f
  return filename


def clean_str(val):
  if val is None:
    return ""
  s = str(val).replace('"', "").replace("'", "").replace("«", "").replace("»", "")
  s = s.lower().replace("ё", "е")
  s = re.sub(r"\s+", " ", s)
  return s.strip()


# 1. Читаем Подразделения (Факультеты/Кафедры -> Шифр)
path_podr = resolve_file(FILE_PODRAZDEL, "подразделения")
print(f"1. Читаем Подразделения: {path_podr}")
wb_podr = openpyxl.load_workbook(path_podr, data_only=True)
kaf_to_shifr = {}

for sheet in wb_podr.worksheets:
  for row in sheet.iter_rows(values_only=True):
    if len(row) >= 3 and row[1] is not None and row[2] is not None:
      shifr = str(row[1]).strip()
      kaf_full = clean_str(row[2])
      if kaf_full and kaf_full != "факультеты/кафедры" and shifr.lower() != "шифр":
        kaf_to_shifr[kaf_full] = shifr

all_podr_names = list(kaf_to_shifr.keys())


def get_shifr_by_kaf_name(kaf_name):
  if not kaf_name:
    return None
  if kaf_name in kaf_to_shifr:
    return kaf_to_shifr[kaf_name]
  for k_podr, s_podr in kaf_to_shifr.items():
    if kaf_name in k_podr or k_podr in kaf_name:
      return s_podr
  close = difflib.get_close_matches(kaf_name, all_podr_names, n=1, cutoff=0.75)
  if close:
    return kaf_to_shifr[close[0]]
  return None


# 2. Читаем Сводную нагрузку:
#    - связка (Дисциплина, Группа) -> счетчик кафедр
#    - связка Дисциплина -> счетчик всех кафедр (чтобы брать основную, а не последнюю строку)
path_nagr = resolve_file(FILE_NAGRUZKA, "svodnaya")
print(f"2. Читаем Сводную нагрузку: {path_nagr}")
wb_nagr = openpyxl.load_workbook(path_nagr, data_only=True)

subj_group_kaf = defaultdict(Counter)
subj_kaf_counter = defaultdict(Counter)

for sheet in wb_nagr.worksheets:
  for row in sheet.iter_rows(values_only=True):
    # В нагрузке: B (1) — Кафедра, C (2) — Дисциплина, E (4) — Группа
    if len(row) >= 3 and row[1] is not None and row[2] is not None:
      kaf_name = clean_str(row[1])
      subj_name = clean_str(row[2])
      if not subj_name or subj_name == "дисциплина" or kaf_name == "кафедра":
        continue

      shifr = get_shifr_by_kaf_name(kaf_name)
      if not shifr:
        continue

      subj_kaf_counter[subj_name][shifr] += 1

      # Вытаскиваем все группы из столбца E (они там могут быть через запятую)
      if len(row) >= 5 and row[4] is not None:
        groups_str = clean_str(row[4])
        for grp in re.split(r"[,;\s]+", groups_str):
          if grp:
            subj_group_kaf[(subj_name, grp)][shifr] += 1

# 3. Обрабатываем основную таблицу
path_main = resolve_file(FILE_MAIN, "result")
print(f"3. Открываем основную таблицу: {path_main}")
wb_main = openpyxl.load_workbook(path_main)
ws_main = wb_main.active

filled_kaf = 0
updated_fio = 0

for row_idx in range(2, ws_main.max_row + 1):
  # Очищаем столбец A от ФМ(ФМ), если он остался в исходнике
  cell_a = ws_main.cell(row=row_idx, column=1)
  if cell_a.value is not None:
    val_a = str(cell_a.value).strip()
    m = re.match(r"^(.+)\(\1\)$", val_a)
    if m:
      cell_a.value = m.group(1)

  cell_grp = ws_main.cell(row=row_idx, column=COL_GROUP)
  cell_subj = ws_main.cell(row=row_idx, column=COL_SUBJECT)
  cell_fio = ws_main.cell(row=row_idx, column=COL_FIO)
  cell_kaf = ws_main.cell(row=row_idx, column=COL_KAFEDRA)

  grp_clean = clean_str(cell_grp.value)
  subj_clean = clean_str(cell_subj.value)
  fio_raw = str(cell_fio.value).strip() if cell_fio.value is not None else ""
  kaf_raw = str(cell_kaf.value).strip() if cell_kaf.value is not None else ""

  # ШАГ 1: Если кафедра пустая — ищем её
  if not kaf_raw and subj_clean:
    shifr = None

    # Приоритет 1: Точное совпадение по Предмету + Группе
    if grp_clean and (subj_clean, grp_clean) in subj_group_kaf:
      shifr = subj_group_kaf[(subj_clean, grp_clean)].most_common(1)[0][0]

    # Приоритет 2: Самая частая (основная) кафедра для этого предмета в нагрузке
    elif subj_clean in subj_kaf_counter:
      shifr = subj_kaf_counter[subj_clean].most_common(1)[0][0]

    if shifr:
      kaf_raw = shifr
      cell_kaf.value = shifr
      filled_kaf += 1

  # ШАГ 2: Приписываем к ФИО преподавателя (кафедра) без пробела
  if fio_raw and kaf_raw and fio_raw.lower() not in ("вакансия", "нет", "-"):
    fio_base = re.sub(r"\s*\([^)]*\)\s*$", "", fio_raw).strip()
    cell_fio.value = f"{fio_base}({kaf_raw})"
    updated_fio += 1

wb_main.save(FILE_OUTPUT)
print(f"\nГотово! Результат сохранен в: {FILE_OUTPUT}")
print(f"— Дозаполнено пустых кафедр: {filled_kaf}")
print(f"— Обновлено ФИО преподавателей со скобками: {updated_fio}")