[PYTHON/COMMON] 입력 디렉토리 경로의 트리 구조 출력하기

icodebroker COMMON 2024-06-18

■ 입력 디렉토리 경로의 트리 구조를 출력하는 방법을 보여준다.

▶ 예제 코드 (PY)


import glob
import os.path

directoryCount = 0
fileCount      = 0

def traverse(parentDirectoryPath, depth):
    global directoryCount
    global fileCount

    if depth == 0:
        print(parentDirectoryPath)

    for childPath in glob.glob(parentDirectoryPath + "/*"):
        if depth == 0:
            prefix = "|--"
        else:
            prefix = "|" + "    " * depth + "|--"

        if os.path.isdir(childPath):
            directoryCount += 1
            print(prefix + os.path.basename(childPath))
            traverse(childPath, depth + 1)
        elif os.path.isfile(childPath):
            fileCount += 1
            print(prefix + os.path.basename(childPath))
        else:
            print(prefix + 'unknown object :', childPath)

if __name__ == "__main__":
    traverse("D:\\DEVELOPMENT.TOOL", 0)
    print('\n', '디렉토리 수 :', directoryCount, '   파일 수 :', fileCount)

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

import glob

import os.path

directoryCount = 0

fileCount = 0

def traverse(parentDirectoryPath, depth):

global directoryCount

global fileCount

if depth == 0:

print(parentDirectoryPath)

for childPath in glob.glob(parentDirectoryPath + "/*"):

if depth == 0:

prefix = "|--"

else:

prefix = "|" + " " * depth + "|--"

if os.path.isdir(childPath):

directoryCount += 1

print(prefix + os.path.basename(childPath))

traverse(childPath, depth + 1)

elif os.path.isfile(childPath):

fileCount += 1

print(prefix + os.path.basename(childPath))

else:

print(prefix + 'unknown object :', childPath)

if __name__ == "__main__":

traverse("D:\\DEVELOPMENT.TOOL", 0)

print('\n', '디렉토리 수 :', directoryCount, ' 파일 수 :', fileCount)