24 lines
482 B
Python
24 lines
482 B
Python
import enum
|
|
from typing import Set
|
|
|
|
|
|
class ShapeType(enum.Enum):
|
|
MARKER = "marker"
|
|
OVAL = "oval"
|
|
RECTANGLE = "rectangle"
|
|
TEXT = "text"
|
|
|
|
|
|
SHAPES: Set[ShapeType] = {ShapeType.OVAL, ShapeType.RECTANGLE}
|
|
|
|
|
|
def is_draw_shape(shape_type: ShapeType) -> bool:
|
|
return shape_type in SHAPES
|
|
|
|
|
|
def is_shape_text(shape_type: ShapeType) -> bool:
|
|
return shape_type == ShapeType.TEXT
|
|
|
|
|
|
def is_marker(shape_type: ShapeType) -> bool:
|
|
return shape_type == ShapeType.MARKER
|