FoBot/modules/pi.py

import os
import re

import traductions as tr


class MainClass:
    name = "pi"

    def __init__(self, guild):
        self.guild = guild
        self.pi_file = "D:\\Users\\louis chauvet\\Documents\\GitHub\\foBot\\modules\\pi\\pi1.txt"

    async def pi(self, msg, command, args):
        start = 0
        if len(args) == 1:
            try:
                start = int(args[0])
            except ValueError:
                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)
            await msg.channel.send(tr.tr[self.guild.config["lang"]]["modules"]["pi"]["pi"].format(debut=start))
            await msg.channel.send(txt)

    async def pi_search(self, msg, command, args):
        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