242 lines
11 KiB
Plaintext
242 lines
11 KiB
Plaintext
async def cpcc_product_add(ns={}):
|
||
"""
|
||
自定义产品
|
||
组合产品
|
||
{type: "cpu", model: "STANDARD", amount: 1}
|
||
customize/combination
|
||
:param ns:
|
||
:return:
|
||
"""
|
||
db = DBPools()
|
||
async with db.sqlorContext('kboss') as sor:
|
||
if ns.get('sor'):
|
||
sor = ns.get('sor')
|
||
try:
|
||
# 拼接自定义产品名称 gpu.cores.memory.cpu
|
||
if ns.get('product_type') == 'combination' and ns.get('spec_note'):
|
||
spec_li = json.loads(ns['spec_note']) if isinstance(ns['spec_note'], str) else ns['spec_note']
|
||
# 创建一个字典来存储每种类型的信息
|
||
components = {}
|
||
for item in spec_li:
|
||
comp_type = item["type"]
|
||
if comp_type == 'disk' and item["model"] == 'SYS':
|
||
comp_type = 'disk-sys'
|
||
if comp_type == 'disk' and item["model"] == 'DATA':
|
||
comp_type = 'disk-data'
|
||
components[comp_type] = {
|
||
"model": item["model"].replace(" ", ""), # 去除空格
|
||
"amount": item["amount"]
|
||
}
|
||
# 按照指定顺序提取信息并拼接字符串
|
||
gpu_model = components['gpu']['model'] if components.get('gpu') else '-'
|
||
cpu_amount = components['cpu']['amount']
|
||
cpu_model = components['cpu']['model']
|
||
memory_amount = components['memory']['amount']
|
||
clusterid = ns.get('clusterid')
|
||
|
||
if not clusterid:
|
||
return {
|
||
'status': False,
|
||
'msg': '添加产品失败, 缺少参数站点ID：clusterid'
|
||
}
|
||
|
||
sys_disk_amount = 10
|
||
data_disk_amount = 0
|
||
if components.get('disk-sys'):
|
||
sys_disk_amount = components['disk-sys']['amount']
|
||
if components.get('disk-data'):
|
||
data_disk_amount = components['disk-data']['amount']
|
||
|
||
# 拼接最终字符串
|
||
ns['name'] = f"{gpu_model}.cpu{cpu_amount}memory{memory_amount}.disks{sys_disk_amount}diskd{data_disk_amount}.{cpu_model}.{clusterid}"
|
||
|
||
ns['ptype'] = 20
|
||
ns['unit'] = 'hour'
|
||
ns['discount'] = 1
|
||
ns['classify'] = 'CPCC-COMBINATION' # k8s product
|
||
|
||
# 自定义产品
|
||
if ns.get('product_type') == 'customize':
|
||
ns['name'] = 'bcc.mncn.ia.customize'
|
||
ns['classify'] = 'CPCC-CUSTOMIZE' # init product
|
||
ns['product_area'] = 'customize-area'
|
||
|
||
# 获取协议
|
||
saleprotocol = await sor.R('saleprotocol',
|
||
{'bid_orgid': ns.get('reseller_orgid'), 'offer_orgid': ns.get('providerid'),
|
||
'del_flg': '0'})
|
||
if not saleprotocol:
|
||
return {
|
||
'status': False,
|
||
'msg': '添加失败, 请把算力中心转换成供应商'
|
||
}
|
||
ns['protocolid'] = saleprotocol[0]['id']
|
||
ns['spec_note'] = json.dumps(ns.get('spec_note')) if ns.get('spec_note') and isinstance(ns.get('spec_note'), list) else ns.get('spec_note')
|
||
nss = {
|
||
'id': uuid(),
|
||
'providerid': ns.get('providerid') or '',
|
||
'providerpid': ns.get('providerpid') or '',
|
||
'name': ns.get('name') or '',
|
||
'description': ns.get('description') or '',
|
||
'ptype': ns.get('ptype') or '',
|
||
'state': ns.get('state') or '',
|
||
# 'effect_date': ns.get('start_date') or '',
|
||
# 'expire_date': ns.get('end_date') or '',
|
||
'salemode': ns.get('salemode') or '',
|
||
'product_code': ns.get('product_code') or '',
|
||
'spec_note': ns.get('spec_note') or '',
|
||
'product_area': ns.get('product_area') or '',
|
||
'specific_pattern': ns.get('specific_pattern') or '',
|
||
'reseller_orgid': ns.get('reseller_orgid') or '',
|
||
'classify': ns.get('classify'),
|
||
'productgroup': ns.get('productgroup'),
|
||
'label': ns.get('label')
|
||
}
|
||
nss_product_salemode = {
|
||
'id': uuid(),
|
||
'protocolid': ns.get('protocolid'),
|
||
'providerid': ns.get('providerid'),
|
||
'productid': nss['id'],
|
||
'discount': ns.get('discount'),
|
||
'price': ns.get('floorprice'),
|
||
'unit': ns.get('unit')
|
||
}
|
||
nss_product_salemode_copy = {
|
||
'id': uuid(),
|
||
'protocolid': ns.get('protocolid'),
|
||
'providerid': ns.get('providerid'),
|
||
'productid': '*',
|
||
'discount': ns.get('discount'),
|
||
'price': ns.get('floorprice'),
|
||
'unit': ns.get('unit')
|
||
}
|
||
# if not ns.get('effect_date'):
|
||
# return {
|
||
# "status": False,
|
||
# "message": "product effect date is empty"
|
||
# }
|
||
#
|
||
# if not ns.get('expire_date'):
|
||
# nss['expire_date'] = "9999-12-31"
|
||
|
||
# 添加产品 如果有底价 触发插入底价表, 同时插入两条数据
|
||
# 1. 买方售方都设置则为底价 2.售方为本机构 买方为空则为售价
|
||
if ns.get('selling') and ns.get('floorprice'):
|
||
if int(ns.get('selling')) <= int(ns.get('floorprice')):
|
||
return {
|
||
'status': False,
|
||
'msg': '售价不能低于底价'
|
||
}
|
||
ns_floor = {
|
||
'id': uuid(),
|
||
'offer_orgid': ns.get('providerid'),
|
||
'bid_orgid': ns.get('reseller_orgid'),
|
||
'productid': nss.get('id'),
|
||
'price': ns.get('floorprice'),
|
||
'begin_date': ns.get('effect_date'),
|
||
'end_date': '9999-12-31'
|
||
}
|
||
ns_selling = {
|
||
'id': uuid(),
|
||
'offer_orgid': ns.get('reseller_orgid'),
|
||
'bid_orgid': '',
|
||
'productid': nss.get('id'),
|
||
'price': ns.get('selling'),
|
||
'begin_date': ns.get('effect_date'),
|
||
'end_date': '9999-12-31'
|
||
}
|
||
await sor.C('floorprice', ns_floor)
|
||
await sor.C('floorprice', ns_selling)
|
||
if ns.get('protocolid'):
|
||
# 获取供应商名称 用于筛选中金设置映射产品
|
||
provider_name_li = await sor.R('organization', {'id': ns.get('providerid')})
|
||
provider_name = provider_name_li[0]['orgname']
|
||
|
||
# 获取协议起始时间
|
||
time_line_li = await sor.R('saleprotocol', {'id': ns.get('protocolid'), 'del_flg': '0'})
|
||
start_time = time_line_li[0]['start_date']
|
||
end_time = time_line_li[0]['end_date']
|
||
nss['effect_date'] = start_time
|
||
nss['expire_date'] = end_time
|
||
nss['salemode'] = time_line_li[0]['salemode']
|
||
|
||
# 检查产品表是否已经存在
|
||
ns_exist = {
|
||
'providerid': ns.get('providerid'),
|
||
'name': ns.get('name'),
|
||
'del_flg': '0'
|
||
}
|
||
product_exist_li = await sor.R('product', ns_exist)
|
||
# 如果已经存在就更新
|
||
if product_exist_li:
|
||
product_table_exits_id = product_exist_li[0]['id']
|
||
nss['id'] = product_table_exits_id
|
||
await sor.U('product', nss)
|
||
# 此时默认协议子表已经存在并更新
|
||
# 查找协议子表id 用于更新
|
||
exists_zibiao = {
|
||
'protocolid': ns.get('protocolid'),
|
||
'providerid': ns.get('providerid'),
|
||
'productid': nss['id']
|
||
}
|
||
exists_zibiao_li = await sor.R('product_salemode', exists_zibiao)
|
||
if exists_zibiao_li:
|
||
zibiao_id = exists_zibiao_li[0]['id']
|
||
nss_product_salemode['id'] = zibiao_id
|
||
nss_product_salemode['productid'] = nss['id']
|
||
if provider_name == '中金超算':
|
||
await sor.U('product_salemode', nss_product_salemode_copy)
|
||
await sor.U('product_salemode', nss_product_salemode)
|
||
else:
|
||
await sor.C('product', nss)
|
||
# 如果时中金协议子表创建两个 一个productid正常用于展示 一个productid为*
|
||
if provider_name == '中金超算':
|
||
zj_offer_yezhu_bind_exist = {
|
||
'protocolid': ns.get('protocolid'),
|
||
'providerid': ns.get('providerid'),
|
||
'productid': '*',
|
||
'del_flg': '0'
|
||
}
|
||
# 如果中金offer bind业主机构 已经配置过* 更新
|
||
zj_exist_li = await sor.R('product_salemode', zj_offer_yezhu_bind_exist)
|
||
if zj_exist_li:
|
||
nss_product_salemode_copy['id'] = zj_exist_li[0]['id']
|
||
await sor.U('product_salemode', nss_product_salemode_copy)
|
||
else:
|
||
await sor.C('product_salemode', nss_product_salemode_copy)
|
||
await sor.C('product_salemode', nss_product_salemode)
|
||
|
||
# 如果是批量添加 salemode=1就是添加产品 不添加统一客户配置
|
||
if ns.get('batch'):
|
||
ns_batch_dic = {
|
||
'bid_orgid': nss['reseller_orgid'],
|
||
'providerid': nss['providerid'],
|
||
'productid': nss['id'],
|
||
'sor': sor
|
||
}
|
||
if ns.get('salemode') == '0':
|
||
ns_batch_dic['discount'] = ns.get('kehu_discount')
|
||
await discount_guanli_jia(ns_batch_dic)
|
||
elif ns.get('salemode') == '2':
|
||
ns_batch_dic['price'] = ns.get('kehu_price')
|
||
await price_guanli_jia(ns_batch_dic)
|
||
else:
|
||
if ns.get('classify') == 'E':
|
||
nss['hotshow'] = '0'
|
||
await sor.C('product',nss)
|
||
return {
|
||
"status": True,
|
||
"msg": "product add success",
|
||
}
|
||
except Exception as e:
|
||
await sor.rollback()
|
||
import traceback
|
||
traceback.print_exc()
|
||
return {
|
||
"status": False,
|
||
"msg": "product add failed, %s" % str(e) + traceback.format_exc()
|
||
}
|
||
|
||
|
||
ret = await cpcc_product_add(params_kw)
|
||
return ret |