Esempio n. 1
0
 async def award_take(self, message, is_award):
     server, channel, author = message.guild, message.channel, message.author
     split_message = message.content.split(" ")
     mode_text, mode_change = ("awarded", 1) if is_award else ("deducted",
                                                               -1)
     if len(split_message) > 2:
         amount = split_message[1]
         if amount.isdigit():
             amount = int(amount)
             user = Utils.get_user(server, split_message[2])
             if user is not None:
                 EconomyUtils.set_cash(
                     server.id, user.id,
                     EconomyUtils.get_cash(server.id, user.id) +
                     amount * mode_change)
                 await Utils.simple_embed_reply(
                     channel, "[%s]" % str(author),
                     "User `%s` was %s %d%s." %
                     (str(user), mode_text, amount, EconomyUtils.currency))
             else:
                 given_role = Utils.get_role(server,
                                             ''.join(split_message[2]))
                 users = []
                 if given_role is not None:
                     for user in server.members:
                         if given_role in user.roles:
                             EconomyUtils.set_cash(
                                 server.id, user.id,
                                 EconomyUtils.get_cash(server.id, user.id) +
                                 amount * mode_change)
                             users.append(user)
                     if len(users) > 0:
                         await Utils.simple_embed_reply(
                             channel, "[%s]" % str(author),
                             "Users with the role `%s` were %s %d%s." %
                             (str(given_role), mode_text, amount,
                              EconomyUtils.currency))
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]",
                             "No users are equipped with that role.")
                 else:
                     await Utils.simple_embed_reply(
                         channel, "[Error]",
                         "Invalid user or role supplied.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Amount parameter is incorrect.")
     else:
         await Utils.simple_embed_reply(
             channel, "[Error]", "Insufficient parameters supplied.")
