diff --git a/libtoorchat.py b/libtoorchat.py index a2b9e58..792aee8 100644 --- a/libtoorchat.py +++ b/libtoorchat.py @@ -36,4 +36,32 @@ class ToorChatMessage(): return self.start + self.xid + self.user + self.data + self.end def to_string(self): - return self.__str__() \ No newline at end of file + return self.__str__() + +class ToorChatParsePacket(): + ''' This is a class for parsing the data of the packet ''' + + def __init__(self): + self.start = ToorChatProtocol.get_packet_start + self.end = ToorChatProtocol.get_packet_end + + def check_packet_start(self, message): + if message.start == self.start + return True + else + return False + + def check_packet_end(self, message): + if message.end == self.end + return True + else + return False + + def get_xid(cls): + return self.xid + + def get_user(cls): + return self.user + + def get_message(cls): + return self.message