wstkeys/API/rok.py

80 lines
1.8 KiB
Python

from dotenv import load_dotenv
import os
load_dotenv()
import requests
partner = os.getenv('partner')
user_name = os.getenv('userName')
password = os.getenv('password')
headers = {
'Content-Type': 'application/json',
}
BASE_URL = "https://partner.rokky.com/api/v1"
json_data = {
'partner': partner,
'userName': user_name,
'password': password,
'scope': 'WebApi',
}
def tokens():
response = requests.post('https://partner.rokky.com/api/v1/tokens', headers=headers, json=json_data)
# print(response.json())
data = response.json()
access_token = data["accessToken"]
return access_token
access_token = tokens()
def get_orders(access_token, page=1, page_size=25):
url = f"{BASE_URL}/orders"
headers = {
"Authorization": f"Bearer {access_token}",
"Content-Type": "application/json"
}
payload = {
"page": page,
"pageSize": page_size
}
response = requests.get(url, headers=headers, json=payload)
response.raise_for_status()
print(response.json())
# return response.json()
def download_products(user_name):
url = f"https://cdn-partners.rokky.com/partner-catalogs/{user_name}_products.json.gz.enc"
response = requests.get(url)
response.raise_for_status()
with open("./files/products.json.gz.enc", "wb") as f:
f.write(response.content)
print("Каталог продуктов скачан.")
download_products(user_name)
# get_orders(access_token)
def download_prices(partner_code):
url = f"https://cdn-partners.rokky.com/partner-catalogs/{partner_code}_prices.json.gz.enc"
response = requests.get(url)
response.raise_for_status()
with open("./files/prices.json.gz.enc", "wb") as f:
f.write(response.content)
print("Каталог цен скачан.")
# download_prices(partner)