def update_file(files, file_path, invoice_data):
isExists = os.path.exists(file_path)
if not isExists:
os.makedirs(file_path)
for file in files:
# 原始文件名
source_fname = files.get(file)[0].get("filename")
# 生成新的文件名, 返回完整路径
full_file = FileUploadOperate.gen_file_name(
file_path, source_fname)
with open(full_file, "wb") as f:
f.write(files.get(file)[0].get("body"))
em = "save {0} to path: {1}".format(file, full_file)
LOG.info(em)
# 更新数据库数据库
upload_db_obj = UploadLog.query.filter(
and_(UploadLog.invoice_uuid == invoice_data.invoice_uuid,
UploadLog.file_type == file)).first()
if not upload_db_obj:
em = "can not found file type: <{0}> with invoice id: <{1}>".format(
file, invoice_data.invoice_uuid)
LOG.exception(em)
continue
upload_db_obj.file = full_file
upload_db_obj.create_time = int(time.time)
upload_db_obj.status = "unapproval"
db_session.flush()
db_session.commit()
return 200
def generate_discount(money, counts=1, code_len=8, valid_date=None):
''' 随机生成8位的优惠码 记录到数据库'''
db_objs = []
for count in range(counts):
code_list = []
# 0-9
for i in range(10):
code_list.append(str(i))
# A-Z
for i in range(65, 91):
code_list.append(chr(i))
# a-z
for i in range(97, 123):
code_list.append(chr(i))
myslice = random.sample(code_list, code_len)
verification_code = ''.join(myslice)
db_obj = Discount(uid=str(uuid.uuid1()),
discount_id=verification_code,
money=money,
gen_time=int(time.time()),
is_used=False,
valid_date=valid_date,
used_time=None,
description=None)
db_objs.append(db_obj)
for db_obj in db_objs:
db_session.add(db_obj)
db_session.flush()
db_session.commit()
return {"result": "successful"}
def update_user_money(money, order):
"""
this method to update user's money(order's money)
:param order: db object with order info
:return:
"""
try:
if float(money) <= 0:
return {"code": 0, "message": ""}
user = User.query.filter(User.user_id == order.user_id).first()
if not user:
em = "can not found user id: <{0}>".format(order.user_id)
LOG.exception(em)
return False
user.money = (float(user.money) if float(user.money) else 0)
new_money = user.money - money
if new_money < 0 and order.order_type == options.order_immed:
em = "user have no enough money. user id: <{0}> resource id: <{1}> " \
"resource name: <{2}>".format(order.user_id,
order.resource_id,
order.resource_name)
LOG.exception(em)
return {"code": 1, "message": em}
User.query.filter(User.user_id == order.user_id).update(
{User.money: new_money})
db_session.flush()
msg = "update user <{0}> money from <{1}> to <{2}>".format(
user.user_id, user.money, new_money)
LOG.debug(msg)
# add log
consume_log = ConsumeLog(str(uuid.uuid1()),
order.user_id,
order.user_name,
order.project_id,
order.resource_name,
order.resource_type,
order.resource_id,
order.start_time,
order.end_time,
money,
int(time.time()),
resource_from=order.resource_from,
order_uid=order.uid,
details=order.resource)
db_session.add(consume_log)
db_session.commit()
# 实时同步数据到cmdb
CMDBHandle.syncdata2cmdb("consume", consume_log.uid)
return {"code": 0, "message": ""}
except Exception as e:
db_session.rollback()
em = "update user money failed. user id: <{0}> " \
"resource id: <{1}> resource name: <{2}> msg: <{3}>".format(order.user_id,
order.resource_id,
order.resource_name,
e)
LOG.exception(em)
return {"code": 1, "message": em}
def add_order(data):
"""添加订单"""
data = orderinfo_plugins.GetOrderAbout(data)
for r_name, used in data.resources.iteritems():
if not used:
continue
order_object = Orders(str(uuid.uuid1()),
start_time=data.time_stamp,
end_time=data.end_time,
resource_from=data.resource_from,
resource_from_provider=data.resource_from_provider,
off=data.off,
used=used,
resource_id=data.resource_id,
resource_name=r_name,
project_id=data.project_id,
project_name=data.project_name,
user_id=data.user_id,
resource=data.resource,
user_name=data.user_name,
order_type=data.order_type,
log_time=int(time.time()),
status=options.order_Settlement,
)
if data.money <= 0:
em = "tencent order. money can not le 0"
LOG.exception(em)
continue
# 更新用户余额，添加消费记录
result = ExprOrders.update_user_money(data.money, order_object)
if result.get("code") != 0:
return result
# 订单状态为已结算
db_session.add(order_object)
db_session.flush()
# 更新用户过期时间
ExprOrders.expr_user_exceed_time(data.user_id)
db_session.commit()
return {"code": 0, "message": ""}
def delete_invoice(uuid):
"""取消发票"""
invoice_obj = Invoice.query.filter(Invoice.uuid == uuid).first()
if not invoice_obj:
em = "can not found invoice with uuid <{0}>".format(uuid)
LOG.exception(em)
raise HTTPError(400, em)
# 如果发票状态为已邮寄，则不能取消和删除
if invoice_obj.status == options.invoice_status.get("posted"):
em = "invoice is posted. can not delete"
LOG.exception(em)
raise HTTPError(400, em)
# 更新充值记
if invoice_obj.recharge_uuids:
recharge_ids = invoice_obj.recharge_uuids.split(",")
for recharge_id in recharge_ids:
RechargeLog.query.filter(RechargeLog.uid == recharge_id).update({RechargeLog.is_invoiced: False})
db_session.flush()
invoice_obj.deleted = True
db_session.commit()
return invoice_obj
def use_discount(id, user_context):
try:
now_time = int(time.time())
# check discount id is it exsit
discount = Discount.query.filter(
Discount.discount_id == id).first()
if not discount:
em = "can not found discount id. id: <0>".format(id)
LOG.exception(em)
return False, 410
# check discount id is is exceed
if discount.valid_date or discount.is_used:
if discount.valid_date <= now_time or discount.is_used:
em = "discount id is not valid. id: <{0}>".format(
discount.discount_id)
LOG.exception(em)
return False, 400
user_id = user_context.get("user").get("user").get("id")
# update user's money
user = User.query.filter(User.user_id == user_id).first()
if not user:
em = "can not found user. user id: <{0}>".format(user_id)
LOG.exception(em)
# try to add user
data = {
"tenant_id":
user_context.get("tenant_id"),
"_context_project_name":
user_context.get("user").get("project").get("name"),
"user_id":
user_context.get("user").get("user").get("id"),
"_context_user_name":
user_context.get("user").get("user").get("id")
}
ret = UserOperate.add_user(data)
if not ret.get("success"):
return False, 500
user = ret.get("success")[0].get("data")
user.money += discount.money
# update discount tables's status
# Discount.query.filter(Discount.discount_id == id).update({Discount.is_used: True})
# add Recharge Log
recharge = RechargeLog(
uid=str(uuid.uuid1()),
user_id=user_id,
user_name=user_context.get("user").get("user").get("name"),
money=discount.money,
log_time=int(time.time()),
out_trade_no=discount.discount_id,
trade_status="success",
recharge_way="discount_id")
db_session.add(recharge)
db_session.flush()
# update discount id
Discount.query.filter(Discount.discount_id == id).update({
Discount.is_used:
True,
Discount.used_time:
now_time
})
db_session.commit()
LOG.debug("discount id used by user id: <{0}>")
# update user's exceed time
ExprOrders.sync_user_exceed_time(user_id=user_id)
return True, 200
except Exception as e:
db_session.rollback()
em = "use discount id error. id: <{0}> user id<{1}>".format(
id, user_id)
LOG.exception(em)
return False, 500
def sync_user_exceed_time(user_id=None):
"""数据同步，根据用户当前订单及余额，更新用户超时时间"""
try:
now_time = int(time.time())
if not user_id:
all_users = User.query.filter().all()
else:
all_users = User.query.filter(User.user_id == user_id).all()
for user in all_users:
total_used = 0
min_used = 0
orders = Orders.query.filter(
and_(Orders.user_id == user.user_id, Orders.status == None,
Orders.resource_type)).all()
# 如果没有订单，余额大于0，更新过期时间为空，未过期
if not orders and float(user.money) > 0:
User.query.filter(User.user_id == user.user_id).update({
User.is_exceed:
False,
User.exceed_time:
None,
})
db_session.flush()
for order in orders:
order.end_time = now_time
order.order_type = options.order_immed
price = Price.query.filter(
Price.price_type == order.resource_type).first()
if not price:
em = "can not found price type. price type: <{0}>".format(
order)
LOG.exception(em)
return False
used = ExprOrders.order_used_money(order,
price,
immed=True)
if used:
total_used += used
order.end_time = None
if total_used > float(user.money):
if not user.is_exceed:
User.query.filter(User.user_id == user.user_id).update(
{
User.is_exceed: True,
User.exceed_time: now_time,
})
db_session.flush()
else:
for order in orders:
price = Price.query.filter(
Price.price_type == order.resource_type).first()
used = ExprOrders.order_used_money(order,
price,
immed=False)
if used:
min_used += used
if total_used > 0 and min_used > 0:
new_money = float(user.money) - total_used
valid_use_sec = new_money / min_used * 60
user_exceed_time = valid_use_sec + now_time
User.query.filter(User.user_id == user.user_id).update(
{
User.exceed_time: user_exceed_time,
User.is_exceed: False
})
db_session.flush()
em = "update <{0}> user's exceed time to <{1}>".format(
user.user_id, user_exceed_time)
LOG.debug(em)
print "==================finish one. user_id: <{0}>==============================".format(
user.user_id)
db_session.commit()
except ExprOrders as e:
db_session.rollback()
print "error. user_id==========>>>, ", user.user_id
print e
def expr_user_exceed_time(user_id, new_orders=None):
"""计算用户过期时间"""
# user_id 用户唯一ID
# new_order 新进来的订单，列表中包含数据库对象[db_obj1, db_obj2, db_obj3, .....]
min_used_money = 0
total_used_money = 0
now_time = int(time.time())
user = User.query.filter(User.user_id == user_id).first()
if not user:
em = "can not found user. user id: <{0}>".format(user_id)
LOG.exception(em)
return {"code": 1, "message": em}
orders = Orders.query.filter(
and_(Orders.user_id == user_id, Orders.status == None,
Orders.resource_type != None)).all()
for order in orders:
price = Price.query.filter(
Price.price_type == order.resource_type).first()
print "start time=======", order.start_time
print "end time=======", order.end_time
# 结束时间为当前时间
order.end_time = now_time
# 获取订单总使用金额
total_used = ExprOrders.order_used_money(order, price, immed=True)
if total_used:
total_used_money += total_used
# 获取订单每分钟使用金额
min_used = ExprOrders.order_used_money(order, price, immed=False)
if min_used:
min_used_money += min_used
order.end_time = None
if new_orders:
for new_order in new_orders:
price = Price.query.filter(
Price.price_type == new_order.resource_type).first()
if price:
# 获取新订单每分钟使用金额
min_used = ExprOrders.order_used_money(new_order,
price,
immed=False)
if min_used:
min_used_money += min_used
# 当前余额 - 之前订单使用总和 = 可用余额
vaild_money = float(user.money) - total_used_money
# 余额小于0，直接更新为过期
if vaild_money <= 0:
if not user.is_exceed:
User.query.filter(User.user_id == user_id).update({
User.is_exceed:
True,
User.exceed_time:
now_time
})
db_session.flush()
em = "user available money le 0. user is exceed with user id: <{0}>".format(
user_id)
LOG.debug(em)
else:
# 可用余额 / 每分钟使用量 * 60 = 可用秒数
if min_used_money > 0:
valid_use_sec = (vaild_money / min_used_money) * 60
# 过期时间 = 当前时间 + 可用秒数
user_exceed_time = now_time + valid_use_sec
User.query.filter(User.user_id == user_id).update(
{User.exceed_time: user_exceed_time})
db_session.flush()
em = "update user's exceed time to <{0}> with user id <{1}>".format(
user_exceed_time, user_id)
LOG.debug(em)
# 用户现在过期状态为`已过期`，如果无订单，余额大于0，更新用户过期状态为`未过期`
if user.is_exceed:
if not orders:
User.query.filter(User.user_id == user_id).update({
User.exceed_time:
None,
User.is_exceed:
False
})
db_session.flush()
db_session.commit()
return True
def add_invoice(data, upfile_uuid):
# 计算所有充值记录之和
if not data.orders:
em = "no any more orders id"
LOG.exception(em)
raise HTTPError(400, em)
money = 0
for order in data.orders:
# 查找所有充值记录
recharge = RechargeLog.query.filter(RechargeLog.uid == order).first()
if not recharge:
em = "user id: <{0}>. can not found recharge log. id: <{1}>".format(data.user_id, order)
LOG.exception(em)
continue
if recharge.trade_status != "success" :
em = "user id: <{0}>. recharge log is not success. id: <{1}>".format(data.user_id, order)
LOG.exception(em)
continue
if recharge.is_invoiced:
em = "user id: <{0}>. recharge log is invoiced. id: <{1}>".format(data.user_id, order)
LOG.exception(em)
continue
# 不允许开取发票的类型
if recharge.recharge_way in options.reject_recharge_type:
em = "user id: <{0}>. recharge type is not allowed id: <{1}>".format(data.user_id, order)
LOG.exception(em)
continue
money += float(recharge.money)
# 更新充值记录为已开发票
RechargeLog.query.filter(RechargeLog.uid == order).update({RechargeLog.is_invoiced: True})
db_session.flush()
if money <= 0:
em = "user id: <{0}>. apply invoice error. user's money is le 0".format(data.user_id)
LOG.exception(em)
raise HTTPError(500, "server internal error")
# add database
invoice_obj = Invoice(uuid=data.invoice_uuid,
title=data.title,
status=options.invoice_status.get("verifying"),
post_address=data.post_address,
post_user=data.post_user,
post_phone=data.post_phone,
money=money,
title_type=data.title_type,
title_mode=data.title_mode,
context=data.context,
corporation_name=data.corporation_name,
taxpayer_dentity=data.taxpayer_dentity,
register_address=data.register_address,
register_phone=data.register_phone,
deposit_bank=data.deposit_bank,
deposit_account=data.deposit_account,
user_name=data.user_name,
user_id=data.user_id,
application_date=get_nowtime(),
description=data.description,
recharge_uuids=','.join(data.orders),
upfile_uuid=upfile_uuid,
)
db_session.add(invoice_obj)
db_session.commit()
return invoice_obj
def add_order(data):
"""添加订单"""
data = orderinfo_plugins.GetOrderAbout(data)
new_orders = []
for r_name, used in data.resources.iteritems():
if not used:
continue
# 检查订单是否存在
check_orders = Orders.query.filter(
and_(Orders.resource_id == data.resource_id,
Orders.project_id == data.project_id,
Orders.user_id == data.user_id,
Orders.resource_name == r_name,
Orders.status == None)).first()
if check_orders:
em = "order is already exist. resource id: <{0}> resource type: <{1}>".format(
data.resource_id, r_name)
LOG.exception(em)
continue
# 获取单价
price_type_obj = Price.query.filter(Price.name == r_name).first()
if not price_type_obj:
n_r_name = r_name.split("/")[-1]
price_type_obj = Price.query.filter(
Price.name == n_r_name).first()
price_type = (price_type_obj.price_type
if price_type_obj else None)
order_object = Orders(
str(uuid.uuid1()),
start_time=data.time_stamp,
end_time=data.end_time,
resource_from=data.resource_from,
resource_from_provider=data.resource_from_provider,
off=data.off,
used=used,
resource_id=data.resource_id,
resource_type=price_type,
resource_name=r_name,
project_id=data.project_id,
project_name=data.project_name,
user_id=data.user_id,
resource=data.resource,
user_name=data.user_name,
order_type=data.order_type,
log_time=int(time.time()),
)
# 订单类型-立即结算
if data.order_type == options.order_immed:
if not price_type_obj:
em = "can not found price type, order type is immediately. Please define first"
LOG.exception(em)
return {"code": 1, "message": em}
else:
# get the order total used money
immed_order_money = ExprOrders.order_used_money(
order_object, price_type_obj, immed=True)
if not immed_order_money:
continue
# 更新用户余额，添加消费记录
result = ExprOrders.update_user_money(
immed_order_money, order_object)
if result.get("code") != 0:
return result
# 订单状态为已结算
order_object.status = options.order_Settlement
db_session.add(order_object)
db_session.flush()
else:
if price_type_obj:
new_orders.append(order_object)
# 更新用户过期时间
ExprOrders.expr_user_exceed_time(data.user_id, new_orders)
# 添加到数据库
for new_order in new_orders:
db_session.add(new_order)
db_session.flush()
msg = "add order successful. resource_id: <{0}>".format(
data.resource_id)
LOG.debug(msg)
db_session.commit()
return {"code": 0, "message": ""}
