How to list all files in a folder using Python?

Question

I am building a program that needs to know the files in its work directory. So I typed the following code.

def encontrar_arq(cadena, caminho):
   encontrado = False
   # lista_arq = ls(caminho) #funcao que lista todos os arquivos do caminho
   for nome_arq in lista_arq:
       if nome_arq == cadena:
           encontrado = True
   return encontrado

Answer 1

There are some forms, one of them is the function os.listdir:

from os import listdir

def encontrar_arq(cadena, caminho):
   encontrado = False
   lista_arq = listdir(caminho)

   for arquivo in lista_arq:
      # Use "arquivo" aqui...

To list files and directories separately, use os.walk:

from os import walk

def encontrar_arq(cadena, caminho):
   encontrado = False

   for path, diretorios, arquivos in walk(caminho):
       for arquivo in arquivos:
           # Use "arquivo" aqui...

Another alternative is the glob.glob:

from glob import glob

def encontrar_arq(cadena, caminho):
   encontrado = False

   arquivos = glob(caminho + '.*') # Para listar somente .txt altere para "*.txt"
   for arquivo in arquivos:
       # Use "arquivo" aqui...