X-Git-Url: https://git.ctpug.org.za/?p=tabakrolletjie.git;a=blobdiff_plain;f=tabakrolletjie%2Fscenes%2Fnight.py;h=0c7688924e7ebd0d2e8e6fb245cab9fe1afabd8c;hp=803827665d534eb2627aef88dc47bb4d42a6d1a6;hb=c8191a153de7d9354c781b915e6aa3c41c5fc94b;hpb=ea91002db68118e01b0d09527783a39b4112ac3e diff --git a/tabakrolletjie/scenes/night.py b/tabakrolletjie/scenes/night.py index 8038276..0c76889 100644 --- a/tabakrolletjie/scenes/night.py +++ b/tabakrolletjie/scenes/night.py @@ -55,6 +55,11 @@ class NightScene(BaseScene): '32', 'exit.png', name='exit', pos=(SCREEN_SIZE[0] - 50, y))) return tools + def add_day_button(self): + y = SCREEN_SIZE[1] - 40 + self._tools.append(ImageButton( + '32', 'day.png', name='day', pos=(SCREEN_SIZE[0] - 200, y))) + @property def turnip_count(self): return len(self._turnips) @@ -116,6 +121,8 @@ class NightScene(BaseScene): elif tool.name == 'exit': from .menu import MenuScene SceneChangeEvent.post(scene=MenuScene()) + elif tool.name == 'day': + self._to_day() def toggle_pause(self): self._paused = not self._paused @@ -167,6 +174,9 @@ class NightScene(BaseScene): self._end_night() if not self._mould.alive(): self._end_night() + if not self.turnip_count: + self.add_day_button() + def exit(self, gamestate): turnip_data = [turnip.serialize() for turnip in self._turnips]