VexBot/VexBot.py at master · Trainmaster2/VexBot · 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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
import asyncio,discord,os
from discord.ext.commands import Bot
from requests import get
from urllib.parse import urlencode as ue
from datetime import datetime as dt

Client = Bot('=')
token='BOT TOKEN HERE'
prefix='-'
matchDict={1:'P',2:'Q',3:'QF',4:'SF',5:'F'}

def teamEmbed(data):
    embed=discord.Embed(title=data['number'],description=data['team_name'],color=0xd9272e)
    embed.add_field(name='City',value=data['city'],inline=True)
    embed.add_field(name='Region',value=data['region'],inline=True)
    embed.add_field(name='Country',value=data['country'],inline=True)
    embed.add_field(name='Organisation',value=data['organisation'],inline=True)
    embed.add_field(name='Grade',value=data['grade'],inline=True)
    return embed

def rankEmbed(data):
    event=eval(get('https://api.vexdb.io/v1/get_events?sku='+data['sku']).content)['result'][0]['name']
    embed=discord.Embed(title=data['team'],description=event+' - '+data['division'],color=0xd9272e)
    embed.add_field(name='Rank',value=data['rank'],inline=False)
    embed.add_field(name='Wins',value=data['wins'],inline=True)
    embed.add_field(name='Losses',value=data['losses'],inline=True)
    embed.add_field(name='Ties',value=data['ties'],inline=True)
    embed.add_field(name='WP',value=data['wp'],inline=True)
    embed.add_field(name='AP',value=data['ap'],inline=True)
    embed.add_field(name='SP',value=data['sp'],inline=True)
    return embed

def matchEmbed(data):
    event=eval(get('https://api.vexdb.io/v1/get_events?sku='+data['sku']).content)['result'][0]['name']
    d=dt.strptime(data['scheduled'],'%Y-%m-%dT%H:%M:%S+00:00')
    date=d.strftime('%I:%M %p UTC on %b. %d, %Y')
    date='{}:{} {} on {}-{}-{}'.format(dt[3],dt[4],d.tzname(),dt[1],dt[2],dt[0])
    if data['round']>2:
        match='{} {}-{}'.format(matchDict[data['round']],data['instance'],data['matchnum'])
    else:
        match='{} {}'.format(matchDict[data['round']],data['matchnum'])
    teams=[[data['red1'],data['red2'],data['red3']],[data['blue1'],data['blue2'],data['blue3']]]
    if not teams[0][2]: teams[0][2]='N/A'
    if not teams[1][2]: teams[1][2]='N/A'
    embed=discord.Embed(title=event+' - '+data['division'],description=match,color=0xd9272e)
    embed.add_field(name='Red 1',value=teams[0][0],inline=True)
    embed.add_field(name='Red 2',value=teams[0][1],inline=True)
    embed.add_field(name='Red 3',value=teams[0][2],inline=True)
    if data['scored']:embed.add_field(name='Red Score',value=data['redscore'],inline=False)
    embed.add_field(name='Blue 1',value=teams[1][0],inline=True)
    embed.add_field(name='Blue 2',value=teams[1][1],inline=True)
    embed.add_field(name='Blue 3',value=teams[1][2],inline=True)
    if data['scored']:embed.add_field(name='Blue Score',value=data['bluescore'],inline=False)
    embed.add_field(name='Field',value=data['field'],inline=True)
    embed.add_field(name='Time',value=date,inline=True)
    return embed

def teamsEmbed(data,sku):
    event=eval(get('https://api.vexdb.io/v1/get_events?sku='+sku).content)['result'][0]['name']
    embeds=[]
    embed=discord.Embed(title='Team List',description=event,color=0xd9272e)
    j=0
    for i in data:
        if j==25:
            embeds.append(embed.to_dict())
            j=0
            embed=discord.Embed(title=i['number'],description=i['team_name'],color=0xd9272e)
        else:
            embed.add_field(name=i['number'],value=i['team_name'],inline=False)
            j+=1
    embeds.append(embed.to_dict())
    return embeds

def awardEmbed(data):
    event=eval(get('https://api.vexdb.io/v1/get_events?sku='+data['sku']).content)['result'][0]['name']
    embed=discord.Embed(title=data['name'],description=event,color=0xd9272e)
    embed.add_field(name='Team',value=data['team'],inline=True)
    return embed

