# main.py

import json
import logging
import requests
from pathlib import Path
from urllib3.exceptions import InsecureRequestWarning
import urllib3

urllib3.disable_warnings()

logging.basicConfig(
    level=logging.INFO,
    filename="logs/parser.log",
    format="%(asctime)s - %(module)s - %(levelname)s - %(funcName)s: %(lineno)d - %(message)s",
    datefmt='%H:%M:%S',
)
logger = logging.getLogger(__name__)

MAIN_URL = 'https://www.mrsk-1.ru/customers/customer-service/power-outage/ajax.php?request='
DISTRICT_CODE = '0ECC8970-2702-4C14-9E4D-956606EFFEE3'
LOCALITY_CODE = '02E9C019-AB4D-4FA0-928E-D6C0A41DC256'
URL_PLAN = f'{MAIN_URL}2&regionCode=31&districtCode={DISTRICT_CODE}&locality=Белгород&localityCode={LOCALITY_CODE}'
URL_VNEREGLAMENT = f'{MAIN_URL}3&regionCode=31&districtCode={DISTRICT_CODE}&locality=Белгород&localityCode={LOCALITY_CODE}'
URL_AVAR = f'{MAIN_URL}4&regionCode=31&districtCode={DISTRICT_CODE}&locality=Белгород&localityCode={LOCALITY_CODE}&Street=-'

PLAN_FILE = Path("plan.txt")
VN_FILE = Path("vnereglament.txt")
AVAR_FILE = Path("avar.txt")

HEADERS = {
    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36",
    "Accept": "application/json, text/plain, */*",
    "Accept-Language": "ru-RU,ru;q=0.9,en-US;q=0.8,en;q=0.7",
    "Referer": "https://www.mrsk-1.ru/customers/customer-service/power-outage/",
    "X-Requested-With": "XMLHttpRequest"
}

def clean_street(street: str) -> str:
    street = street.replace('Белгород г; ', '').replace('\n\n', '\n').strip()
    lines = [line.strip() for line in street.split('\n') if line.strip()]
    return '\n'.join(lines)

def format_time(dt_str: str) -> str:
    return dt_str.replace('T', ' ')

def parse_and_save(data, file_path: Path, mode: str):
    if not data or data == 0:
        file_path.write_text("[Нет отключений]", encoding='utf-8')
        return

    records = data if isinstance(data, list) else [data]
    lines = []

    for item in records:
        if mode in ('plan', 'vnereglament'):
            street = clean_street(item.get("DisconnectionObject", ""))
            time_down = format_time(item.get("DisconnectionDateTime", ""))
            time_up = format_time(item.get("EnergyOnPlanningDateTime", ""))
            lines.append(f"{street}\nОтключение: {time_down} — {time_up}\n{'─' * 30}")
        elif mode == 'avar':
            street = clean_street(item.get("StreetHome", ""))
            time_up = format_time(item.get("ScheduledTimeRemoval", ""))
            lines.append(f"{street}\nВосстановление: ~{time_up}\n{'─' * 30}")

    file_path.write_text('\n\n'.join(lines), encoding='utf-8')

def get_plan():
    try:
        r = requests.get(url=URL_PLAN, headers=HEADERS, verify=False, timeout=15)
        r.raise_for_status()
        data = r.json()
        parse_and_save(data, PLAN_FILE, 'plan')
        logger.info("[Плановые отключения] обновлены")
    except Exception as e:
        logger.error(f"Ошибка при получении плановых отключений: {e}")
        if 'r' in locals():
            logger.error(f"Статус: {r.status_code}, Ответ: {r.text[:300]}")
        PLAN_FILE.write_text("[Ошибка загрузки данных]", encoding='utf-8')

def get_vnereglament():
    try:
        r = requests.get(url=URL_VNEREGLAMENT, headers=HEADERS, verify=False, timeout=15)
        r.raise_for_status()
        data = r.json()
        parse_and_save(data, VN_FILE, 'vnereglament')
        logger.info("[Внерегламентные отключения] обновлены")
    except Exception as e:
        logger.error(f"Ошибка при получении внерегламентных отключений: {e}")
        if 'r' in locals():
            logger.error(f"Статус: {r.status_code}, Ответ: {r.text[:300]}")
        VN_FILE.write_text("[Ошибка загрузки данных]", encoding='utf-8')

def get_avar():
    try:
        r = requests.get(url=URL_AVAR, headers=HEADERS, verify=False, timeout=15)
        r.raise_for_status()
        data = r.json()
        parse_and_save(data, AVAR_FILE, 'avar')
        logger.info("[Аварийные отключения] обновлены")
    except Exception as e:
        logger.error(f"Ошибка при получении аварийных отключений: {e}")
        if 'r' in locals():
            logger.error(f"Статус: {r.status_code}, Ответ: {r.text[:300]}")
        AVAR_FILE.write_text("[Ошибка загрузки данных]", encoding='utf-8')

def start_parser():
    get_plan()
    get_vnereglament()
    get_avar()
    logger.info("Парсер отработал")