Pythonのライブラリに処理を追加したい

Question

前提・実現したいこと

camelotのimage_processing.pyのadaptive_thresholdを変更して自作の処理を追加したいのですが

ライブラリの処理を変更したい場合はどのようにすればよろしいのでしょうか教えてください

よろしくお願いいたします。

image_processing.pyのadaptive_thresholdを書き換え下記のソースコードの処理に変更したい
https://github.com/camelot-dev/camelot/blob/420d5aa6242fc19100f285de5a1d4781d1cd6254/camelot/image_processing.py#L7

adaptive_thresholdの呼び出し側
https://github.com/camelot-dev/camelot/blob/420d5aa6242fc19100f285de5a1d4781d1cd6254/camelot/parsers/lattice.py#L27

発生している問題・エラーメッセージ

該当のソースコード

import camelot

# パッチ

def my_method(imagename, process_background=False, blocksize=15, c=-2):

    print("処理")

    img = cv2.imread(imagename)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    if process_background:
        threshold = cv2.adaptiveThreshold(
            gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, blocksize, c
        )
    else:
        threshold = cv2.adaptiveThreshold(
            np.invert(gray),
            255,
            cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
            cv2.THRESH_BINARY,
            blocksize,
            c,
        )
    return img, threshold

# ライブラリを直接変更せずにパッチ摘要できる方法があれば

tables = camelot.read_pdf("data.pdf", pages="all", split_text=True)

試したこと

Pythonでライブラリの処理を上書きする
https://qiita.com/Asayu123/items/8d8da9911dd0c3296a81

# パターン1
camelot.image_processing.adaptive_threshold = my_method
camelot.image_processing.adaptive_threshold("data.png")

# パターン2
camelot.adaptive_threshold = my_method
camelot.adaptive_threshold("data.png")

両方を試してみましたが直接呼び出す場合は変更されているのですが

tables = camelot.read_pdf("data.pdf", pages="all", split_text=True)

のように内部で呼び出される場合は変更されていません

追記

• image_processing.pyを書き換え

https://github.com/camelot-dev/camelot/blob/420d5aa6242fc19100f285de5a1d4781d1cd6254/camelot/image_processing.py#L7

を直接変更後に「import camelot」は成功

• lattice.pyを書き換え

https://github.com/camelot-dev/camelot/blob/420d5aa6242fc19100f285de5a1d4781d1cd6254/camelot/parsers/lattice.py#L36

パッチを追加後

「adaptive_threshold = my_threshold」

で「import camelot」は成功しました

ライブラリを直接変更せずにパッチを適用することはできないのでしょうか？

補足情報（FW/ツールのバージョンなど）

Python 3.8.6
Google Colaboratry

Answer 1

read_pdf を呼び出した時に、
何処で adaptive_threshold が呼ばれているのかを探しましょう。

• 原因: Lattice (camelot/parsers/lattice.py) の import 時 に

　image_processingの adaptive_threshold を import し参照してます。
後からimage_processingの adaptive_threshold を変更しても、
lattice.py 内で参照してる adaptive_threshold は変わりません。

• 解決策: adaptive_threshold が呼ばれるのは、Lattice インスタンス生成時なので、

　それ迄に camelot.parsers.lattice.adaptive_threshold を変更。

import camelot.parsers.lattice
camelot.parsers.lattice.adaptive_threshold = my_method

変更したのは、「image_processing.pyのadaptive_threshold」ですが、
read_pdf 内部で使われてるのは、それ迄に import した元の関数。
変更した関数は、「変更後のimport」にのみ適応されます。

他には、import 順序を工夫する事でも解決可能かもしれません（未検証）

Comments

[barobaro]

ありがとうございます
lattice.pyを変更してパッチを適用することはできました

camelotを呼び出し側で書き換えできるといいのですが
もしご存じできたらよろしくお願いします

[barobaro]

無事作成することができました
ありがとうございます