# -*- coding: utf-8 -*- def widget_pricing(rows): """ 核心汇算部件价格逻辑 """ # 初始化资源字典 resources = { "cpu": {}, "memory": {}, "disk": {}, "gpu": {} } # 处理查询结果并填充资源字典 for row in rows: resource_type, model, unit_price, resource_unit, stock = row if model == 'STANDARD': model = 'STANDARD' else: model = model.strip() if resource_type in resources: resources[resource_type][model] = { "unit_price": float(unit_price), "resource_unit": resource_unit, "stock": int(stock) } # 将资源字典转换为 JSON 字符串 return json.dumps(resources, indent=4) def apply_discount(price, duration): """折扣策略""" if duration >= 720: # 使用超过30天 return price * 0.9 # 9折优惠 return price def get_memory_price(amount): """内存阶梯定价""" if amount <= 4096: # 4GB以下 return 0.00012 # 较高单价 return 0.0001 # 量大优惠 def calculate_price(request_data, RESOURCE_PRICING, DURATION_FACTOR): """ 计算多种资源组合的总价格 返回计算结果，包含总价和各资源明细 """ # 初始化结果 result = { "total_price": 0.0, "currency": "元", "price_details": [], "status": True, "message": "汇算正常" } # 获取资源配置 resources = request_data.get("resources", []) if not resources: result["status"] = False result["message"] = "异常:请求中必须包含资源信息" return result # 获取使用时长并转换为小时 duration = request_data.get("duration", 1) duration_unit = request_data.get("duration_unit", "小时") duration_hours = duration * DURATION_FACTOR.get(duration_unit, 1) result["duration"] = duration result["duration_unit"] = duration_unit # 遍历计算每种资源的价格 for resource in resources: resource_type = resource.get("type") amount = resource.get("amount", 0) if not resource_type or amount <= 0: continue try: # 获取资源定价信息 if resource_type in ["cpu", "gpu", "disk", "memory"]: # 需要查询型号的部件类型 model = resource.get("model") if not model: raise ValueError(f"{resource_type} 必须指定部件型号/系列") if not RESOURCE_PRICING.get(resource_type): raise ValueError(f"没有该类型组件") resource_info = RESOURCE_PRICING[resource_type].get(model) if not resource_info: raise ValueError(f"无效的 {resource_type} 部件型号: {model}") resource_name = f"{resource_type} {model}" else: raise ValueError(f"不支持的资源类型: {resource_type}") # 检查库存 if resource_info["stock"] < amount: raise ValueError(f'{resource_name} 库存不足 (库存: {resource_info["stock"]}, 请求: {amount})') # 计算价格 unit_price = resource_info["unit_price"] subtotal = unit_price * float(amount) * float(duration_hours) # 更新总价和明细 result["total_price"] += subtotal result["price_details"].append({ "resource_name": resource_name, "resource_type": resource_type, "amount": amount, "resource_unit": resource_info["resource_unit"], "unit_price": unit_price, "duration": duration, "duration_unit": duration_unit, "subtotal": round(subtotal, 2) }) except Exception as e: # 记录错误并继续处理其他资源 result["price_details"].append({ "resource_type": resource_type, "error": str(e) }) if result["status"] == True: result["status"] = False result["message"] = str(e) # 结果格式化 result["total_price"] = round(result["total_price"], 2) return result # 使用示例 if __name__ == "__main__": # RESOURCE_PRICING = widget_pricing() # 模拟资源定价数据（包含多种资源类型和型号） RESOURCE_PRICING2 = { # CPU 资源（按核计价，不同型号单价不同） "cpu": { "AMD EPYC 7542 32-Core Processor": {"unit_price": 0.8, "resource_unit": "核", "stock": 100}, "INTEL(R) XEON(R) PLATINUM 8582C": {"unit_price": 1.5, "resource_unit": "核", "stock": 50}, "STANDARD": {"unit_price": 2.2, "resource_unit": "核", "stock": 0} }, # 内存资源（按 Mi 计价，统一单价 "memory": { "Kingston FURY": # 高性能-便宜 {"unit_price": 0.0001, "resource_unit": "Mi", "stock": 1048576}, # 1024 Gi / 1TB "Crucial P-series": # 性能一般-价格一般 {"unit_price": 0.0005, "resource_unit": "Mi", "stock": 1048576}, "Samsung DDR4 DIMMs": # 高性能-贵 {"unit_price": 0.005, "resource_unit": "Mi", "stock": 1048576}, "STANDARD": {"unit_price": 0.0004, "resource_unit": "Mi", "stock": 0}, # Kyy自营 }, # 云盘资源（按 Gi 计价，不同类型单价不同） "disk": { "DATA": # 共享数据盘 {"unit_price": 0.01, "resource_unit": "Gi", "stock": 20480}, # 20 TiB "SYS": # 系统盘 {"unit_price": 0.005, "resource_unit": "Gi", "stock": 51200}, # 50 TiB }, # GPU 资源（按卡计价，不同型号单价不同） "gpu": { "RTX4090-24G": {"unit_price": 4.0, "resource_unit": "卡", "stock": 12}, "RTX5090-32G": {"unit_price": 6.0, "resource_unit": "卡", "stock": 10}, "A100-80G": {"unit_price": 18.0, "resource_unit": "卡", "stock": 5}, "V100-32G": {"unit_price": 12.0, "resource_unit": "卡", "stock": 8}, "STANDARD-8G": {"unit_price": 2.0, "resource_unit": "卡", "stock": 1}, } } RESOURCE_PRICING = { "cpu": { "AMD EPYC 7542 32-Core Processor": { "unit_price": 0.8, "resource_unit": "核", "stock": 100 }, "STANDARD-9G": { "unit_price": 1.22, "resource_unit": "核", "stock": 2 }, "STANDARD": { "unit_price": 2.2, "resource_unit": "核", "stock": 0 }, "INTEL(R) XEON(R) PLATINUM 8582C": { "unit_price": 1.5, "resource_unit": "核", "stock": 50 } }, "memory": { "Kingston FURY": { "unit_price": 0.0001, "resource_unit": "Mi", "stock": 1048576 }, "Crucial P-series": { "unit_price": 0.0005, "resource_unit": "Mi", "stock": 1048576 }, "STANDARD": { "unit_price": 0.0004, "resource_unit": "Mi", "stock": 0 }, "Samsung DDR4 DIMMs": { "unit_price": 0.005, "resource_unit": "Gi", "stock": 1048576 } }, "disk": { "SYS": { "unit_price": 0.005, "resource_unit": "Gi", "stock": 51200 }, "DATA": { "unit_price": 0.01, "resource_unit": "Gi", "stock": 20480 } }, "gpu": { "V100-32G": { "unit_price": 12, "resource_unit": "卡", "stock": 8 }, "RTX4090-24G": { "unit_price": 4, "resource_unit": "卡", "stock": 12 }, "RTX5090-32G": { "unit_price": 6, "resource_unit": "卡", "stock": 10 }, "A100-80G": { "unit_price": 18, "resource_unit": "卡", "stock": 5 }, "STANDARD-8G": { "unit_price": 2, "resource_unit": "卡", "stock": 1 } } } # 计费周期转换系数（统一转换为小时） DURATION_FACTOR = { "小时": 1, # 按量付费（按小时计费） "天": 24, "周": 168, "月": 720 # 按30天计算(按月计费) } # HTTP请求示例： # 4核CPU-INTEL(R) XEON(R) PLATINUM 8582C + # 8GB-Samsung DDR4 DIMMs内存 + # 512GB-DATA数据盘 + # 2张RTX5090-32G + # 使用1天 params = { "resources": [ {"type": "cpu", "model": "INTEL(R) XEON(R) PLATINUM 8582C", "amount": 4}, {"type": "memory", "model": "Samsung DDR4 DIMMs", "amount": 8192}, # 8GB必须转成8192Mi {"type": "disk", "model": "DATA", "amount": 512}, {"type": "gpu", "model": "RTX5090-32G", "amount": 2} ], "duration": 2, "duration_unit": "天" } # 计算价格 result = calculate_price(params, RESOURCE_PRICING, DURATION_FACTOR) # 打印结果 print(f"状态: {result['status']}") if result["message"]: print(f"提示: {result['message']}") print(f"实时总价: {result['total_price']}{result['currency']}/{result.get('duration')}{result.get('duration_unit')}") print(f"价格明细: {result}") for detail in result["price_details"]: if "error" in detail: print(f"❌ {detail['resource_type']}: {detail['error']}") else: print(f"✅ {detail['resource_name']} ({detail['amount']}{detail['resource_unit']}): " f"{detail['subtotal']} 元 ({detail['unit_price']}元/{detail['resource_unit']}/小时 × " f"{detail['duration']}{detail['duration_unit']})")