|
|
|
|
|
|
|
|
from discord.ext.commands import group |
|
|
from discord.ext.commands import group |
|
|
|
|
|
|
|
|
import checks |
|
|
import checks |
|
|
from main import blacklisted |
|
|
|
|
|
from config.server_config import ServerConfig |
|
|
from config.server_config import ServerConfig |
|
|
|
|
|
|
|
|
|
|
|
def blacklisted(user): |
|
|
|
|
|
with open("config/blacklist.txt", "r") as fp: |
|
|
|
|
|
for line in fp.readlines(): |
|
|
|
|
|
if user.id+"\n" == line: |
|
|
|
|
|
return True |
|
|
|
|
|
return False |
|
|
|
|
|
|
|
|
class Twitch(): |
|
|
class Twitch(): |
|
|
""" |
|
|
""" |
|
|
|
|
|
|
|
|
typeb = False |
|
|
typeb = False |
|
|
if member_a.game: |
|
|
if member_a.game: |
|
|
if member_a.game.type and not typeb: # Hopefully this fucking fixes it |
|
|
if member_a.game.type and not typeb: # Hopefully this fucking fixes it |
|
|
ts_enabled = self.serverconfig[member_a.server.id]["twitch"]["enabled"] |
|
|
|
|
|
ts_whitelist = self.serverconfig[member_a.server.id]["twitch"]["whitelist"]["enabled"] |
|
|
|
|
|
|
|
|
ts_enabled = self.servers[member_a.server.id]["twitch"]["enabled"] |
|
|
|
|
|
ts_whitelist = self.servers[member_a.server.id]["twitch"]["whitelist"]["enabled"] |
|
|
if ts_enabled: |
|
|
if ts_enabled: |
|
|
if not ts_whitelist or member_a.id in \ |
|
|
if not ts_whitelist or member_a.id in \ |
|
|
self.serverconfig[member_a.server.id]["twitch"]["whitelist"]["list"]: |
|
|
|
|
|
channel = discord.Object(self.serverconfig[member_a.server.id]["twitch"]["twitch-channel"]) |
|
|
|
|
|
|
|
|
self.servers[member_a.server.id]["twitch"]["whitelist"]["list"]: |
|
|
|
|
|
channel = discord.Object(self.servers[member_a.server.id]["twitch"]["twitch-channel"]) |
|
|
return await self.bot.send_message(channel, |
|
|
return await self.bot.send_message(channel, |
|
|
content=":video_game:** {} is live!** :video_game:\n{}\n{}".format( |
|
|
content=":video_game:** {} is live!** :video_game:\n{}\n{}".format( |
|
|
member_a.name, member_a.game.name, member_a.game.url)) |
|
|
member_a.name, member_a.game.name, member_a.game.url)) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@twitch.command(pass_context=True, hidden=True) |
|
|
@twitch.command(pass_context=True, hidden=True) |
|
|
async def enablewhitelist(self, ctx): |
|
|
async def enablewhitelist(self, ctx): |
|
|
self.serverconfig = self.con.load_config() |
|
|
|
|
|
if not self.serverconfig[ctx.server.id]["twitch"]["whitelist"]["enabled"]: |
|
|
|
|
|
self.serverconfig[ctx.server.id]["twitch"]["whitelist"]["enabled"] = 1 |
|
|
|
|
|
self.con.update_config(self.serverconfig) |
|
|
|
|
|
|
|
|
self.servers = self.con.load_config() |
|
|
|
|
|
if not self.servers[ctx.server.id]["twitch"]["whitelist"]["enabled"]: |
|
|
|
|
|
self.servers[ctx.server.id]["twitch"]["whitelist"]["enabled"] = 1 |
|
|
|
|
|
self.con.update_config(self.servers) |
|
|
return await self.bot.reply("Whitelist for Twitch shilling has been enabled.") |
|
|
return await self.bot.reply("Whitelist for Twitch shilling has been enabled.") |
|
|
else: |
|
|
else: |
|
|
self.serverconfig[ctx.server.id]["twitch"]["whitelist"]["enabled"] = 0 |
|
|
|
|
|
self.con.update_config(self.serverconfig) |
|
|
|
|
|
|
|
|
self.servers[ctx.server.id]["twitch"]["whitelist"]["enabled"] = 0 |
|
|
|
|
|
self.con.update_config(self.servers) |
|
|
return await self.bot.reply("Whitelist for Twitch shilling has been disabled.") |
|
|
return await self.bot.reply("Whitelist for Twitch shilling has been disabled.") |
|
|
|
|
|
|
|
|
@twitch.command(pass_context=True, hidden=True) |
|
|
@twitch.command(pass_context=True, hidden=True) |
|
|
|
|
|
|
|
|
return await self.bot.say('Invalid option "%s" specified, use +, -, add, or remove' % option, expire_in=20) |
|
|
return await self.bot.say('Invalid option "%s" specified, use +, -, add, or remove' % option, expire_in=20) |
|
|
|
|
|
|
|
|
if option in ['+', 'add']: |
|
|
if option in ['+', 'add']: |
|
|
self.serverconfig = self.con.load_config() |
|
|
|
|
|
|
|
|
self.servers = self.con.load_config() |
|
|
for user in ctx.message.mentions: |
|
|
for user in ctx.message.mentions: |
|
|
self.serverconfig[ctx.message.server.id]["twitch"]["whitelist"]["list"].append(user.id) |
|
|
|
|
|
self.con.update_config(self.serverconfig) |
|
|
|
|
|
|
|
|
self.servers[ctx.message.server.id]["twitch"]["whitelist"]["list"].append(user.id) |
|
|
|
|
|
self.con.update_config(self.servers) |
|
|
whitelist_count += 1 |
|
|
whitelist_count += 1 |
|
|
return await self.bot.say('{} user(s) have been added to the whitelist'.format(whitelist_count)) |
|
|
return await self.bot.say('{} user(s) have been added to the whitelist'.format(whitelist_count)) |
|
|
|
|
|
|
|
|
elif option in ['-', 'remove']: |
|
|
elif option in ['-', 'remove']: |
|
|
self.serverconfig = self.con.load_config() |
|
|
|
|
|
|
|
|
self.servers = self.con.load_config() |
|
|
for user in ctx.message.mentions: |
|
|
for user in ctx.message.mentions: |
|
|
if user.id in self.serverconfig[ctx.message.server.id]["twitch"]["whitelist"]["list"]: |
|
|
|
|
|
self.serverconfig[ctx.message.server.id]["twitch"]["whitelist"]["list"].remove(user.id) |
|
|
|
|
|
self.con.update_config(self.serverconfig) |
|
|
|
|
|
|
|
|
if user.id in self.servers[ctx.message.server.id]["twitch"]["whitelist"]["list"]: |
|
|
|
|
|
self.servers[ctx.message.server.id]["twitch"]["whitelist"]["list"].remove(user.id) |
|
|
|
|
|
self.con.update_config(self.servers) |
|
|
whitelist_count += 1 |
|
|
whitelist_count += 1 |
|
|
return await self.bot.say('{} user(s) have been removed to the whitelist'.format(whitelist_count)) |
|
|
return await self.bot.say('{} user(s) have been removed to the whitelist'.format(whitelist_count)) |
|
|
|
|
|
|
|
|
elif option == 'list': |
|
|
elif option == 'list': |
|
|
return await self.bot.say( |
|
|
return await self.bot.say( |
|
|
self.serverconfig[ctx.message.server.id]["twitch"]["whitelist"]["list"]) |
|
|
|
|
|
|
|
|
self.servers[ctx.message.server.id]["twitch"]["whitelist"]["list"]) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def setup(Bot): |
|
|
def setup(Bot): |