async def on_guild_join(self, g: discord.Guild):
guild = session.query(Guild).filter(Guild.guild_id == g.id).first()
if not guild:
guild = Guild(guild_id=g.id)
session.add(guild)
session.commit()
async def nick(self, ctx: commands.Context, *, new_nick: str = ""):
guild = session.query(Guild).filter(
Guild.guild_id == ctx.guild.id).first()
if not guild:
guild = Guild(guild_id=ctx.guild.id)
session.add(guild)
session.commit()
if not guild.custom_nick:
await ctx.send(
f"<{EMOJIS['XMARK']}> This server has disabled custom nicknames!"
)
return
discord_user = session.query(DiscordUser).filter(
DiscordUser.user_id == ctx.author.id).first()
verified = [acc.username for acc in discord_user.reddit_accounts]
if new_nick != "":
end = f"({'unverified' if len(verified) <= 0 else '/u/' + verified[0]})"
new_nick = f"{new_nick[:32 - len(end) - 1]} {end}"
else:
new_nick = f"/u/{verified[0]}" if len(
verified) > 0 else ctx.author.username
try:
await ctx.author.edit(nick=new_nick)
except discord.errors.Forbidden:
await ctx.send(
f"<{EMOJIS['XMARK']}> I don't have the permissions to edit your nickname!"
)
async def addemoji(self, ctx: commands.Context, emoji: str):
try:
await ctx.message.add_reaction(emoji)
except BaseException:
await ctx.send(f"<{EMOJIS['XMARK']}> Uh-oh, looks like that emoji doesn't work!")
return
channel = session.query(Channel).filter(Channel.channel_id == ctx.channel.id).first()
if not channel or not channel.poll_channel:
msg = f"<{EMOJIS['XMARK']}> This channel isn't setup as a poll channel! Please use `!togglepoll` to enable the feature!"
await ctx.send(msg)
return
channel_emoji = session.query(ChannelEmoji).filter(
and_(
ChannelEmoji.channel_id == ctx.channel.id,
ChannelEmoji.emoji == emoji)).first()
if not channel_emoji:
channel_emoji = ChannelEmoji(channel_id=ctx.channel.id, emoji=emoji)
session.add(channel_emoji)
session.commit()
await ctx.send(f"<{EMOJIS['CHECK']}> Successfully added {emoji} to the list of emojis to add in this channel!", delete_after=3)
async for msg in ctx.history(limit=None):
await msg.add_reaction(emoji)
else:
await ctx.send(f"<{EMOJIS['XMARK']}> That emoji is already set for this channel!")
async def toggle(self, ctx: commands.Context, toggle: str):
guild = session.query(Guild).filter(
Guild.guild_id == ctx.guild.id).first()
if not guild:
guild = Guild(guild_id=ctx.guild.id)
session.add(guild)
session.commit()
if toggle == "username":
guild.set_username = not guild.set_username
session.commit()
if guild.set_username:
for member in ctx.guild.members:
if discord_user := session.query(DiscordUser).filter(
DiscordUser.user_id == member.id).first():
await self.bot.update_guild_user(guild, discord_user)
await ctx.send(
f"<{EMOJIS['CHECK']}> Reddit usernames enabled for this server!"
)
else:
await ctx.send(
f"<{EMOJIS['CHECK']}> Reddit usernames disabled for this server!"
)
async def addtags(self, ctx: commands.Context, *, key: str = ""):
guild = self.bot.get_db_guild(ctx.guild.id)
if ctx.author.id not in AUTHORIZED_USERS:
if not guild.melon_role:
await ctx.send(
f"<{EMOJIS['XMARK']}> Custom Melons for this guild haven't been enabled!"
)
return
roles = [role.id for role in ctx.author.roles]
if guild.melon_role not in roles:
await ctx.send(
f"<{EMOJIS['XMARK']}> You aren't authorized to edit Melons in this guild!"
)
return
if not key:
key = await self.ask_prompt(
ctx, "❓ What is the key used for the Melon?")
if not key:
return
else:
key = key.lower()
if ctx.author.id in AUTHORIZED_USERS:
melon = session.query(Melon).filter(
and_(
func.lower(Melon.key) == key,
or_(Melon.guild_id == guild.guild_id,
Melon.guild_id.is_(None)))).first()
else:
melon = session.query(Melon).filter(
and_(
func.lower(Melon.key) == key,
Melon.guild_id == guild.guild_id)).first()
if not melon:
await ctx.send(f"<{EMOJIS['XMARK']}> Melon '{key}' doesn't exist!")
tags = await self.ask_prompt(
ctx, f"❓ Which tags should be added to the Melon '{key}'?")
if not tags:
return
tags = tags.split(" ")
for t in tags:
tag = session.query(Tag).filter(
func.lower(Tag.value) == t.lower()).first()
if not tag:
tag = Tag(value=t)
session.add(tag)
elif tag in melon.tags:
continue
melon.tags.append(tag)
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully added tags `{', '.join(tags)}` to Melon '{melon.key}'!"
)
def get_db_guild(self, guild_id: int) -> Guild:
guild = session.query(Guild).filter(Guild.guild_id == guild_id).first()
if not guild:
categories = session.query(Category).filter(
Category.default == true()).all()
guild = Guild(guild_id=guild_id, categories=categories)
session.add(guild)
session.commit()
return guild
async def addcat(self, ctx: commands.Context, *, cat: str):
category = session.query(Category).filter(
func.lower(Category.name) == cat.lower()).first()
if category:
await ctx.send(
f"<{EMOJIS['XMARK']}> '{cat}' is already a category in the Melons!"
)
else:
category = Category(name=cat)
session.add(category)
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully added '{cat}' as a category in the Melons!"
)
async def setchannelsub(self, ctx: commands.Context, sub: str):
s = await self.bot.reddit.subreddit(sub)
channel = session.query(Channel).filter(
Channel.channel_id == ctx.channel.id).first()
if not channel:
channel = Channel(channel_id=ctx.channel.id)
session.add(channel)
channel.subreddit = sub
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully set /r/{sub} as this channels's subreddit!"
)
async def togglepoll(self, ctx: commands.Context):
channel = session.query(Channel).filter(Channel.channel_id == ctx.channel.id).first()
if not channel:
channel = Channel(channel_id=ctx.channel.id, poll_channel=True)
session.add(channel)
else:
channel.poll_channel = not channel.poll_channel
session.commit()
if channel.poll_channel:
await ctx.send(f"<{EMOJIS['CHECK']}> Successfully registered channel as a poll channel!", delete_after=3)
else:
await ctx.send(f"<{EMOJIS['CHECK']}> Successfully removed polling feature from this channel!", delete_after=3)
await ctx.message.delete()
async def status(self, ctx: commands.Context):
guild = session.query(Guild).filter(
Guild.guild_id == ctx.guild.id).first()
if not guild:
guild = Guild(guild_id=ctx.guild.id)
session.add(guild)
session.commit()
set_role = "on" if guild.set_role else "off"
set_username = "******" if guild.set_username else "off"
set_nickname = "on" if guild.custom_nick else "off"
role_name = ctx.guild.get_role(guild.role).name if guild.role else None
await ctx.channel.send(
f"Adding '{role_name}' role: {set_role}\n" +
f"Changing nickname to /u/username: {set_username}\n" +
f"Allowing custom nicknames: {set_nickname}")
async def unverify(self, ctx: commands.Context, user: str):
if reddit_user := session.query(RedditUser).filter(
and_(
func.lower(RedditUser.username) == user,
RedditUser.discord_user == ctx.author.id)).first():
session.delete(reddit_user)
session.commit()
guild = session.query(Guild).filter(
Guild.guild_id == ctx.guild.id).first()
if not guild:
guild = Guild(guild_id=ctx.guild.id)
session.add(guild)
session.commit()
if discord_user := session.query(DiscordUser).filter(
DiscordUser.user_id == ctx.author.id).first():
await self.bot.update_guild_user(guild, discord_user)
async def adddefault(self, ctx: commands.Context, emoji: str):
try:
await ctx.message.add_reaction(emoji)
except BaseException:
await ctx.send(f"<{EMOJIS['XMARK']}> Uh-oh, looks like that emoji doesn't work!")
return
channel = session.query(Channel).filter(Channel.channel_id == ctx.channel.id).first()
if not channel or not channel.poll_channel:
msg = f"<{EMOJIS['XMARK']}> This channel isn't setup as a poll channel! Please use `!togglepoll` to enable the feature!"
await ctx.send(msg)
return
channel_emoji = ChannelEmoji(channel_id=channel.channel_id, emoji=emoji, default_emoji=True)
session.add(channel_emoji)
session.commit()
await ctx.send(f"<{EMOJIS['CHECK']}> Added {emoji} to the list of default emojis for this poll channel!", delete_after=3)
await ctx.message.delete()
async def setguildsub(self, ctx: commands.Context, sub: str):
guild = session.query(Guild).filter(
Guild.guild_id == ctx.guild.id).first()
if not guild:
guild = Guild(guild_id=ctx.guild.id)
session.add(guild)
session.commit()
try:
s = await self.bot.reddit.subreddit(sub)
except BaseException:
await ctx.send(
"❗ /r/{} doesn't exist or isn't visible to me!".format(sub))
else:
guild.subreddit = sub
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully set /r/{sub} as this guild's subreddit!"
)
async def verify(self, ctx: commands.Context, user: str):
user = user.replace("u/", "").replace("/", "").strip().lower()
discord_user = session.query(DiscordUser).filter(
DiscordUser.user_id == ctx.author.id).first()
if not discord_user:
discord_user = DiscordUser(user_id=ctx.author.id)
session.add(discord_user)
session.commit()
if session.query(RedditUser).filter(
and_(
func.lower(RedditUser.username) == user,
RedditUser.discord_user != ctx.author.id)).first():
await ctx.message.channel.send(
f"❗ /u/{user} is already linked to another Discord account! " +
f"If you believe this is an error, please contact {MAINTAINER}."
)
return
if session.query(RedditUser).filter(
and_(
func.lower(RedditUser.username) == user,
RedditUser.discord_user == ctx.author.id)).first():
await ctx.message.channel.send(
f"❗ /u/{user} is already linked to your Discord account!")
return
if reddit_user := session.query(RedditUser).filter(
and_(RedditUser.discord_user == ctx.author.id,
RedditUser.verified == false())).first():
await ctx.message.channel.send(
f"❗ /u/{reddit_user.username} is still awaiting verification on Reddit. "
+
f"If you would like to unlink it, use `!unverify {reddit_user.username}`."
)
return
def add():
import pdb; pdb.set_trace()
m = Movie(**{x: y for x, y in request.form.iteritems()})
session.add(m)
session.commit()
return 'Movie "%s" was added.\n' % (m.title)
def create(self):
session.add(self)
session.commit()
return self
async def addmelon(self, ctx: commands.Context, *, arg: str = ""):
def check(m):
return m.author == ctx.author and m.channel == ctx.channel
if arg and ctx.author.id in AUTHORIZED_USERS:
category = session.query(Category).filter(
func.lower(Category.name) == arg.lower()).first()
if not category:
msg = await ctx.send(
f"<{EMOJIS['XMARK']}> Category '{arg}' doesn't exist! Would you like to create it?"
)
await msg.add_reaction(f"<{EMOJIS['CHECK']}>")
await msg.add_reaction(f"<{EMOJIS['XMARK']}>")
try:
reaction = await bot.wait_for("reaction_add",
check=check,
timeout=2 * 60)
if reaction[0].custom_emoji and reaction[
0].emoji.id == int(EMOJIS['CHECK'].split(":")[2]):
category = Category(name=arg)
session.add(category)
await ctx.send(
f"<{EMOJIS['CHECK']}> Category '{arg}' successfully created!"
)
except asyncio.exceptions.TimeoutError:
await ctx.send(
f"<{EMOJIS['XMARK']}> That took too long! " +
"You can restart the process by calling this command again."
)
return
key = await self.ask_prompt(
ctx, "❓ What should the key of the Melon be?")
if not key:
return
else:
key = key.lower()
melon = session.query(Melon).filter(
and_(
func.lower(Melon.key) == key,
Melon.category_id == category.category_id)).first()
if melon:
await ctx.send(
f"❗ A Melon with that key already exists in {arg}! " +
"You can edit it with `!editmelon`.")
return
await ctx.send(f"❓ What should the value of {key} be?")
message = await self.bot.wait_for("message", check=check)
melon = Melon(key=key,
category=category,
value=message.content,
created_by=ctx.author.id)
session.add(melon)
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully added Melon '{key}' to the global Melons!"
)
else:
guild = self.bot.get_db_guild(ctx.guild.id)
if not guild.melon_role:
await ctx.send(
f"<{EMOJIS['XMARK']}> Custom Melons for this guild haven't been enabled!"
)
return
roles = [role.id for role in ctx.author.roles]
if guild.melon_role not in roles:
await ctx.send(
f"<{EMOJIS['XMARK']}> You aren't authorized to edit Melons in this guild!"
)
return
if not arg:
key = await self.ask_prompt(
ctx, "❓ What should the key of the Melon be?")
if not key:
return
else:
key = key.lower()
else:
key = arg.lower()
melon = session.query(Melon).filter(
and_(
func.lower(Melon.key) == key,
Melon.guild_id == guild.guild_id)).first()
if melon:
await ctx.send(
f"❗ A Melon with that key already exists in this guild! " +
"You can edit it with `!editmelon`.")
return
await ctx.send(f"❓ What should the value of {key} be?")
message = await self.bot.wait_for("message", check=check)
melon = Melon(key=key,
guild=guild,
value=message.content,
created_by=ctx.author.id)
session.add(melon)
session.commit()
await ctx.send(
f"<{EMOJIS['CHECK']}> Successfully added Melon '{key}' to the guild Melons!"
)
class UserCog(commands.Cog):
def __init__(self, bot: 'Reddify'):
self.bot = bot
@commands.command(
help=
"Sends a PM to the Reddit user to await confirmation of the account ownership."
)
async def verify(self, ctx: commands.Context, user: str):
user = user.replace("u/", "").replace("/", "").strip().lower()
discord_user = session.query(DiscordUser).filter(
DiscordUser.user_id == ctx.author.id).first()
if not discord_user:
discord_user = DiscordUser(user_id=ctx.author.id)
session.add(discord_user)
session.commit()
if session.query(RedditUser).filter(
and_(
func.lower(RedditUser.username) == user,
RedditUser.discord_user != ctx.author.id)).first():
await ctx.message.channel.send(
f"❗ /u/{user} is already linked to another Discord account! " +
f"If you believe this is an error, please contact {MAINTAINER}."
)
return
if session.query(RedditUser).filter(
and_(
func.lower(RedditUser.username) == user,
RedditUser.discord_user == ctx.author.id)).first():
await ctx.message.channel.send(
f"❗ /u/{user} is already linked to your Discord account!")
return
if reddit_user := session.query(RedditUser).filter(
and_(RedditUser.discord_user == ctx.author.id,
RedditUser.verified == false())).first():
await ctx.message.channel.send(
f"❗ /u/{reddit_user.username} is still awaiting verification on Reddit. "
+
f"If you would like to unlink it, use `!unverify {reddit_user.username}`."
)
return
try:
redditor = await self.bot.reddit.redditor(user)
except Exception as e:
await ctx.message.channel.send(
f"❌ Redditor /u/{user} doesn't exist!\n\nError:{e}")
else:
await redditor.message(
"Account Ownership Confirmation",
f"Please reply to this message with 'verify' to confirm that you are Discord user '{ctx.message.author}'."
)
msg = await ctx.message.channel.send(
f"Check your Reddit PMs to verify you are the owner of /u/{redditor.name} by responding with `verify`!"
)
reddit_user = RedditUser(user_id=redditor.fullname,
discord_user=ctx.author.id,
username=redditor.name)
session.add(reddit_user)
auth_user = await self.bot.reddit.user.me()
async for message in auth_user.unread.stream(skip_existing=True):
if not isinstance(message, apraw.models.Message):
continue
if not "Account Ownership Confirmation" in message.subject:
continue
try:
author = await message.author()
if author.name.lower() != reddit_user.username.lower():
continue
except Exception as e:
await self.bot.send_error(e)
if "verify" in message.body.lower(
) and not "unverify" in message.body.lower():
reddit_user.verified = True
user = self.bot.get_user(reddit_user.discord_user)
if user:
await msg.add_reaction(EMOJIS["CHECK"])
rep = await message.reply(
f"Confirmation of {user} successful!")
for guild in session.query(Guild).all():
await self.bot.update_guild_user(
guild, discord_user)
else:
message.reply(
"Discord user unlinked from your Reddit account. " +
"Please run `!verify` again if this is an error and respond to the new PM."
)
session.delete(reddit_user)
break
session.commit()