Итерация по каталогам выдает ошибку без итератора - PullRequest
Новая
       17

Итерация по каталогам выдает ошибку без итератора

0 голосов
/ 17 апреля 2019

Я попробовал точный код на этой странице , чтобы перебрать подкаталоги. Однако я получаю следующую ошибку: 

  File "dir_iterator.py", line 29, in <module>
    for x in it:
TypeError: iter() returned non-iterator of type 'iterdir'

Где проблема и как ее можно решить?

Примечание: я использую Python версии 3.5.3 на Debian Stable Linux

Редактировать: как предложено @ DroidX86 в комментариях ниже, я публикую дословно код, скопированный с этой ссылки : 

import os

class iterdir(object):
    def __init__(self, path, deep=False):
    self._root = path
    self._files = None
    self.deep = deep
    def __iter__(self):
    return self
    def next(self):
    if self._files:
        join = os.path.join
        d = self._files.pop()
        r = join(self._root, d)
        if self.deep and os.path.isdir(r):
        self._files += [join(d,n) for n in os.listdir(r)]
    elif self._files is None:
        self._files = os.listdir(self._root)
    if self._files:
        return self._files[-1]
    else:
        raise StopIteration


# sample:
#   a deep traversal of directories which starts with a vowel
#
it = iterdir('.')
for x in it:
    p = os.path.basename(x)
    it.deep = p[0].lower() in "aeiou"
    print x

Ответы [ 2 ]

0 голосов
/ 17 апреля 2019

Код написан для python2. По любой причине, если вы хотите, чтобы он работал с python3, измените def next: на def __next__: и print x на print(x). Эти два изменения необходимы для преобразования кода python2 в ссылке в python3. 

import os


class iterdir(object):
    def __init__(self, path, deep=False):
        self._root = path
        self._files = None
        self.deep = deep

    def __iter__(self):
        return self

    def __next__(self):
        if self._files:
            join = os.path.join
            d = self._files.pop()
            r = join(self._root, d)
            if self.deep and os.path.isdir(r):
                self._files += [join(d, n) for n in os.listdir(r)]
        elif self._files is None:
            self._files = os.listdir(self._root)

        if self._files:
            return self._files[-1]
        else:
            raise StopIteration


# sample:
#   a deep traversal of directories which starts with a vowel
#
it = iterdir('.')
for x in it:
    p = os.path.basename(x)
    it.deep = p[0].lower() in "aeiou"
    print(x)
0 голосов
/ 17 апреля 2019

Связанный код написан для python2.

вы используете Python3.

либо вам придется внести изменения в код, чтобы он работал в python3, либо вы можете использовать python2.

...