async def message_role_reactions(message, data):
for line in data:
if (discord.utils.get(message.guild.roles, name=line[0]) is None):
await message.channel.send("{} {} nie je rola".format(
utils.generate_mention(message.author.id), line[0]))
else:
try:
await message.add_reaction(line[1])
except discord.errors.HTTPException:
await message.channel.send(
"{} {} pre rolu {} nie je emote".format(
utils.generate_mention(message.author.id), line[1],
line[0]))
async def botroom_check(message):
room = await get_room(message)
if room != "bot-room" and room is not None:
await message.channel.send("{} <:sadcat:576171980118687754> 👉 "
"<#461549842896781312>\n".format(
utils.generate_mention(
message.author.id)))
async def add_role_on_reaction(role, member, message):
role = discord.utils.get(message.guild.roles, name=role)
max_role = discord.utils.get(message.guild.roles, name="Rubbergod")
if role is not None:
if role < max_role:
await member.add_roles(role)
else:
await message.channel.send(
"{} na pridanie role {} nemáš práva".format(
utils.generate_mention(member.id), role.name))
async def get_join_role_data(message):
input_string = message.content
input_string = input_string.replace(" - ", " ")
input_string = input_string.replace(": ", " ")
input_string = input_string.replace("**", "")
output = []
try:
input_string = (input_string[input_string.index('\n') +
1:].strip().split('\n'))
except ValueError:
await message.channel.send("{} nesprávny formát. Použi !god".format(
utils.generate_mention(message.author.id)))
return output
for line in input_string:
line = line.split()
if len(line) > 1:
line = [line[0], line[1]]
output.append(line)
else:
await message.channel.send("{} {} je neplatný riadok".format(
utils.generate_mention(message.author.id), line[0]))
return output
async def send_code(message):
if len(str(message.content).split(" ")) != 2:
await message.channel.send("Očekávám 1 argument (login)\n")
return
if not user.has_role(message, config.verification_role):
if str(message.content).split(" ")[1] == "xlogin00":
guild = client.get_guild(config.guild_id)
fp = await guild.fetch_emoji(585915845146968093)
await message.channel.send("Tvuj login {} {}".format(
str(fp), utils.generate_mention(message.author.id)))
return
db_record = user.find_login_to_mail(message)
if db_record:
# get server permit role
code = ''.join(
random.choices(string.ascii_uppercase + string.digits, k=20))
login = str(message.content).split(" ")[1]
email_message = "!verify " + login + " " + code
password = config.email_pass
port = 465
context = ssl.create_default_context()
sender_email = "*****@*****.**"
receiver_email = login + "@stud.fit.vutbr.cz"
subject = "FIT Discord verifikace"
mail_content = 'Subject: {}\n\n{}'.format(subject, email_message)
with smtplib.SMTP_SSL("smtp.gmail.com", port,
context=context) as server:
server.login("*****@*****.**", password)
server.sendmail(sender_email, receiver_email, mail_content)
user.save_mail(message, code)
await message.channel.send(
("Kód byl odeslán " + "na tvůj mail " +
"(@stud.fit.vutbr.cz)! {}\n" + "Pro verifikaci použij:\n" +
"!verify xlogin00 kód").format(
utils.generate_mention(message.author.id)))
else:
await message.channel.send(
("Login nenalezen nebo jsi již " +
"prošel tímhle krokem {} {}").format(
utils.generate_mention(message.author.id),
utils.generate_mention(config.admin_id)))
else:
await message.channel.send("Už si byl verifikován {} {}".format(
utils.generate_mention(message.author.id),
utils.generate_mention(config.admin_id)))
try:
await message.delete()
except discord.errors.Forbidden:
return
async def pick(message):
""""Pick an option"""
option = rng.pick_option(message)
if option:
await message.channel.send("{} {}".format(
option, utils.generate_mention(message.author.id)))
async def verify(message):
""""Verify if VUT login is from database"""
if len(str(message.content).split(" ")) != 3:
await message.channel.send(
"Očekávám 2 argumenty (login a kód)\n" +
"Pro získaní kódu použij `!getcode xlogin00`")
return
if not user.has_role(message, config.verification_role):
if str(message.content).split(" ")[1] == "xlogin00":
guild = client.get_guild(config.guild_id)
fp = await guild.fetch_emoji(585915845146968093)
await message.channel.send("Tvůj login {} {}".format(
str(fp), utils.generate_mention(message.author.id)))
return
if str(message.content).split(" ")[2] == "kód":
guild = client.get_guild(config.guild_id)
fp = await guild.fetch_emoji(585915845146968093)
await message.channel.send(
"Kód který ti přišel na mail {} {}".format(
str(fp), utils.generate_mention(message.author.id)))
return
db_record = user.find_login(message)
if db_record:
db_record = db_record[2].split()
year = None
if len(db_record) == 3:
if db_record[0] == "FIT":
db_record[2] = int(db_record[2][:-1])
if db_record[1] == "BIT":
year = "BIT"
if db_record[2] < 4:
year = str(db_record[2]) + year
else:
year = "4BIT+"
elif db_record[1] in [
"MBS", "MBI", "MIS", "MIN", "MMI", "MMM", "MGM",
"MGMe", "MPV", "MSK"
]:
year = "MIT"
if db_record[2] < 3:
year = str(db_record[2]) + year
else:
year = "3MIT+"
elif db_record[1] == "DVI4":
year = "PhD+"
if year is None:
await message.channel.send(
"Čauec {}, nechám {} aby to udělal manuálne\nYear:`{}`".
format(utils.generate_mention(message.author.id),
utils.generate_mention(config.admin_id),
str(db_record)))
return
try:
# get server permit role
verify = discord.utils.get(message.guild.roles,
name=config.verification_role)
year = discord.utils.get(message.guild.roles, name=year)
member = message.author
except AttributeError:
# jsme v PM
guild = client.get_guild(config.guild_id)
verify = discord.utils.get(guild.roles,
name=config.verification_role)
year = discord.utils.get(guild.roles, name=year)
member = guild.get_member(message.author.id)
await member.add_roles(verify)
await member.add_roles(year)
user.save_record(message)
await message.channel.send(
"Gratuluji, byl si verifikován! {}".format(
utils.generate_mention(message.author.id)))
else:
await message.channel.send(
("Login nenalezen nebo jsi j*z " +
"prosel timhle krokem {} {}").format(
utils.generate_mention(message.author.id),
utils.generate_mention(config.admin_id)))
else:
await message.channel.send("Už si byl verifikován {} {}".format(
utils.generate_mention(message.author.id),
utils.generate_mention(config.admin_id)))
try:
await message.delete()
except discord.errors.Forbidden:
return
async def on_message(message):
if message.author == client.user:
return
elif message.content.startswith("!verify"):
await verify(message)
elif message.content.startswith("!getcode"):
await send_code(message)
elif message.content.startswith("!roll"):
await message.channel.send(rng.generate_number(message))
await botroom_check(message)
elif message.content.startswith("!flip"):
await message.channel.send(rng.flip())
await botroom_check(message)
elif message.content.startswith("!week"):
await message.channel.send(rng.week())
elif message.content.startswith("!pick"):
await pick(message)
await botroom_check(message)
elif message.content.startswith("!karma get"):
if not await guild_check(message):
await message.channel.send("{}".format(config.server_warning))
else:
try:
await karma.get(message)
await botroom_check(message)
except discord.errors.Forbidden:
return
elif message.content.startswith("!karma revote"):
if not await guild_check(message):
await message.channel.send("{}".format(config.server_warning))
else:
if message.channel.id == config.vote_room:
try:
await message.delete()
await karma.revote(message, config)
except discord.errors.Forbidden:
return
else:
await message.channel.send("Tohle funguje jen v {}".format(
discord.utils.get(message.guild.channels,
id=config.vote_room)))
elif message.content.startswith("!karma vote"):
if not await guild_check(message):
await message.channel.send("{}".format(config.server_warning))
else:
if message.channel.id == config.vote_room:
try:
await message.delete()
await karma.vote(message, config)
except discord.errors.Forbidden:
return
else:
await message.channel.send("Tohle funguje jen v {}".format(
discord.utils.get(message.guild.channels,
id=config.vote_room)))
elif message.content.startswith("!karma give"):
if message.author.id == config.admin_id:
await karma.karma_give(message)
else:
await message.channel.send(
"{} na použitie tohto príkazu nemáš práva".format(
utils.generate_mention(message.author.id)))
elif message.content.startswith("!karma"):
await show_karma(message)
await botroom_check(message)
elif message.content.startswith("!leaderboard"):
await karma_leaderboard(message, 'DESC')
await botroom_check(message)
elif message.content.startswith("!bajkarboard"):
await karma_leaderboard(message, 'ASC')
await botroom_check(message)
elif message.content.startswith("!god"):
await message.channel.send(config.info())
elif message.content.startswith("!diceroll"):
await message.channel.send(roll_dice.roll_dice(message, config))
elif message.content.startswith(config.role_string):
role_data = await get_join_role_data(message)
await message_role_reactions(message, role_data)