fomys/FoBot
1
0
Fork 0
Code Issues Pull Requests Activity

Add fpi command to search in pi with regex

Browse Source
This commit is contained in:
louis chauvet 2018-09-01 17:39:14 +02:00
parent 22c7a61342
commit 3c325ca9e0
2 changed files with 38 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
modules/pi.py
View File

@ -1,4 +1,6 @@
import os
import re
import traductions as tr
@ -15,7 +17,7 @@ class MainClass:
try:
start = int(args[0])
except ValueError:
await msg.channel.send(tr.tr[self.guild.config["lang"]]["Errors"]["TooBigNumberPiError"])
await msg.channel.send(tr.tr[self.guild.config["lang"]]["errors"]["TooBigNumberPiError"])
with open(self.pi_file) as pi_file:
pi_file.read(start)
txt = pi_file.read(2000)
@ -23,11 +25,42 @@ class MainClass:
await msg.channel.send(txt)
async def pi_search(self, msg, command, args):
pass
if len(args) == 0:
await msg.channel.send(tr.tr[self.guild.config["lang"]]["errors"]["NotEnoughParamError"])
return
try:
to_search = re.compile(args[0])
if "*" in args[0] or "+" in args[0]:
await msg.channel.send(tr.tr[self.guild.config["lang"]]["errors"]["ForbiddenRegexError"])
return
elif len(args[0]) > 50:
await msg.channel.send(tr.tr[self.guild.config["lang"]]["errors"]["RegexTooLongError"])
except re.error:
await msg.channel.send(tr.tr[self.guild.config["lang"]]["errors"]["RegexError"])
return
with open(self.pi_file) as pi_file:
pi = pi_file.readline()
results = to_search.finditer(pi)
texts = []
for result in results:
texts.append(
"Une occurence de votre recherche a été trouvée à la {debut}ème place: `{before}`{find}`{after}`".format(
debut=result.start(0),
before=pi[result.start(0) - 10:result.start(0)],
find=pi[result.start(0):result.end(0)],
after=pi[result.end(0):result.end(0) + 10]
))
if texts:
for text in texts[:10]:
await msg.channel.send(text)
else:
await msg.channel.send("Pas de résultats")
async def on_message(self, msg):
if msg.content.startswith(self.guild.config["prefix"]):
command, *args = msg.content.lstrip(self.guild.config["prefix"]).split(" ")
if command == "pi":
await self.pi(msg, command, args)
elif command == "fpi":
await self.pi_search(msg, command, args)
return

3
traductions.py
View File

@ -147,6 +147,9 @@ tr = {
"du bot.",
"CommandNotFoundError": "La commande {command} n'existe pas.",
"TooBigNumberPiError": "Vous devez spécifier un nombre inferieur a 10000.",
"RegexError": "La regex que vous avez utilisé n'est pas valide.",
"ForbiddenRegexError": "Vous n'avez pas le droit d'utiliser les caractères `*` et `+` dans une regex.",
"RegexTooLongError": "La regex ne doit pas faire plus e 50 caractères",
},
},
}