File: fields.py

package info (click to toggle)
python-django-tree-queries 0.20-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 388 kB
  • sloc: python: 2,269; makefile: 26; sh: 6
file content (13 lines) | stat: -rw-r--r-- 420 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
 
from django.db import models

from tree_queries.forms import TreeNodeChoiceField


class TreeNodeForeignKey(models.ForeignKey):
    def deconstruct(self):
        name, _path, args, kwargs = super().deconstruct()
        return (name, "django.db.models.ForeignKey", args, kwargs)

    def formfield(self, **kwargs):
        kwargs.setdefault("form_class", TreeNodeChoiceField)
        return super().formfield(**kwargs)