import os
import sys
import shutil as st

def criarPastas(pastas):
    curdir = os.getcwd()
    dirList = [ i for i in os.listdir(curdir) if os.path.isdir(os.path.join(curdir,i))]
    count = 0

    for pasta in pastas:
        if not pasta in dirList:
            path = os.path.join(curdir, pasta)

            os.mkdir(path)
            print('pasta "%s" criada' %pasta)
            count += 1
    return count

def buscarArquivos(formatos, pasta_destino, pasta_atual):
    arquivos = []
    pastas = []
    numpastas = 0
    numarqs = 0

    # separando pastas e arquivos
    for arquivo in os.listdir(pasta_atual):
        caminhoAbsoluto = os.path.join(pasta_atual, arquivo)

        # se for pasta e não começar com .
        if os.path.isdir(caminhoAbsoluto):
            # ignora pastas que iniciam com . e a pasta destino
            if arquivo[0] != '.' and arquivo != os.path.split(pasta_destino)[-1]:
                pastas.append(caminhoAbsoluto)
                numpastas += 1
            continue

        # se for arquivo e estiver na lista de formatos
        elif (arquivo.split('.')[-1] in formatos):
            arquivos.append(caminhoAbsoluto)
            numarqs += 1
            #print(arquivo, 'adicionado')
    print("---------------------------------------------------------")
    print("Diretorio:", pasta_atual)
    print('Pastas encontradas:', numpastas, '\nArquivos compativeis:', numarqs)

    # movendo arquivos da pasta atual
    for arq in arquivos:
        nome = os.path.split(arq)[-1]
        try:
            comando = 'sudo mv ' + arq + ' ' + pasta_destino
            print(comando)
            os.system(comando)
        except:
            print('arquivo não movido:' + nome)

    for pasta in pastas:
        buscarArquivos(formatos, pasta_destino, pasta)

def main():
    pastas = ['pyImagens', 'pyVideos', 'pyDocum']

    fimg = ['jpg', 'png', 'jpeg', 'gif']
    fvid = ['mp4', 'avi', 'mov', 'wmv', 'flv', '3gp', 'mkv', 'rm', 'webm']
    fdoc = ['doc', 'pdf', 'ppt', 'pptx']
    formatos = [fimg, fvid, fdoc]

    ###### execução do programa
    if not criarPastas(pastas):
        print("Nenhuma pasta foi criada\n")
    input("Pressione qualquer tecla para iniciar")

    for i in range(3):
        destino = os.path.join(os.getcwd(), pastas[i])
        print('Buscando', pastas[i][2:].lower())
        buscarArquivos(formatos[i], destino, os.getcwd())


main()