LBI/modules/modules.py

import discord
import asyncio
import os
import importlib
import shutil
import random
import time
temp_dir="temp_load"
moduleFiles="modules"
class MainClass():
    def __init__(self, client, modules, owners, prefix):
        if os.path.exists("storage/"+moduleFiles+"/"+temp_dir):
            shutil.rmtree("storage/"+moduleFiles+"/"+temp_dir)
        if not os.path.exists("storage/"+moduleFiles):
            os.mkdir("storage/"+moduleFiles)
        os.mkdir("storage/"+moduleFiles+"/"+temp_dir)
        self.states={}
        for moduleName in os.listdir('modules'):
            if moduleName.endswith(".py"):
                self.states.update({moduleName[:-3:]:'not loaded'})
        for moduleName in list(modules.keys()):
            if len(modules[moduleName]) == 2 :
                self.states.update({moduleName:'initialized'})
        self.client = client
        self.modules = modules
        self.owners = owners
        self.prefix=prefix
        self.events=['on_message', 'on_ready'] #events list
        self.command="%smodule"%self.prefix #command prefix (can be empty to catch every single messages)

        self.name="Modules"
        self.description="Module de gestion des modules"
        self.interactive=True
        self.authlist=[549662392120901633]
        self.color=0x8f3196
        self.help="""\
 </prefix>modules list
 => Liste les modules ainsi que leurs états
 
 </prefix>modules enable <module/modules>
 => Charge et active le / les module(s) spécifié(s)
 
 </prefix>modules disable <module/modules>
 => Désactive et décharge le / les module(s) spécifié(s)
 
 </prefix>modules reload <module/modules>
 => Désactive, décharge, puis recharge et réactive le / les module(s) spécifié(s)
 
 => <module/modules>
 ==> Unique module ou liste de module séparés par des virgules
"""
        self.states.update({'modules': 'initialized'})
    async def on_message(self, message):
        error=None
        args = message.content.split(" ")
        if len(args) == 2 and args[1]=='list':
            await message.channel.send(embed=discord.Embed(title="[Modules] - Modules list", description="```PYTHON\n{0}```".format(str(self.states).replace(',', '\n,'))))
        elif len(args) == 3 and args[1] in ['enable', 'disable', 'reload']:
            if args[1]=='enable':
                for moduleName in args[2].split(','):
                    if moduleName + '.py' in os.listdir('modules'):
                        try:
                            self.enable_module(moduleName)
                            await message.channel.send(message.author.mention + ", le module {0} a été activé".format(moduleName))
                        except Exception as e:
                            error=e
                            await message.channel.send(message.author.mention + ", le module {0} **n'a pas pu être activé**".format(moduleName))
                    else:
                        await message.channel.send(message.author.mention + ", le module {0} n'existe pas.".format(moduleName))
            elif args[1]=='disable':
                for moduleName in args[2].split(','):
                    if moduleName == 'modules':
                        await message.channel.send(message.author.mention + ", le module {0} ne peut pas être désactivé car il est nécéssaire pour gérer les modules.".format(moduleName))
                    else:
                        if moduleName + '.py' in os.listdir('modules'):
                            self.unload_module(moduleName)
                            await message.channel.send(message.author.mention + ", le module {0} a été désactivé.".format(moduleName))
                        else:
                            await message.channel.send(message.author.mention + ", le module {0} n'existe pas.".format(moduleName))
            elif args[1]=='reload':
                for moduleName in args[2].split(','):
                    if moduleName == 'modules':
                        await message.channel.send(message.author.mention + ", le module {0} ne peut pas être rechargé car il est nécéssaire pour gérer les modules.".format(moduleName))
                    else:
                        if moduleName in self.modules.keys():
                            self.unload_module(moduleName)
                            await message.channel.send(message.author.mention + ", le module {0} a été désactivé.".format(moduleName))
                        else:
                            await message.channel.send(message.author.mention + ", le module {0} n'est pas chargé.".format(moduleName))
                        if moduleName + '.py' in os.listdir('modules'):
                            try:
                                self.enable_module(moduleName)
                                await message.channel.send(message.author.mention + ", le module {0} a été activé".format(moduleName))
                            except Exception as e:
                                error=e
                                await message.channel.send(message.author.mention + ", le module {0} **n'a pas pu être activé**".format(moduleName))
                        else:
                            await message.channel.send(message.author.mention + ", le module {0} n'existe pas.".format(moduleName))
        else:
            await self.modules['help'][1].send_help(message.channel, self)
        if error:
            raise error
    async def on_ready(self):
        error=None
        for fileName in os.listdir('modules'):
            try:
                self.load_module(fileName[:-3:])
                self.init_module(fileName[:-3:])
            except Exception as e:
                error=e
        if error:
            raise error
    def enable_module(self, moduleName):
        self.load_module(moduleName)
        self.init_module(moduleName)

    def load_module(self, moduleName):
        if moduleName + ".py" in os.listdir('modules'):
            if not moduleName in list(self.states.keys()) or self.states[moduleName] == 'not loaded':
                try:
                    tmpstr=str("storage/"+moduleFiles+"/"+temp_dir+"/"+moduleName+"-%s.py")%random.randint(1,100000000000000000000000000000000)
                    shutil.copy2("modules/%s.py"%moduleName, tmpstr)
                    time.sleep(0.1)
                    self.modules.update({moduleName:[importlib.import_module(tmpstr.replace('/','.')[:-3:])]})
                    print("Module {0} chargé.".format(moduleName))
                    self.states[moduleName] = 'loaded'
                except:
                    print("[ERROR] Le module {0} n'a pas pu être chargé.".format(moduleName))
                    self.unload_module(moduleName)
                    raise
    def init_module(self, moduleName):
        if moduleName + ".py" in os.listdir('modules'):
            if self.states[moduleName] == 'loaded':
                try:
                    self.modules[moduleName].append(self.modules[moduleName][0].MainClass(self.client, self.modules, self.owners, self.prefix))
                    print("Module {0} initialisé.".format(moduleName))
                    self.states[moduleName] = 'initialized'
                except:
                    print("[ERROR] Le module {0} n'a pas pu être initialisé.".format(moduleName))
                    self.unload_module(moduleName)
                    raise
    def unload_module(self, moduleName):
        if moduleName + ".py" in os.listdir('modules'):
            self.states[moduleName] = 'not loaded'
            self.modules.pop(moduleName, None)
            print("Module {0} déchargé.".format(moduleName))