làm cho các đoạn trong Python tự động tìm các gói

Question

Khi mô tả gói python trong setup.py trong Python, có cách nào để tự động lấy mọi thư mục có một số __init__.py trong đó và bao gồm đó là gói phụ?

tức là nếu cấu trúc là:

mypackage/__init__.py 
mypackage/a/__init__.py 
mypackage/b/__init__.py 

Tôi muốn tránh thực hiện:

packages = ['mypackage', 'mypackage.a', 'mypackage.b'] 

và thay vào đó chỉ làm:

packages = ['mypackage'] 

và có nó tự động tìm những thứ như a và b vì chúng có tệp init. cảm ơn.

Answer 1

Cách đơn giản nhất (mà tôi biết) là sử dụng pkgutil.walk_packages để mang lại gói:

from distutils.core import setup 
from pkgutil import walk_packages 

import mypackage 

def find_packages(path=__path__, prefix=""): 
    yield prefix 
    prefix = prefix + "." 
    for _, name, ispkg in walk_packages(path, prefix): 
     if ispkg: 
      yield name 

setup(
    # ... snip ... 
    packages = list(find_packages(mypackage.__path__, mypackage.__name__)), 
    # ... snip ... 
) 

Answer 2

Tôi muốn giới thiệu cách sử dụng find_packages() chức năng có sẵn với setuptools như:

from setuptools import setup, find_packages 

và sau đó làm

packages=find_packages() 

Answer 3

import re, os 
def find_packages(path='.'): 
    ret = [] 
    for root, dirs, files in os.walk(path): 
     if '__init__.py' in files: 
      ret.append(re.sub('^[^A-z0-9_]+', '', root.replace('/', '.'))) 
    return ret 

setup(
... 
packages = find_packages() 
... 
) 

Answer 4

Tương tự như câu trả lời dm76, chỉ là tôi cũng có các bài kiểm tra trong repo của tôi, vì vậy tôi sử dụng:

from setuptools import find_packages 

packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),