From 33b66af7812c20a596e0943077473a4bee6b9e9a Mon Sep 17 00:00:00 2001 From: miloszowi Date: Tue, 28 Sep 2021 23:14:14 +0200 Subject: [PATCH] added more validation on group name input --- src/handler/vo/updateData.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/src/handler/vo/updateData.py b/src/handler/vo/updateData.py index 8739480..2af6f48 100644 --- a/src/handler/vo/updateData.py +++ b/src/handler/vo/updateData.py @@ -22,10 +22,17 @@ class UpdateData(): chat_id = str(update.effective_chat.id) if context.args and context.args[0]: - if not context.args[0].isalpha() or context.args[0] == Group.default_name: - raise InvalidArgumentException(re.escape(f'Group name must contain only letters and can not be `{Group.default_name}`.')) - else: - chat_id += f'~{context.args[0]}'.lower() + group_name = str(context.args[0]) + if not context.args[0].isalpha(): + raise InvalidArgumentException(re.escape('Group name must contain only letters.')) + + if context.args[0] == Group.default_name: + raise InvalidArgumentException(re.escape(f'Group can not be `{Group.default_name}`.')) + + if len(context.args[0]) > 20: + raise InvalidArgumentException(re.escape(f'Group name length can not be greater than 20.')) + + chat_id += f'~{context.args[0]}'.lower() user_id = str(update.effective_user.id)