tfa/Chef_mu
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: tfa:94a51fcaf8fe46a9fb5a346fee68e37bea0a436c
Branches Tags
tfa:master
...
compare: tfa:3718a745b24e1bbd5f0da3db9d48aa0aca5b5ca3
Branches Tags
tfa:master

2 Commits
94a51fcaf8 ... 3718a745b2

Author SHA1 Message Date
tfa
3718a745b2 Ajout liste exclusion + logs 2022-09-07 15:17:35 +02:00
tfa
df382d7f53 Fix message d'erreur 2022-09-07 14:40:48 +02:00
1 changed files with 10 additions and 3 deletions
Expand all files Collapse all files

13
chef_mou.py
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,10 +94,12 @@ 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')
idx = 9 idx = 25
if chunks[1].lower() == "souba" or chunks[1].lower() == "soubassophone" or chunks[1].lower() == "sb": if chunks[1].lower() == "souba" or chunks[1].lower() == "soubassophone" or chunks[1].lower() == "sb":
idx = 0 idx = 0
elif chunks[1].lower() == "basse" or chunks[1].lower() == "bs" : elif chunks[1].lower() == "basse" or chunks[1].lower() == "bs" :
@@ -112,7 +119,7 @@ class ChefMou(Plugin):
try : try :
await evt.respond("["+originalList[idxTitre[0]]+ " " + chunks[1]+"]"+"(<https://partoches.pustule.org"+t[idx].get("href")+">)") await evt.respond("["+originalList[idxTitre[0]]+ " " + chunks[1]+"]"+"(<https://partoches.pustule.org"+t[idx].get("href")+">)")
except IndexError: except IndexError:
await evt.respond("Le morceau "+ strTitre[0] + " existe mais pas ta partoche") await evt.respond("Le morceau "+ originalList[idxTitre[0]] + " existe mais pas ta partoche")
else: else:
await evt.respond("Morceau non trouvé") await evt.respond("Morceau non trouvé")