re-implement toggle()'s add=True|False

1 files changed, 4 insertions, 3 deletions

diff --git a/helpers.py b/helpers.py
index 310984c..c386faa 100644
--- a/helpers.py
+++ b/helpers.py
@@ -198,14 +198,15 @@ def save_json_file(filename, obj):
     error("Failed to write to %s: %s" % (filename, e))
 
 
-def toggle(player, ls, name = "Toggle"):
+def toggle(player, ls, name = "Toggle", add = None):
   """
   Toggles presence of a player's UUID in a list
+  If add is given, True explicitely adds it whereas False removes it
   """
   pid = uid(player)
-  if pid in ls:
+  if pid in ls or add == False:
     ls.remove(pid)
     msg(player, "&a%s turned off!" % name)
-  else:
+  elif add != False:
     ls.append(pid)
     msg(player, "&a%s turned on!" % name)
\ No newline at end of file