seller.py

import io
import logging.config
import os
import re
import zipfile
from environs import Env

import pandas as pd
import requests

logger = logging.getLogger(__file__)


def get_product_list(last_id, client_id, seller_token):
    """Получить список товаров магазина озон.

    Отправить запрос на сайт озон и получить список товаров.

    Аргументы:
        last_id (str): идентификатор прошлого запроса ("" в момент отправки первого)
        client_id (str): уникальный идентификатор клиента озон
        seller_token (str): уникальный идентификатор продавца озон

    Возвращаемые значения:
        dict: Список продуктов

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    url = "https://api-seller.ozon.ru/v2/product/list"
    headers = {
        "Client-Id": client_id,
        "Api-Key": seller_token,
    }
    payload = {
        "filter": {
            "visibility": "ALL",
        },
        "last_id": last_id,
        "limit": 1000,
    }
    response = requests.post(url, json=payload, headers=headers)
    response.raise_for_status()
    response_object = response.json()
    return response_object.get("result")


def get_offer_ids(client_id, seller_token):
    """Получить артикулы товаров магазина озон.

    Из общего списка товаров магазина озон получить список артикулов.

    Аргументы:
        client_id (str): уникальный идентификатор клиента озон
        seller_token (str): уникальный идентификатор продавца озон

    Возвращаемые значения:
        list: список артикулов
    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    last_id = ""
    product_list = []
    while True:
        some_prod = get_product_list(last_id, client_id, seller_token)
        product_list.extend(some_prod.get("items"))
        total = some_prod.get("total")
        last_id = some_prod.get("last_id")
        if total == len(product_list):
            break
    offer_ids = []
    for product in product_list:
        offer_ids.append(product.get("offer_id"))
    return offer_ids


def update_price(prices: list, client_id, seller_token):
    """Обновить цены товаров на сайте.

    Обновить цены товаров, отправив запрос на сайт озон.

    Аргументы:
        prices (list): список цен
        client_id (str): уникальный идентификатор клиента озон
        seller_token (str): уникальный идентификатор продавца озон

    Возвращаемые значения:
        dict: ответ от сайта
    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    url = "https://api-seller.ozon.ru/v1/product/import/prices"
    headers = {
        "Client-Id": client_id,
        "Api-Key": seller_token,
    }
    payload = {"prices": prices}
    response = requests.post(url, json=payload, headers=headers)
    response.raise_for_status()
    return response.json()


def update_stocks(stocks: list, client_id, seller_token):
    """Обновить остатки на сайте.

    Обновить остатки товаров, отправив запрос на сайт озон.

    Аргументы:
        stocks (list): список остатков
        client_id (str): уникальный идентификатор клиента озон
        seller_token (str): уникальный идентификатор продавца озон

    Возвращаемые значения:
        dict: ответ от сайта

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    url = "https://api-seller.ozon.ru/v1/product/import/stocks"
    headers = {
        "Client-Id": client_id,
        "Api-Key": seller_token,
    }
    payload = {"stocks": stocks}
    response = requests.post(url, json=payload, headers=headers)
    response.raise_for_status()
    return response.json()


def download_stock():
    """Скачать файл ostatki с сайта casio.

    Скачать файл ostatki.zip с сайта casio, получить из него данные и удалить файл.

    Возвращаемые значения:
        list: данные об остатках из файла

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    # Скачать остатки с сайта
    casio_url = "https://timeworld.ru/upload/files/ostatki.zip"
    session = requests.Session()
    response = session.get(casio_url)
    response.raise_for_status()
    with response, zipfile.ZipFile(io.BytesIO(response.content)) as archive:
        archive.extractall(".")
    # Создаем список остатков часов:
    excel_file = "ostatki.xls"
    watch_remnants = pd.read_excel(
        io=excel_file,
        na_values=None,
        keep_default_na=False,
        header=17,
    ).to_dict(orient="records")
    os.remove("./ostatki.xls")  # Удалить файл
    return watch_remnants


def create_stocks(watch_remnants, offer_ids):
    """Структуризовать список остатков.

    Структуризовать список остатков, добавить недостающее и привести список к единому виду.

    Аргументы:
        watch_remnants (list): данные об остатках с сайта
        offer_ids (list): список артикулов

    Возвращаемые значения:
        list: дополненный список остатков

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    # Уберем то, что не загружено в seller
    stocks = []
    for watch in watch_remnants:
        if str(watch.get("Код")) in offer_ids:
            count = str(watch.get("Количество"))
            if count == ">10":
                stock = 100
            elif count == "1":
                stock = 0
            else:
                stock = int(watch.get("Количество"))
            stocks.append({"offer_id": str(watch.get("Код")), "stock": stock})
            offer_ids.remove(str(watch.get("Код")))
    # Добавим недостающее из загруженного:
    for offer_id in offer_ids:
        stocks.append({"offer_id": offer_id, "stock": 0})
    return stocks


