在修复一个用户的answer on AskUbuntu时,我发现了一个小问题。代码本身很简单:os.walk,递归地获取目录中所有文件的总和。
但它打破了符号链接:
$ python test_code2.py $HOME
Traceback (most recent call last):
File "test_code2.py", line 8, in <module>
space += os.stat(os.path.join(subdir, f)).st_size
OSError: [Errno 2] No such file or directory: '/home/xieerqi/.kde/socket-eagle'
问题是,如何告诉python忽略这些文件并避免对它们求和?
解决方案:
正如评论中所建议的那样,我添加了os.path.isfile()
检查,现在它完美无缺,并为我的主目录提供了正确的大小
$> cat test_code2.py
#! /usr/bin/python
import os
import sys
space = 0L # L means "long" - not necessary in Python 3
for subdir, dirs, files in os.walk(sys.argv[1]):
for f in files:
file_path = os.path.join(subdir, f)
if os.path.isfile(file_path):
space += os.stat(file_path).st_size
sys.stdout.write("Total: {:d}\n".format(space))
$> python test_code2.py $HOME
Total: 76763501905
答案 0 :(得分:4)
正如Antti Haapala在评论中已经提到的那样,脚本不会在符号链接上中断,而是在损坏的符号链接上。避免这种情况的一种方法是将现有脚本作为起点,使用try/except
:
#! /usr/bin/python2
import os
import sys
space = 0L # L means "long" - not necessary in Python 3
for root, dirs, files in os.walk(sys.argv[1]):
for f in files:
fpath = os.path.join(root, f)
try:
space += os.stat(fpath).st_size
except OSError:
print("could not read "+fpath)
sys.stdout.write("Total: {:d}\n".format(space))
作为副作用,它会为您提供有关可能断开的链接的信息。
答案 1 :(得分:1)
是的,os.path.isfile
是可行的方法。但是,以下版本可能更节省内存。
for subdir, dirs, files in os.walk(sys.argv[1]):
paths = (os.path.join(subdir, f) for f in files)
space = sum(os.stat(path).st_size for path in paths if os.path.isfile(path))