import os, sys
import shutil
from pathlib import Path
import re

#-----------------------------------------------------------------------------------------------------------------------------
caminhoteste="/Users/Public/Pictures/Sample Pictures/"
sim= Path("/Users/Public/Pictures/Sample Pictures/sim")
nao= Path("/Users/Public/Pictures/Sample Pictures/nao")

#-----------------------------------------------------------------------------------------------------------------------------
print("\n")
print("---------------------------Divisão de arquivos por conteudo--------------------------------")
print("\n")
caminhoinput = input("Entre com o caminho da pasta principal onde deseja buscar buscar: \nex:/Users/acpasso/Documents/anapassos\nobs:nao de espacos\n")
print("\n")

path = os.path.abspath(caminhoinput)
encontra=Path(input("Entre com o caminho da pasta onde os arquivos em que foi encontrado o elemento busca serao movidos: \nex:/Users/acpasso/Documents/anapassos\nobs:nao de espacos\n"))
print("\n")

nao_encontra= Path(input("Entre com o caminho da pasta onde os arquivos em que nao foi encontrado o elemento busca serao movidos: \nex:/Users/acpasso/Documents/anapassos\nobs:nao de espacos\n"))
print("\n")

search = input("Entre com a(s) palavra(s) que deseja buscar: \nobs(no caso de mais de uma palavra separe-as com virgulas, nao use espacos) ")
searches = re.split(',', search)
print("\n")

#-----------------------------------------------------------------------------------------------------------------------------

print("---------------------------------Estamos buscando por--------------------------------------")
print("\n")
for search in searches:
    print(search)
print("\n")
print("-----------------------------------Realizando busca----------------------------------------")
print("\n")

#-----------------------------------------------------------------------------------------------------------------------------

for root, dirs, f in os.walk(path):
    print("\nEstamos em:", root, "\nPastas contidas:", dirs, "\nArquivos contidos:", f, "\n")
    print("--------------------------------------------------------------------------------------------")
    for file in f:
        if file.endswith((".txt",".docx")):
            src = os.path.join(root,file)
            with open(src) as ana:
                print("\nEstamos no arquivo", file, "\n")
                for search in searches:
                    print("\nBusca: ", search)
                    if any(search in line for line in ana):
                        if not os.path.exists(encontra):
                            os.mkdir(encontra)
                        dst = os.path.join(encontra)
                        print("Status: encontrada\n")
                        break
                    else:
                        if not os.path.exists(nao_encontra):
                            os.mkdir(nao_encontra)
                        dst=os.path.join(nao_encontra)
                        print("Status: não encontrada \n")
                shutil.copy(src, dst)
                print("\nO arquivo ", file, " foi movido para ", dst,"\n")
                print("--------------------------------------------------------------------------------------------")
#ana.close()
#os.remove(src)