import requests
import hashlib
import hmac
import base64
import time
from urllib.parse import quote

def get_product_categories(seller_id, mws_auth_token, marketplace_id, sku):
    # 基础参数
    method = 'GET'
    host = 'mws.amazonservices.com'
    uri = '/Products/2011-10-01'
    params = {
        'AWSAccessKeyId': 'YOUR_ACCESS_KEY',
        'Action': 'GetProductCategoriesForSKU',
        'SellerId': seller_id,
        'MWSAuthToken': mws_auth_token,
        'MarketplaceId': marketplace_id,
        'SellerSKU': sku,
        'Timestamp': time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
        'SignatureMethod': 'HmacSHA256',
        'SignatureVersion': '2',
        'Version': '2011-10-01'
    }
    
    # 生成签名
    query_string = '&'.join(f"{k}={quote(str(v), safe='')}" for k, v in sorted(params.items()))
    string_to_sign = f"{method}\n{host}\n{uri}\n{query_string}"
    signature = base64.b64encode(hmac.new('YOUR_SECRET_KEY'.encode('utf-8'), string_to_sign.encode('utf-8'), hashlib.sha256).digest())
    params['Signature'] = signature
    
    # 发送请求
    url = f"https://{host}{uri}?{query_string}&Signature={quote(signature)}"
    response = requests.get(url)
    return response.text

def get_product_details(seller_id, mws_auth_token, marketplace_id, id_type, id_value):
    # 基础参数
    method = 'GET'
    host = 'mws.amazonservices.com'
    uri = '/Products/2011-10-01'
    params = {
        'AWSAccessKeyId': 'YOUR_ACCESS_KEY',
        'Action': 'GetMatchingProductForId',
        'SellerId': seller_id,
        'MWSAuthToken': mws_auth_token,
        'MarketplaceId': marketplace_id,
        'IdType': id_type,  # 可以是ASIN, UPC, EAN等
        'IdList.Id.1': id_value,
        'Timestamp': time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
        'SignatureMethod': 'HmacSHA256',
        'SignatureVersion': '2',
        'Version': '2011-10-01'
    }
    
    # 生成签名和发送请求（同上）
    # ...

from xml.etree import ElementTree as ET

def parse_product_response(xml_response):
    root = ET.fromstring(xml_response)
    
    # 检查是否有销售限制
    restrictions = root.find('.//Restrictions')
    if restrictions is not None:
        return "需要审批"
    
    # 检查类别是否需要审批
    categories = root.findall('.//ProductCategory')
    for category in categories:
        category_name = category.find('.//ProductCategoryName').text
        if "受限" in category_name or "需要审批" in category_name:
            return "需要审批"
    
    return "可以销售"