from enum import Enum