Implementing reset

Author: ndbzika

.gitignore

@@ -7,7 +7,7 @@ __pycache__/
 *.so
 
 # Projeto
-/aquivos_aleatorios
+/arquivos_aleatorios
 /diretorio_temporario
 contagem_final
 

README.md

@@ -31,7 +31,7 @@ Baixe ou Clone o Repositório: Se ainda não tiver o repositório, clone-o ou ba
 git clone https://github.com/Cabereca/MapReduce.git
 cd MapReduce
 ```
-Prepare os Arquivos de Entrada O projeto inclui um diretório /aquivos_aleatorios onde os arquivos de texto com palavras serão armazenados. Os arquivos podem ser gerados com o módulo Gerador de Arquivos. Certifique-se de que o diretório aquivos_aleatorios contenha arquivos de texto com palavras geradas aleatoriamente.
+Prepare os Arquivos de Entrada O projeto inclui um diretório /arquivos_aleatorios onde os arquivos de texto com palavras serão armazenados. Os arquivos podem ser gerados com o módulo Gerador de Arquivos. Certifique-se de que o diretório arquivos_aleatorios contenha arquivos de texto com palavras geradas aleatoriamente.
 
 Execute o Código Para rodar o código, basta executar o arquivo main.py. Este arquivo irá gerar os arquivos de entrada e executar o processo de MapReduce.
 

controller.py

@@ -2,21 +2,12 @@
 import os
 from map import map_function
 from reduce import reduce_function
-
-def limpar_arquivos(output_dir):
-    # Apagar arquivo temporário se existir
-    tmp_file = os.path.join(output_dir, 'arquivo_temporario.tmp')
-    if os.path.exists(tmp_file):
-        os.remove(tmp_file)
-
-    # Apagar o arquivo final se existir
-    final_file = 'contagem_final'
-    if os.path.exists(final_file):
-        os.remove(final_file)
+from reset import limpar_arquivos
 
 def controller(input_files, output_dir):
      # Apagar arquivos antigos antes de começar
-    limpar_arquivos(output_dir)
+    if os.path.exists(output_dir):
+        limpar_arquivos(output_dir)
 
     map_threads = []
 

gerador_de_arquivos.py

@@ -1,5 +1,6 @@
 import random
 import os
+from reset import deletar_arquivos
 
 class FileGenerator:
     def __init__(self, split, n, alphabet, min_size, max_size):
@@ -21,10 +22,7 @@ def gerar_arquivos(self, output_dir):
         # Se o diretório já existe, apagamos seu conteúdo (se houver arquivos antigos)
         if os.path.exists(output_dir):
             # Apagar os arquivos antigos dentro do diretório
-            for file_name in os.listdir(output_dir):
-                file_path = os.path.join(output_dir, file_name)
-                if os.path.isfile(file_path):
-                    os.remove(file_path)
+            deletar_arquivos(output_dir)
         else:
             # Se o diretório não existe, criamos ele
             os.makedirs(output_dir)
@@ -42,13 +40,13 @@ def gerar_arquivos(self, output_dir):
 if __name__ == "__main__":
     # Parâmetros de configuração
     gerador_de_arquivos = FileGenerator(
-        split=3,              # Dividir o texto em 3 arquivos
-        n=300,                # Gerar 100 palavras
+        split=1,              # Dividir o texto em 3 arquivos
+        n=100,                # Gerar 100 palavras
         alphabet=['a', 'b', 'c', 'd', 'e'],  # Definir as letras permitidas
         min_size=3,           # Tamanho mínimo de palavras: 4 letras
         max_size=6            # Tamanho máximo de palavras: 8 letras
     )
 
     # Criar e executar o gerador de arquivos
-    aquivos_aleatorios_dir = 'aquivos_aleatorios'
-    gerador_de_arquivos.gerar_arquivos(aquivos_aleatorios_dir)              
+    arquivos_aleatorios_dir = 'arquivos_aleatorios'
+    gerador_de_arquivos.gerar_arquivos(arquivos_aleatorios_dir)              

main.py

@@ -3,11 +3,11 @@
 
 if __name__ == "__main__":
     # Diretório onde os arquivos estão localizados
-    aquivos_aleatorios_dir = 'aquivos_aleatorios'
+    arquivos_aleatorios_dir = 'arquivos_aleatorios'
     output_dir = 'diretorio_temporario'
 
     # Obter os arquivos gerados pelo Gerador de Arquivos
-    input_files = [os.path.join(aquivos_aleatorios_dir, f) for f in os.listdir(aquivos_aleatorios_dir) if f.endswith('.txt')]
+    input_files = [os.path.join(arquivos_aleatorios_dir, f) for f in os.listdir(arquivos_aleatorios_dir) if f.endswith('.txt')]
 
     # Executar o controlador MapReduce com os arquivos gerados
     controller(input_files, output_dir)

reset.py

@@ -0,0 +1,25 @@
+import os
+
+def deletar_arquivos(output_dir='arquivos_aleatorios'):
+    if os.path.exists(output_dir):
+        for file_name in os.listdir(output_dir):
+            file_path = os.path.join(output_dir, file_name)
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+        os.rmdir(output_dir)
+
+def limpar_arquivos(output_dir='diretorio_temporario'):
+    # Apagar arquivo temporário se existir
+    tmp_file = os.path.join(output_dir, 'arquivo_temporario.tmp')
+    if os.path.exists(tmp_file):
+        os.remove(tmp_file)
+        os.rmdir(output_dir)
+
+    # Apagar o arquivo final se existir
+    final_file = 'contagem_final'
+    if os.path.exists(final_file):
+        os.remove(final_file)
+
+if __name__ == "__main__":
+    limpar_arquivos()
+    deletar_arquivos()