Pythonのクラスをそのまま保存する方法について

Question

前提

save.py、load.pyの２ファイルと、生成ファイル cfg.pickleで動かすことが前提です。

やりたいこと

Pythonのクラスとその変数をそのまま、pickleなどに保存して、別ファイルから読み込む方法はありますか？

不具合のプログラム

save.py

class TEST():
    p = 100 # 静的な変数。その他多数あり。

    def __init__(self):
        # 基本的にはインスタンス化しないクラスです。よって、
        # self.p = 100 とはしません。
        pass

print(TEST.p) # 100

import pickle
with open('TEST.pickle', mode='wb') as f:
    pickle.dump(TEST, f)

load.py

import pickle
with open('TEST.pickle', mode='rb') as f:
    TEST = pickle.load(f)

print(TEST.p)

エラー内容

AttributeError: 'TEST' object has no attribute 'p'

実現したいこと

なるべくクラスをいじらないで、静的なクラス変数を読み込ませたいです。
また、クラスだけ外部ファイルにすることも仕様上できません。
なにか良い方法はないでしょうか。

Comments

[TakaiY]

「静的な変数」というのは値を変更しない変数を意味していますか？ pickleをloadする場合、そのオブジェクのクラス定義は読み込み先でも読み込む必要があり、定義を読めば変数は手に入るので、pickleをやりとりする必要はなくなります。

なぜ、変更されないオブジェクトをpickleでやりとりする必要があるのでしょうか。

[quickquip]

TakaiYさんの指摘と同じくなぜこんなことをしなければならないか? がはっきりしません。

すでに回答がついていますが、書いてある save.py と load.py をそのまま実行すると load.py はエラーを起こします。load.py にも TESTクラスの定義がないといけないです。（完全名 __main__.TEST が要ります）
実際のコードは"こうなってはいない"のでしょう。となると、質問に抽出された説明に"本当に解決したいこと"が反映されていないのでは? という疑いが出てきます。

[teefpc]

＞なぜ、変更されないオブジェクトをpickleでやりとりする必要があるのでしょうか。
普通はconfig.py のような外部ファイルを読み込めばいいのですが、試しに、save.pyに直接config値を、マジックナンバー同然に書き込んで、そのクラスを別ファイルで読み込むことをたくらみました。ですので、普通の開発ではなく、個人的な研究課題でした。

[quickquip]

> また、クラスだけ外部ファイルにすることも仕様上できません。
save.py の存在自体が「クラスだけ外部ファイルになっている」状況なので、矛盾しています。
極めて不合理な前提条件かと。

load.py に from save import TEST と書けばよい が解決方法にならない理由が不明です。

[TakaiY]

試しに、save.pyに直接config値を、マジックナンバー同然に書き込んで、そのクラスを別ファイルで読み込むことをたくらみました。

であれば僕も「 from save import TEST」が正解だと思いますね。

[teefpc]

なるほど。そうですね。ありがとうございました。

Answer 1

pickleの公式ドキュメントの例に、type型のインスタンスをシリアル化してloadする時にクラス定義を必要としない例が載っています。
保存する属性を自分で列挙しているのでやりたいこととはずれているだろうとは思いますが、ただの研究なら目を通して置いていいかと思います。
https://docs.python.org/ja/3/library/pickle.html#custom-reduction-for-types-functions-and-other-objects

Comments

[teefpc]

ありがとうございます！参考にさせていただきます！

Answer 2

load.py にも class TEST(): ... のクラス定義が必要です。

同じ定義を複数のファイルに書くのは無駄なので、class TEST の定義だけの
ファイル test.py を作り、save.py と load.py の両方の先頭に
from test import TEST を置けばよいでしょう。

追記
外部ファイルがダメなら、最初に述べたように
load.py にclass TEST の定義を書くしかないのでは？

追記2
load.py に from save import TEST と書けばよいようです。