Add minute of day / minute number.

[kiosk.git] / grab_bag.py

diff --git a/grab_bag.py b/grab_bag.py
index a427256ca0f1b1530a0e757391dc027b1391099a..798ebcfb154b58c3fdbf9588bcc01bc4b69d140a 100644 (file)
--- a/grab_bag.py
+++ b/grab_bag.py
@@ -1,28 +1,31 @@
+#!/usr/bin/env python3
+
 import random
+from typing import Iterable, List, Optional, Set
 
 
 class grab_bag(object):
-    def __init__(self):
-        self.contents = set()
+    def __init__(self) -> None:
+        self.contents: Set[str] = set()
 
-    def clear(self):
+    def clear(self) -> None:
         self.contents.clear()
 
-    def add(self, item):
+    def add(self, item: str) -> None:
         if item not in self.contents:
             self.contents.add(item)
 
-    def add_all(self, collection):
+    def add_all(self, collection: Iterable[str]) -> None:
         for x in collection:
             self.add(x)
 
-    def subset(self, count):
+    def subset(self, count: int) -> Optional[List[str]]:
         if len(self.contents) < count:
             return None
         subset = random.sample(self.contents, count)
         return subset
 
-    def size(self):
+    def size(self) -> int:
         return len(self.contents)