获取当前目录中的所有文件和目录python

我正在编写一个python包管理器，并且我正在根据当前目录中的程序/包生成__all__对象，但我也想忽略任何非包。即，如果我有以下目录结构：获取当前目录中的所有文件和目录python

project 
|------__init__.py 
|------prog1.py 
|------package 
    |------__init__.py 
    |------prog2.py 
|------resources 
    |------ping.png

我想__all__变量如下所示：

__all__ = ['prog1', 'package']

我曾尝试过各种东西，但在所有诚实，我不知道我在做什么。

这是我最后一次尝试：

import os 
from os import path 


def config_all(directory): 
    __all__ = [] 
    directory = path.abspath(directory) 
    for root, dirs, files in os.walk(directory, topdown=False): 
     for d in dirs: 
      os.chdir("/czar/".join(path.abspath(d).rsplit("/", 1))) 
      v = list(os.walk('.'))[0][2] 
      if "__init__.py" in v: 
       __all__.append(d) 
      for i in v: 
       while i in __all__: 
        __all__.remove(i) 
      os.chdir('..') 
     for f in files: 
      __all__.append(path.join(root, f).rsplit("/", 1)[1].rsplit(".", 1)[0]) 
    return __all__ 

__all__ = config_all('czar/') 
print(__all__)

来源

2016-05-31 mcchucklezz

请出示你已经尝试过了，你已经阅读过哪些文档。 –

@BryanOakley我没有展示我已经尝试过的东西，因为它没有像我想要的那样工作，我不希望这篇文章的回复是切题。 – mcchucklezz

不要这样做：不要使'__all__'动态。只需*显式*指定要从模块中导出哪些函数和变量。 – Evert

你希望所有的.py，文件和目录都用一个__init__.py文件指定的目录，使用glob：

def config_all(directory): 
    result = [] 
    for filename in glob.iglob(os.path.join(directory, '*', '__init__.py')): 
     result.append(os.path.basename(os.path.dirname(filename)) 
    for filename in glob.iglob(os.path.join(directory, '*.py'): 
     result.append(os.path.splitext(os.path.basename(filename))[0]) 
    return result

来源

2016-05-31 22:38:06 Daniel

获取当前目录中的所有文件和目录python

回答

相关问题