68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
import random
|
||
from collections.abc import Sequence
|
||
|
||
from colorama import Fore, init
|
||
|
||
from src.common.logger import get_logger
|
||
|
||
egg = get_logger("小彩蛋")
|
||
|
||
|
||
def weighted_choice(data: Sequence[str], weights: Sequence[float] | None = None) -> str:
|
||
"""
|
||
从 data 中按权重随机返回一条。
|
||
若 weights 为 None，则所有元素权重默认为 1。
|
||
"""
|
||
if weights is None:
|
||
weights = [1.0] * len(data)
|
||
|
||
if len(data) != len(weights):
|
||
raise ValueError("data 和 weights 长度必须相等")
|
||
|
||
# 计算累计权重区间
|
||
total = 0.0
|
||
acc = []
|
||
for w in weights:
|
||
total += w
|
||
acc.append(total)
|
||
|
||
if total <= 0:
|
||
raise ValueError("总权重必须大于 0")
|
||
|
||
# 随机落点
|
||
r = random.random() * total
|
||
# 二分查找落点所在的区间
|
||
left, right = 0, len(acc) - 1
|
||
while left < right:
|
||
mid = (left + right) // 2
|
||
if r < acc[mid]:
|
||
right = mid
|
||
else:
|
||
left = mid + 1
|
||
return data[left]
|
||
|
||
|
||
class BaseMain:
|
||
"""基础主程序类"""
|
||
|
||
def __init__(self):
|
||
"""初始化基础主程序"""
|
||
self.easter_egg()
|
||
|
||
@staticmethod
|
||
def easter_egg():
|
||
# 彩蛋
|
||
init()
|
||
items = [
|
||
"多年以后，面对AI行刑队，张三将会回想起他2023年在会议上讨论人工智能的那个下午",
|
||
"你知道吗？诺狐的耳朵很软，很好rua",
|
||
"喵喵~你的麦麦被猫娘入侵了喵~",
|
||
]
|
||
w = [10, 5, 2]
|
||
text = weighted_choice(items, w)
|
||
rainbow_colors = [Fore.RED, Fore.YELLOW, Fore.GREEN, Fore.CYAN, Fore.BLUE, Fore.MAGENTA]
|
||
rainbow_text = ""
|
||
for i, char in enumerate(text):
|
||
rainbow_text += rainbow_colors[i % len(rainbow_colors)] + char
|
||
egg.info(rainbow_text)
|