Mise à jour de l'aide pour la fonction pattern

wgamesolv.py

@@ -3,6 +3,7 @@
 
 import json
 import argparse
+from random import shuffle
 
 
 def load_dict(dict_file):
@@ -19,13 +20,9 @@ def noDouble(mots):
     # Fonction d'élimination de mots contenant plusieurs occurences de lettres
     ndList = []
     for mot in mots:
-        llist = []
-        for lettre in mot:
-            if lettre not in llist:
-                llist.append(lettre)
-        if len(mot) == len(llist):
+        if len(list(mot)) == len(set(list(mot))):
             ndList.append(mot)
-    return(ndList)
+    return ndList
 
 
 def firstTryFilter(mots, max_mots):
@@ -35,18 +32,21 @@ def firstTryFilter(mots, max_mots):
     '''
     bestLetters = ['E', 'A', 'I', 'R', 'S', 'N', 'T', 'O', 'L', 'U']
     bestWords = []
+    shuffle(mots)
+    mots = mots[:100]
     for mot in mots:
         for bestLetter in bestLetters:
             if bestLetter in mot:
                 bestWords.append(mot)
             bestWords = list(set(bestWords))
     bestWords = noDouble(bestWords)  # Autant maximiser les chances
-    bestWords = bestWords[0:max_mots]
+    bestWords = bestWords[:max_mots]
     return(bestWords)
 
 
 def resolv_fl(liste, firstletter):
     # On ne garde que les mots qui commence par FirstLetter
+    firstletter = firstletter.upper()
     return [mot for mot in liste if mot[0] == firstletter]
 
 
@@ -57,10 +57,12 @@ def resolv_len(liste, lenght):
 
 def resolv_pattern(liste, pattern):
     # Pattern : élimination des mots ne satisfaisant pas la pattern
-    pattern_dict = {}
+    pattern_dict, not_pattern_dict = {}, {}
     for lettre in pattern:
-        if lettre != ".":
+        if lettre != "." and lettre.isupper():
             pattern_dict.update({pattern.index(lettre): lettre})
+        if lettre != "." and lettre.islower():
+            not_pattern_dict.update({pattern.index(lettre): lettre.upper()})
     MotsOK, MotsKO = [], []
     for mot in liste:
         for key in pattern_dict:
@@ -68,6 +70,10 @@ def resolv_pattern(liste, pattern):
                 MotsOK.append(mot)
             else:
                 MotsKO.append(mot)
+        for key in not_pattern_dict:
+            if mot[key] == not_pattern_dict[key]:
+                MotsKO.append(mot)
+    MotsKO = set(MotsKO)
     liste = [mot for mot in MotsOK if mot not in MotsKO]
     liste = list(set(liste))
     return liste
@@ -78,7 +84,7 @@ def resolv_kl(liste, lettres):
     Known Letters : élimination des mots ne comprenant pas
     les lettres validées (hors pattern)
     '''
-    lettres = list(set(list(lettres)))
+    lettres = list(set(list(lettres.upper())))
     MotsRestants = []
     for mot in liste:
         if len(set(lettres).intersection(mot)) == len(lettres):
@@ -91,18 +97,13 @@ def resolv_bl(liste, lettres):
     Bad Letters : élimination des mots contenant
     des lettres invalidées (hors pattern)
     '''
-    lettres = list(set(list(lettres)))
+    lettres = list(set(list(lettres.upper())))
     MotsSuppr = []
     for mot in liste:
         for lettre in lettres:
             if lettre in mot:
                 MotsSuppr.append(mot)
-    for mot in MotsSuppr:
-        try:
-            liste.pop(liste.index(mot))
-        except ValueError:
-            pass
-    return liste
+    return [mot for mot in liste if mot not in MotsSuppr]
 
 
 if __name__ == "__main__":
@@ -119,8 +120,7 @@ if __name__ == "__main__":
         '-fl',
         '--firstLetter',
         help='Première lettre',
-        default=False,
-        required=True
+        default=False
     )
     parser.add_argument(
         '-nb',
@@ -146,7 +146,7 @@ if __name__ == "__main__":
         '-p',
         '--pattern',
         default=False,
-        help='Placement, avec des "." pour les inconnues : -p A...T.I.'
+        help='Placement : bien placée en majuscule, mal placée en minuscule et non existante par un \'.\'. E.g. : Ra..e.T'
     )
     parser.add_argument(
         '-nd',
@@ -171,9 +171,16 @@ if __name__ == "__main__":
 
     liste = load_dict(args.dictionary)
     liste = resolv_len(liste, args.nbLetters)
-    liste = resolv_fl(liste, args.firstLetter)
+    if args.firstLetter:
+        liste = resolv_fl(liste, args.firstLetter)
+    if args.pattern:
+        liste = resolv_pattern(liste, args.pattern)
     if args.noDoubleLetters:
         liste = noDouble(liste)
+    if args.knownLetters:
+        liste = resolv_kl(liste, args.knownLetters)
+    if args.badLetters:
+        liste = resolv_bl(liste, args.badLetters)
     if args.firstTry:
         liste = firstTryFilter(liste, args.firstTry)
         if len(liste) == 1:
@@ -183,12 +190,6 @@ if __name__ == "__main__":
         else:
             print("Hmmm... Je crois qu'on s'est perdu...")
         exit(0)
-    if args.knownLetters:
-        liste = resolv_kl(liste, args.knownLetters)
-    if args.badLetters:
-        liste = resolv_bl(liste, args.badLetters)
-    if args.pattern:
-        liste = resolv_pattern(liste, args.pattern)
     if len(liste) > 1:
         print(liste)
         print(f"Il y a {len(liste)} mots dans la liste")