Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2024-12-26 06:57:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add posting status for discordbots.org

Browse Source
This commit is contained in:
Jeremy Zhang 2018-01-05 09:21:25 +00:00
parent 01e1412af1
commit 71097eca8b
3 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
discordbot/config.example.py
View File

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

8
discordbot/titanembeds/bot.py
View File

@ -2,6 +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 collections import deque from collections import deque
import discord import discord
import aiohttp import aiohttp
@ -24,6 +25,8 @@ class Titan(discord.Client):
self.socketio = SocketIOInterface(self, config["redis-uri"]) self.socketio = SocketIOInterface(self, config["redis-uri"])
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
def _cleanup(self): def _cleanup(self):
try: try:
@ -91,6 +94,9 @@ class Titan(discord.Client):
await self.database.remove_unused_guilds(self.servers) await self.database.remove_unused_guilds(self.servers)
else: else:
print("Skipping indexing server due to no-init flag") print("Skipping indexing server due to no-init flag")
self.discordBotsOrg = DiscordBotsOrg(self.user.id, config.get("discord-bots-org-token", None))
await self.discordBotsOrg.post(len(self.servers))
async def on_message(self, message): async def on_message(self, message):
await self.database.push_message(message) await self.database.push_message(message)
@ -128,9 +134,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))
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))
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)

12
discordbot/titanembeds/poststats.py Normal file
View File

@ -0,0 +1,12 @@
import aiohttp
class DiscordBotsOrg(): # https://discordbots.org
def __init__(self, client_id, token):
self.url = "https://discordbots.org/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:
await aioclient.post(self.url, data=payload, headers=headers)