Podemos ler os arquivos de um diretório usando o módulo os e a função listdir(). Observe como:
import os
# entrada
pasta = '.' # '.' = diretorio atual, '..' => anterior
extensoes = ['csv'] # deixe em branco para todos
# lê arquivos na pasta
arquivos = os.listdir(pasta)
# para cada arquivo na pasta
for i in arquivos:
# se vazio, exibir todos
if extensoes == []:
print(i)
else:
extensao = i.split('.')[-1]
if extensao in extensoes:
print(i)