通过Python打印基本名称

Question

如何在主文件夹和子文件夹中打印Python文件的基本名称？

我的尝试

#!/usr/bin/python    

import os    
import sys    

def dir_basename (dir_name):    
        for dirpath, dirnames, filenames in os.walk(dir_name):
                for fname in filenames:
                        print os.path.basename(fname)            // Problem here!

if len(sys.argv) != 1:
        u = "Usage: dir_basename <dir_name>\n"
        sys.stderr.write(u)                       
        sys.exit(1)            

dir_basename ( sys.argv[1] )

---

第一个问题通过一个错误解决了

---

第二个问题:代码输出失败了

man.aux
about_8php.tex
refman.pdf
successful_notice.php
...

我希望得到一个输出

 aux
 tex
 pdf
 php
 ...

Answer 1

让我稍微解释一下调试方法.

当你遇到这种情况时len(sys.argv) != 1,你应该问自己:"实际价值是len(sys.argv)多少？为什么会这样？".答案是:

>>> len(sys.argv)
2
>>> sys.argv
['/tmp/basename.py', '/path/to/home/Desktop/pgCodes/']

我想现在问题应该变得更清楚了.

编辑:要解决第二个问题,您感兴趣的内容称为文件扩展名或后缀,而不是基本名称.这是完整的解决方案:

import sys, os

def iflatten(xss):
    'Iterable(Iterable(a)) -> Iterable(a)'
    return (x for xs in xss for x in xs)

def allfiles(dir):
    'str -> Iterable(str)'
    return iflatten(files for path, dirs, files in os.walk(dir))

def ext(path):
    'str -> str'
    (root, ext) = os.path.splitext(path)
    return ext[1:]

def main():
    assert len(sys.argv) == 2, 'usage: progname DIR'
    dir = sys.argv[1]

    exts = (ext(f) for f in allfiles(dir))
    for e in exts:
        print e

if __name__ == '__main__':
    main()