api_valerdat.py

import requests
from datetime import date, datetime
from dto import OrderIdsResponseDto, OrderResponseDto
from typing import List
from uuid import UUID


class ValerdatToken:
    def __init__(self, username: str, password: str, path: str):
        self.__username = username
        self.__password = password
        self.path = path
        self.__token = self.__get_token()

    def __get_token(self):
        response = requests.post(
            url=f"{self.path}/auth/login/",
            json={"username": self.__username, "password": self.__password},
        )
        response.raise_for_status()
        return response.json().get("key")

    def get_authorization_header(self):
        return {"Authorization": f"Token {self.__token}"}


class ApiValerdat(ValerdatToken):
    BASE_URL = "https://apipanel.valerdat.com/customer"
    VERSION = "v1"

    def __init__(self, customer: str, username: str, password: str):
        self.customer = customer
        super().__init__(username, password, self.get_customer_path())

    def get_customer_path(self):
        return f"{self.BASE_URL}/{self.customer}/{self.VERSION}"

    def get_orders_ids(self, created_at_gte: date) -> List[OrderIdsResponseDto]:
        created_at_gte = created_at_gte.strftime("%Y-%m-%d")
        response = requests.get(
            url=f"{self.path}/data/getorderids/?createdAtGT={created_at_gte}",
            headers=self.get_authorization_header(),
        )
        response.raise_for_status()
        # Transforming json strings to correct format
        orders = [
            OrderIdsResponseDto(
                UUID(i["uuid"]),
                datetime.fromisoformat(i["created_at"]),
                float(i["gap"]),
            )
            for i in response.json()
        ]
        return sorted(orders, key=lambda o: o.gap, reverse=True)

    def get_order(self, order_uuid: UUID, include_up_to_date: bool = False) -> List[OrderResponseDto]:
        response = requests.get(
            url=f"{self.path}/data/order/?uuid={order_uuid}&includeUpToDate={str(include_up_to_date).lower()}",
            headers=self.get_authorization_header(),
        )
        response.raise_for_status()
        return [OrderResponseDto(
            UUID(i["paqueteId"]),
            UUID(i["compraId"]),
            i["proveedorRef"],
            i["proveedorName"],
            date.fromisoformat(i["fechaEmision"]),
            i["productoRef"],
            i["productoNombre"],
            i["almacenRef"],
            float(i["importeBase"]),
            float(i["cantidadSolicitada"]),
            float(i["precioUnitario"]),
            date.fromisoformat(i["fechaEntregaPrevista"]),
            i["moneda"],
        ) for i in response.json()]