chom

Author: Qelxiros

conditional/util/context_processors.py

@@ -1,13 +1,7 @@
 # pylint: disable=bare-except
 
-import hashlib
-import urllib
-from functools import lru_cache
-from datetime import datetime
-from typing import Callable
-
-from conditional import app, packet_bp
-from conditional.models.models import FreshmanAccount, Freshman, UpperSignature
+from conditional import app
+from conditional.models.models import FreshmanAccount
 from conditional.util.cache import service_cache
 from conditional.util.ldap import ldap_get_member, ldap_is_current_student
 
@@ -45,84 +39,3 @@ def utility_processor():
         "get_member_name": get_member_name,
         "check_current_student": check_current_student,
     }
-
-
-# pylint: disable=bare-except
-@lru_cache(maxsize=128)
-def packet_get_csh_name(username: str) -> str:
-    try:
-        member = ldap_get_member(username)
-        return member.cn + " (" + member.uid + ")"
-    except:
-        return username
-
-
-def get_roles(sig: UpperSignature) -> dict[str, str]:
-    """
-    Converts a signature's role fields to a dict for ease of access.
-    :return: A dictionary of role short names to role long names
-    """
-    out = {}
-    if sig.eboard:
-        out["eboard"] = sig.eboard
-    if sig.active_rtp:
-        out["rtp"] = "RTP"
-    if sig.three_da:
-        out["three_da"] = "3DA"
-    if sig.w_m:
-        out["wm"] = "Wiki Maintainer"
-    if sig.webmaster:
-        out["webmaster"] = "Webmaster"
-    if sig.c_m:
-        out["cm"] = "Constitutional Maintainer"
-    if sig.drink_admin:
-        out["drink"] = "Drink Admin"
-    return out
-
-
-# pylint: disable=bare-except
-@lru_cache(maxsize=256)
-def get_rit_name(username: str) -> str:
-    try:
-        freshman = Freshman.query.filter_by(rit_username=username).first()
-        return freshman.name + " (" + username + ")"
-    except:
-        return username
-
-
-# pylint: disable=bare-except
-@lru_cache(maxsize=256)
-def get_rit_image(username: str) -> str:
-    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"
-            )
-            try:
-                with urllib.request.urlopen(url) as gravatar:
-                    if gravatar.getcode() == 200:
-                        return url
-            except:
-                continue
-    return "https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
-
-
-def log_time(label: str) -> None:
-    """
-    Used during debugging to log timestamps while rendering templates
-    """
-    print(label, datetime.now())
-
-
-@packet_bp.context_processor
-def packet_utility_processor() -> dict[str, Callable]:
-    return {
-        "get_csh_name": packet_get_csh_name,
-        "get_rit_name": get_rit_name,
-        "get_rit_image": get_rit_image,
-        "log_time": log_time,
-        "get_roles": get_roles,
-    }

conditional/util/packet_context_processors.py

@@ -0,0 +1,92 @@
+# pylint: disable=bare-except
+
+import hashlib
+import urllib
+from functools import lru_cache
+from datetime import datetime
+from typing import Callable
+
+from conditional import packet_bp
+from conditional.models.models import Freshman, UpperSignature
+from conditional.util.ldap import ldap_get_member
+
+
+# pylint: disable=bare-except
+@lru_cache(maxsize=128)
+def packet_get_csh_name(username: str) -> str:
+    try:
+        member = ldap_get_member(username)
+        return member.cn + " (" + member.uid + ")"
+    except:
+        return username
+
+
+def get_roles(sig: UpperSignature) -> dict[str, str]:
+    """
+    Converts a signature's role fields to a dict for ease of access.
+    :return: A dictionary of role short names to role long names
+    """
+    out = {}
+    if sig.eboard:
+        out["eboard"] = sig.eboard
+    if sig.active_rtp:
+        out["rtp"] = "RTP"
+    if sig.three_da:
+        out["three_da"] = "3DA"
+    if sig.w_m:
+        out["wm"] = "Wiki Maintainer"
+    if sig.webmaster:
+        out["webmaster"] = "Webmaster"
+    if sig.c_m:
+        out["cm"] = "Constitutional Maintainer"
+    if sig.drink_admin:
+        out["drink"] = "Drink Admin"
+    return out
+
+
+# pylint: disable=bare-except
+@lru_cache(maxsize=256)
+def get_rit_name(username: str) -> str:
+    try:
+        freshman = Freshman.query.filter_by(rit_username=username).first()
+        return freshman.name + " (" + username + ")"
+    except:
+        return username
+
+
+# pylint: disable=bare-except
+@lru_cache(maxsize=256)
+def get_rit_image(username: str) -> str:
+    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"
+            )
+            try:
+                with urllib.request.urlopen(url) as gravatar:
+                    if gravatar.getcode() == 200:
+                        return url
+            except:
+                continue
+    return "https://www.gravatar.com/avatar/freshmen?d=mp&f=y"
+
+
+def log_time(label: str) -> None:
+    """
+    Used during debugging to log timestamps while rendering templates
+    """
+    print(label, datetime.now())
+
+
+@packet_bp.context_processor
+def packet_utility_processor() -> dict[str, Callable]:
+    return {
+        "get_csh_name": packet_get_csh_name,
+        "get_rit_name": get_rit_name,
+        "get_rit_image": get_rit_image,
+        "log_time": log_time,
+        "get_roles": get_roles,
+    }