From f76851ba23cdd341f0784ac860e169752d641d2a Mon Sep 17 00:00:00 2001
From: Guillaume Dott <guillaume+github@dott.fr>
Date: Mon, 2 Oct 2017 09:48:20 +0200
Subject: [PATCH] Hide button when disabled

---
 photoboite.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/photoboite.py b/photoboite.py
index d1af037..15c64ad 100755
--- a/photoboite.py
+++ b/photoboite.py
@@ -16,6 +16,7 @@ class Button:
         self.rect.topleft = topleft
 
         self.pressed = False
+        self.enabled = True
 
     def draw_on(self, screen):
         if self.pressed:
@@ -47,7 +48,8 @@ class CaptureButton(Button):
         self.count = 0
 
     def draw_on(self, screen):
-        super(CaptureButton, self).draw_on(screen)
+        if self.enabled:
+            super(CaptureButton, self).draw_on(screen)
         if self.capture_mode:
             if self.elapsed >= CaptureButton.countdown:
                 text = "CLIC!"
@@ -61,6 +63,7 @@ class CaptureButton(Button):
     def run(self):
         pygame.event.set_blocked((pygame.MOUSEBUTTONDOWN, pygame.MOUSEBUTTONUP))
         self.capture_mode = True
+        self.enabled = False
 
         capture = Capture(self.count)
         self.count += 1
@@ -81,6 +84,7 @@ class CaptureButton(Button):
             pygame.time.wait(1000)
         pygame.event.set_allowed((pygame.MOUSEBUTTONDOWN, pygame.MOUSEBUTTONUP))
         self.capture_mode = False
+        self.enabled = True
 
 
 class Capture: