Skip to content

Commit ec400f1

Browse files
committed
refactor(hash): replace lambda decoder with b64digest function
1 parent da0d2d1 commit ec400f1

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/redis_func_cache/mixins/hash.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
import json
55
import pickle
66
from abc import ABC
7-
from base64 import b64decode
87
from dataclasses import dataclass
98
from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple
109

@@ -154,7 +153,7 @@ class JsonMd5Base64HashMixin(AbstractHashMixin):
154153
__hash_config__ = HashConfig(
155154
algorithm="md5",
156155
serializer=lambda x: json.dumps(x).encode(),
157-
decoder=lambda x: b64decode(x.digest().decode().rstrip("=")),
156+
decoder=b64digest,
158157
)
159158

160159

@@ -196,7 +195,7 @@ class JsonSha1Base64HashMixin(AbstractHashMixin):
196195
__hash_config__ = HashConfig(
197196
algorithm="sha1",
198197
serializer=lambda x: json.dumps(x).encode(),
199-
decoder=lambda x: b64decode(x.digest().decode().rstrip("=")),
198+
decoder=b64digest,
200199
)
201200

202201

@@ -238,7 +237,7 @@ class JsonSha256Base64HashMixin(AbstractHashMixin):
238237
__hash_config__ = HashConfig(
239238
algorithm="sha256",
240239
serializer=lambda x: json.dumps(x).encode(),
241-
decoder=lambda x: b64decode(x.digest().decode().rstrip("=")),
240+
decoder=b64digest,
242241
)
243242

244243

@@ -280,7 +279,7 @@ class JsonSha512Base64HashMixin(AbstractHashMixin):
280279
__hash_config__ = HashConfig(
281280
algorithm="sha512",
282281
serializer=lambda x: json.dumps(x).encode(),
283-
decoder=lambda x: b64decode(x.digest().decode().rstrip("=")),
282+
decoder=b64digest,
284283
)
285284

286285

0 commit comments

Comments
 (0)