Расширение встроенных классов в Python
-
20-08-2019 - |
Вопрос
Как я могу расширить встроенный класс в Python?Я хотел бы добавить метод в класс str.
Я немного поискал, но нашел только старые сообщения. Надеюсь, кто-нибудь знает что-нибудь новое.
Решение
Просто создайте подкласс типа
>>> class X(str):
... def my_method(self):
... return int(self)
...
>>> s = X("Hi Mom")
>>> s.lower()
'hi mom'
>>> s.my_method()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 3, in my_method
ValueError: invalid literal for int() with base 10: 'Hi Mom'
>>> z = X("271828")
>>> z.lower()
'271828'
>>> z.my_method()
271828
Другие советы
Одним из способов может быть использование концепции «повторного открытия класса» (существующей изначально в Ruby), которую можно реализовать в Python с помощью декоратора класса.Пример приведен на этой странице:http://www.ianbicking.org/blog/2007/08/opening-python-classes.html
Я цитирую:
Я думаю, что с декораторами классов вы могли бы сделать это:
@extend(SomeClassThatAlreadyExists)
class SomeClassThatAlreadyExists:
def some_method(self, blahblahblah):
stuff
Реализовано так:
def extend(class_to_extend):
def decorator(extending_class):
class_to_extend.__dict__.update(extending_class.__dict__)
return class_to_extend
return decorator
Предполагая, что вы не можете изменить встроенные классы.Чтобы имитировать «повторное открытие класса», например Ruby в Python3, где __dict__
является объектом MappingProxy, а не объектом dict:
def open(cls):
def update(extension):
for k,v in extension.__dict__.items():
if k != '__dict__':
setattr(cls,k,v)
return cls
return update
class A(object):
def hello(self):
print('Hello!')
A().hello() #=> Hello!
#reopen class A
@open(A)
class A(object):
def hello(self):
print('New hello!')
def bye(self):
print('Bye bye')
A().hello() #=> New hello!
A().bye() #=> Bye bye
Я также мог бы написать функцию-декоратор open:
def open(cls):
def update(extension):
namespace = dict(cls.__dict__)
namespace.update(dict(extension.__dict__))
return type(cls.__name__,cls.__bases__,namespace)
return update