1 files changed, 10 insertions, 7 deletions
diff --git a/frozen_deps/base58/__main__.py b/frozen_deps/base58/__main__.py
index b76fdad..3a6f5d2 100644
--- a/frozen_deps/base58/__main__.py
+++ b/frozen_deps/base58/__main__.py
@@ -1,10 +1,18 @@
 import argparse
 import sys
+from typing import Callable, Dict, Tuple
 
 from base58 import b58decode, b58decode_check, b58encode, b58encode_check
 
+_fmap = {
+ (False, False): b58encode,
+ (False, True): b58encode_check,
+ (True, False): b58decode,
+ (True, True): b58decode_check
+} # type: Dict[Tuple[bool, bool], Callable[[bytes], bytes]]
 
-def main():
+
+def main() -> None:
 '''Base58 encode or decode FILE, or standard input, to standard output.'''
 
 stdout = sys.stdout.buffer
@@ -26,12 +34,7 @@ def main():
 help='append a checksum before encoding')
 
 args = parser.parse_args()
- fun = {
- (False, False): b58encode,
- (False, True): b58encode_check,
- (True, False): b58decode,
- (True, True): b58decode_check
- }[(args.decode, args.check)]
+ fun = _fmap[(args.decode, args.check)]
 
 data = args.file.buffer.read()