from django.db.models import ForeignKey
from ajaxfkey.forms import AjaxForeignKeyFormField


class AjaxForeignKey(ForeignKey):

    def __init__(self, *args, **kwargs):
        ForeignKey.__init__(self, *args, **kwargs)


    def get_manipulator_fields(self, *args, **kwargs):
        return [AjaxForeignKeyFormField(field_name = self.name,
                                        field      = self)]