#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import pandas as pd

USR = os.path.expanduser('~/.hermes/profiles/zone-b/cache/documents/doc_6f16ee249d8e_刘子研三季度数据-截止至9.30(1).xls')


def load(sheet):
    d = pd.read_excel(USR, sheet_name=sheet, header=0, engine='xlrd')
    d.columns = [str(c).strip() for c in d.columns]
    hk = '货号#' if '货号#' in d.columns else '货号'
    d['_sku'] = d[hk].fillna('（未填货号）').astype(str).str.strip()
    d['_d'] = pd.to_datetime(d['合同日期'], errors='coerce')
    d['_q'] = d['_d'].dt.quarter
    d['_amt'] = pd.to_numeric(d['价税合计'], errors='coerce').fillna(0)
    g = d['集团客户#'].fillna('').astype(str).str.strip().replace({'nan': ''})
    d['_grp'] = g.where(g != '', d['客户'].astype(str).str.strip())
    return d[d['_q'].isin([1, 2, 3])]


a = pd.concat([load('2025年销售数据').assign(y=2025), load('2026年销售数据').assign(y=2026)])
for g in ['苏州华益美生物科技有限公司', '中国人民解放军陆军军医大学第一附属医院', '深圳市雅为世纪科技有限公司', '安瑞科集团']:
    print('=' * 70)
    print('▶', g)
    sub = a[a['_grp'] == g]
    t = sub.groupby(['y', '_sku']).agg(额=('_amt', 'sum'), 量=('销售数量', 'sum'), 名=('物料名称', 'first'),
                                       客户=('客户', lambda s: '/'.join(sorted(set(s.astype(str))))))
    print(t.to_string())