Automatizovaný billing podle RFID karet

#!/usr/bin/env python3
"""
MyBox RFID Billing - automatizovaný billing energie podle RFID karet.

Použití:
    python rfid_billing.py                      # Billing za poslední měsíc
    python rfid_billing.py --months 3           # Billing za poslední 3 měsíce
    python rfid_billing.py --from 2025-09-01 --to 2025-09-30
"""

import requests
from requests.auth import HTTPBasicAuth
from datetime import datetime, timedelta, timezone
from collections import defaultdict
import csv
import argparse
import json

# ─── Konfigurace ──────────────────────────────────────────────
API_KEY = "váš_api_key"
API_SECRET = "váš_api_secret"
BASE_URL = "https://cloud.mybox.pro/admin-panel/v1"
AUTH = HTTPBasicAuth(API_KEY, API_SECRET)
HEADERS = {"Accept": "application/json"}


# ─── API funkce ───────────────────────────────────────────────

def get_all_devices():
    """Získá seznam všech nabíjecích stanic."""
    response = requests.get(
        f"{BASE_URL}/external/device",
        auth=AUTH,
        headers=HEADERS
    )
    response.raise_for_status()
    return response.json()['data']


def get_charging_reports(device_id, start_date, end_date):
    """
    Získá všechny charging reporty pro zařízení.
    Automaticky stránkuje pomocí meta.totalCount.
    """
    all_reports = []
    offset = 0
    limit = 100

    while True:
        response = requests.get(
            f"{BASE_URL}/external/charging-reports/device/{device_id}",
            auth=AUTH,
            params={
                'startDate': start_date,
                'endDate': end_date,
                'limit': limit,
                'offset': offset
            },
            headers=HEADERS
        )
        response.raise_for_status()
        data = response.json()
        reports = data['data']
        total = data['meta']['totalCount']

        all_reports.extend(reports)

        if len(all_reports) >= total:
            break
        offset += limit

    return all_reports


# ─── Billing logika ──────────────────────────────────────────

def generate_billing(start_date, end_date):
    """
    Vygeneruje billing seskupený podle RFID karet.
    Projde všechna zařízení a stáhne reporty.
    """
    billing = defaultdict(lambda: {
        'name': '',
        'sessions': 0,
        'energy_kwh': 0.0,
        'cost': 0.0,
        'currency': '',
        'charging_seconds': 0,
        'devices': set(),
        'first_charge': None,
        'last_charge': None,
        'sessions_detail': []
    })

    devices = get_all_devices()
    print(f"Nalezeno {len(devices)} zařízení")
    print(f"Období: {start_date} -- {end_date}\n")

    for i, device in enumerate(devices):
        device_id = device['identifier']
        device_title = device.get('user_title') or device.get('title', device_id)

        reports = get_charging_reports(device_id, start_date, end_date)

        if reports:
            print(f"  [{i+1}/{len(devices)}] {device_title}: {len(reports)} relací")

        for r in reports:
            if r['status'] != 'Finished':
                continue

            rfid = r.get('user_id', 'unknown')
            energy = float(r.get('energy_delivered', 0) or 0)
            cost = float(r.get('cost', 0) or 0)
            charging_time = int(r.get('charging_time', 0) or 0)
            start_time = r.get('start_time', '')

            billing[rfid]['name'] = r.get('user_tag_title', 'Neznámý')
            billing[rfid]['sessions'] += 1
            billing[rfid]['energy_kwh'] += energy
            billing[rfid]['cost'] += cost
            billing[rfid]['currency'] = r.get('currency', 'CZK')
            billing[rfid]['charging_seconds'] += charging_time
            billing[rfid]['devices'].add(device_id)

            # Sledování časového rozsahu
            if start_time:
                if billing[rfid]['first_charge'] is None or start_time < billing[rfid]['first_charge']:
                    billing[rfid]['first_charge'] = start_time
                if billing[rfid]['last_charge'] is None or start_time > billing[rfid]['last_charge']:
                    billing[rfid]['last_charge'] = start_time

            # Detail relace
            billing[rfid]['sessions_detail'].append({
                'date': start_time[:10] if start_time else '',
                'device': device_title,
                'energy_kwh': energy,
                'cost': cost,
                'duration_min': round(charging_time / 60, 1)
            })

    return billing


# ─── Výstup ──────────────────────────────────────────────────

def format_duration(seconds):
    """Převede sekundy na lidsky čitelný formát."""
    hours = seconds // 3600
    minutes = (seconds % 3600) // 60
    if hours > 0:
        return f"{hours}h {minutes}m"
    return f"{minutes}m"


def print_billing_summary(billing):
    """Vypíše souhrnný billing do konzole."""
    print(f"\n{'='*95}")
    print(f"{'RFID karta':<16} {'Jméno':<22} {'Relací':>7} "
          f"{'Energie (kWh)':>14} {'Náklady':>11} {'Doba':>10} {'Stanic':>7}")
    print(f"{'='*95}")

    total_energy = 0.0
    total_cost = 0.0
    total_sessions = 0
    currency = 'CZK'

    for rfid, info in sorted(billing.items(), key=lambda x: x[1]['energy_kwh'], reverse=True):
        currency = info['currency']
        duration = format_duration(info['charging_seconds'])
        print(f"{rfid:<16} {info['name']:<22} {info['sessions']:>7} "
              f"{info['energy_kwh']:>14.2f} {info['cost']:>8.2f} {currency} "
              f"{duration:>10} {len(info['devices']):>7}")
        total_energy += info['energy_kwh']
        total_cost += info['cost']
        total_sessions += info['sessions']

    print(f"{'='*95}")
    print(f"{'CELKEM':<16} {'':<22} {total_sessions:>7} "
          f"{total_energy:>14.2f} {total_cost:>8.2f} {currency}")
    print()


