Using freshmen gravitar images

Author: devinmatte

packet/context_processors.py

@@ -1,7 +1,8 @@
 """
 Context processors used by the jinja templates
 """
-
+import hashlib
+import urllib
 from functools import lru_cache
 from datetime import datetime
 
@@ -18,6 +19,7 @@ def get_csh_name(username):
     except:
         return username
 
+
 def get_roles(sig):
     """
     Converts a signature's role fields to a dict for ease of access.
@@ -49,6 +51,18 @@ def get_rit_name(username):
         return username
 
 
+@lru_cache(maxsize=128)
+def get_rit_image(username):
+    if username:
+        addresses = [username + "@rit.edu", username + "@g.rit.edu"]
+        for addr in addresses:
+            url = "https://gravatar.com/avatar/" + hashlib.md5(addr.encode('utf8')).hexdigest() + ".jpg?d=404&s=250"
+            gravatar = urllib.request.urlopen(url)
+            if gravatar.getcode() == 200:
+                return url
+    return "https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
+
+
 def log_time(label):
     """
     Used during debugging to log timestamps while rendering templates

packet/templates/active_packets.html

@@ -42,7 +42,7 @@ <h4 class="page-title">Active Packets</h4>
                                                 <a href="{{ url_for('freshman_packet', packet_id=packet.id) }}">
                                                     <img class="eval-user-img"
                                                          alt="{{ get_rit_name(packet.freshman_username) }}"
-                                                         src="https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
+                                                         src="{{ get_rit_image(packet.freshman_username) }}"
                                                          width="25"
                                                          height="25"/> {{ get_rit_name(packet.freshman_username) }}
                                                 </a>

packet/templates/packet.html

@@ -110,7 +110,7 @@ <h5>Upperclassmen Score - {{ '%0.2f' % upper_score }}%</h5>
                                     <tr {% if sig.signed %}style="background-color: #4caf505e" {% endif %}>
                                         <td>
                                             <img class="eval-user-img" alt="{{ sig.freshman_username }}"
-                                                 src="https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
+                                                 src="{{ get_rit_image(packet.freshman_username) }}"
                                                  width="25" height="25"/>
                                             {{ get_rit_name(sig.freshman_username) }}
                                         </td>

packet/templates/upperclassman.html

@@ -33,7 +33,7 @@ <h3 class="page-title">
                                             <a href="{{ url_for('freshman_packet', packet_id=packet.id) }}">
                                                 <img class="eval-user-img"
                                                      alt="{{ get_rit_name(packet.freshman_username) }}"
-                                                     src="https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
+                                                     src="{{ get_rit_image(packet.freshman_username) }}"
                                                      width="25"
                                                      height="25"/>
                                                 {{ get_rit_name(packet.freshman_username) }}