Python monkey transpching

Question

Potrzebuję klasy odpowiedzi żądania monkeypatch (wersja 1.0.4, aktualna na podstawie tego pytania), aby dodać dodatkowe metody.

mam ten kod:

import requests 

class Response(requests.models.Response): 
    def hmm(self): 
     return 'ok' 

requests.models.Response = Response 

r = requests.get('http://bbc.co.uk') 

print r 

on zawiedzie, gdy oryginalny Response nazywa super() - https://github.com/kennethreitz/requests/blob/master/requests/models.py#L391

Myślę, że to dlatego, że pogubi, bo zastąpiły klasę, czuję jak robię coś głupiego, jakieś pomysły? Z góry dziękuję.

Answer 1

Byłbyś lepiej po prostu dodając swoją funkcję bezpośrednio do klasy:

def hmm(self): 
    return 'ok' 
requests.models.Response.hmm = hmm 

to działa dobrze:

>>> import requests 
>>> def hmm(self): 
...  return 'ok' 
... 
>>> requests.models.Response.hmm = hmm 
>>> r = requests.get('http://bbc.co.uk') 
>>> print r 
<Response [200]> 
>>> r.hmm() 
'ok' 
>>> requests.__version__ 
'1.0.4' 

Answer 2

Nie sądzę można małpa-łata, że ​​typ tak. Podczas importowania requests, wszystkie modułysą inicjowane. A ponieważ cała biblioteka używa wielokrotnie biblioteki from xy import Request, będzie ona zawierała dokładne odniesienie do rzeczywistego typu. Dopiero potem zastępujesz typ odpowiedzi w module modeli, więc wpływa to tylko na kolejne importowanie.

Jeśli nie przejdziesz przez wszystkie moduły i ręcznie zastąpisz odniesienie do odpowiedzi swoim nowym typem, nadal będą one używać oryginalnego, co sprawi, że twój patch będzie bezużyteczny.

Zamiast tego powinieneś zachować oryginalny typ, ale rozwinąć go bezpośrednio, jak sugerował Martijn.

Answer 3

Wystarczy szybkie ustalenie za pomocą setattr, ale to trochę brzydkie (ale semantycznie równoważne @Martijn answer):

def hmm(self): 
    return 'OK - %s' % self.status_code 

setattr(requests.models.Response, 'hmm', hmm) 

r = requests.get('http://bbc.co.uk') 
print r.hmm() 
# prints 
# OK - 200