Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # coding: utf-8
- from glob import glob
- from os import chdir, system
- from functools import lru_cache
- @lru_cache(maxsize=None)
- def enc_arquivos(extensao, diretorio): #função feita para encontrar arquivos e retornar numa lista, senão retorna False
- extensao = "*" + extensao
- try:
- try:
- try:
- chdir(diretorio)
- arquivos = {}
- arquivos_achados = []
- # print("Estamos no diretório %s" % diretorio)
- for arquivo in glob(extensao):
- arquivos_achados.append(arquivo)
- arquivos[extensao] = arquivo
- for key in arquivos:
- exte = key[key.find('.') + 1::]
- filename = arquivos[key]
- filename = filename[:filename.find('.'):]
- print('Arquivo com extensão', exte, 'com nome de', filename)
- except IndexError:
- print('Você não digitou nada!')
- return False
- return arquivos_achados
- except FileNotFoundError:
- print("Não conseguimos acessar o diretório.\nTalvez ele não exista...")
- return False
- except IndexError:
- print("Você não digitou nada!\n")
- return False
- # Demonstração
- extensions = ['.gif', '.exe', '.ico', '.txt', '.png', '.jpg', '.iso', '.py', '.zip', '.url', '.msi', '.zip', 'dll']
- total=0
- for extension in extensions:
- path = "C:\\Windows\\System32" # Use \\ ao invés de só \ (input é mais fácil e você não precisa digitar \\, :D)
- arquivos_encontrados = enc_arquivos(extension, path)
- total+=len(arquivos_encontrados)
- print("Arquivos encontrados:", total)
Advertisement
Add Comment
Please, Sign In to add comment