import xml.etree.ElementTree as ET
from pathlib import Path

def parse_catalog(filepath: str) -> dict:
    """Парсинг каталога CommerceML. Возвращает группы + товары."""
    tree = ET.parse(filepath)
    root = tree.getroot()

    # Группы (категории) — рекурсивная иерархия
    groups = {}
    def walk_groups(parent_el, parent_id=None):
        for g in parent_el.findall('Группа'):
            gid = g.findtext('Ид')
            groups[gid] = {
                'name': g.findtext('Наименование'),
                'parent_id': parent_id
            }
            sub = g.find('Группы')
            if sub is not None:
                walk_groups(sub, gid)

    classifier = root.find('Классификатор')
    if classifier is not None:
        top_groups = classifier.find('Группы')
        if top_groups is not None:
            walk_groups(top_groups)

    # Свойства (характеристики) — справочник
    properties = {}
    props_el = classifier.find('Свойства') if classifier is not None else None
    if props_el is not None:
        for prop in props_el.findall('Свойство'):
            pid = prop.findtext('Ид')
            variants = {}
            vv = prop.find('ВариантыЗначений')
            if vv is not None:
                for v in vv.findall('Справочник'):
                    variants[v.findtext('ИдЗначения')] = v.findtext('Значение')
            properties[pid] = {
                'name': prop.findtext('Наименование'),
                'type': prop.findtext('ТипЗначений', 'Строка'),
                'variants': variants
            }

    # Товары
    products = []
    catalog = root.find('Каталог')
    if catalog is not None:
        for item in catalog.findall('.//Товар'):
            product = {
                'id': item.findtext('Ид'),
                'name': item.findtext('Наименование'),
                'sku': item.findtext('Артикул', ''),
                'description': item.findtext('Описание', ''),
                'group_ids': [g.text for g in item.findall('Группы/Ид')],
                'images': [img.text for img in item.findall('Картинка')],
                'properties': {},
                'requisites': {},
                'characteristics': []
            }
            # Значения свойств
            for zs in item.findall('ЗначенияСвойств/ЗначенияСвойства'):
                prop_id = zs.findtext('Ид')
                value = zs.findtext('Значение', '')
                product['properties'][prop_id] = value

            # Реквизиты
            for zr in item.findall('ЗначенияРеквизитов/ЗначениеРеквизита'):
                product['requisites'][zr.findtext('Наименование')] = zr.findtext('Значение', '')

            # Характеристики товара
            for ch in item.findall('ХарактеристикиТовара/ХарактеристикаТовара'):
                product['characteristics'].append({
                    'id': ch.findtext('Ид'),
                    'name': ch.findtext('Наименование', '')
                })

            products.append(product)

    return {'groups': groups, 'properties': properties, 'products': products}

def parse_offers(filepath: str) -> dict:
    """Парсинг предложений: цены и остатки по товарам."""
    tree = ET.parse(filepath)
    root = tree.getroot()
    packet = root.find('ПакетПредложений')
    if packet is None:
        return {'price_types': {}, 'offers': []}

    # Типы цен
    price_types = {}
    for pt in packet.findall('ТипыЦен/ТипЦены'):
        price_types[pt.findtext('Ид')] = {
            'name': pt.findtext('Наименование'),
            'currency': pt.findtext('Валюта', 'RUB')
        }

    # Предложения
    offers = []
    for offer in packet.findall('Предложения/Предложение'):
        offer_id = offer.findtext('Ид', '')
        # Разделение товар#характеристика
        parts = offer_id.split('#')
        product_id = parts[0]
        char_id = parts[1] if len(parts) > 1 else None

        prices = {}
        for price in offer.findall('Цены/Цена'):
            pt_id = price.findtext('ИдТипаЦены')
            prices[pt_id] = {
                'value': float(price.findtext('ЦенаЗаЕдиницу', '0')),
                'currency': price.findtext('Валюта', 'RUB'),
                'unit': price.findtext('Единица', 'шт')
            }

        # Остатки — общий или по складам
        quantity = float(offer.findtext('Количество', '0'))
        warehouses = {}
        for wh in offer.findall('Склад'):
            wh_id = wh.get('ИдСклада', '')
            warehouses[wh_id] = float(wh.get('КоличествоНаСкладе', '0'))

        offers.append({
            'product_id': product_id,
            'characteristic_id': char_id,
            'prices': prices,
            'quantity': quantity,
            'warehouses': warehouses
        })

    return {'price_types': price_types, 'offers': offers}

