vontor-cz/backend/thirdparty/zasilkovna/client.py

from zeep import Client
from zeep.exceptions import Fault
from zeep.transports import Transport
from zeep.cache import SqliteCache

from configuration.models import Configuration

import tempfile
import base64
import logging
import os

from functools import lru_cache

logger = logging.getLogger(__name__)

WSDL_URL = os.getenv("PACKETA_WSDL_URL", "https://www.zasilkovna.cz/api/soap.wsdl")
PACKETA_API_PASSWORD = os.getenv("PACKETA_API_PASSWORD")

Configuration.g



# --- 1. Singleton pro klienta (aby se WSDL stáhlo jen jednou pro celý proces) ---
@lru_cache(maxsize=1)
def get_shared_client():
    """
    Tato funkce vrátí instanci klienta a drží ji v tempfile na jednu hodinu.
    """
    if not PACKETA_API_PASSWORD:
        raise ValueError("Packeta API password is not set.")
    
    
    tempFpath = os.path.join(tempfile.gettempdir(), 'zeep_packeta_cache.db')
    transport = Transport(cache=SqliteCache(path=tempFpath, timeout=3600))
    
    return Client(wsdl=WSDL_URL, transport=transport)


class PacketaAPI:
    # --- 2. Property pro lazy loading ---
    @property
    def client(self):
        """
        Při zavolání self.client se zavolá tento kód.
        Vrátí již existujícího klienta z cache.
        """
        return get_shared_client()
        
    
    # ---------- CREATE PACKET METHODS ----------

    def create_packet(self, number: str, name: str, surname: str, email: str, phone: str,
                     address_id: int, value: float, currency: str = "CZK",
                     weight: float = 1.0, eshop: str = "MyEshop", send_email_to_customer: bool = False):
        """
        Jednoduché uložení balíku. Vrací packetId, pod kterým lze později tahat veškeré info.

        Parametry:
            number: číslo objednávky
            name, surname, email, phone: údaje zákazníka
            address_id: ID výdejního místa
            value: cena/částka zásilky
            currency: měna
            weight: hmotnost balíku
            eshop: název e-shopu
            send_email_to_customer: jestli poslat potvrzovací email

        Návrat:
            dict: {
                "packetId": str,
                "number": str,
                "email": str,
                "phone": str,
                "value": float,
                "currency": str
            }
        """
        attributes = {
            "number": number,
            "name": name,
            "surname": surname,
            "email": email,
            "phone": phone,
            "addressId": address_id,
            "value": value,
            "currency": currency,
            "weight": weight,
            "eshop": eshop,
            "sendEmailToCustomer": send_email_to_customer
        }

        try:
            # Použijeme createPacketClaimWithPassword, protože umožňuje ukládat email a telefon
            result = self.client.service.createPacketClaimWithPassword(PACKETA_API_PASSWORD, attributes)
            return result

        except Fault as e:
            logger.error(f"Packeta store_packet error: {e}")
            raise Exception(f"Chyba při ukládání balíku: {e}")

    # ---------- CANCEL SENDING FROM STORE PACKET METHODS ----------

    def cancel_packet(self, packet_id: int):
        """
        Zrušení zásilky (pokud ještě nebyla fyzicky odevzdána).
        packet_id = 1234567890
        """
        try:
            self.client.service.cancelPacket(PACKETA_API_PASSWORD, packet_id)

            return {"status": "ok", "message": f"Zásilka {packet_id} byla zrušena."}
        
        except Fault as e:
            logger.error(f"Packeta cancelPacket error: {e}")
            raise Exception(f"Chyba při rušení zásilky: {e}")


    # ---------- INFO PACKET METHODS ----------

    def packet_state_ready_to_pickup(self, packet_id: int):
        """
        Vrací datum do kdy je uložená zásilka.
        """
        try:
            request = self.client.service.packetGetStoredUntil(PACKETA_API_PASSWORD, packet_id)
            if request is None:
                raise Exception(f"Zásilka {packet_id} nebyla ještě doručena.")
            
            else:
                return request  # vrací datum do kdy je zásilka uložena
            
        except Fault as e:
            logger.error(f"Packeta packetGetStoredUntil error: {e}")
            raise Exception(f"Chyba při získávání data uložení zásilky: {e}")
        
    def get_packet_label_pdf(self, packet_id: int, format: str = "A6 on A6", offset: int = 0):
        """
        Získání PDF štítku k zásilce.

        Parametry:
            packet_id (int): ID zásilky (např. 1234567890)
            format (str): jeden z formátů:
                - "A6 on A6"
                - "A7 on A7"
                - "A6 on A4"
                - "A7 on A4"
                - "A8 on A8"
            offset (int): pozice na stránce (0 = vlevo nahoře)

        Návrat:
            bytes: PDF soubor (base64 dekódovaný)
        """
        try:
            pdf_base64 = self.client.service.packetLabelPdf(
                PACKETA_API_PASSWORD, packet_id, format, offset
            )

            return base64.b64decode(pdf_base64)
        
        except Fault as e:
            logger.error(f"Packeta packetLabelPdf error: {e}")
            raise Exception(f"Chyba při získávání štítku: {e}")


    # ---------- RETURNING PACKET METHODS ----------

    def get_return_routing(self, sender_label: str):
        """
        Získá dva návratové routing stringy pro vrácení zásilky.

        Args:
            sender_label (str): Čárový kód původní zásilky (např. 'Z123456789')

        Returns:
            list[str]: Dva řetězce, které se mají vytisknout pro návratovou zásilku.
        """
        response = self.client.service.senderGetReturnRouting(sender_label)
        return list(response)

    # --------- SHIPMENT METHODS ---------

    def create_shipment(self, packet_ids: list):
        """
        Vytvoření zásilky (shipment) z více balíků.

        packet_ids = ["1234567890", "1234567891", "1234567892"]
        """
        try:
            result = self.client.service.createShipment(PACKETA_API_PASSWORD, packet_ids)

            return result
        
        except Fault as e:
            logger.error(f"Packeta createShipment error: {e}")
            raise Exception(f"Chyba při vytváření shipmentu: {e}")
        
    def get_shipment_packets(self, shipment_id: str):
        """
        Získá seznam balíků ve shipmentu podle jeho shipmentId.
        """
        try:
            result = self.client.service.shipmentPackets(PACKETA_API_PASSWORD, shipment_id)
            return result
        except Fault as e:
            logger.error(f"Packeta shipmentPackets error: {e}")
            raise Exception(f"Chyba při získávání balíků ve shipmentu: {e}")