Поиск файлов на основе расширения с использованием os.path

#python #os.path

#python #os.path

Вопрос:

Мне было интересно, как искать файлы на основе их расширения. Если бы вы искали py или txt или какое-либо другое расширение, пользователь ввел бы его, а затем он искал бы его. Мне не разрешено использовать glob или os.walk

Это метод, который у меня есть. При необходимости я опубликую весь код.

Пользователь сначала вводит каталог, который находится в другом методе, после этого они скажут

E txt или E .txt (оба работают), и он вернет файлы в зависимости от того, какое расширение вы использовали. Я не уверен, что не так с моим кодом.

Ответ №1:

В вашем коде,

  for file in directory 
  

определенно неверно, поскольку directory — это строка с указанным в ней путем. это должно быть что-то вроде

 for file in os.listdir(directory):
  

чтобы получить список файлов в каталоге