From 71097eca8bd23fa93329cf58b4484438b8078bdc Mon Sep 17 00:00:00 2001
From: Jeremy Zhang <jkzhang@uw.edu>
Date: Fri, 5 Jan 2018 09:21:25 +0000
Subject: [PATCH] Add posting status for discordbots.org

---
 discordbot/config.example.py        |  2 ++
 discordbot/titanembeds/bot.py       |  8 ++++++++
 discordbot/titanembeds/poststats.py | 12 ++++++++++++
 3 files changed, 22 insertions(+)
 create mode 100644 discordbot/titanembeds/poststats.py

diff --git a/discordbot/config.example.py b/discordbot/config.example.py
index b600ae4..228c259 100644
--- a/discordbot/config.example.py
+++ b/discordbot/config.example.py
@@ -5,5 +5,7 @@ config = {
     
     'redis-uri': "redis://",
     
+    'discord-bots-org-token': "DiscordBots.org Post Stats Token",
+    
     'logging-location': "/home/titan/Titan/discordbot/titanbot.log",
 }
diff --git a/discordbot/titanembeds/bot.py b/discordbot/titanembeds/bot.py
index 0624674..e54fc9f 100644
--- a/discordbot/titanembeds/bot.py
+++ b/discordbot/titanembeds/bot.py
@@ -2,6 +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 collections import deque
 import discord
 import aiohttp
@@ -24,6 +25,8 @@ class Titan(discord.Client):
         self.socketio = SocketIOInterface(self, config["redis-uri"])
         
         self.delete_list = [] # List of msg ids to prevent duplicate delete
+        
+        self.discordBotsOrg = None
 
     def _cleanup(self):
         try:
@@ -91,6 +94,9 @@ class Titan(discord.Client):
             await self.database.remove_unused_guilds(self.servers)
         else:
             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):
         await self.database.push_message(message)
@@ -128,9 +134,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))
 
     async def on_server_remove(self, guild):
         await self.database.remove_guild(guild)
+        await self.discordBotsOrg.post(len(self.servers))
 
     async def on_server_update(self, guildbefore, guildafter):
         await self.database.update_guild(guildafter)
diff --git a/discordbot/titanembeds/poststats.py b/discordbot/titanembeds/poststats.py
new file mode 100644
index 0000000..50568e3
--- /dev/null
+++ b/discordbot/titanembeds/poststats.py
@@ -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)
\ No newline at end of file