Adding "Users Served" counter. Resolves #2

bot.py

@@ -3,6 +3,8 @@
 import fate
 from setting_storage import settings_from_user_id, target_settings_from_user_id
 import setting_storage
+import stats_storage
+import stats
 import settings
 import json
 import help
@@ -17,6 +19,7 @@ async def on_ready():
     print(client.user.id)
     print('------')
     setting_storage.load_settings()
+    stats_storage.load_stats()
 
 @client.event
 async def on_message(message):
@@ -43,6 +46,9 @@ async def on_message(message):
     if message.content.startswith('!tviewsettings'):
         await settings.handle_view_settings(client, message, message.author.name, message.channel, target_settings_from_user_id(message.author))
 
+    if message.content.startswith('!stats'):
+        await stats.handle_view_stats(client, message, message.author.name, message.channel)
+
     if message.content.startswith('!save'):
         await setting_storage.save_settings()
 
@@ -54,6 +60,7 @@ async def autosave_task():
     while not client.is_closed:
         await asyncio.sleep(300)
         await setting_storage.save_settings()
+        await stats_storage.save_stats()
 
 with open('.connections.json') as json_data:
     connections = json.load(json_data)

fate.py

@@ -4,6 +4,7 @@
 import asyncio
 import random
 import json
+import stats_storage
 
 delay = 0.9
 
@@ -14,6 +15,7 @@ async def handle_fate(client, message, name, channel, settings):
     for string in get_fate_strings(settings):
         await client.send_message(channel, string)
         await asyncio.sleep(delay)
+    stats_storage.increment_fate()
 
 async def handle_target_fate(client, message, name, channel, settings):
     words = message.content.split(None, 1)
@@ -36,6 +38,8 @@ async def handle_target_fate(client, message, name, channel, settings):
     for string in get_fate_strings(settings):
         await client.send_message(channel, string)
         await asyncio.sleep(delay)
+    stats_storage.increment_fate()
+
 
 async def handle_whisper_fate(client, message, name, channel, settings):
     words = message.content.split(None, 1)
@@ -64,6 +68,7 @@ async def handle_whisper_fate(client, message, name, channel, settings):
         target_fate_string = target_fate_string + string + "\n"
 
     await client.send_message(message.author, target_fate_string)    
+    stats_storage.increment_fate()
 
 def get_fate_strings(settings: Settings) -> list:
     strings = []

setting_storage.py

@@ -25,6 +25,12 @@ def find_settings_from_name(name) -> Settings:
             return settings
     return None
 
+def get_users_count():
+    return len(settings_dict.keys())
+
+def get_helpless_users_count():
+    return len([item for item in settings_dict.values() if item.helpless])
+
 async def save_settings():
     output = open('settings.pkl', 'wb')
 

stats.py

@@ -0,0 +1,10 @@
+import stats_storage
+import setting_storage
+
+async def handle_view_stats (client, message, name, channel):
+    stats_string = f"""
+**Fates Served:** {stats_storage.stats["fates_delivered"]}
+**Users:** {setting_storage.get_users_count()}
+**Helpless Users:** {setting_storage.get_helpless_users_count()}
+"""
+    await client.send_message(channel, stats_string)    

stats_storage.py

@@ -0,0 +1,23 @@
+import pickle
+import os
+
+stats = {}
+
+def increment_fate():
+    stats["fates_delivered"] += 1
+
+async def save_stats():
+    output = open('stats.pkl', 'wb')
+
+    pickle.dump(stats, output)
+    output.close()
+
+def load_stats():
+    global stats
+    if (os.path.isfile('stats.pkl')):
+        input = open('stats.pkl', 'rb')
+        stats = pickle.load(input)
+        input.close()
+    else:
+        stats = {}
+        stats["fates_delivered"] = 0