marathonbot/bot.py at main · BriGuy7727/marathonbot · 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
import json
import os
from datetime import datetime
from discord.utils import get
import copy

import discord
import requests
from discord.ext import tasks
import tokens

client = discord.Client(intents=discord.Intents.default())


async def purge_channels():
    def is_me(m):
        return m.author == client.user
    with open('current_marathons.json', 'w') as purgefile:
        purgefile.write(json.dumps({}))
    try:
        guilds = [guild async for guild in client.fetch_guilds()]
        for x in guilds:
            clean_channel = get(client.get_all_channels(), guild=x, name='marathon-alerts')
            await clean_channel.purge(check=is_me)
    except AttributeError:
        print("dang")

def format_time(ts):
    try:
        newtime = datetime.strptime(ts, '%Y-%m-%dT%H:%M:%SZ').strftime("%m/%d/%Y")
        return newtime
    except TypeError:
        return ts


@client.event
async def on_ready():
    print('We have logged in as {0.user}'.format(client))
    await purge_channels()
    sub_messages.start()


@tasks.loop(hours=1)
async def sub_messages():
    if not os.path.exists('current_marathons.json'):
        with open('current_marathons.json', 'w') as newfile:
            newfile.write(json.dumps({}))
    with open('current_marathons.json') as marathon_file:
        marathon_dict = json.load(marathon_file)
    sub_channel = get(client.get_all_channels(), name="marathon-alerts")
    response = requests.get("https://oengus.io/api/v1/marathons")
    x = json.loads(response.text)
    max_subs = len(x['open'])
    sub_num = 0
    check_list = []
    for xx, yy in x.items():
        if xx == "open":
            for z in yy:
                check_list.append(z['id'])
            for y, v in copy.deepcopy(marathon_dict).items():
                if y not in check_list:
                    message = await sub_channel.fetch_message(v['msg_id'])
                    await message.delete()
                    del marathon_dict[y]
            while sub_num < max_subs:
                embed = discord.Embed()
                current = yy[sub_num]
                if current['id'] in marathon_dict:
                    pass
                else:
                    details = requests.get(f"https://oengus.io/api/v1/marathons/{current['id']}")
                    d = json.loads(details.text)
                    if current['onsite']:
                        try:
                            eventloc = ', '.join([current['location'], current['country']])
                        except TypeError:
                            eventloc = "Online"
                    else:
                        eventloc = "Online"
                    embed.title = f"{current['name']}"
                    embed.url = f'https://oengus.io' \
                                f'/marathon/{current["id"]}'
                    if d['description'] and len(d['description']) > 500:
                        embed.description = ''.join([str(d['description'])[0:500], "..."])
                    else:
                        embed.description = d['description']
                    embed.add_field(name='Start Date', value=format_time(current['startDate']))
                    embed.add_field(name="End Date", value=format_time(current['endDate']))
                    embed.add_field(name="Submissions Open Until", value=format_time(current['submissionsEndDate']),
                                    inline=False)
                    embed.add_field(name="Location", value=eventloc)
                    embed.add_field(name="Language", value=current['language'].upper())
                    embed.add_field(name="Max Runners", value=d['maxNumberOfScreens'])
                    if d['emulatorAuthorized']:
                        embed.add_field(name="Emulators Okay?", value="Yes")
                    else:
                        embed.add_field(name="Emulators Okay?", value="No")
                    if d['discordRequired']:
                        embed.add_field(name="Required to Join Discord?",
                                        value=''.join(["Yes\nhttps://discord.gg/", d['discord']]))
                    else:
                        embed.add_field(name="Required to Join Discord?", value="No")
                    embed.colour = discord.Colour.random()
                    sub_msg_id = await sub_channel.send(embed=embed)
                    marathon_dict[current['id']] = {"msg_id": sub_msg_id.id}
                sub_num += 1
    with open('current_marathons.json', 'w') as newfile:
        newfile.write(json.dumps(marathon_dict))


client.run(tokens.DISCORD_TOKEN)