From eaee1acf21d903f4346c47f96bf42a9db9870d07 Mon Sep 17 00:00:00 2001
From: aikaterna <20862007+aikaterna@users.noreply.github.com>
Date: Sun, 18 Mar 2018 15:50:08 -0700
Subject: [PATCH] Add discriminator to user name, shorten by length

---
 chatchart/chatchart.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/chatchart/chatchart.py b/chatchart/chatchart.py
index 0092dc4..e15e4b4 100644
--- a/chatchart/chatchart.py
+++ b/chatchart/chatchart.py
@@ -19,6 +19,7 @@ class ChatChart:
 
     def __init__(self, bot):
         self.bot = bot
+
     def create_chart(self, top, others, channel):
         plt.clf()
         sizes = [x[1] for x in top]
@@ -69,14 +70,19 @@ class ChatChart:
         msg_data = {'total count': 0, 'users': {}}
 
         for msg in history:
+            if len(msg.author.name) >= 20:
+                short_name = '{}...'.format(msg.author.name[:20])
+            else:
+                short_name = msg.author.name
+            whole_name = '{}#{}'.format(short_name, msg.author.discriminator)
             if msg.author.bot:
                 pass
-            elif msg.author.name in msg_data['users']:
-                msg_data['users'][msg.author.name]['msgcount'] += 1
+            elif whole_name in msg_data['users']:
+                msg_data['users'][whole_name]['msgcount'] += 1
                 msg_data['total count'] += 1
             else:
-                msg_data['users'][msg.author.name] = {}
-                msg_data['users'][msg.author.name]['msgcount'] = 1
+                msg_data['users'][whole_name] = {}
+                msg_data['users'][whole_name]['msgcount'] = 1
                 msg_data['total count'] += 1
 
         for usr in msg_data['users']: