make more robust: trial can be shorter than the solution

1 files changed, 4 insertions, 1 deletions

diff --git a/server.py b/server.py
index c504f97..a3429f5 100755
--- a/server.py
+++ b/server.py
@@ -31,7 +31,10 @@ class TrialHandler(BaseHandler):
         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 + ''.join([' ' for i in range(len(self.trial))]), 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 + ''.join([' ' for i in range(len(self.solution))])
+        ) 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)