Esempio n. 2
0
 async def command_called(self, message, command):
     split_message = message.content.split(" ")
     server, channel, author = message.guild, message.channel, message.author
     if command is self.commands["Set Shop Command"]:
         if len(split_message) > 1:
             shop_name = split_message[1]
             if is_valid_shop_name(shop_name):
                 # Drop old shop table if needed
                 self.delete_shop_by_channel_id(channel.id)
                 # Create new tables and rows
                 self.shops_database.execute(
                     """CREATE TABLE IF NOT EXISTS '%s'(role_id TEXT UNIQUE, 
                     price NUMERIC, time_added REAL, duration REAL)""" %
                     shop_name)
                 self.shop_info_database.execute(
                     "REPLACE INTO shops VALUES('%s', '%s', 0)" %
                     (channel.id, shop_name))
                 await Utils.simple_embed_reply(
                     channel, "[Shop Created]",
                     "`%s` has been assigned `%s.`" %
                     (str(channel), shop_name))
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop parameter incorrect.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["List Shops Command"]:
         shop_names = []
         info = self.shop_info_database.execute(
             "SELECT channel_id, shop_name FROM shops")
         known_channels = [channel.id for channel in server.channels]
         for i in range(0, len(info), 2):
             if info[i] in known_channels:
                 shop_names.append(info[i + 1])
         if len(shop_names) > 0:
             shop_text = ''.join(
                 [shop_name + "\n" for shop_name in shop_names])[:-1]
             await Utils.simple_embed_reply(channel, "[Shops]", shop_text)
         else:
             await Utils.simple_embed_reply(channel, "[Shops]",
                                            "No shops exist.")
     elif command is self.commands["Delete Shop Command"]:
         if len(split_message) > 1:
             shop_name = split_message[1]
             if is_valid_shop_name(shop_name):
                 shop_names = self.shop_info_database.execute(
                     "SELECT shop_name FROM shops")
                 if len(shop_names) > 0:
                     if shop_name in shop_names:
                         self.shops_database.execute(
                             "DROP TABLE IF EXISTS '%s'" % shop_name)
                         self.shop_info_database.execute(
                             "DELETE FROM shops WHERE shop_name='%s'" %
                             shop_name)
                         self.shop_info_database.execute(
                             "DELETE FROM messages WHERE shop_name='%s'" %
                             shop_name)
                         await Utils.simple_embed_reply(
                             channel, "[Shops]",
                             "Shop `%s` was deleted." % shop_name)
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]", "That shop doesn't exist.")
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "No shops exist.")
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop parameter incorrect.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["Set Shop Role Command"]:
         if len(split_message) > 4:
             shop_name, price, duration, role_text = split_message[1], split_message[2].lower(), split_message[3], \
                                                     split_message[4]
             if is_valid_shop_name(shop_name):
                 if self.shop_exists(shop_name):
                     if Utils.isfloat(duration) or duration == "permanent":
                         duration = -1 if duration == "permanent" or 0 else float(
                             duration)
                         if price.isdigit():
                             price = int(price)
                             role = Utils.get_role(server, role_text)
                             if role is not None:
                                 role_ids = [
                                     name.lower()
                                     for name in self.shops_database.
                                     execute("SELECT role_id FROM '%s'" %
                                             shop_name)
                                 ]
                                 role_names = [
                                     str(
                                         Utils.get_role_by_id(
                                             server, role_id))
                                     for role_id in role_ids
                                 ]
                                 if str(role).lower() not in role_names:
                                     self.shops_database.execute(
                                         "INSERT OR IGNORE INTO '%s' VALUES('%s', '%d', '%s', '%s')"
                                         % (shop_name, role.id, int(price),
                                            time.time(), duration))
                                     await Utils.simple_embed_reply(
                                         channel, "[Shops]",
                                         "`%s` has been assigned to `%s` at the price of `%s` for `%s` hours."
                                         %
                                         (str(role), shop_name, str(price),
                                          str("infinite" if duration ==
                                              -1 else duration)))
                                     await self.update_messages()
                                 else:
                                     if role.id in role_ids:
                                         self.shops_database.execute(
                                             "REPLACE INTO '%s' VALUES('%s', '%d', '%s', '%s')"
                                             % (shop_name, role.id,
                                                int(price), time.time(),
                                                duration))
                                         await Utils.simple_embed_reply(
                                             channel, "[Shops]",
                                             "The role `%s` within `%s` now has a price of `%s` for `%s` hours."
                                             %
                                             (str(role), shop_name,
                                              str(price),
                                              str("infinite" if duration ==
                                                  -1 else duration)))
                                         await self.update_messages()
                                     else:
                                         await Utils.simple_embed_reply(
                                             channel, "[Error]",
                                             "Duplicate role names not allowed. (lowercase-checked)"
                                         )
                             else:
                                 await Utils.simple_embed_reply(
                                     channel, "[Error]",
                                     "That role doesn't exist.")
                         else:
                             await Utils.simple_embed_reply(
                                 channel, "[Error]",
                                 "Price parameter incorrect.")
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]",
                             "Duration parameter incorrect.")
                 else:
                     await Utils.simple_embed_reply(
                         channel, "[Error]", "That shop doesn't exist.")
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop parameter incorrect.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["Toggle Shop Autodelete Command"]:
         if len(split_message) > 1:
             shop_name = split_message[1]
             if is_valid_shop_name(shop_name):
                 if self.shop_exists(shop_name):
                     new_value = int(
                         self.shop_info_database.execute(
                             "SELECT is_purge FROM shops WHERE shop_name='%s'"
                             % shop_name)[0]) ^ 1
                     self.shop_info_database.execute(
                         "UPDATE shops SET is_purge='%d' WHERE shop_name='%s'"
                         % (new_value, shop_name))
                     await Utils.simple_embed_reply(
                         channel, "[Shops]",
                         "`%s`'s delete mode set to `%r`" %
                         (shop_name, bool(new_value)))
                 else:
                     await Utils.simple_embed_reply(
                         channel, "[Error]", "That shop doesn't exist.")
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop parameter incorrect.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["Shop Command"]:
         if len(split_message) > 1:
             shop_name = split_message[1]
             if self.shop_exists(shop_name):
                 embed = await self.get_shop_embed(shop_name, server)
                 shop_message = await channel.send(embed=embed)
                 self.shop_info_database.execute(
                     "REPLACE INTO messages VALUES('%s', '%s', '%s')" %
                     (shop_name, shop_message.id, channel.id))
             else:
                 await Utils.simple_embed_reply(channel, "[Error]",
                                                "That shop doesn't exist.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["Buy Command"]:
         if len(split_message) > 1:
             given_name = ' '.join(split_message[1:]).lower()
             shop = self.shop_info_database.execute(
                 "SELECT shop_name FROM shops where channel_id='%s'" %
                 message.channel.id)
             if len(shop) > 0:
                 shop = shop[0]
                 role_ids = self.shops_database.execute(
                     "SELECT role_id FROM '%s'" % shop)
                 role_costs = self.shops_database.execute(
                     "SELECT price FROM '%s'" % shop)
                 if len(role_ids) > 0:
                     role_names = [
                         str(Utils.get_role_by_id(server, role_id))
                         for role_id in role_ids
                     ]
                     if given_name in [
                             str(name).lower() for name in role_names
                     ]:
                         for i in range(len(role_ids)):
                             role_id, role_name, role_cost = role_ids[
                                 i], role_names[i], role_costs[i]
                             if role_name.lower() == given_name:
                                 user_cash = EconomyUtils.get_cash(
                                     server.id, author.id)
                                 if user_cash >= role_cost:
                                     EconomyUtils.set_cash(
                                         server.id, author.id,
                                         user_cash - role_cost)
                                     role = Utils.get_role_by_id(
                                         server, role_id)
                                     if role not in author.roles:
                                         await Utils.client.add_roles(
                                             author, role)
                                         await Utils.simple_embed_reply(
                                             channel, "[%s]" % shop,
                                             "You have purchased `%s`." %
                                             role_name)
                                     else:
                                         await Utils.simple_embed_reply(
                                             channel, "[Error]",
                                             "You already have that role.")
                                 else:
                                     await Utils.simple_embed_reply(
                                         channel, "[Error]",
                                         "You don't have enough cash to do that."
                                     )
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]", "Role not found.")
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "No roles found.")
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop not found for this channel.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
     elif command is self.commands["Delete Shop Role Command"]:
         if len(split_message) > 1:
             given_name = ' '.join(split_message[1:]).lower()
             shop = self.shop_info_database.execute(
                 "SELECT shop_name FROM shops where channel_id='%s'" %
                 channel.id)
             if len(shop) > 0:
                 shop = shop[0]
                 role_ids = self.shops_database.execute(
                     "SELECT role_id FROM '%s'" % shop)
                 if len(role_ids) > 0:
                     role_names = [
                         str(Utils.get_role_by_id(server, role_id))
                         for role_id in role_ids
                     ]
                     if given_name in [
                             str(name).lower() for name in role_names
                     ]:
                         for i in range(len(role_ids)):
                             self.shops_database.execute(
                                 "DELETE FROM '%s' WHERE role_name='%s'" %
                                 (shop, role_names[i]))
                             await Utils.simple_embed_reply(
                                 channel, "[%s]" % shop,
                                 "Role `%s` has been deleted from `%s`." %
                                 (role_names[i], shop))
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]", "Role not found.")
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "No roles found.")
             else:
                 await Utils.simple_embed_reply(
                     channel, "[Error]", "Shop not found for this channel.")
         else:
             await Utils.simple_embed_reply(
                 channel, "[Error]", "Insufficient parameters supplied.")
