from django import forms

from django.forms import ModelForm
from .models import Employee, Contract, ChargeNumber

class EmployeeForm(ModelForm):
    class Meta:
        model = Employee
        # TODO: fix slary to be salary
        fields = ["primaryAddress","workAddress", "slary"]

class ContractForm(ModelForm):
    class Meta:
        model = Contract
        fields = ["contract_type","name","proposed_amount","baseline_amount","funded_amount","baseline_start","baseline_end"]

class ChargeNumberForm(ModelForm):
    class Meta:
        model = ChargeNumber
        fields = ["charge_number_type","amount", "start_date","end_date"]