def parse_orders(filepath: str) -> list[dict]:
    """Парсинг заказов CommerceML."""
    tree = ET.parse(filepath)
    root = tree.getroot()
    orders = []

    for doc in root.findall('Документ'):
        order = {
            'id': doc.findtext('Ид'),
            'number': doc.findtext('Номер'),
            'date': doc.findtext('Дата'),
            'time': doc.findtext('Время', ''),
            'operation': doc.findtext('ХозОперация', ''),
            'role': doc.findtext('Роль', ''),
            'currency': doc.findtext('Валюта', 'RUB'),
            'total': float(doc.findtext('Сумма', '0')),
            'comment': doc.findtext('Комментарий', ''),
            'counterparties': [],
            'items': [],
            'requisites': {}
        }

        # Контрагенты
        for cp in doc.findall('Контрагенты/Контрагент'):
            party = {
                'id': cp.findtext('Ид'),
                'name': cp.findtext('Наименование'),
                'full_name': cp.findtext('ПолноеНаименование', ''),
                'role': cp.findtext('Роль', ''),
                'inn': cp.findtext('ИНН', ''),
                'address': {},
                'contacts': []
            }
            for af in cp.findall('АдресРегистрации/АдресноеПоле'):
                party['address'][af.findtext('Тип')] = af.findtext('Значение', '')
            for ct in cp.findall('Контакты/Контакт'):
                party['contacts'].append({
                    'type': ct.findtext('Тип'),
                    'value': ct.findtext('Значение')
                })
            order['counterparties'].append(party)

        # Товары
        for item in doc.findall('Товары/Товар'):
            parts = item.findtext('Ид', '').split('#')
            order['items'].append({
                'product_id': parts[0],
                'characteristic_id': parts[1] if len(parts) > 1 else None,
                'name': item.findtext('Наименование', ''),
                'price': float(item.findtext('ЦенаЗаЕдиницу', '0')),
                'quantity': float(item.findtext('Количество', '0')),
                'total': float(item.findtext('Сумма', '0')),
                'unit': item.findtext('БазоваяЕдиница', 'шт')
            })

        # Реквизиты
        for zr in doc.findall('ЗначенияРеквизитов/ЗначениеРеквизита'):
            order['requisites'][zr.findtext('Наименование')] = zr.findtext('Значение', '')

        orders.append(order)

    return orders

def merge_catalog_offers(catalog: dict, offers: dict) -> list[dict]:
    """Объединение товаров с ценами/остатками. Найти товары без цен."""
    offer_map = {}
    for o in offers['offers']:
        key = o['product_id']
        if o['characteristic_id']:
            key += '#' + o['characteristic_id']
        offer_map[key] = o

    result = []
    for product in catalog['products']:
        pid = product['id']
        if product['characteristics']:
            for ch in product['characteristics']:
                key = pid + '#' + ch['id']
                offer = offer_map.get(key)
                result.append({
                    **product,
                    'characteristic': ch['name'],
                    'prices': offer['prices'] if offer else {},
                    'quantity': offer['quantity'] if offer else 0,
                    'has_offer': offer is not None
                })
        else:
            offer = offer_map.get(pid)
            result.append({
                **product,
                'characteristic': None,
                'prices': offer['prices'] if offer else {},
                'quantity': offer['quantity'] if offer else 0,
                'has_offer': offer is not None
            })
    return result

# Проверка и перекодировка
with open(filepath, 'rb') as f:
    raw = f.read(100)
if b'encoding="windows-1251"' in raw or b'encoding="Windows-1251"' in raw:
    with open(filepath, 'r', encoding='windows-1251') as f:
        content = f.read()
    # Перезаписать в UTF-8 для стандартного парсинга
    content = content.replace('encoding="windows-1251"', 'encoding="utf-8"')
    content = content.replace('encoding="Windows-1251"', 'encoding="utf-8"')

offer_id = "aaa-bbb-ccc#ddd-eee-fff"
product_id, char_id = offer_id.split('#') if '#' in offer_id else (offer_id, None)

from pathlib import Path

files = sorted(Path(directory).glob('import*.xml'))
all_products = []
for f in files:
    result = parse_catalog(str(f))
    all_products.extend(result['products'])

value = zs.findtext('Значение')  # Может быть None или ''
# None = тег отсутствует, '' = значение сброшено

import xml.etree.ElementTree as ET

def parse_large_catalog(filepath: str):
    """Потоковый парсинг больших каталогов."""
    for event, elem in ET.iterparse(filepath, events=('end',)):
        if elem.tag == 'Товар':
            yield {
                'id': elem.findtext('Ид'),
                'name': elem.findtext('Наименование'),
                'sku': elem.findtext('Артикул', '')
            }
            elem.clear()  # Освобождаем память