make more robust: trial can be longer than the solution

1 files changed, 3 insertions, 1 deletions

diff --git a/server.py b/server.py
index 8bb520b..c504f97 100755
--- a/server.py
+++ b/server.py
@@ -24,12 +24,14 @@ class TrialHandler(BaseHandler):
     def is_solved(self):
         return self.solution == self.trial
     def letter_is_green(self, index):
+        if index >= len(self.solution):
+            return False
         return self.solution[index] == self.trial[index]
     def letter_is_green_or_yellow(self, zip_list, index):
         char = self.trial[index]
         return [ c[0] for c in zip_list ].count(char) >= [ c[1] for c in zip_list ][0:index+1].count(char)
     def get_colors(self):
-        zip_list = [ c for c in zip(self.solution, self.trial) if c[0] != c[1] ]
+        zip_list = [ c for c in zip(self.solution + ''.join([' ' for i in range(len(self.trial))]), self.trial) if c[0] != c[1] ]
         return [
             2 if self.letter_is_green(index)
             else 1 if self.letter_is_green_or_yellow(zip_list, index)