Update ensiee.py

ensiee.py

@@ -3,13 +3,17 @@
 
 from random import shuffle, randrange
 
-# renommons l'école. 
+def double(chaine) :
+    res = ''
+    for f in chaine :
+        res = res + f + f
+    return res
 
-def ensiee() : 
-    base = "E"
-    length = randrange(4,8)
-    rest = list('NNSSIIIIEE')
-    shuffle(rest)
+# renommons l'école. 
+def ensiee(nom = "ENSIIE"):
+    base = nom[0]
+    rest = list(double(nom[1:]))
+    length = randrange(len(nom) - 2, len(nom) + 3)
      res = ''
     for f in rest : 
         if randrange(0,2) == 1 :  
@@ -17,7 +21,6 @@ def ensiee() :
         else :
             res = res + f 
     res = base + ''.join(res[:length])
-    if res == "ENSIIE" :
+    if res == nom :
         res = res + "(Waw, une chance sur 1 million !)"
     return res
\ No newline at end of file
-    
\ No newline at end of file