bites 228 - Create a Gravatar URL with hashlib and string formatting

Author: [email protected]

228/gravatar.py

@@ -0,0 +1,21 @@
+import hashlib
+
+GRAVATAR_URL = ("https://www.gravatar.com/avatar/"
+                "{hashed_email}?s={size}&r=g&d=robohash")
+
+
+def create_gravatar_url(email, size=200):
+    """Use GRAVATAR_URL above to create a gravatar URL.
+
+       You need to create a hash of the email passed in.
+
+       PHP example: https://en.gravatar.com/site/implement/hash/
+
+       For Python check hashlib check out (md5 / hexdigest):
+       https://docs.python.org/3/library/hashlib.html#hashlib.hash.hexdigest
+    """
+    email = email.strip().lower()
+    m = hashlib.md5()
+    m.update(str.encode(email))
+    hashed_email = m.hexdigest()
+    return GRAVATAR_URL.format(hashed_email=hashed_email, size=size)

228/test_gravatar.py

@@ -0,0 +1,31 @@
+import pytest
+
+from gravatar import create_gravatar_url
+
+BASE_URL = "https://www.gravatar.com/avatar/"
+HASHED_INFO_EMAIL = "ff2de96de035e1ccdb39ca31d3fe4a5c"
+HASHED_SUPPORT_EMAIL = "1c888a408baa7a685c7c06155e48de84"
+SIZE, DEFAULT = 200, 'robohash'
+
+EXPECTED = iter([  # make an iterator
+    f'{BASE_URL}{HASHED_INFO_EMAIL}?s={SIZE}&r=g&d={DEFAULT}',
+    f'{BASE_URL}{HASHED_INFO_EMAIL}?s={SIZE}&r=g&d={DEFAULT}',
+    f'{BASE_URL}{HASHED_INFO_EMAIL}?s=40&r=g&d={DEFAULT}',
+    f'{BASE_URL}{HASHED_SUPPORT_EMAIL}?s={SIZE}&r=g&d={DEFAULT}',
+    f'{BASE_URL}{HASHED_SUPPORT_EMAIL}?s={SIZE}&r=g&d={DEFAULT}',
+    f'{BASE_URL}{HASHED_SUPPORT_EMAIL}?s=1000&r=g&d={DEFAULT}',
+])
+
+
+@pytest.mark.parametrize("email, size", [
+    ("[email protected]", 200),
+    ("[email protected] ", 200),
+    ('[email protected]', 40),
+    ('[email protected]', 200),
+    ('[email protected]', 200),
+    (' [email protected]', 1000),
+])
+def test_gravatar(email, size):
+    actual = create_gravatar_url(email, size=size)
+    expected = next(EXPECTED)
+    assert actual == expected