From 8ca52c68220bdb3b0c9698e225ba6ec66456c6a8 Mon Sep 17 00:00:00 2001 From: Forest Belton <65484+forestbelton@users.noreply.github.com> Date: Tue, 22 Jun 2021 13:43:51 -0400 Subject: [PATCH] Remove redundant argument check --- paircd/message.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/paircd/message.py b/paircd/message.py index 118bc6c..1796d77 100644 --- a/paircd/message.py +++ b/paircd/message.py @@ -3,13 +3,6 @@ from typing import List, Optional MAX_MESSAGE_SIZE = 512 -EXPECTED_ARG_COUNT = { - "NICK": 1, - "USER": 4, - "JOIN": 1, - "PRIVMSG": 2, -} - class ParsingError(Exception): message: Optional[str] @@ -56,10 +49,5 @@ def parse_message(raw: bytes) -> Message: raise ParsingError("Message has no command") cmd = tokens[0].upper() - if cmd in EXPECTED_ARG_COUNT and EXPECTED_ARG_COUNT[cmd] != len(tokens) - 1: - raise ParsingError( - f"{cmd} had {len(tokens)-1} arguments, expected {EXPECTED_ARG_COUNT[cmd]}" - ) - tokens[-1] = tokens[-1].strip() - return Message(cmd=tokens[0], args=tokens[1:]) + return Message(cmd=cmd, args=tokens[1:])