# -*- coding:utf-8 -*-
from models import Staff, Relation
from django.contrib import admin
from django import forms

class RelationForm(forms.ModelForm):

    def __init__(self, *args, **kwargs):
        super(RelationForm, self).__init__(*args, **kwargs)
        qs = Staff.objects.filter(company_name = self.instance.company_name)
        self.fields['who'].queryset = qs
        self.fields['with_whom '].queryset = qs

class RelationAdmin(admin.ModelAdmin):
    form = RelationForm

admin.site.register(Relation, RelationAdmin)