企业组织年末优惠活动

from flask import Flask, request, jsonify

app = Flask(__name__)

# 模拟商品数据库
products = {
    '1': {'name': 'Laptop', 'price': 1000},
    '2': {'name': 'Smartphone', 'price': 500},
}

# 模拟优惠活动
discounts = {
    '1': 0.1,  # Laptop 10% off
    '2': 0.2,  # Smartphone 20% off
}

@app.route('/calculate_price', methods=['POST'])
def calculate_price():
    data = request.json
    product_id = data.get('product_id')
    quantity = data.get('quantity')

    if product_id not in products or product_id not in discounts:
        return jsonify({'error': 'Invalid product ID'}), 400

    original_price = products[product_id]['price']
    discount = discounts[product_id]
    discounted_price = original_price * (1 - discount)
    total_price = discounted_price * quantity

    return jsonify({
        'product_name': products[product_id]['name'],
        'original_price': original_price,
        'discounted_price': discounted_price,
        'total_price': total_price
    })

if __name__ == '__main__':
    app.run(debug=True)