import os
def disk_size(path):
total = os.path.getsize(path)
if os.path.isdir(path):
pathd = os.listdir(path)
for f in pathd:
total = total + disk_size(os.path.join(path, f))
elif os.path.isfile(path):
print "{0:<7} {1}".format(total, path)
return total
path = raw_input()
if os.path.exists(path):
print disk_size(path)
아까와 달라진 포인트
1. if문 조건 고침
2. os.path.join : os.listdir의 리턴값은 전체 경로명이 아닌 파일명만 나옴. 그러니 현재 경로와 join이 필요함
다른 것들도 좀 손보고 싶지만 달라진 포인트만 강조하기 위해 그대로 놔둠..