|
|
@@ -22,6 +22,7 @@ from services.utils import current_date
|
|
|
LIMITER_DATE = "lm_date" # 当前日期
|
|
|
LIMITER_FORECAST_WARNING = "lm_forecast_warning" # 上限预告
|
|
|
LIMITER_EXCEEDED_WARNING = "lm_exceeded_warning" # 超限告警
|
|
|
+PENV = setting.PLATFORM_ENVIRONMENT # 当前运行环境
|
|
|
|
|
|
redis_db = redis.RedisDB() # redis实例
|
|
|
|
|
|
@@ -62,7 +63,8 @@ class PayCaptchaLimiter(Limiter):
|
|
|
usages_count = self._storage.get(item.key_for(*identifiers))
|
|
|
val = '{:.2f}'.format(usages_count / item.amount * 100)
|
|
|
if float(val) > 80 and int(redis_db.hget(table, LIMITER_FORECAST_WARNING)):
|
|
|
- send_msg("超级鹰", item.amount, usages_count, f"使用次数已超过{val}%")
|
|
|
+ tips = "".join([f"[{PENV}]", f"使用次数已超过{val}%"])
|
|
|
+ send_msg("超级鹰", item.amount, usages_count, tips)
|
|
|
redis_db.hset(table, LIMITER_FORECAST_WARNING, 0)
|
|
|
|
|
|
def send_finished(self, limit):
|
|
|
@@ -70,8 +72,8 @@ class PayCaptchaLimiter(Limiter):
|
|
|
amount = limit.limit.amount
|
|
|
self.logger.debug(f"今日接口调用次数 {amount} 已达上限！")
|
|
|
if int(redis_db.hget(setting.LIMITER_REDIS_KEY, LIMITER_EXCEEDED_WARNING)):
|
|
|
- msg = "今日接口调用次数已达上限！\n 继续使用请点击"
|
|
|
- send_msg("超级鹰", max_limit, amount, msg, allow_reset=True)
|
|
|
+ tips = "".join([f"[{PENV}]", "今日接口调用次数已达上限！\n 继续使用请点击"])
|
|
|
+ send_msg("超级鹰", max_limit, amount, tips, allow_reset=True)
|
|
|
# 关闭超限告警
|
|
|
redis_db.hset(setting.LIMITER_REDIS_KEY, LIMITER_EXCEEDED_WARNING, 0)
|
|
|
|
dzr