diff options
Diffstat (limited to 'src/sliceitoff/mainmenu')
| -rw-r--r-- | src/sliceitoff/mainmenu/__init__.py | 2 | ||||
| -rw-r--r-- | src/sliceitoff/mainmenu/mainmenu.py | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/sliceitoff/mainmenu/__init__.py b/src/sliceitoff/mainmenu/__init__.py index 3dd1fe3..9259d1f 100644 --- a/src/sliceitoff/mainmenu/__init__.py +++ b/src/sliceitoff/mainmenu/__init__.py @@ -1,2 +1,2 @@ """ mainmenu - Menu where user can new game, quit etc """ -from .mainmenu import Mainmenu +from .mainmenu import Mainmenu, MenuItems diff --git a/src/sliceitoff/mainmenu/mainmenu.py b/src/sliceitoff/mainmenu/mainmenu.py index c907443..e0466a5 100644 --- a/src/sliceitoff/mainmenu/mainmenu.py +++ b/src/sliceitoff/mainmenu/mainmenu.py @@ -1,12 +1,12 @@ """ mainmenu.mainmenu - Let's user choose """ -from enum import Enum +from enum import IntEnum import pygame from screens import mainmenu_screen from game.anykey import anykey -class MenuItems(Enum): +class MenuItems(IntEnum): """ Items in the menu. Should match mainmenu_screen """ NEWGAME = 0 HISCORES = 1 |