def create_prices(watch_remnants, offer_ids):
    """Получить список цен.

    Из общего списка данных получить список цен в нужной форме

    Аргументы:
        watch_remnants (list): общий список данных
        offer_ids (list): список артикулов

    Возвращаемые значения:
        list: структурированный список цен

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    prices = []
    for watch in watch_remnants:
        if str(watch.get("Код")) in offer_ids:
            price = {
                "auto_action_enabled": "UNKNOWN",
                "currency_code": "RUB",
                "offer_id": str(watch.get("Код")),
                "old_price": "0",
                "price": price_conversion(watch.get("Цена")),
            }
            prices.append(price)
    return prices


def price_conversion(price: str) -> str:
    """Преобразовать цену.

    Преобразовывает произвольную цену в однотипную строку.

    Аргументы:
        price (str): произвольная цена.

    Возвращаемые значения:
        str: преобразованная строка.

    Примеры:
        \>>> price = "5'990.00 руб"
        \>>> price_conversion(price)
        5990

        \>>> price = "1wd4"
        \>>>price_conversion(price)
        14
    """
    return re.sub("[^0-9]", "", price.split(".")[0])


def divide(lst: list, n: int):
    """Разделить список lst на части по n элементов"""
    for i in range(0, len(lst), n):
        yield lst[i : i + n]


async def upload_prices(watch_remnants, client_id, seller_token):
    """Структурировать список цен и отправить их на сайт озон

    Аргументы:
        watch_remnants (list): общий список данных
        client_id (str): уникальный идентификатор клиента озон
        seller_token (str): уникальный идентификатор продавца озон
    Возвращаемые значения:
        list: структурированный список цен

    Исключения:
        HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    offer_ids = get_offer_ids(client_id, seller_token)
    prices = create_prices(watch_remnants, offer_ids)
    for some_price in list(divide(prices, 1000)):
        update_price(some_price, client_id, seller_token)
    return prices


async def upload_stocks(watch_remnants, client_id, seller_token):
    """Структурировать список остатков и отправить их на сайт озон

        Аргументы:
            watch_remnants (list): общий список данных
            client_id (str): уникальный идентификатор клиента озон
            seller_token (str): уникальный идентификатор продавца озон
        Возвращаемые значения:
            list: список товаров с положительными остатками
            list: структурированный список остатков

        Исключения:
            HTTPSConnectionPool(host='api-seller.ozon.ru', port=443): Max retries exceeded with url: /v2/product/list (Caused by NewConnectionError('<urllib3.connection.HTTPSConnection object at 0x000001743C7C53D0>: Failed to establish a new connection: [Errno 11001] getaddrinfo failed')) Ошибка соединения.
    """
    offer_ids = get_offer_ids(client_id, seller_token)
    stocks = create_stocks(watch_remnants, offer_ids)
    for some_stock in list(divide(stocks, 100)):
        update_stocks(some_stock, client_id, seller_token)
    not_empty = list(filter(lambda stock: (stock.get("stock") != 0), stocks))
    return not_empty, stocks


def main():
    env = Env()
    seller_token = env.str("SELLER_TOKEN")
    client_id = env.str("CLIENT_ID")
    try:
        offer_ids = get_offer_ids(client_id, seller_token)
        watch_remnants = download_stock()
        # Обновить остатки
        stocks = create_stocks(watch_remnants, offer_ids)
        for some_stock in list(divide(stocks, 100)):
            update_stocks(some_stock, client_id, seller_token)
        # Поменять цены
        prices = create_prices(watch_remnants, offer_ids)
        for some_price in list(divide(prices, 900)):
            update_price(some_price, client_id, seller_token)
    except requests.exceptions.ReadTimeout:
        print("Превышено время ожидания...")
    except requests.exceptions.ConnectionError as error:
        print(error, "Ошибка соединения")
    except Exception as error:
        print(error, "ERROR_2")


if __name__ == "__main__":
    main()