From d8254fd9186c3d1c08664fc8a4b3c103c598c836 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonah=20Br=C3=BCchert?= <jbb@kaidan.im>
Date: Tue, 13 Feb 2024 01:33:23 +0100
Subject: Add back base64 compat

---
 src/lib/Bcfg2/Compat.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'src')

diff --git a/src/lib/Bcfg2/Compat.py b/src/lib/Bcfg2/Compat.py
index 2e13301f8..251c80f4c 100644
--- a/src/lib/Bcfg2/Compat.py
+++ b/src/lib/Bcfg2/Compat.py
@@ -102,6 +102,23 @@ except ImportError:
         """ implementation of functools.wraps() for python 2.4 """
         return lambda f: f
 
+# base64 compat
+if sys.hexversion >= 0x03000000:
+    from base64 import b64encode as _b64encode, b64decode as _b64decode
+
+    @wraps(_b64encode)
+    def b64encode(val, **kwargs):  # pylint: disable=C0111
+        try:
+            return _b64encode(val, **kwargs)
+        except TypeError:
+            return _b64encode(val.encode('UTF-8'), **kwargs).decode('UTF-8')
+
+    @wraps(_b64decode)
+    def b64decode(val, **kwargs):  # pylint: disable=C0111
+        return _b64decode(val.encode('UTF-8'), **kwargs).decode('UTF-8')
+else:
+    from base64 import b64encode, b64decode
+
 try:
     input = raw_input
 except NameError:
-- 
cgit v1.2.3-1-g7c22