bot.py

#!/usr/bin/env python3
#-*- coding:utf-8 -*-
import discord
from discord.ext import commands

import geo
import interface as itf

"""
Bot discord basé sur les données hospitalieres du covid disponibles sur data.gouv.fr

exemples d'utilisation :
    +Coro Indre 08-2020 09-2020
    +Coro Indre 08-2020
    +Coro 36
    +Coro Île-de-France
    +Coro help
    +Coro départements (départements/dep)
"""

bot = commands.Bot(command_prefix='!')

@bot.event
async def on_ready():
    print("CoroBot connecté")
    await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.listening, name="!coro help"))

@bot.command(hidden=True)
async def exit(ctx):
    if ctx.author.id == 404395089389944832 :
        await ctx.send("Fermeture du CoroBot...")
        print("CoroBot fermé avec succès")
        await bot.close()
    else :
        await ctx.send("nope")


#******************************************************************************
@commands.cooldown(1, 5, commands.BucketType.guild)
@bot.group()
async def coro(ctx):
    if ctx.invoked_subcommand is None :
        await ctx.send("Pour de l'aide sur la commande Coro, utilises `!coro help`")

@coro.command()
async def help(ctx):
    await ctx.send(itf.help_message)

@coro.command()
async def plot(ctx, *args):
    args = itf.parseArgs(args)
    # print(args)
    file = discord.File("fig.jpg")
    if args["type"] == "hospi" :
        async with ctx.channel.typing():
            infos = itf.plotFromArgs(args)
            embed = discord.Embed(
                title=infos["Titre"],
                description=infos["Description"],
                colour=discord.Colour.magenta())
            embed.set_image(url="attachment://fig.jpg")
            if len(infos["Erreurs"]) > 0 :
                for info in infos["Erreurs"] :
                    embed.add_field(name="Information : ", value=info)
            await ctx.send(file=file, embed=embed)
    elif args["type"] == "tests" :
        if args["dep"] == "FR" :
            await ctx.send("Il faut spécifier un département (pour l'instant)")
        else :
            async with ctx.channel.typing():
                dep, d1, d2 = geo.plotGivenDep(dep=args["dep"], params=args, d1=args["d1"], d2=args["d2"])
                embed = discord.Embed(
                    title=f"Courbe des dépistages positifs : {dep}",
                    description=f"Entre le {d1} et le {d2}\nMoyenne flottante sur 15 jours",
                    colour=discord.Colour.magenta())
                embed.set_image(url="attachment://fig.jpg")
                await ctx.send(file=file, embed=embed)
            
@plot.error
async def plot_error(ctx, error):
    if isinstance(error, commands.errors.CommandInvokeError):
        await ctx.send("La commande n'a pas été invoquée correctement, utilises `!coro help` pour plus d'informations !")

@coro.command()
async def dep(ctx, *args):
    args = itf.parseArgs(args)
    if args["type"] == "dep" :
        await ctx.message.author.send(itf.sendDeps())
        await ctx.send(f"Je t'ai envoyé la liste des département en message privé {ctx.author.mention}")

@coro.command()
async def carte(ctx, typeD=None, days=40, fmean=15):
    if typeD not in ["tests", "hospi"] :
        await ctx.send("Il faut donner un type de données : `!coro carte tests` pour les données de dépistage et `!coro carte hospi`pour les données hospitalières")
    else :
        async with ctx.channel.typing():
            if typeD == "tests" :
                df = geo.donnesDepistage(days=days, floatingMean=fmean)
                typeD = "taux de tests positifs"
            else :
                df = geo.donnesHosp(days=days, floatingMean=fmean)
                typeD = "nombre de décès répertoriés (données hospitalières)"
            geo.mapDepInfection(df)
            file = discord.File("fig.jpg")
            embed = discord.Embed(
                title=f"Progression du Covid : augmentation du {typeD}",
                description=f"Départements ou le Covid progresse le plus vite depuis {days} jours. \nMoyenne flottante sur {fmean} jours",
                colour=discord.Colour.magenta())
            embed.set_image(url="attachment://fig.jpg")
            await ctx.send(file=file, embed=embed)

@coro.error
async def coro_error(ctx, error):
    if isinstance(error, commands.errors.CommandOnCooldown) :
        await ctx.send(f"Minute papillon, laisses moi quelques secondes ! :rage: {ctx.author.mention}")
    else :
        with open("id.txt", "r") as idtxt :
            id = int(idtxt.read())
        MYSELF = bot.get_user(id)
        await MYSELF.send(f"Une erreur non anticipée est advenue : \n'{error}'\n\
Serv : {ctx.guild.name}\n\
Salon : {ctx.channel.mention}")

#******************************************************************************
with open("token.txt", "r") as token :
    t = token.read()
bot.run(t)