clear selection at end of game

1 files changed, 12 insertions, 5 deletions

diff --git a/window.py b/window.py
index e4f1728..2b6ce14 100644
--- a/window.py
+++ b/window.py
@@ -176,11 +176,13 @@ def _button_release_cb(win, event, tw):
            # test to see if the game is over
            if tw.deck.count-tw.deck.index == 0:
                if find_a_match(tw) is False:
+                   tw.activity.deck_label.set_text("")
                    tw.activity.status_label.set_text(_("Game over") + \
                                                      " (" + \
                                                      str(int(tw.total_time)) + \
                                                      " " + _("seconds") + ")")
                    gobject.source_remove(tw.timeout_id)
+                   unselect(tw)
                    return True
            tw.matches += 1
            tw.activity.status_label.set_text(_("Match"))
@@ -190,20 +192,25 @@ def _button_release_cb(win, event, tw):
            else:
                tw.activity.match_label.set_text(
                   _("%d matches") % (tw.matches))
-
            # reset the timer
            tw.start_time = gobject.get_current_time()
            tw.timeout_id = None
            _counter(tw)
-
        else:
            tw.activity.status_label.set_text(_("No match"))
-       tw.clicked = [None, None, None]
-       for a in tw.selected:
-           a.hide_card()
+       unselect(tw)
    return True
 
 #
+# unselect the cards
+#
+def unselect(tw):
+    tw.clicked = [None, None, None]
+    for a in tw.selected:
+        a.hide_card()
+
+
+#
 # Keypress
 #
 def _keypress_cb(area, event, tw):