diff options
author | Determinant <[email protected]> | 2021-01-11 14:20:43 -0500 |
---|---|---|
committer | Determinant <[email protected]> | 2021-01-11 14:20:43 -0500 |
commit | e5b01bb04331676cbc5e202993e4bf883cb97633 (patch) | |
tree | 2882b010ca13746aa58e23265d15f0c4e1ddc5a2 /frozen_deps/base58/__main__.py | |
parent | aa9206a264b43f5d538fc39c2c1532d279e6f775 (diff) |
upgrade to Python 3.9
Diffstat (limited to 'frozen_deps/base58/__main__.py')
-rw-r--r-- | frozen_deps/base58/__main__.py | 17 |
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() |