Esempio n. 3
0
 async def command_called(self, message, command):
     # Slip the message on space to help parse the command
     split_message = message.content.split(" ")
     # Extract the server, channel and author from the message
     server, channel, author = message.guild, message.channel, message.author
     if command is self.commands["I Am Command"]:
         if len(split_message) > 1:
             given_name = split_message[1].lower()
             if is_valid_name(given_name):
                 role_id = self.roles_db.execute(
                     "SELECT role_id FROM '%s' where name='%s'" %
                     (server.id, given_name))
                 if len(role_id) > 0:
                     role = Utils.get_role_by_id(server, role_id[0])
                     if role is not None:
                         await Utils.client.add_roles(author, role)
                         await Utils.simple_embed_reply(
                             channel, "[SelfRoles]",
                             "You now have the `%s` role!" % given_name)
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]", "Role not found.")
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "Role not found.")
             else:
                 await Utils.simple_embed_reply(channel, "[Error]",
                                                "Invalid role supplied.")
         else:
             await Utils.simple_embed_reply(channel, "[Error]",
                                            "Invalid parameters supplied.")
     elif command is self.commands["I Am Not Command"]:
         if len(split_message) > 1:
             given_name = split_message[1].lower()
             if is_valid_name(given_name):
                 role_id = self.roles_db.execute(
                     "SELECT role_id FROM '%s' where name='%s'" %
                     (server.id, given_name))
                 if len(role_id) > 0:
                     role = Utils.get_role_by_id(server, role_id[0])
                     if role is not None:
                         await Utils.client.remove_roles(author, role)
                         await Utils.simple_embed_reply(
                             channel, "[SelfRoles]",
                             "You no longer have the `%s` role!" %
                             given_name)
                     else:
                         await Utils.simple_embed_reply(
                             channel, "[Error]", "Role not found.")
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "Role not found.")
             else:
                 await Utils.simple_embed_reply(channel, "[Error]",
                                                "Invalid role supplied.")
         else:
             await Utils.simple_embed_reply(channel, "[Error]",
                                            "Invalid parameters supplied.")
     elif command is self.commands["Add Self Role Command"]:
         if len(split_message) > 2:
             role_name = split_message[1].lower()
             if is_valid_name(role_name):
                 role = Utils.get_role(server, split_message[2])
                 if role is not None:
                     self.roles_db.execute(
                         "INSERT OR IGNORE INTO '%s' VALUES ('%s', '%s')" %
                         (server.id, role.id, role_name))
                     await Utils.simple_embed_reply(
                         channel, "[SelfRoles]",
                         "Role `%s` was added as `%s`." %
                         (role.name, role_name))
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "Role not found.")
             else:
                 await Utils.simple_embed_reply(channel, "[Error]",
                                                "Invalid name supplied.")
         else:
             await Utils.simple_embed_reply(channel, "[Error]",
                                            "Invalid parameters supplied.")
     elif command is self.commands["Delete Self Role Command"]:
         if len(split_message) > 1:
             role_name = split_message[1].lower()
             if is_valid_name(role_name):
                 if len(
                         self.roles_db.execute(
                             "SELECT name FROM `%s` WHERE name='%s'" %
                             (server.id, role_name))) > 0:
                     self.roles_db.execute(
                         "DELETE FROM '%s' where name='%s'" %
                         (server.id, role_name))
                     await Utils.simple_embed_reply(
                         channel, "[SelfRoles]",
                         "Role `%s` was deleted." % role_name)
                 else:
                     await Utils.simple_embed_reply(channel, "[Error]",
                                                    "Role not found.")
             else:
                 await Utils.simple_embed_reply(channel, "[Error]",
                                                "Invalid name supplied.")
         else:
             await Utils.simple_embed_reply(channel, "[Error]",
                                            "Invalid parameters supplied.")
     elif command is self.commands["List Self Roles Command"]:
         role_names = self.roles_db.execute("SELECT name from '%s'" %
                                            server.id)
         if len(role_names) > 0:
             roles_string = ''.join([i + "\n" for i in role_names])
             # TODO: Add role listing cap (to not break the embed max)
             # List the roles
             await channel.send(
                 embed=discord.Embed(title="[Self Roles]",
                                     description=roles_string[0:-1],
                                     color=self.embed_color))
         else:
             await Utils.simple_embed_reply(
                 channel, "[Self Roles]",
                 "There are currently no self roles.")