diff --git a/src/fiber_package/signup/enums.py b/src/fiber_package/signup/enums.py index 6dff12b..d62f853 100644 --- a/src/fiber_package/signup/enums.py +++ b/src/fiber_package/signup/enums.py @@ -1,5 +1,5 @@ from enum import StrEnum -from typing import Literal +from typing import Literal, get_args ''' This category is used to differentiate types of form elements @@ -17,3 +17,6 @@ FormTextCategory = Literal[ FormElementCategory.text_piece, FormElementCategory.title_text ] + +def is_question_category(category: FormElementCategory) -> bool: + return category in get_args(FormQuestionCategory)