Encode.py

from itertools import groupby
from heapq import *

# Node para abstração dos nós da heap que vai ser gerada
class Node(object):
 esquerda = None
 direita = None
 dado = None
 peso = 0

 # Metódo construtor pro object Node iniciando dado e peso de cada nó
 def __init__(self, d, p):
 self.dado = d
 self.peso = p

 # Método para setar os nós filhos(esquerdo e direito) de cada nó da pilha
 def setChildren(self, ne, nd):
 self.esquerda = ne
 self.direita = nd

 # Método sobreposto pra visualização das informações(print) dentro do nó
 def __repr__(self):
 return "Dado: "+str(self.dado)+" - Peso: "+str(self.peso)+" - esquerda: "+str(self.esquerda)+" - direita: "+str(self.direita)+"\n"

 # Implementaçã de sobrecarga de método LESS THAN necessaria pra verificação da queue no Python3
 def __lt__(self, other):
 return self.peso < other.peso

def EncodeHuffman(input, nomeArquivo):
 itemqueue = [Node(a,len(list(b))) for a,b in groupby(sorted(input))]
 itemqueue = sorted(itemqueue,reverse=True)
 while len(itemqueue) > 1:
 e = heappop(itemqueue)
 d = heappop(itemqueue)
 n = Node(None, d.peso+e.peso)
 n.setChildren(e,d)
 heappush(itemqueue, n)

 codes = {}

 def codeIt(s, node):
 if node.dado:
 if not s:
 codes[node.dado] = "0"
 else:
 codes[node.dado] = s
 else:
 codeIt(s+"0", node.esquerda)
 codeIt(s+"1", node.direita)

 codeIt("",itemqueue[0])

 try:
 f = open(nomeArquivo+".enc", 'w')
 f.write(str(codes)+"\n")
 f.write("".join([codes[a] for a in input]))
 f.close()
 print("Arquivo salvo com sucesso com o nome: "+nomeArquivo+".enc")
 except:
 print("Erro ao abrir arquivo, tente novamente")