55 строки
1.6 KiB
Python
55 строки
1.6 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT license.
|
|
|
|
from functools import wraps
|
|
|
|
class DocInherit(object):
|
|
"""
|
|
Docstring inheriting method descriptor
|
|
The class itself is also used as a decorator doc_inherit decorator
|
|
|
|
"""
|
|
|
|
def __init__(self, mthd):
|
|
self.mthd = mthd
|
|
self.name = mthd.__name__
|
|
|
|
def __get__(self, obj, cls):
|
|
if obj:
|
|
return self.get_with_inst(obj, cls)
|
|
else:
|
|
return self.get_no_inst(cls)
|
|
|
|
def get_with_inst(self, obj, cls):
|
|
#overridden = getattr(super(cls, obj), self.name, None)
|
|
for parent in cls.__mro__[1:]:
|
|
overridden = getattr(parent, self.name, None)
|
|
if overridden:
|
|
break
|
|
|
|
@wraps(self.mthd, assigned=('__name__', '__module__', '__doc__'))
|
|
def f(*args, **kwargs):
|
|
return self.mthd(obj, *args, **kwargs)
|
|
|
|
return self.use_parent_doc(f, overridden)
|
|
|
|
def get_no_inst(self, cls):
|
|
for parent in cls.__mro__[1:]:
|
|
overridden = getattr(parent, self.name, None)
|
|
if overridden:
|
|
break
|
|
|
|
@wraps(self.mthd, assigned=('__name__', '__module__', '__doc__'))
|
|
def f(*args, **kwargs):
|
|
return self.mthd(*args, **kwargs)
|
|
|
|
return self.use_parent_doc(f, overridden)
|
|
|
|
def use_parent_doc(self, func, source):
|
|
if source is None:
|
|
raise NameError("Can't find '%s' in parents" % self.name)
|
|
if func.__doc__ is None:
|
|
func.__doc__ = source.__doc__
|
|
return func
|
|
|