python隐藏类的属性 重写__dir___在python中重写__dir__方法的正确方法

Python 2.7,3.3类mixin，简化__dir__方法在子类中的实现。希望会有所帮助。

Gist。

import six

class DirMixIn:

""" Mix-in to make implementing __dir__ method in subclasses simpler

"""

def __dir__(self):

if six.PY3:

return super(DirMixIn, self).__dir__()

else:

# code is based on

# http://www.quora.com/How-dir-is-implemented-Is-there-any-PEP-related-to-that

def get_attrs(obj):

import types

if not hasattr(obj, '__dict__'):

return [] # slots only

if not isinstance(obj.__dict__, (dict, types.DictProxyType)):

raise TypeError("%s.__dict__ is not a dictionary"

"" % obj.__name__)

return obj.__dict__.keys()

def dir2(obj):

attrs = set()

if not hasattr(obj, '__bases__'):

# obj is an instance

if not hasattr(obj, '__class__'):

# slots

return sorted(get_attrs(obj))

klass = obj.__class__

attrs.update(get_attrs(klass))

else:

# obj is a class

klass = obj

for cls in klass.__bases__:

attrs.update(get_attrs(cls))

attrs.update(dir2(cls))

attrs.update(get_attrs(obj))

return list(attrs)

return dir2(self)