Как добавить кода в чужую функцию?

Question

[marataziat]

Уменя идея сделать requests модуль безопасным к атаке ssrf. Я хочу сделать проверку ip ресурса на которой будет посылатся запрос, если он из локальной сети то отменить запрос.
Сюда я добавлю просто свою проверку: https://github.com/kennethreitz/requests/blob/bedd...
Я знаю что можно просто сделать requests.models.prepare_url=moyafunctia но тогда мне придется полностью переписать всю функцию! Есть способы чтобы я добавил кода а не заменял всю функцию полностью?

Comments

[Влад Григорьев]

форкните репу.
поправьте в форке.
используйте форкнутую версию.

Answer 1

[marataziat]

https://stackoverflow.com/a/2789542/7415288

Comments

[Влад Григорьев]

это плохое решение.

[marataziat]

Влад Григорьев, Почему? Нормальное решение вроде

[Влад Григорьев]

marataziat, monkey patching - это решение которое добавляет неопределенности, возможности выстрелить себе в ногу. усложняет отладку и разбор другим людям.
Если есть потребность менять код то лучше или отнаследоватся от класса и использовать свой или форкать либу.

[Влад Григорьев]

class A:
    def blabla(self):
        print('blabla')


class С:
    def blabla(self):
        print('blabla')


def decor(fn):
    def verified(*args, **kwargs):
        print('verified: blabla')
        return fn(*args, **kwargs)

    return verified


# Зачем то добавили декоратор, но все про это забуд пока не сломается.
A.blabla = decor(A.blabla)


class B(С):
    @decor
    def blabla(self):
        """
        Переопредили метод, для понятных целей. при использовании кастомного класса
        будет понятнее что поведение может отличатся от задокументированного.
        :return:
        """
        super().blabla()


B().blabla()
A().blabla()

Answer 2

[LODIII]

То что Вы предлагаете называется monkey patching/
Думаю Вам все же лучше найти место в своем коде для такого анализа