72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
import uuid
|
|
from apiclient import APIClient, endpoint, JsonRequestFormatter, JsonResponseHandler, HeaderAuthentication
|
|
from apiclient_pydantic import serialize
|
|
|
|
from .models.order import Order, Orders
|
|
from .models.product import Product
|
|
from .models.recipe import Recipe
|
|
from .models.cart import Cart, AddItem, Add
|
|
from .endpoints import Endpoints
|
|
|
|
|
|
class Rohlik(APIClient):
|
|
def __init__(self, api_key: str):
|
|
super().__init__(request_formatter=JsonRequestFormatter, response_handler=JsonResponseHandler, authentication_method=HeaderAuthentication(token=api_key,parameter="x-api-authorization",scheme="token"))
|
|
|
|
@serialize()
|
|
def get_product(self, product_id: int) -> Product:
|
|
product = self.get(Endpoints.product.format(product_id=product_id))
|
|
prices = self.get(Endpoints.prices.format(product_id=product_id))
|
|
stock = self.get(Endpoints.stock.format(product_id=product_id))
|
|
description = self.get(Endpoints.description.format(product_id=product_id))
|
|
composition = self.get(Endpoints.composition.format(product_id=product_id))
|
|
product['prices'] = prices
|
|
product['stock'] = stock
|
|
product['description'] = description['description']
|
|
product['composition'] = composition
|
|
return product
|
|
|
|
@serialize()
|
|
def get_order(self, order_id: int) -> Order:
|
|
return self.get(Endpoints.order.format(order_id=order_id))
|
|
|
|
@serialize()
|
|
def get_orders(self) -> Orders:
|
|
return self.get(Endpoints.orders_delivered)
|
|
|
|
@serialize()
|
|
def get_recipe(self, recipe_id: int) -> Recipe:
|
|
recipe_response = self.get(Endpoints.recipe.format(recipe_id=recipe_id))
|
|
return recipe_response['data']
|
|
|
|
@serialize()
|
|
def get_cart(self) -> Cart:
|
|
return self.get(Endpoints.cart)
|
|
|
|
@serialize()
|
|
def add_to_cart(self, product_id) -> Cart:
|
|
add = Add()
|
|
add_item=AddItem()
|
|
add_item.uuid=str(uuid.uuid4())
|
|
add_item.product_id = product_id
|
|
add_item.action_type = "modify"
|
|
add_item.amount = 1
|
|
add_item.source = "hp-v2"
|
|
add_item.sale_id = 0
|
|
add.items = []
|
|
add.items.append(add_item)
|
|
return self.patch(Endpoints.cart, add.dict(by_alias=True))
|
|
|
|
@serialize()
|
|
def remove_from_cart(self, product_id) -> Cart:
|
|
add = Add()
|
|
add_item=AddItem()
|
|
add_item.uuid=str(uuid.uuid4())
|
|
add_item.product_id = product_id
|
|
add_item.action_type = "modify"
|
|
add_item.amount = -1
|
|
add_item.source = "hp-v2"
|
|
add_item.sale_id = 0
|
|
add.items = []
|
|
add.items.append(add_item)
|
|
return self.patch(Endpoints.cart, add.dict(by_alias=True))
|