huffman.py

   1 import utils
   2 import sys
   3
   4 _opt_type = sys.argv[1]
   5 _file_path = sys.argv[2]
   6
   7
   8 if _opt_type == "c":
   9     _file = utils.get_file(_file_path)
  10     _pbm_content = utils.BitMapFile(_file)
  11     _result = utils.huffman_encode_image(_pbm_content)
  12     with open(sys.argv[2]+".bin", "wb") as file:
  13         file.write(_result)
  14     print(_result)
  15 elif _opt_type == "d":
  16     _result = utils.huffman_decode_file(_file_path)
  17     with open(sys.argv[2].strip('.bin'), "w") as file:
  18         file.write(_result)