Update main.py

main.py

@@ -27,13 +27,34 @@ def mul_mod(a,b): # (a*b) mod 17 // p.3
     return res
 
 
-def xor(a,b): 
-# ne transforme pas les int en str, 
-# accepte qu'en binaire str et de même taille
+def xor(a,b):#a XOR b = str(binaire)
+# CODE SPAGHETTI | À OPTIMISER SI POSSIBLE
+# accepte des nombres (décimaux & binaires) qui n'ont pas la même taille (100 et 7 par exemple)
     res = ''
-    size = len(a)
+
+    if(type(a) != str): #transforme les int en str (5 = 101 ; pas 0b101)
+        a = format(a, "b")
+
+    if(type(b) != str): #transforme les int en str (7 = 111 ; pas 0b101)
+        b = format(b, "b")
+
+    if(len(a) > len(b)):#si len(a) > len(b), met b --> a & a --> b pour étape suivante où len(a) pourra être comme len(b)
+        tmp = a
+        a = b
+        b = tmp
+    
+    if(len(a) < len(b)): #partie ou on met les 2 strings à la même taille.
+        l_a = len(a) #int
+        l_b = len(b) #int
+        list = []
+        list = ((l_b - l_a) * '0', a) #à gauche, le nombre de '0' nécessaire pour avoir la même taille que 'b'
+        # & à droite 'a' qui est le plus petit str
+        a = ''.join(list) #fusion des '0' et du str(a)
+
     i = 0
-    while i < size:
+    size = len(a) #choisis 'a', mais on aurait pu mettre 'b'; same
+
+    while i < size: #fonction XOR
         if a[i] == b[i]:
             res += '0'
         else: