使用os模块实现展示目录下的文件和文件夹

Windows 10家庭中文版，Python 3.6.4

今天学习了os模块，下面是使用它开发的一个展示目录下的文件和文件夹的函数，代码如下：

 import os

 # deep大于等于1的整数，默认为1；
 # deep为1时，用于显示当前目录下的文件和文件夹；
 # deep控制显示的当前目录下的层级，当前为1，每深入一层（进入下一级文件夹），加1；
 def listdir(deep = 1):
     # path is an absolute path
     def listdir_inner(path, ilevel = 0):
         if ilevel == deep:
             return

         # 添加到每条文件信息前面的缩进，每深入一层增加4个空格
         prefix = ' ' * (4 * ilevel)

         # 检查目录是否存在
         if not os.path.exists(path):
             print(prefix, 'Error: Path (%s) does not exist.' % path, sep = '')
             return

         # 如果path代表文件，立即返回
         if os.path.isfile(path):
             print(prefix, 'Warning: ', path, ' is a file!', sep = '')
             return

         # 统一使用Linux风格的路径
         path = path.replace('\\', '/')

         if not path.endswith('[/\\]'):
             path += '/'

         # 判断目录path是否可以执行os.listdir操作
         # 在Windows下产生了PermissionError错误，但用os.access检查却是可以访问的
         # Linux上可能使用os.stat()来检查文件或文件夹的权限
         perm_ok = True
         try:
             dlist = os.listdir(path)
         except PermissionError as err:
             perm_ok = False
             print(prefix, 'Error: ', err.strerror, sep = '')
         except Exception as err:
             perm_ok = False
             print(prefix, 'Error: ', err.strerror, sep = '')

         # 展示目录下的文件和文件夹
         if perm_ok:
             for item in dlist:
                 if os.path.isfile(path + item):
                     # 文件，直接输出
                     print(prefix, 'F-', item, sep = '')
                 elif os.path.isdir(path + item):
                     # 子目录，输出信息
                     print(prefix, 'D-', item, sep = '')
                     # 寻找子目录下的文件和文件夹
                     # 注意，层级加1了
                     listdir_inner(path + item + '/', ilevel + 1)
                 else:
                     # 其它类型，可能是link、mount等
                     print(prefix, 'X-', item, sep = '')

     return listdir_inner

 # 测试
 if __name__ == '__main__':
     d2 = listdir(deep = 2)

     print('展示目录下的文件：')
     # 展示D盘下的文件及文件夹目录，深度为2
     d2('d:/')

使用方法见测试——调用listdir函数建立显示不同深度的函数，再使用新建的函数 展示传入的目录（绝对路径）下的文件和文件夹。

测试可以更多的，在目前的代码下，使用d:、d:/、d:\\三个方式都是访问D盘下的文件夹和目录。

调试过程中遇到错误：

C盘有下一个识别为文件夹的目录无法执行listdir：

使用os.access(...)函数检查其读写执行权限时，都是可以的。

使用os.stat(...)函数检测其权限，也是可以的，但却无法执行listdir，而且，如果此文件夹后面带着反斜杠传入os.stat(...)，os.stat(...)执行就会报错。

后来网上找到一篇博文，其中使用win32security来检测文件权限，但发生错误：

此问题后面也没有继续深究了，先这样。

可能的测试输入：使用unitest模块或Flask学习时用到的pytest（都没用过）

d:

d:/

d:\\

c

d

e

d:/existed_file

d:/unexisted_file

c:/

...

下面是一个测试结果：

更改deep为1：

函数改进方向：

添加测试；

添加更专业的注释；

更改函数名为ulistdir，开头的u表示为用户自定义；

deep参数没有检测；

print(...)函数的使用方法需要改进，每个都有sep=''这个参数；

目前使用嵌套函数，可以改造为类来实现；

测试过检测本机Windows上的目录，扩展展示网络主机上的文件夹；

还有什么呢？

os.stat(...)函数还需要深入学习，以及Linux下的stat命令，以及ls命令。