def update_admin(member, server, delete=False):
if (delete):
session.query(Admin).filter_by(discord_id=member.id).delete()
session.commit()
return
new_data = {'server_id': server.id, 'username': member.name}
session.query(Admin).filter_by(discord_id=member.id).update(new_data)
session.commit()
def update_role(target_role_id, server_id, required_role_id=None, delete=False):
if (delete):
session.query(Role).filter_by(target_role_id = target_role_id).delete()
session.commit()
return
new_data = {
'server_id': server_id,
'required_role_id': required_role_id
}
def update_active_listener(caller_msg, target_msg, delete=False):
if (delete):
session.query(ActiveListener).filter_by(caller_message_id = caller_msg.id).delete()
session.commit()
return
new_data = {
'caller_message_id': caller_msg.id,
'target_message_id': target_msg.id
}
session.query(ActiveListener).filter_by(caller_message_id = caller_msg.id).update(new_data)
session.commit()
def update_command(triggerkey, response, count, server, author_id, delete=False):
if (delete):
session.query(Command).filter_by(trigger = server.id + triggerkey).delete()
session.commit()
return
new_data = {
'server_id': server.id,
'response': response,
'count': count,
'author_id': int(author_id)
}
session.query(Command).filter_by(trigger = server.id + triggerkey).update(new_data)
session.commit()
def update_user(disc_id, delete=False):
if (delete):
session.query(User).filter_by(discord_id=disc_id).delete()
session.commit()
return
new_data = {
'aut_score': karma_dict[disc_id][0],
'norm_score': karma_dict[disc_id][1],
'nice_score': karma_dict[disc_id][2],
'toxic_score': karma_dict[disc_id][3]
}
session.query(User).filter_by(discord_id=disc_id).update(new_data)
session.commit()
def initialize_roles():
role_list = session.query(Role).all()
for server in client.servers:
roles.setdefault(int(server.id), [])
for role in role_list:
roles.setdefault(role.server_id, [])
roles[role.server_id].append((role.target_role_id, role.required_role_id))
def initialize_admins():
admin_list = session.query(Admin).all()
for server in client.servers:
admins[int(server.id)] = [int(server.owner.id)]
admins[int(server.id)].append(int(JOHNYS_ID))
for admin in admin_list:
admins.setdefault(admin.server_id, [])
admins[admin.server_id].append(admin.discord_id)
async def initialize_listeners():
listener_list = session.query(ActiveListener).all()
for listener in listener_list:
for channel in self.server.channels:
try: caller_msg = await self.client.get_message(channel, id=listener.caller_message_id)
except: caller_msg = None
if caller_msg:
target_msg = await self.client.get_message(channel, id=listener.target_message_id)
timestamp = target_msg.timestamp
timeout = (60*60*24*7) - (datetime.utcnow() - timestamp)
if any(x in caller_msg for x in poll_command.get_aliases()):
await poll_command.run_listener(self.client, channel, caller_msg, target_msg, init=False, timeout=timeout)
else:
def initialize_commands():
command_list = session.query(Command).all()
for server in client.servers:
smart_commands.setdefault(int(server.id), [])
#settings = server_settings(session, server)
#settings.admins_ids = [JOHNYS_ID]
#settings.bot_commands_channels = []
#print(settings.admins_ids)
for command in command_list:
smart_commands.setdefault(command.server_id, [])
smart_commands[command.server_id].append(smart_command(command.trigger.replace(str(command.server_id), '', 1), command.response, command.count, client.get_server(str(command.server_id)), command.author_id))
for server, cmds in smart_commands.items():
smart_commands[server].sort()
def initialize_commands():
command_list = session.query(Command).all()
for server in client.servers:
smart_commands.setdefault(int(server.id), [])
for command in command_list:
smart_commands.setdefault(command.server_id, [])
smart_commands[command.server_id].append(
smart_command(
command.trigger.replace(str(command.server_id), '', 1),
command.response, command.count,
client.get_server(str(command.server_id)), command.author_id))
for server, cmds in smart_commands.items():
smart_commands[server].sort()
def initialize_scores():
users = session.query(User).all()
for user in users:
karma_dict[user.discord_id] = user.as_entry()
