2019-02-25 20:29:13 +01:00
|
|
|
#!/usr/bin/python3
|
2019-04-07 21:31:13 +02:00
|
|
|
|
2019-02-25 20:29:13 +01:00
|
|
|
import importlib
|
2019-04-07 21:31:13 +02:00
|
|
|
import json
|
|
|
|
import logging
|
|
|
|
import logging.config
|
2019-03-02 14:38:37 +01:00
|
|
|
import os
|
2019-02-25 20:29:13 +01:00
|
|
|
import traceback
|
2019-03-02 14:38:37 +01:00
|
|
|
|
|
|
|
import discord
|
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
from modules.base import BaseClass
|
2019-02-25 20:29:13 +01:00
|
|
|
|
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
def setup_logging(default_path='config/log_config.json', default_level=logging.INFO, env_key='LBI_LOG_CONFIG'):
|
|
|
|
"""Setup logging configuration
|
|
|
|
"""
|
|
|
|
path = default_path
|
|
|
|
value = os.getenv(env_key, None)
|
|
|
|
if value:
|
|
|
|
path = value
|
|
|
|
if os.path.exists(path):
|
|
|
|
with open(path, 'rt') as f:
|
|
|
|
config = json.load(f)
|
|
|
|
logging.config.dictConfig(config)
|
2019-02-25 20:29:13 +01:00
|
|
|
else:
|
2019-04-07 21:31:13 +02:00
|
|
|
logging.basicConfig(level=default_level)
|
|
|
|
|
|
|
|
|
|
|
|
def modules_edit(func):
|
|
|
|
def wrapper(self, *args, **kwargs):
|
|
|
|
print(func.__name__, ":", self.reloading)
|
|
|
|
if self.reloading:
|
|
|
|
return func(self, *args, **kwargs)
|
|
|
|
else:
|
|
|
|
self.reloading = True
|
|
|
|
a = func(self, *args, **kwargs)
|
|
|
|
self.reloading = False
|
|
|
|
return a
|
|
|
|
|
|
|
|
return wrapper
|
|
|
|
|
|
|
|
|
|
|
|
def event(func):
|
|
|
|
def wrapper(self, *args, **kwargs):
|
|
|
|
if self.reloading:
|
|
|
|
return lambda: None
|
|
|
|
else:
|
|
|
|
return func(self, *args, **kwargs)
|
|
|
|
|
|
|
|
return wrapper
|
|
|
|
|
|
|
|
|
|
|
|
setup_logging()
|
|
|
|
|
|
|
|
log_discord = logging.getLogger('discord')
|
|
|
|
log_LBI = logging.getLogger('LBI')
|
|
|
|
|
|
|
|
debug = log_LBI.debug
|
|
|
|
info = log_LBI.info
|
|
|
|
warning = log_LBI.warning
|
|
|
|
error = log_LBI.error
|
|
|
|
critical = log_LBI.critical
|
|
|
|
|
|
|
|
|
|
|
|
class LBI(discord.Client):
|
|
|
|
base_path = "storage"
|
|
|
|
debug = log_LBI.debug
|
|
|
|
info = log_LBI.info
|
|
|
|
warning = log_LBI.warning
|
|
|
|
error = log_LBI.error
|
|
|
|
critical = log_LBI.critical
|
|
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.reloading = False
|
|
|
|
self.id = ClientById(self)
|
|
|
|
self.ready = False
|
|
|
|
# Content: {"module_name": {"module": imported module, "class": initialized class}}
|
|
|
|
self.modules = {}
|
|
|
|
self.config = {
|
|
|
|
"modules": ["modules"],
|
|
|
|
"prefix": "%",
|
|
|
|
}
|
|
|
|
self.owners = [281166473102098433, 318866596502306816]
|
|
|
|
self.load_config()
|
|
|
|
self.load_modules()
|
|
|
|
|
|
|
|
def load_config(self, config_file="config/config.json"):
|
|
|
|
if os.path.exists(config_file):
|
|
|
|
with open(config_file, 'rt') as f:
|
|
|
|
config = json.load(f)
|
|
|
|
self.config.update(config)
|
|
|
|
info("Config successfully loaded.")
|
|
|
|
else:
|
|
|
|
with open(config_file, 'w') as f:
|
|
|
|
json.dump(self.config, f)
|
|
|
|
info("Config successfully created.")
|
|
|
|
|
|
|
|
def save_config(self, config_file="config/config.json"):
|
|
|
|
with open(config_file, "w") as f:
|
|
|
|
json.dump(self.config, f)
|
|
|
|
info("Config successfully saved.")
|
|
|
|
|
|
|
|
@modules_edit
|
|
|
|
def load_modules(self):
|
|
|
|
info("Starts to load modules...")
|
|
|
|
e = {}
|
|
|
|
for module in self.config["modules"]:
|
|
|
|
e.update({module: self.load_module(module)})
|
|
|
|
info("Finished to load all modules.")
|
|
|
|
return e
|
|
|
|
|
|
|
|
@modules_edit
|
|
|
|
def load_module(self, module):
|
|
|
|
try:
|
|
|
|
info("Start loading module {module}...".format(module=module))
|
|
|
|
imported = importlib.import_module('modules.' + module)
|
|
|
|
if issubclass(imported.MainClass, BaseClass):
|
|
|
|
initialized_class = imported.MainClass(self)
|
|
|
|
self.modules.update({module: {"imported": imported, "initialized_class": initialized_class}})
|
|
|
|
info("Module {module} successfully imported.".format(module=module))
|
|
|
|
initialized_class.on_load()
|
|
|
|
if module not in self.config["modules"]:
|
|
|
|
self.config["modules"].append(module)
|
|
|
|
self.save_config()
|
|
|
|
else:
|
|
|
|
error("Module {module} isn't an instance of BaseClass.".format(module=module))
|
|
|
|
except AttributeError as e:
|
|
|
|
error("Module {module} doesn't have MainClass.".format(module=module))
|
|
|
|
return e
|
|
|
|
|
|
|
|
@modules_edit
|
|
|
|
def unload_module(self, module):
|
|
|
|
info("Start unload module {module}...".format(module=module))
|
|
|
|
try:
|
|
|
|
if module in self.config["modules"]:
|
|
|
|
self.config["modules"].remove(module)
|
|
|
|
self.save_config()
|
|
|
|
self.unload_all()
|
|
|
|
self.load_modules()
|
|
|
|
except KeyError as e:
|
|
|
|
error("Module {module} not loaded.").format(module=module)
|
|
|
|
return e
|
|
|
|
|
|
|
|
@modules_edit
|
|
|
|
def reload(self):
|
|
|
|
del self.modules
|
|
|
|
self.load_modules()
|
|
|
|
|
|
|
|
@modules_edit
|
|
|
|
def unload_all(self):
|
|
|
|
del self.modules
|
|
|
|
self.modules = {}
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_socket_raw_receive(self, message):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_socket_raw_receive(message)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_socket_raw_send(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_socket_raw_send(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_typing(self, channel, user, when):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_typing(channel, user, when)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_message(self, message):
|
|
|
|
print(message.content)
|
|
|
|
try:
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"]._on_message(message)
|
|
|
|
except RuntimeError:
|
|
|
|
info("Liste des modules changée pendant l'execution d'un on_message")
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_message_delete(self, message):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_message_delete(message)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_message_delete(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_message_delete(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_bulk_message_delete(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_bulk_message_delete(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_message_edit(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_message_edit(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_message_edit(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_message_edit(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_reaction_add(self, reaction, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_reaction_add(reaction, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_reaction_add(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_reaction_add(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_reaction_remove(self, reaction, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_reaction_remove(reaction, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_reaction_remove(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_reaction_remove(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_reaction_clear(self, message, reactions):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_reaction_clear(message, reactions)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_raw_reaction_clear(self, payload):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_raw_reaction_clear(payload)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_private_channel_delete(self, channel):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_private_channel_delete(channel)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_private_channel_create(self, channel):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_private_channel_create(channel)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_private_channel_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_private_channel_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_private_channel_pins_update(self, channel, last_pin):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_private_channel_pins_update(channel, last_pin)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_channel_delete(self, channel):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_channel_delete(channel)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_channel_create(self, channel):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_channel_create(channel)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_channel_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_channel_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_channel_pins_update(self, channel, last_pin):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_channel_pins_update(channel, last_pin)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_member_join(self, member):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_member_join(member)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_member_remove(self, member):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_member_remove(member)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_member_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_member_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_join(self, guild):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_join(guild)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_remove(self, guild):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_remove(guild)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_role_create(self, role):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_role_create(role)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_role_delete(self, role):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_role_delete(role)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_role_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_role_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_emojis_update(self, guild, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_emojis_update(guild, before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_available(self, guild):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_available(guild)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_unavailable(self, guild):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_unavailable(guild)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_voice_state_update(self, member, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_voice_state_update(member, before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_member_ban(self, guild, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_member_ban(guild, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_member_unban(self, guild, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_member_unban(guild, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_group_join(self, channel, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_group_join(channel, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_group_remove(self, channel, user):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_group_remove(channel, user)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_relationship_add(self, relationship):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_relationship_add(relationship)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_relationship_remove(self, relationship):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_relationship_remove(relationship)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_relationship_update(self, before, after):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_relationship_update(before, after)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_connect(self):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_connect()
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_shard_ready(self):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_shard_ready()
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_resumed(self):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_resumed()
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_error(self, event_, *args, **kwargs):
|
|
|
|
print(event_, *args, **kwargs)
|
|
|
|
print(traceback.format_exc())
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_error(event_, *args, **kwargs)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_guild_integrations_update(self, guild):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_guild_integrations_update(guild)
|
|
|
|
|
|
|
|
@event
|
|
|
|
async def on_webhooks_update(self, channel):
|
|
|
|
for module in self.modules.values():
|
|
|
|
await module["initialized_class"].on_webhooks_update(channel)
|
|
|
|
|
|
|
|
|
|
|
|
class ClientById:
|
|
|
|
client: LBI
|
|
|
|
|
|
|
|
def __init__(self, client_):
|
|
|
|
self.client = client_
|
|
|
|
|
|
|
|
async def get_message(self, id_, *args, **kwargs):
|
|
|
|
"""Find a message by id
|
|
|
|
|
|
|
|
:param id_: Id of message to find
|
|
|
|
:type id_: int
|
|
|
|
|
|
|
|
:raises discord.NotFound: This exception is raised when a message is not found (or not accessible by bot)
|
|
|
|
|
|
|
|
:rtype: discord.Message
|
|
|
|
:return: discord.Message instance if message is found.
|
|
|
|
"""
|
|
|
|
msg = None
|
|
|
|
for channel in self.client.get_all_channels():
|
|
|
|
try:
|
|
|
|
return await channel.get_message(id_, *args, **kwargs)
|
|
|
|
except discord.NotFound:
|
|
|
|
continue
|
|
|
|
if msg is None:
|
|
|
|
raise discord.NotFound(None, "Message not found")
|
2019-02-25 20:29:13 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
async def edit_message(self, id_, *args, **kwargs):
|
|
|
|
"""Edit message by id_
|
2019-03-02 14:38:37 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
:param id_: Id of the message to edit
|
|
|
|
:type id_: int"""
|
|
|
|
message = await self.get_message(id_)
|
|
|
|
return await message.edit(**kwargs)
|
2019-02-25 20:29:13 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
async def remove_reaction(self, id_message, *args, **kwargs):
|
|
|
|
"""Remove reaction from message by id
|
2019-03-02 14:38:37 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
:param id_message: Id of message
|
|
|
|
:type id_message: int"""
|
|
|
|
message = await self.get_message(id_message)
|
|
|
|
return await message.remove_reaction(*args, **kwargs)
|
2019-02-25 20:29:13 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
async def send_message(self, id_, *args, **kwargs):
|
|
|
|
"""Send message by channel id
|
2019-03-02 14:38:37 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
:param id_: Id of channel where to send message
|
|
|
|
:type id_: int"""
|
|
|
|
channel = self.client.get_channel(id_)
|
|
|
|
return channel.send(*args, **kwargs)
|
2019-02-25 20:29:13 +01:00
|
|
|
|
2019-03-02 14:38:37 +01:00
|
|
|
|
2019-04-07 21:31:13 +02:00
|
|
|
client = LBI()
|
|
|
|
client.run("NTUwMDkxOTAyMDY2ODg0NjA4.XKfETw.v150BqB-fMo2wCRuSsTLuKXAOyE", max_messages=500000)
|