implement generate login interface and login

src/user_interface.py

@@ -1,4 +1,5 @@
 from pydantic import BaseModel
+from secrets import choice
 from src.utils import list_to_matrix, secure_fisher_yates_shuffle, matrix_to_list
 
 
@@ -18,7 +19,7 @@ class UserInterface(BaseModel):
     def disperse_interface(self):
         user_interface_matrix = list_to_matrix(self.interface_index, self.numb_sets)
         shuffled_keys = secure_fisher_yates_shuffle(user_interface_matrix)
-        dispersed_interface = self._random_attribute_rotation(shuffled_keys)
+        dispersed_interface = self._random_attribute_rotation(shuffled_keys, list(range(self.numb_sets)))
         self.interface_index = matrix_to_list(dispersed_interface)
 
     @staticmethod
@@ -26,13 +27,28 @@ class UserInterface(BaseModel):
         return [list(row) for row in zip(*interface)]
 
     def shuffle_interface(self):
-        pass
+        """just like dispersion but only half the sets are rotated"""
+        numb_of_selected_sets = self.numb_sets // 2
+        # randomly shuffle half the sets. if numb_sets is odd, randomly add one 50% of the time
+        numb_of_selected_sets += choice([0, 1]) if (self.numb_sets & 1) == 1 else 0
+        selected_sets = secure_fisher_yates_shuffle(list(range(self.numb_sets)))[:numb_of_selected_sets]
+        user_interface_matrix = list_to_matrix(self.interface_index, self.numb_sets)
+        shuffled_keys = secure_fisher_yates_shuffle(user_interface_matrix)
+        interface_by_sets = []
+        for idx, attrs in enumerate(self.matrix_transpose(shuffled_keys)):
+            if idx in selected_sets:
+                interface_by_sets.append(secure_fisher_yates_shuffle(attrs))
+            else:
+                interface_by_sets.append(attrs)
+        self.interface_index = matrix_to_list(self.matrix_transpose(interface_by_sets))
 
-    def _random_attribute_rotation(self, user_interface: list[list[int]]) -> list[list[int]]:
+    def _random_attribute_rotation(self, user_interface: list[list[int]], selected_sets: list[int]) -> list[list[int]]:
         attr_rotation = secure_fisher_yates_shuffle(list(range(self.numb_keys)))[:self.numb_sets]
         transposed_user_interface = self.matrix_transpose(user_interface)
         assert (len(attr_rotation) == len(transposed_user_interface))
         for idx, attr_set in enumerate(transposed_user_interface):
+            if idx not in selected_sets:
+                continue
             rotation = attr_rotation[idx]
             transposed_user_interface[idx] = attr_set[rotation:] + attr_set[:rotation]
         return self.matrix_transpose(transposed_user_interface)
@@ -46,3 +62,8 @@ class UserInterface(BaseModel):
                 graph[attr].remove(attr)
 
         return graph
+
+    def get_key_attr_idxs(self, key_numb: int) -> list[int]:
+        assert (0 <= key_numb < self.numb_keys)
+        keypad_attr_idx = list_to_matrix(self.interface_index, self.numb_sets)
+        return keypad_attr_idx[key_numb]