Djangoのモデルはすごく便利なのですが、
固定の定義を作らないとダメなので、
何かの設定とかに応じてモデルを作りたい場合はちょっと使えない。
SQLを自前で書いてゴニョゴニョしてもいいんだけど、
ORMは出来ればDjangoの使いたいのでどうにかしたい。
ということで、モデルを動的に作りたくなった場合。
あと、動的に作ったモデルのテーブルも動的に作りたい場合についても。
参考：https://code.djangoproject.com/wiki/DynamicModels
※すごく古いのでちょっと微妙かも。。。
まず、動的にモデル作る本体は下の感じ。
from django.apps import apps
from django.db import models
def create_model_class(cls, name: str, fields=None, app_label: str = "", module: str = "", options=None):
"""指定されたモデルのClassを作成する"""
class Meta(object):
pass
if app_label:
setattr(Meta, "app_label", app_label)
if options is not None:
for key, value in options.items():
setattr(Meta, key, value)
attrs = {"__module__": module, "Meta": Meta}
if fields:
attrs.update(fields)
model = type(name, (models.Model,), attrs)
return model
普通にクラス定義するときにやってる事を、コードで書いてるだけ。肝は↓のとこ。
type(name, (models.Model,), attrs)
models.Modelを継承したクラス定義にする。
で、どう使うかというと↓の感じ。
app_label = "app"
model_name = "DynamicModelName"
ops = {"db_table": "table_name"}
fields = {}
fields["id"] = models.AutoField(primary_key=True, null=False)
fields["test01"] = models.IntegerField(null=False, default=0)
fields["test02"] = models.CharField(null=False, max_length=255, default="")
ret = create_model_class(
name=model_name,
app_label=app_label,
module=".".join([app_label, model_name]),
options=ops,
fields=fields
)
obj = ret()
obj.test01 = 123
obj.test02 = "abc"
obj.save()
q = ret.objects.all()
q.values()
ただ、動的に作ったモデルはDBにはまだテーブル作られてないので、
save()とか、objects.all()とかはもちろん動ない。
テーブルをこの作ったモデル定義が作成する場合はschema-editorを使う。
こんな感じ。
from django.db import connections
connection = connections[con_label]
with connection.schema_editor(collect_sql=False, atomic=True) as se:
se.create_model(model) # ここでCreate Tableが発行される
SQLだけ拾いたいときは下の感じ
with connection.schema_editor(collect_sql=True, atomic=True) as se:
se.create_model(model)
"\n".join(se.collected_sql)
collect_sqlをTrueにすると、操作に対してのSQLがcollected_sqlのlistに
たまってくので、最後に「"\n".join(se.collected_sql)」で全部取るみたいな。
collect_sqlをTrueにした場合は、create_model()とかしても、SQLは発行されない。
あと、作ったモデルは↓の感じで使いまわせる。
from django.apps import apps
django_model_name = model_name.lower()
if django_model_name in apps.all_models[app_label].keys():
ret = apps.all_models[app_label][django_model_name]
q = ret.objects.all()
q.values()
「apps.all_models」がDjangoがmodel定義を全部キャッシュしてある場所。
削除したい場合はここから消せばいい気がするんだけど、
ちゃんと調べてないので、要調査。今のとこ使い終わった場合は↓の感じで消す。
def unload(cls, app_label, model_name):
django_model_name = model_name.lower()
if app_label in apps.all_models:
mdic = apps.all_models[app_label]
if django_model_name in mdic.keys():
mdic.pop(django_model_name)
apps.clear_cache() # これないとapps.get_models()から消えてない
あと、外部キー制約がどうもうまくいかなかった。
他にも上手くいかないオプションはあると思うのですが、
単純なテーブルのORMやるだけなら十分。。。
と、ここまで書いてて、こんなパッケージがある事に気づいた。
後でコード見ておこ。。。
