Как вернуть результат рекурсивной функции в переменную?

Question

[etetetetetete]

Написал рекурсивную функцию по поиску вложенных каталогов в корне.
Проблема в том что требуется вернуть список в переменную, но только не так как сделал я, с помощью global, а c помощью return`а.

import os


def SearchDir(path):
    global z
    if os.path.exists(path):
        if os.listdir(path):
            for d in os.listdir(path):
                if os.path.isdir(os.path.join(path, d)):
                    z += [(os.path.join(path, d))]
                    SearchDir(os.path.join(path, d))
        else:
            return print('The directory is empty')
    else:
        return print('The directory does not exist')

path = r'C:\test'
z = []

SearchDir(path)
print(z)

ЗЫ. Python 3

Comments

[abcd0x00]

Сформулируй задачу правильно, а то по быдлокоду догадываться, в чём именно задача, - последнее дело.

Answer 1

[Tark]

А если очень хочется разобраться с рекурсией, то вот:

import os


def SearchDir(path, z=[]):
    if os.path.exists(path):
        if os.listdir(path):
            for d in os.listdir(path):
                if os.path.isdir(os.path.join(path, d)):
                    z += [(os.path.join(path, d))]
                    SearchDir(os.path.join(path, d), z)
        else:
            return print('The directory is empty')
    else:
        return print('The directory does not exist')

    return z

path = r'f:\Python-pyramid\farm'
print(SearchDir(path))

Но вы видите, что код очень непричёсанный, да? Слишком много лишнего написано, да и return print удивляет.

Comments

[etetetetetete]

Спасабо большое!

Answer 2

[sim3x]

пхп стайл

используй https://docs.python.org/3/library/os.html#os.walk

принт не нужно писать в return

Comments

[etetetetetete]

Встроенная функция не интересует так как преследовалаь учебная цель)
Не нужно было бы писать если бы вызов был обёрнут принтом, это не принципиально

[sim3x]

etetetetetete:
не пайтон вей
выбери задачу, без встроенных функций

[etetetetetete]

sim3x: Учту)

Answer 3

[Алексей Черемисин]

Надеюсь, понятно.

def SearchDir(path):
   z= []
   def _SearchDir(path):
      ...
      z.append("abc")
      _SearchDir(newpath)
      ...
   _SearchDir(path)
   return z

Comments

[RedSkotina]

глобальная переменная - вид сбоку.

[Алексей Черемисин]

Не спорю, вид с изнинки.