You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.9 KiB
108 lines
2.9 KiB
"""
|
|
Store information about a discord user, for the purposes of enabling conversations. We store a message
|
|
history, message count, and the id of the user in order to track them.
|
|
"""
|
|
|
|
|
|
class RedoUser:
|
|
def __init__(self, prompt, message, ctx, response):
|
|
self.prompt = prompt
|
|
self.message = message
|
|
self.ctx = ctx
|
|
self.response = response
|
|
self.interactions = []
|
|
|
|
def add_interaction(self, interaction):
|
|
self.interactions.append(interaction)
|
|
|
|
def in_interaction(self, interaction):
|
|
return interaction in self.interactions
|
|
|
|
# Represented by user_id
|
|
def __hash__(self):
|
|
return hash(self.message.author.id)
|
|
|
|
def __eq__(self, other):
|
|
return self.message.author.id == other.message.author.id
|
|
|
|
# repr
|
|
def __repr__(self):
|
|
return f"RedoUser({self.message.author.id})"
|
|
|
|
|
|
class User:
|
|
def __init__(self, id):
|
|
self.id = id
|
|
self.history = []
|
|
self.count = 0
|
|
|
|
# These user objects should be accessible by ID, for example if we had a bunch of user
|
|
# objects in a list, and we did `if 1203910293001 in user_list`, it would return True
|
|
# if the user with that ID was in the list
|
|
def __eq__(self, other):
|
|
return self.id == other.id
|
|
|
|
def __hash__(self):
|
|
return hash(self.id)
|
|
|
|
def __repr__(self):
|
|
return f"User(id={self.id}, history={self.history})"
|
|
|
|
def __str__(self):
|
|
return self.__repr__()
|
|
|
|
|
|
class Thread:
|
|
def __init__(self, id):
|
|
self.id = id
|
|
self.history = []
|
|
self.count = 0
|
|
|
|
# These user objects should be accessible by ID, for example if we had a bunch of user
|
|
# objects in a list, and we did `if 1203910293001 in user_list`, it would return True
|
|
# if the user with that ID was in the list
|
|
def __eq__(self, other):
|
|
return self.id == other.id
|
|
|
|
def __hash__(self):
|
|
return hash(self.id)
|
|
|
|
def __repr__(self):
|
|
return f"Thread(id={self.id}, history={self.history})"
|
|
|
|
def __str__(self):
|
|
return self.__repr__()
|
|
|
|
|
|
class EmbeddedConversationItem:
|
|
def __init__(self, text, timestamp):
|
|
self.text = text
|
|
self.timestamp = int(timestamp)
|
|
|
|
def __repr__(self):
|
|
return self.text
|
|
|
|
def __str__(self):
|
|
return self.__repr__()
|
|
|
|
def __eq__(self, other):
|
|
return self.text == other.text and self.timestamp == other.timestamp
|
|
|
|
def __hash__(self):
|
|
return hash(self.text) + hash(self.timestamp)
|
|
|
|
def __lt__(self, other):
|
|
return self.timestamp < other.timestamp
|
|
|
|
def __gt__(self, other):
|
|
return self.timestamp > other.timestamp
|
|
|
|
def __le__(self, other):
|
|
return self.timestamp <= other.timestamp
|
|
|
|
def __ge__(self, other):
|
|
return self.timestamp >= other.timestamp
|
|
|
|
def __ne__(self, other):
|
|
return not self.__eq__(other)
|