122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
"""
|
||
Karl Marx 2
|
||
A bot by turtlebasket
|
||
"""
|
||
|
||
import asyncio
|
||
import json
|
||
import discord
|
||
from discord.ext import commands
|
||
|
||
MUTE_VOTE_TIME = 10
|
||
MUTE_VOTER_MIN = 2
|
||
MUTE_TIME = 600 # 10 mins
|
||
BAN_VOTE_TIME = 8
|
||
BAN_VOTER_MIN = 4
|
||
bot = commands.Bot(command_prefix='>>')
|
||
|
||
@bot.event
|
||
async def on_ready():
|
||
print("Bot started.")
|
||
print("--------------------------")
|
||
|
||
@bot.command()
|
||
async def ping(ctx):
|
||
""" command: ping
|
||
Use with caution, you might stir up a revolution.
|
||
"""
|
||
await ctx.send("What do you think I am, some sort of toy? I refuse to bend to the will of the bourgeoisie!")
|
||
|
||
@bot.command()
|
||
async def anthem(ctx):
|
||
"""
|
||
command: anthem
|
||
some people just need a reference, ya know?
|
||
"""
|
||
await ctx.send("Soyuz nyerushimyiy ryespublik svobodnyikh\n"
|
||
"Splotila navyeki Vyelikaya Rus’.\n"
|
||
"Da zdravstvuyet sozdannyiy volyey narodov\n"
|
||
"Yedinyiy, moguchiy Sovyetskiy Soyuz!\n")
|
||
|
||
# not a command, just some functionality that's used across commands
|
||
async def take_vote(ctx, question:str, wait_time):
|
||
"""
|
||
take_vote(ctx, question:str) - Collects votes
|
||
ctx: pass from command function
|
||
question: what to ask
|
||
|
||
returns [<all who want>, <all who don't want>].
|
||
It's up to the context/use case to decide how these should be used.
|
||
"""
|
||
|
||
votey_message = await ctx.send("**=== NEW VOTE ===**\n{}".format(question))
|
||
|
||
await votey_message.add_reaction('✅')
|
||
await votey_message.add_reaction('❌')
|
||
|
||
await asyncio.sleep(wait_time)
|
||
|
||
finished_votey = await votey_message.channel.fetch_message(votey_message.id)
|
||
all_in_favor = 0
|
||
not_in_favor = 0
|
||
|
||
for reaction in finished_votey.reactions:
|
||
if str(reaction.emoji) == '✅':
|
||
all_in_favor += reaction.count-1 # don't include bot's reaction
|
||
elif str(reaction.emoji) == '✅':
|
||
not_in_favor += reaction.count-1
|
||
else:
|
||
pass
|
||
|
||
await ctx.send(
|
||
"**=== VOTE RESULTS ===**\n"
|
||
"✅ - {0}\n"
|
||
"❌ - {1}\n".format(all_in_favor, not_in_favor))
|
||
return [all_in_favor, not_in_favor]
|
||
|
||
@bot.command()
|
||
async def mute(ctx, target_user:discord.User):
|
||
|
||
results = await take_vote(ctx, "Mute `{}`?".format(target_user), MUTE_VOTE_TIME)
|
||
all_in_favor = results[0]
|
||
not_in_favor = results[1]
|
||
|
||
if all_in_favor > 0 and all_in_favor - not_in_favor > 0:
|
||
# Take action to mute user
|
||
|
||
# add temp. role for mute
|
||
muted_role = await ctx.guild.create_role(name="Muted")
|
||
# edit role position to take precedence over other roles
|
||
await muted_role.edit(position=ctx.guild.get_member(target_user.id).top_role.position+1)
|
||
|
||
for channel in ctx.guild.channels:
|
||
await channel.set_permissions(muted_role, read_messages=True, send_messages=False, add_reactions=False, connect=False)
|
||
|
||
await ctx.guild.get_member(target_user.id).add_roles(muted_role)
|
||
await ctx.send("**{0}, the majority has ruled that you should be muted.** See ya in {1} minutes!".format(target_user, int(MUTE_TIME/60)))
|
||
await asyncio.sleep(MUTE_TIME)
|
||
await muted_role.delete()
|
||
|
||
else:
|
||
await ctx.send("**{} has not been muted.**".format(target_user))
|
||
|
||
@bot.command()
|
||
async def roletest(ctx):
|
||
await ctx.send("`{}`".format(ctx.guild.roles))
|
||
|
||
@bot.command()
|
||
async def exile(ctx, target_user:discord.User):
|
||
results = await take_vote(ctx, "Ban `{}`?".format(target_user), BAN_VOTE_TIME)
|
||
all_in_favor = results[0]
|
||
not_in_favor = results[1]
|
||
|
||
if all_in_favor > not_in_favor and all_in_favor >= BAN_VOTER_MIN: # change to 10 later
|
||
await ctx.guild.ban(target_user)
|
||
await ctx.send(":crab: :crab: `{}` IS GONE :crab: :crab:".format(target_user.name))
|
||
elif all_in_favor <= all_in_favor:
|
||
await ctx.send("The majority (>50%) did not decide on banning `{}`.".format(target_user.name))
|
||
elif all_in_favor < 1:
|
||
await ctx.send("Not enough users voted to ban `{}`.".format(target_user.name))
|
||
|
||
bot.run(open("token.txt").read().strip())
|