TwitchBot/UserData.py
2019-12-31 01:59:39 +11:00

36 lines
1001 B
Python

import datetime
MSG_TIMEOUT = 30
MSG_LIMIT = 5
class UserData:
def __init__(self, username, ismod):
self.username = username
self.mod = ismod
self.messageCount = 0
self.messageTime = 0
self.messageLimit = MSG_LIMIT
self.URLPermit = False
def setUserTimeout(self):
if self.messageTime == 0:
self.messageTime = int(datetime.datetime.now().timestamp()) #Sets messagetime to now (Unix Timestamp)
def checkUserMessageTimout(self):
if self.messageTime >= self.messageTime + MSG_TIMEOUT:
self.messageTime = 0
return False
else:
return True
def setUserMessageCount(self, amt = 1):
if self.checkUserMessageTimout():
self.messageCount += amt
def checkUserMessageCount(self):
return self.messageCount >= MSG_LIMIT
def addURLPermit(self):
self.URLPermit = True
def removeURLPermit(self):
self.URLPermit = False