@Client.event
async def on_ready():
    print('Invite link: https://discordapp.com/oauth2/authorize?client_id={}&scope=bot&permissions=8\n'.format(Client.user.id))
    embed=discord.Embed(title='Commands List',description='Make sure that "Allow DMs from server members" is enabled.',color=0xd9272e)
    embed.add_field(name=prefix+'team [team #]', value='Returns information about a team.', inline=False)
    embed.add_field(name=prefix+'rank [team #]', value='Returns rank information about a team in a competition.', inline=False)
    embed.add_field(name=prefix+'matches [team #]', value='Returns a list of matches for a team in a competition.', inline=False)
    embed.add_field(name=prefix+'teams', value='Returns a list of teams in a competition.', inline=False)
    embed.add_field(name=prefix+'match (team #)', value='Returns the current match in a competition. Optionally limit to a team.', inline=False)
    embed.add_field(name=prefix+'scored (team #)', value='Returns the most recent scored match for a competition. Optionally limit to a team.', inline=False)
    embed.add_field(name=prefix+'awards',value='Returns a list of all given awards.',inline=False)
    embed.set_footer(text='Make sure that you enter commands in the appropriate channel for your competition.')
    comps=eval(get('https://api.vexdb.io/v1/get_events?status=current&country=United%20States').content)['result']
    info=discord.Embed(title='Welcome to VexBot',description='{} {} currently avaliable.'.format(len(comps),'competitions-'[:-((len(comps)==1)+1)]),color=0xd9272e)
    feedback=discord.Embed(title='Feedback',description='Please leave feedback to help improve this bot.',color=0xd9272e)
    feedback.add_field(name='Reddit',value='https://www.reddit.com/r/vex/comments/846262/vexbot_discord_server/',inline=False)
    feedback.add_field(name='VEX Forum',value='https://www.vexforum.com/index.php/32558-vexbot-discord-server',inline=False)
    feedback.add_field(name='Github',value='https://github.com/Trainmaster2/VexBot',inline=False)
    for server in Client.servers:
        everyone=server.default_role
        perms=everyone.permissions
        perms.send_messages=False
        Client.edit_role(server,everyone,permissions=perms)
        infoFound=False
        for i in list(server.channels).copy():
            if i.name=='info':
              await Client.purge_from(i)
              await Client.send_message(i,embed=info)
              await Client.send_message(i,embed=feedback)
              infoFound=True
              continue
            await Client.delete_channel(i)
        if not infoFound:
            channel=await Client.create_channel(server, 'Info', (everyone,discord.PermissionOverwrite(send_messages=False)))
            await Client.send_message(channel,embed=info)
            await Client.send_message(channel,embed=feedback)
        for i in comps:
            channel = await Client.create_channel(server,i['name'])
            await Client.edit_channel(channel,topic=i['sku'])
            await Client.send_message(channel,embed=embed)
        perms.send_messages=True
        Client.edit_role(server,everyone,permissions=perms)

@Client.event
async def on_message(msg):
    if msg.server and msg.author.id!='422439927825825793':
        await Client.delete_message(msg)
        if msg.content.startswith(prefix):
            cmd=msg.content[len(prefix):].split()
            sku=msg.channel.topic
            if cmd[0]=='team':
                data=eval(get('https://api.vexdb.io/v1/get_teams?team='+cmd[1]).content)['result']
                if len(data)==1:
                    await Client.send_message(msg.author,embed=teamEmbed(data[0]))
                else:
                    await Client.send_message(msg.author,content='Could not find Team '+cmd[1].upper()+'.')
            if cmd[0]=='rank':
                data=eval(get('https://api.vexdb.io/v1/get_rankings?'+ue((('team',cmd[1]),('sku',sku)))).content)['result']
                if len(data)==1:
                    await Client.send_message(msg.author,embed=rankEmbed(data[0]))
                else:
                    await Client.send_message(msg.author,content='Could not find Team '+cmd[1].upper()+'.')
            if cmd[0]=='matches':
                data=eval(get('https://api.vexdb.io/v1/get_matches?'+ue((('team',cmd[1]),('sku',sku)))).content)['result']
                if len(data)>0:
                    for i in data:
                        await Client.send_message(msg.author,embed=matchEmbed(i))
                else:
                    await Client.send_message(msg.author,content='No matches found for Team '+cmd[1].upper()+'.')
            if cmd[0]=='match':
                cmd.append('')
                data=eval(get('https://api.vexdb.io/v1/get_matches?scored=0&'+ue((('sku',sku),('team',cmd[1])))).content)['result']
                if len(data)>0:
                    await Client.send_message(msg.author,embed=matchEmbed(data[0]))
                else:
                    await Client.send_message(msg.author,content='No upcoming matches found.')
            if cmd[0]=='scored':
                cmd.append('')
                data=eval(get('https://api.vexdb.io/v1/get_matches?scored=1&'+ue((('sku',sku),('team',cmd[1])))).content)['result']
                if len(data)>0:
                    await Client.send_message(msg.author,embed=matchEmbed(data[-1]))
                else:
                    await Client.send_message(msg.author,content='No scored matches found.')
            if cmd[0]=='teams':
                data=eval(get('https://api.vexdb.io/v1/get_teams?sku='+sku).content)['result']
                if len(data)>0:
                    for i in teamsEmbed(data,sku):
                        await Client.send_message(msg.author,embed=discord.Embed.from_data(i))
                else:
                    await Client.send_message(msg.author,content='No teams found.')
            if cmd[0]=='awards':
                data=eval(get('https://api.vexdb.io/v1/get_awards?sku='+sku).content)['result']
                if len(data)>0:
                    for i in data:
                        await Client.send_message(msg.author,embed=awardEmbed(i))
                else:
                    await Client.send_message(msg.author,content='No awards found for this competition.')

Client.run(token)