python实现发红包redenv_python实现红包裂变算法
# -*- coding: utf-8 -*-
from decimal import Decimal, InvalidOperation
import random
def money_val(min, max):
return min if min > max else Decimal(str(random.randint(min, max)))
def money_random(total, num, min=0.01):
"""
:param total=10; # 红包总额 10 元
:param num=8; # 分成 8 个红包，支持 8 人随机领取
:param min=0.01; # 每个人最少能收到 0.01 元
"""
money_list = []
try:
total = Decimal(str(total))
except InvalidOperation as e:
return money_list, e.message
try:
if isinstance(num, float) and int(num) != num:
raise ValueError(u'Invalid value for Num: \'{0}\''.format(num))
num = Decimal(str(int(num)))
except ValueError as e:
return money_list, e.message
try:
min = Decimal(str(min))
except InvalidOperation as e:
return money_list, e.message
if total < num * min:
return money_list, u'Invalid value for Total-{0}, Num-{1}, Min-{2}'.format(total, num, min)
for i in xrange(1, num):
safe_total = (total - (num - i) * min) / (num - i) # 随机安全上限
money = money_val(min * 100, int(safe_total * 100)) / 100
total -= money
money_list.append(money)
money_list.append(total)
random.shuffle(money_list) # 随机打乱
return money_list, u'Success'
if __name__ == '__main__':
print money_random(1, 10)
print money_random(0.1, 10)
print money_random(0.11, 10)
print money_random(0.12, 10)
4.2、函数使用4.2.1、使用 money_random 预先将红包裂变，存放在 memcache 或者 redis 中
total —— 红包总额
num —— 裂变个数
min —— 非必须，红包最小面额，默认 0.01
4.2.2、当用户抢红包的时候，直接 pop 出来一个，直到将所有红包 pop 完
4.3、库使用说明因为涉及到浮点运算，所以使用了 Decimal
5、源码示例5.1、运行