21 lines
528 B
Python
21 lines
528 B
Python
|
import time
|
||
|
|
from ..config import config
|
||
|
|
|
||
|
|
|
||
|
|
class DayLimit():
|
||
|
|
day: int = time.localtime(time.time()).tm_yday
|
||
|
|
data: dict = {}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def count(cls, user: str, num):
|
||
|
|
day_ = time.localtime(time.time()).tm_yday
|
||
|
|
if day_ != cls.day:
|
||
|
|
cls.day = day_
|
||
|
|
cls.data = {}
|
||
|
|
count: int = cls.data.get(user, 0)+num
|
||
|
|
if count > config.novelai_daylimit:
|
||
|
|
return -1
|
||
|
|
else:
|
||
|
|
cls.data[user] = count
|
||
|
|
return config.novelai_daylimit-count
|