Save requirements.

This commit is contained in:
Danila Fedorin 2020-12-03 03:21:00 -08:00
revīzija eb0f3cec07
2 mainīti faili ar 104 papildinājumiem un 0 dzēšanām

95
bot.py Normal file
Parādīt failu

@ -0,0 +1,95 @@
import discord
from discord.ext import tasks
import os
import sqlite3
from aiohttp import ClientSession
import json
client = discord.Client()
conn = sqlite3.connect('data.db')
def setup():
conn.execute('''
CREATE TABLE IF NOT EXISTS
guild(id int primary key, channel int, leaderboard int, last_sync text)
''')
@client.event
async def on_ready():
print('We have logged in as {0.user}'.format(client))
@client.event
async def on_message(message):
if message.content.startswith("!setup"):
leaderboard = int(message.content[7:])
conn.execute('''
insert into guild(id, channel, leaderboard, last_sync)
values (?, ?, ?, ?) on conflict(id) do
update set channel=excluded.channel, leaderboard=excluded.leaderboard''',
(message.guild.id, message.channel.id, leaderboard, None))
conn.commit()
await message.channel.send(
'Updated {0.guild} to use channel {0.channel} and leaderboard {1}!'
.format(message, leaderboard))
def gather_stars(member):
star_list = []
for day in member['completion_day_level']:
stars = member['completion_day_level'][day]
if "1" in stars:
star_list.append((member['name'], int(day), 1, stars['1']['get_star_ts']))
if "2" in stars:
star_list.append((member['name'], int(day), 2, stars["2"]['get_star_ts']))
return star_list
def perform_member_diff(old_member, new_member):
old_stars = gather_stars(old_member)
new_stars = gather_stars(new_member)
return [star for star in new_stars if star not in old_stars]
def perform_diff(old_json, new_json):
if old_json is None:
old_json = {'members': {}}
else:
old_json = json.loads(old_json)
new_json = json.loads(new_json)
diff = { 'join': [], 'new_stars': [] }
for new_member in new_json['members'].keys():
if new_member not in old_json['members']:
diff['join'].append(new_json['members'][new_member]['name'])
continue
member_diff = perform_member_diff(old_json['members'][new_member], new_json['members'][new_member])
diff['new_stars'] += member_diff
return diff
async def send_diff(id, channel, diff):
announcements = []
for user in diff['join']:
full_text = "User {0} joined the leaderboard!".format(user)
announcements.append(full_text)
for (user, day, star, ts) in diff['new_stars']:
star_name = "silver" if star == 1 else "gold"
full_text = "{0} won the {1} star from day {2}!".format(user, star_name, day)
announcements.append(full_text)
if len(announcements) != 0:
channel = await client.fetch_channel(channel)
await channel.send("\n".join(announcements))
@tasks.loop(minutes=1.0)
async def update_aoc():
url_string = "https://adventofcode.com/2020/leaderboard/private/view/{0}.json"
cookies = { 'session': os.getenv("AOC_SESSION") }
async with ClientSession(cookies=cookies) as session:
for (id, channel, leaderboard, json) in conn.execute('select * from guild'):
async with session.get(url_string.format(leaderboard)) as resp:
new_json = await resp.text()
diff = perform_diff(json, new_json)
await send_diff(id, channel, diff)
conn.execute('update guild set last_sync=? where id=?', (new_json, id))
conn.commit()
setup()
update_aoc.start()
client.run(os.environ['AOC_BOT_TOKEN'])

9
required.txt Normal file
Parādīt failu

@ -0,0 +1,9 @@
aiohttp==3.6.3
async-timeout==3.0.1
attrs==20.3.0
chardet==3.0.4
discord==1.0.1
discord.py==1.5.1
idna==2.10
multidict==4.7.6
yarl==1.5.1