autoSW/flowers.py at master · Rug-Dealer/autoSW · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import discord
from discord.ext import commands
import random
import json
import asyncio
import operator
from utils import checks


def flower_db_edit(user, json_data, count):
    user_id = str(user.id)
    json_data[user_id] = int(count)
    with open('flowers.json', 'w') as json_file:
        json_file.write(json.dumps(json_data, indent=2))


async def do_flower(message, bot,json_data, reward=1, flower_type='flower'):
    flower_embed = discord.Embed(description='A {} has appeared! '
                                             'Type {}pick to pick it!'.format(flower_type, bot.command_prefix))
    flower_embed.set_image(url='https://i.imgur.com/1pIkDl2.jpg')
    flower_message = await bot.send_message(message.channel, embed=flower_embed)
    response = await bot.wait_for_message(timeout=30, content='{}pick'.format(bot.command_prefix))
    if response is None:
        wilt_message = await bot.send_message(message.channel, 'The flower wilted!')
        await asyncio.sleep(5)
        await bot.delete_message(flower_message)
        await bot.delete_message(wilt_message)
    else:
        flower_db_edit(response.author, json_data, json_data.get(response.author.id, 0) + 1)
        pick_message = await bot.send_message(message.channel,
                                              "{} has picked the {}! They now have {} flowers!"
                                              .format(response.author.display_name,
                                                      flower_type,
                                                      json_data.get(response.author.id, 0)))
        await asyncio.sleep(10)

        def check(message):
            return '{}pick'.format(bot.command_prefix) in message.content

        await bot.purge_from(message.channel, after=flower_message, check=check)
        await bot.delete_message(flower_message)
        await bot.delete_message(pick_message)


class Flowers():
    def __init__(self, bot):
        with open('flowers.json') as json_file:
            json_data = json.load(json_file)
        self.bot = bot
        self.json_data = json_data

    async def on_message(self, message):
        if message.author == message.server.me:
            return
        if self.bot.command_prefix + 'pick' in message.content:
            return
        if message.channel.name == 'shitposting':
            if random.randint(0, 100) == 1:
                await do_flower(message, self.json_data, self.bot)
        elif random.randint(0, 2000) == 1:
            await do_flower(message, self.json_data, self.bot, 10)

    @commands.command(pass_context=True, name='flowers')
    async def _flowers(self, ctx):
        """Gets the flower leaderboards, if users are mentioned in the command, gets their stats specifically"""
        with open('flowers.json') as json_file:
            json_data = json.load(json_file)
        if not ctx.message.mentions:
            flower_board_message = ['Flower leaderboards!\n```']

            sorted_json = sorted(json_data.items(), key=operator.itemgetter(1), reverse=True)

            for data in sorted_json[:10]:
                msg_to_append = '{} : {}'.format(discord.utils.get(ctx.message.server.members, id=data[0]), data[1])
                flower_board_message.append(msg_to_append)

            flower_board_message.append('```')

            flower_board_message = '\n'.join(flower_board_message)
            await self.bot.say(flower_board_message)

        else:
            message_list = []
            for member in ctx.message.mentions[:10]:
                try:
                    message_list.append('{} has {} flowers!'.format(member.display_name, json_data[member.id]))
                except KeyError:
                    message_list.append('{} has no flowers!'.format(member.display_name))
            await self.bot.say('\n'.join(message_list))

    @commands.command()
    @checks.mod_or_permissions(manage_roles=True)
    async def editflowers(self, user: discord.Member, count):
        """Admin command to set the flowers of a specific user"""
        json_data = self.json_data
        user_id = str(user.id)
        try:
            await self.bot.say(
                '{} had {} flowers, now they have {}'.format(user.display_name, json_data.get(user.id, 0), count))
            json_data[user_id] = int(count)
        except KeyError:
            return
        except ValueError:
            await self.bot.say('{} is not an integer!'.format(count))
        flower_db_edit(user,json_data,count)

    @commands.command(pass_context=True)
    @checks.is_owner()
    async def makeflower(self,ctx):
        await do_flower(ctx.message, self.bot, self.json_data)


def setup(bot):
    bot.add_cog(Flowers(bot))