def print_user_detail(billing, rfid):
    """Vypíše detail relací pro konkrétní RFID kartu."""
    if rfid not in billing:
        print(f"RFID karta {rfid} nenalezena.")
        return

    info = billing[rfid]
    print(f"\nDetail: {info['name']} (RFID: {rfid})")
    print(f"Období: {info['first_charge'][:10]} -- {info['last_charge'][:10]}")
    print(f"Celkem: {info['energy_kwh']:.2f} kWh | {info['cost']:.2f} {info['currency']} | "
          f"{info['sessions']} relací | {format_duration(info['charging_seconds'])}")
    print()

    print(f"  {'Datum':<12} {'Stanice':<25} {'Energie':>10} {'Náklady':>10} {'Doba':>8}")
    print(f"  {'-'*70}")
    for s in sorted(info['sessions_detail'], key=lambda x: x['date']):
        print(f"  {s['date']:<12} {s['device']:<25} {s['energy_kwh']:>8.2f} kWh"
              f" {s['cost']:>8.2f} {s['duration_min']:>6.0f} min")


# ─── Export ──────────────────────────────────────────────────

def export_summary_csv(billing, filename='billing_summary.csv'):
    """Exportuje souhrnný billing do CSV."""
    with open(filename, 'w', newline='', encoding='utf-8') as f:
        writer = csv.writer(f, delimiter=';')
        writer.writerow([
            'RFID karta', 'Jméno', 'Počet relací',
            'Dodaná energie (kWh)', 'Náklady', 'Měna',
            'Doba nabíjení', 'Počet stanic',
            'První nabíjení', 'Poslední nabíjení'
        ])
        for rfid, info in sorted(billing.items(), key=lambda x: x[1]['energy_kwh'], reverse=True):
            writer.writerow([
                rfid,
                info['name'],
                info['sessions'],
                f"{info['energy_kwh']:.2f}",
                f"{info['cost']:.2f}",
                info['currency'],
                format_duration(info['charging_seconds']),
                len(info['devices']),
                info['first_charge'][:10] if info['first_charge'] else '',
                info['last_charge'][:10] if info['last_charge'] else ''
            ])
    print(f"Souhrnný CSV: {filename}")


def export_detail_csv(billing, filename='billing_detail.csv'):
    """Exportuje detailní billing (každá relace) do CSV."""
    with open(filename, 'w', newline='', encoding='utf-8') as f:
        writer = csv.writer(f, delimiter=';')
        writer.writerow([
            'RFID karta', 'Jméno', 'Datum', 'Stanice',
            'Energie (kWh)', 'Náklady', 'Doba (min)'
        ])
        for rfid, info in sorted(billing.items(), key=lambda x: x[1]['name']):
            for s in sorted(info['sessions_detail'], key=lambda x: x['date']):
                writer.writerow([
                    rfid,
                    info['name'],
                    s['date'],
                    s['device'],
                    f"{s['energy_kwh']:.2f}",
                    f"{s['cost']:.2f}",
                    f"{s['duration_min']:.0f}"
                ])
    print(f"Detailní CSV:  {filename}")


def export_json(billing, filename='billing.json'):
    """Exportuje billing do JSON."""
    export = {}
    for rfid, info in billing.items():
        export[rfid] = {
            'name': info['name'],
            'sessions': info['sessions'],
            'energy_kwh': round(info['energy_kwh'], 2),
            'cost': round(info['cost'], 2),
            'currency': info['currency'],
            'charging_seconds': info['charging_seconds'],
            'devices_count': len(info['devices']),
            'first_charge': info['first_charge'],
            'last_charge': info['last_charge'],
            'sessions_detail': info['sessions_detail']
        }

    with open(filename, 'w', encoding='utf-8') as f:
        json.dump(export, f, indent=2, ensure_ascii=False)
    print(f"JSON export:   {filename}")


# ─── CLI ─────────────────────────────────────────────────────

def main():
    parser = argparse.ArgumentParser(description='MyBox RFID Billing')
    parser.add_argument('--months', type=int, default=1, help='Počet měsíců zpět (default: 1)')
    parser.add_argument('--from', dest='date_from', help='Počáteční datum (YYYY-MM-DD)')
    parser.add_argument('--to', dest='date_to', help='Koncové datum (YYYY-MM-DD)')
    parser.add_argument('--detail', help='Zobrazit detail pro RFID kartu')
    parser.add_argument('--export', choices=['csv', 'json', 'all'], default='csv',
                        help='Formát exportu (default: csv)')
    args = parser.parse_args()

    # Časový rozsah
    if args.date_from and args.date_to:
        start_date = f"{args.date_from}T00:00:00Z"
        end_date = f"{args.date_to}T23:59:59Z"
    else:
        end = datetime.now(timezone.utc)
        start = end - timedelta(days=args.months * 30)
        start_date = start.strftime('%Y-%m-%dT00:00:00Z')
        end_date = end.strftime('%Y-%m-%dT23:59:59Z')

    # Generování billingu
    billing = generate_billing(start_date, end_date)
    print_billing_summary(billing)

    # Detail pro konkrétní kartu
    if args.detail:
        print_user_detail(billing, args.detail)

    # Export
    if args.export in ('csv', 'all'):
        export_summary_csv(billing)
        export_detail_csv(billing)
    if args.export in ('json', 'all'):
        export_json(billing)


if __name__ == '__main__':
    main()

# .env soubor - uložte vedle skriptu
MYBOX_API_KEY=váš_api_key
MYBOX_API_SECRET=váš_api_secret

import os
from dotenv import load_dotenv

load_dotenv()
API_KEY = os.getenv('MYBOX_API_KEY')
API_SECRET = os.getenv('MYBOX_API_SECRET')