나는 종종 첫 번째 인수가 많은 다른 클래스 중 하나를 가리키는 명령 줄 유틸리티를 작성한다는 것을 알게되었습니다. 예를 들어 ./something.py feature command —-arguments
, where Feature
는 클래스이고 command
해당 클래스의 메소드입니다. 이것을 쉽게 만드는 기본 클래스가 있습니다.
이 기본 클래스는 모든 서브 클래스와 함께 디렉토리에 상주한다고 가정합니다. 그런 다음 전화를 걸어 ArgBaseClass(foo = bar).load_subclasses()
사전을 반환합니다. 예를 들어 디렉토리가 다음과 같은 경우
- arg_base_class.py
- feature.py
feature.py
구현을 가정하면 class Feature(ArgBaseClass)
위의 호출 load_subclasses
이 반환 { 'feature' : <Feature object> }
됩니다. 같은 kwargs
( foo = bar
)이 Feature
수업에 전달됩니다 .
#!/usr/bin/env python3
import os, pkgutil, importlib, inspect
class ArgBaseClass():
# Assign all keyword arguments as properties on self, and keep the kwargs for later.
def __init__(self, **kwargs):
self._kwargs = kwargs
for (k, v) in kwargs.items():
setattr(self, k, v)
ms = inspect.getmembers(self, predicate=inspect.ismethod)
self.methods = dict([(n, m) for (n, m) in ms if not n.startswith('_')])
# Add the names of the methods to a parser object.
def _parse_arguments(self, parser):
parser.add_argument('method', choices=list(self.methods))
return parser
# Instantiate one of each of the subclasses of this class.
def load_subclasses(self):
module_dir = os.path.dirname(__file__)
module_name = os.path.basename(os.path.normpath(module_dir))
parent_class = self.__class__
modules = {}
# Load all the modules it the package:
for (module_loader, name, ispkg) in pkgutil.iter_modules([module_dir]):
modules[name] = importlib.import_module('.' + name, module_name)
# Instantiate one of each class, passing the keyword arguments.
ret = {}
for cls in parent_class.__subclasses__():
path = cls.__module__.split('.')
ret[path[-1]] = cls(**self._kwargs)
return ret