Posting stats for bots.discord.pw

This commit is contained in:
Jeremy Zhang 2018-01-06 09:59:04 +00:00
parent 4ae37fa854
commit ae12b8fa22
3 changed files with 26 additions and 5 deletions

View File

@ -7,5 +7,7 @@ config = {
'discord-bots-org-token': "DiscordBots.org Post Stats Token", 'discord-bots-org-token': "DiscordBots.org Post Stats Token",
'bots-discord-pw-token': "bots.discord.pw Post Stats Token",
'logging-location': "/home/titan/Titan/discordbot/titanbot.log", 'logging-location': "/home/titan/Titan/discordbot/titanbot.log",
} }

View File

@ -2,7 +2,7 @@ from config import config
from titanembeds.database import DatabaseInterface from titanembeds.database import DatabaseInterface
from titanembeds.commands import Commands from titanembeds.commands import Commands
from titanembeds.socketio import SocketIOInterface from titanembeds.socketio import SocketIOInterface
from titanembeds.poststats import DiscordBotsOrg from titanembeds.poststats import DiscordBotsOrg, BotsDiscordPw
from collections import deque from collections import deque
import discord import discord
import aiohttp import aiohttp
@ -27,6 +27,7 @@ class Titan(discord.Client):
self.delete_list = [] # List of msg ids to prevent duplicate delete self.delete_list = [] # List of msg ids to prevent duplicate delete
self.discordBotsOrg = None self.discordBotsOrg = None
self.botsDiscordPw = None
def _cleanup(self): def _cleanup(self):
try: try:
@ -74,7 +75,8 @@ class Titan(discord.Client):
return return
self.discordBotsOrg = DiscordBotsOrg(self.user.id, config.get("discord-bots-org-token", None)) self.discordBotsOrg = DiscordBotsOrg(self.user.id, config.get("discord-bots-org-token", None))
await self.discordBotsOrg.post(len(self.servers)) self.botsDiscordPw = BotsDiscordPw(self.user.id, config.get("bots-discord-pw-token", None))
await self.postStats()
if "no-init" not in sys.argv: if "no-init" not in sys.argv:
for server in self.servers: for server in self.servers:
@ -134,11 +136,11 @@ class Titan(discord.Client):
continue continue
async for message in self.logs_from(channel, limit=50, reverse=True): async for message in self.logs_from(channel, limit=50, reverse=True):
await self.database.push_message(message) await self.database.push_message(message)
await self.discordBotsOrg.post(len(self.servers)) await self.postStats()
async def on_server_remove(self, guild): async def on_server_remove(self, guild):
await self.database.remove_guild(guild) await self.database.remove_guild(guild)
await self.discordBotsOrg.post(len(self.servers)) await self.postStats()
async def on_server_update(self, guildbefore, guildafter): async def on_server_update(self, guildbefore, guildafter):
await self.database.update_guild(guildafter) await self.database.update_guild(guildafter)
@ -240,3 +242,8 @@ class Titan(discord.Client):
if msg.id == msg_id: if msg.id == msg_id:
return True return True
return False return False
async def postStats(self):
count = len(self.servers)
await self.discordBotsOrg.post(count)
await self.botsDiscordPw.post(count)

View File

@ -1,4 +1,5 @@
import aiohttp import aiohttp
import json
class DiscordBotsOrg(): # https://discordbots.org class DiscordBotsOrg(): # https://discordbots.org
def __init__(self, client_id, token): def __init__(self, client_id, token):
@ -9,4 +10,15 @@ class DiscordBotsOrg(): # https://discordbots.org
headers = {"Authorization": self.token} headers = {"Authorization": self.token}
payload = {"server_count": count} payload = {"server_count": count}
async with aiohttp.ClientSession() as aioclient: async with aiohttp.ClientSession() as aioclient:
await aioclient.post(self.url, data=payload, headers=headers) await aioclient.post(self.url, data=payload, headers=headers)
class BotsDiscordPw(): # https://bots.discord.pw/
def __init__(self, client_id, token):
self.url = "https://bots.discord.pw/api/bots/{}/stats".format(client_id)
self.token = token
async def post(self, count):
headers = {"Authorization": self.token}
payload = {"server_count": count}
async with aiohttp.ClientSession() as aioclient:
t = await aioclient.post(self.url, json=payload, headers=headers)