X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=letter_compress.py;h=6cb6b74e87c54928a5aeb7e184114e81ff877b02;hb=532df2c5b57c7517dfb3dddd8c1358fbadf8baf3;hp=b5d326471e8c9412133e94e07737a7d91496dc41;hpb=eb9e6df32ed696158bf34dba6464277b648f5c74;p=python_utils.git diff --git a/letter_compress.py b/letter_compress.py index b5d3264..6cb6b74 100644 --- a/letter_compress.py +++ b/letter_compress.py @@ -1,5 +1,9 @@ #!/usr/bin/env python3 +# © Copyright 2021-2022, Scott Gasch + +"""A simple compression helper for lowercase ascii text.""" + import bitstring from collect.bidict import BiDict @@ -32,9 +36,9 @@ def compress(uncompressed: str) -> bytes: """ compressed = bitstring.BitArray() - for (n, letter) in enumerate(uncompressed): + for letter in uncompressed: if 'a' <= letter <= 'z': - bits = ord(letter) - ord('a') + 1 # 1..26 + bits = ord(letter) - ord('a') + 1 # 1..26 else: if letter not in special_characters: raise Exception(f'"{uncompressed}" contains uncompressable char="{letter}"') @@ -100,4 +104,5 @@ def decompress(kompressed: bytes) -> str: if __name__ == '__main__': import doctest + doctest.testmod()