Ajout liste exclusion + logs

This commit is contained in:
tfa
2022-09-07 15:17:35 +02:00
parent df382d7f53
commit 3718a745b2

View File

@@ -11,6 +11,7 @@ from difflib import SequenceMatcher
from heapq import nlargest as _nlargest from heapq import nlargest as _nlargest
class ChefMou(Plugin): class ChefMou(Plugin):
listRemove = ["", "Souba", "Basse", "Trombone", "Trompette", "Sax Mib", "Clarinette", "Midi", "Original", "Enregistrement", "Paroles", "Commentaire"]
async def start(self) -> None: async def start(self) -> None:
self.on_external_config_update() self.on_external_config_update()
@@ -56,6 +57,7 @@ class ChefMou(Plugin):
""" """
) )
else: else:
self.log.info("Commande : " + pattern)
if pattern == "morceau": if pattern == "morceau":
soup = await self.get_site_content() soup = await self.get_site_content()
s = soup.findAll('th', background='/fond3.gif') s = soup.findAll('th', background='/fond3.gif')
@@ -74,11 +76,14 @@ class ChefMou(Plugin):
return return
soup = await self.get_site_content() soup = await self.get_site_content()
# Récuperer liste morceaux # Récuperer liste morceaux
s = soup.select('th', background='/fond3.gif') #s = soup.select('th', background='/fond3.gif')
s = soup.select('th')
# Object bts en str # Object bts en str
parsedList=[] parsedList=[]
originalList=[] originalList=[]
for ts in s: for ts in s:
if ts.text in self.listRemove :
continue
fullTitre = ts.text fullTitre = ts.text
#Supression des () #Supression des ()
parsedList.append(re.sub("[\(\[].*?[\)\]]", "", fullTitre)) parsedList.append(re.sub("[\(\[].*?[\)\]]", "", fullTitre))
@@ -89,6 +94,8 @@ class ChefMou(Plugin):
idxTitre = self.get_close_matches_indexes(chunks[2], parsedList, cutoff=0.4) idxTitre = self.get_close_matches_indexes(chunks[2], parsedList, cutoff=0.4)
# Si trouvé le chercher sinon erreur # Si trouvé le chercher sinon erreur
if len(idxTitre) > 0 : if len(idxTitre) > 0 :
for el in idxTitre :
self.log.info(parsedList[el] + ': ' + str(SequenceMatcher(None, chunks[2], parsedList[el]).ratio()))
#Et on recherche dans la liste originale avec le titre original #Et on recherche dans la liste originale avec le titre original
s = soup.find('th', string=originalList[idxTitre[0]]) s = soup.find('th', string=originalList[idxTitre[0]])
t = s.parent.findAll('a') t = s.parent.findAll('a')