Encode IDs using URL- and filesystem-safe alphabet (#35)

Author: Cito

src/graphql_relay/utils/base64.py

@@ -1,13 +1,13 @@
-from base64 import b64encode, b64decode
+from base64 import urlsafe_b64encode, urlsafe_b64decode
 
 __all__ = ["base64", "unbase64"]
 
 
 def base64(s: str) -> str:
     """"Encode the string s using Base64."""
-    return b64encode(s.encode("utf-8")).decode("utf-8")
+    return urlsafe_b64encode(s.encode("utf-8")).decode("utf-8")
 
 
 def unbase64(s: str) -> str:
     """"Decode the string s using Base64."""
-    return b64decode(s).decode("utf-8")
+    return urlsafe_b64decode(s).decode("utf-8")

tests/utils/test_base64.py

@@ -2,7 +2,7 @@
 
 
 example_unicode = "Some examples: ❤😀"
-example_base64 = "U29tZSBleGFtcGxlczog4p2k8J+YgA=="
+example_base64 = "U29tZSBleGFtcGxlczog4p2k8J-YgA=="
 
 
 def describe_base64_conversion():