kboss/b/capital/get_ecs_price.dspy
2025-07-16 14:27:17 +08:00

274 lines
14 KiB
Plaintext

async def get_discount_or_price(ns={}):
db = DBPools()
async with db.sqlorContext('kboss') as sor:
try:
# 查找供应商id
providerid_li = await sor.R('organization', {'orgname': ns.get('orgname'), 'del_flg': '0'})
if providerid_li:
providerid = providerid_li[0]['id']
provider_parentid = providerid_li[0]['parentid']
else:
return {
'status': False,
'msg': '没有找到指定名字的供应商'
}
# 查找产品id
productid_li = await sor.R('product', {'providerid': providerid, 'name': ns.get('productname')})
if productid_li:
productid = productid_li[0]['id']
else:
return {
'status': False,
'msg': '没有找到provider: %s对应的产品' % providerid
}
# 查找用户id, orgid, parentid
if ns.get('user_orgid'):
user_orgid = ns.get('user_orgid')
else:
user_orgid = (await sor.R('users', {'id': await get_user(), 'del_flg': '0'}))[0]['orgid']
user_parentid_li = await sor.R('organization', {'id': user_orgid, 'del_flg': '0'})
if user_parentid_li:
user_parentid = user_parentid_li[0]['parentid']
else:
return {
'status': False,
'msg': '没有找到对应用户的机构'
}
# 查找供应商和所属机构得协议
init_protocol_li = await sor.R('saleprotocol', {'offer_orgid': providerid, 'bid_orgid': provider_parentid, 'del_flg': '0'})
if init_protocol_li:
init_protocol = init_protocol_li[0]['salemode']
else:
return {
'status': False,
'msg': '没有找到供应商: %s和用户所在机构: %s的协议' % (providerid, user_parentid)
}
if init_protocol == '2':
real_price = 1000000000000.0
person_price_sql = """select * from saleprotocol where offer_orgid = '%s' and bid_orgid = '%s' and
salemode = '2' and del_flg = '0' and CURRENT_DATE <= end_date and
CURRENT_DATE >= start_date""" % (user_parentid, user_orgid)
tongyi_price_sql = """select * from saleprotocol where offer_orgid = '%s' and bid_orgid = '*' and
salemode = '2' and del_flg = '0' and CURRENT_DATE <= end_date and
CURRENT_DATE >= start_date""" % user_parentid
for sql_detail in [person_price_sql, tongyi_price_sql]:
person_price_li = await sor.sqlExe(sql_detail, {})
for person_price in person_price_li:
protocolid = person_price['id']
xieyi_zi_search_li = await sor.R('product_salemode', {'protocolid': protocolid, 'providerid': providerid, 'del_flg': '0'})
for xieyi_zi in xieyi_zi_search_li:
provider_zi_id = xieyi_zi['providerid']
zi_price = xieyi_zi['price']
zi_productid = xieyi_zi['productid']
if provider_zi_id == providerid and zi_productid == productid:
# 判断产品是否存在 有可能在产品表已经删除
prd_res_exists_li = await sor.R('product', {'id': xieyi_zi['productid'], 'del_flg': '0'})
if not prd_res_exists_li:
continue
real_price = float(zi_price)
break
if real_price != 1000000000000.0:
break
if real_price == 1000000000000.0:
return {'status': False, 'msg': '没有找到产品价格'}
else:
return {'status': True, 'price': real_price, 'productid': productid}
# 如果签属的协议是折扣
if init_protocol == '0':
# 查找对应的产品id
if ns.get('prefix'):
productid_li = await sor.R('product', {'providerid': ns.get('providerid'), 'providerpid': ns.get('prefix') + '_' + ns.get('servicetype'), 'del_flg': '0'})
if productid_li:
productid_soure = productid_li[0]['id']
else:
return {'discount': 1.0}
else:
productid_soure = ns.get('productid')
real_discount = 1.0
person_discount_sql = """select * from saleprotocol where offer_orgid = '%s' and bid_orgid = '%s' and
salemode = '0' and del_flg = '0' and CURRENT_DATE <= end_date and
CURRENT_DATE >= start_date""" % (user_parentid, user_orgid)
tongyi_discount_sql = """select * from saleprotocol where offer_orgid = '%s' and bid_orgid = '*' and
salemode = '0' and del_flg = '0' and CURRENT_DATE <= end_date and
CURRENT_DATE >= start_date""" % user_parentid
# 没有个人配置 获取产品统一配置折扣
for sql_detail in [person_discount_sql, tongyi_discount_sql]:
person_discount_li = await sor.sqlExe(sql_detail, {})
for person_discount in person_discount_li:
protocolid = person_discount['id']
xieyi_zi_search_li = await sor.R('product_salemode', {'protocolid': protocolid, 'del_flg': '0'})
for xieyi_zi in xieyi_zi_search_li:
provider_zi_id = xieyi_zi['providerid']
zi_discount = xieyi_zi['discount']
zi_productid = xieyi_zi['productid']
if provider_zi_id == ns.get('providerid') and zi_productid == productid_soure:
# 判断产品是否存在 有可能在产品表已经删除
prd_res_exists_li = await sor.R('product', {'id': xieyi_zi['productid'], 'del_flg': '0'})
if not prd_res_exists_li:
continue
real_discount = float(zi_discount)
break
if real_discount != 1.0:
break
return {'status': True, 'discount': real_discount, 'productid': productid}
if init_protocol == '2':
pass
except Exception as e:
return {'status': False, 'discount_price': 1.0, 'error_msg': str(e)}
async def get_ecs_price(ns={}):
import aiohttp
"""
获取云服务器价格
"""
ecs_url = 'http://api.capitalonline.net/ecs/v1'
action = "DescribePrice"
method = "POST"
param = {}
if ns.get('IsToMonth') == 0 or ns.get('IsToMonth') == '0':
istomonth = 0
else:
istomonth = 1
if ns.get('BillingMethod') == 0 or ns.get('BillingMethod') == '0':
time_unit = '天'
else:
time_unit = '个月'
systemdiskinfo = ns.get('SystemDiskInfo')
if isinstance(systemdiskinfo, str):
systemdiskinfo = json.loads(systemdiskinfo)
datadiskinfo = ns.get('DataDiskInfo')
if datadiskinfo:
if isinstance(datadiskinfo, str):
datadiskinfo = json.loads(datadiskinfo)
db = DBPools()
async with db.sqlorContext('kboss') as sor:
# 查找供应商id
providerid_li = await sor.R('organization', {'orgname': '北京首都在线科技股份有限公司', 'del_flg': '0'})
if providerid_li:
providerid = providerid_li[0]['id']
provider_parentid = providerid_li[0]['parentid']
else:
return {
'status': False,
'msg': '没有找到指定名字的供应商'
}
exist_prducts = await sor.R('product', {'providerid': providerid, 'name': ns['EcsFamilyName'], 'del_flg': '0'})
if exist_prducts:
productid = None
for exist_prduct in exist_prducts:
spec_notes = json.loads(exist_prduct['spec_note'])
spec_dic = {}
for note in spec_notes:
spec_dic.update({note['configName']: note['value']})
if str(ns['Cpu']) == spec_dic['CPU'] and str(ns['Ram']) == spec_dic['内存']:
productid = exist_prduct['id']
nss = {
'orgname': '北京首都在线科技股份有限公司',
'user_orgid': ns.get('user_orgid'),
'productname': ns['EcsFamilyName']
}
ecs_price_dic = await get_discount_or_price(nss)
if ecs_price_dic['status']:
ecs_productid = ecs_price_dic['productid']
ecs_price = float(ecs_price_dic['price']) * int(ns.get('Duration'))
else:
return {
'status': False,
'msg': '无法获取配置的ecs价格 %s' % ecs_price_dic.get('msg')
}
nss['productname'] = '云硬盘'
disk_price_dic = await get_discount_or_price(nss)
if disk_price_dic['status']:
disk_productid = disk_price_dic['productid']
disk_price = float(disk_price_dic['price']) * int(ns.get('Duration'))
else:
return {
'status': False,
'msg': '无法获取配置的云硬盘价格 %s' % disk_price_dic.get('msg')
}
datadisk_total = sum([int(datadisk['Size']) for datadisk in datadiskinfo])
disk_total = int(systemdiskinfo['Size']) + int(datadisk_total)
return {
"RequestId": "7eedccd5-4c14-43a9-9afb-ef3a66dc1af0",
"status": True,
"data": {
"PriceUnit": "%s%s" % (ns.get('Duration'), time_unit),
"PriceSymbol": "¥",
"TotalPrice": float(ecs_price) + (float(disk_price) * disk_total),
"EcsPrice": ecs_price,
'price_detail': [
{
'config_name': 'disk_unit',
'productid': disk_productid,
'price': float(disk_price),
},
{
'config_name': 'disk',
'productid': disk_productid,
'price': float(disk_price) * disk_total,
#'orgin': disk_price_dic['origin_price']
},
{
'config_name': 'ecs',
'productid': ecs_productid,
'price': ecs_price,
#'orgin': ecs_price_dic['origin_price']
}
]
},
"msg": "获取计费信息成功!"
}
# else:
# return {
# 'status': False,
# 'msg': '有产品名称, 没有对应配置'
# }
# else:
# return {
# 'status': False,
# 'msg': '没有找到对应产品'
# }
body = {
"AvailableZoneCode": ns.get("AvailableZoneCode"),
"EcsFamilyName": ns.get("EcsFamilyName") or '',
"Cpu": int(ns.get("Cpu")),
"Ram": int(ns.get("Ram")),
"Gpu": int(ns.get("Gpu")) if ns.get("Gpu") else 0,
"BillingMethod": ns.get('BillingMethod'),
"Duration": int(ns.get("Duration")) if ns.get("Duration") else 1,
"IsToMonth": istomonth,
"SystemDiskInfo": systemdiskinfo,
"Number": int(ns.get("Number")) if ns.get("Number") else 1,
"DataDiskInfo": datadiskinfo if datadiskinfo else [],
}
url = get_signature(action, method, ecs_url, param)
async with aiohttp.ClientSession() as session:
async with session.request(method, url, json=body) as response:
resp = await response.text()
result = json.loads(resp)
if 'Message' in result.keys():
key_word = 'Message'
else:
key_word = 'Msg'
if result['Code'] == 'Success':
result['status'] = True
result.pop('Code')
result['data'] = result.pop('Data')
result['msg'] = result.pop(key_word)
else:
result['status'] = False
result['msg'] = result.pop(key_word)
return result
ret = await get_ecs_price(params_kw)
return ret