diff --git a/output/IOutput.py b/output/IOutput.py index 7006c06..84b2601 100644 --- a/output/IOutput.py +++ b/output/IOutput.py @@ -1,6 +1,7 @@ from messages.base import Base from messages.chat import ChatMessage from messages.control import ControlMessage +from messages.fansclub import FansclubMessage from messages.gift import GiftMessage from messages.like import LikeMessage from messages.member import MemberMessage @@ -9,6 +10,9 @@ from messages.social import SocialMessage class IOutput(): + def __del__(self): + self.terminate() + def output(self, message_type: str, message_obj: Base): ... @@ -33,8 +37,14 @@ class IOutput(): def control_output(self, message: ControlMessage): ... + def fansclub_output(self, message: FansclubMessage): + ... + def other_output(self, message_type: str, message_raw: bytes): ... def error_output(self, message_type: str, message_raw: bytes, exception: Exception): + ... + + def terminate(self): ... \ No newline at end of file diff --git a/output/print.py b/output/print.py index 3669a11..192a8a8 100644 --- a/output/print.py +++ b/output/print.py @@ -31,3 +31,9 @@ class Print(IOutput): def userseq_output(self, msg): print(f"\n{YELLOW}[+] {msg} {RESET}") + + def control_output(self, msg): + print(f"\n{CYAN}[+] {msg} {RESET}") + + def fansclub_output(self, msg): + print(f"\n{GREEN}[+] {msg} {RESET}")