Address coverage for Python 2, tidy up variable names in _is_valid_dkim_key

2024-05-11 05:55:00 +00:00 · 2020-11-25 23:29:49 +08:00
parent fa266c23d2
commit 7f89c621a1
1 changed files with 4 additions and 5 deletions
@@ -371,12 +371,11 @@ class OvhProvider(BaseProvider):
    @staticmethod
    def _is_valid_dkim_key(key):
        result = True
+        base64_decode = getattr(base64, 'decodestring', None)
+        base64_decode = getattr(base64, 'decodebytes', base64_decode)
+
        try:
-            decode = base64.decodestring
-        except AttributeError:
-            decode = base64.decodebytes
-        try:
-            result = decode(bytearray(key, 'utf-8'))
+            result = base64_decode(bytearray(key, 'utf-8'))
        except binascii.Error:
            result = False
        return result