summaryrefslogtreecommitdiff
path: root/ATRI/utils/limit.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/utils/limit.py')
-rw-r--r--ATRI/utils/limit.py35
1 files changed, 10 insertions, 25 deletions
diff --git a/ATRI/utils/limit.py b/ATRI/utils/limit.py
index c6fae19..7442dfb 100644
--- a/ATRI/utils/limit.py
+++ b/ATRI/utils/limit.py
@@ -1,7 +1,6 @@
import datetime
from random import choice
-from ATRI.config import Config
from ATRI.service import Service as sv
from .list import count_list, del_list_aim
from .apscheduler import scheduler, DateTrigger
@@ -9,15 +8,6 @@ from .apscheduler import scheduler, DateTrigger
exciting_user_temp = []
exciting_user = []
-exciting_repo = [
- "歇歇8,。咱8能再快了",
- "太快惹,太快惹嗯",
- "你吼辣么快干什么!",
- "其实吧我觉得你这速度去d个vup挺适合",
- "我不接受!你太快了",
- "我有点担心,因为你太快了",
- "请稍等!您冲得太快了!",
-]
def del_list(user: str) -> None:
@@ -25,33 +15,28 @@ def del_list(user: str) -> None:
exciting_user = del_list_aim(exciting_user, user)
-async def is_too_exciting(user: int, group: int, times: int, repo: bool) -> bool:
+def is_too_exciting(user: int,
+ times: int,
+ seconds: float = 0,
+ hours: float = 0,
+ days: float = 0) -> bool:
global exciting_user
-
+
if user in exciting_user:
- if repo:
- await sv.NetworkPost.send_msg(
- user_id=user, group_id=group, message=choice(exciting_repo)
- )
return False
else:
if count_list(exciting_user_temp, user) == times:
delta = datetime.timedelta(
- seconds=Config.BotSelfConfig.session_exciting_time
- )
- trigger = DateTrigger(run_date=datetime.datetime.now() + delta)
-
+ seconds=seconds, hours=hours, days=days)
+ trigger = DateTrigger(
+ run_date=datetime.datetime.now() + delta)
+
scheduler.add_job(
func=del_list,
trigger=trigger,
args=(user,),
misfire_grace_time=1,
)
-
- if repo:
- await sv.NetworkPost.send_msg(
- user_id=user, group_id=group, message=choice(exciting_repo)
- )
return False
else:
exciting_user_temp.append(user)