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": ""}