Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2025-11-04 07:47:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

Posting stats for bots.discord.pw

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

15
discordbot/titanembeds/bot.py
View File

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

14
discordbot/titanembeds/poststats.py
View File

@@ -1,4 +1,5 @@
import aiohttp
import json
class DiscordBotsOrg(): # https://discordbots.org
def __init__(self, client_id, token):
@@ -9,4 +10,15 @@ class DiscordBotsOrg(): # https://discordbots.org
headers = {"Authorization": self.token}
payload = {"server_count": count}
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)