from rest_framework_filters import FilterSet, RelatedFilter, AllLookupsFilter
from .models import User, Team, ContractorResumeFiles, ContractorResume


class UserFilterSet(FilterSet):
    id = AllLookupsFilter()
    email = AllLookupsFilter()
    first_name = AllLookupsFilter()
    is_active = AllLookupsFilter()
    is_admin = AllLookupsFilter()
    last_name = AllLookupsFilter()
    projects = RelatedFilter('projects.filters.ProjectFilterSet')
    
    class Meta:
        model = User


class TeamFilterSet(FilterSet):
    id = AllLookupsFilter()
    name = AllLookupsFilter()
    owner = RelatedFilter('users.filters.UserFilterSet')

    class Meta:
        model = Team


class ContractorResumeFilterSet(FilterSet):
    id = AllLookupsFilter()
    text = AllLookupsFilter()
    resume_file = AllLookupsFilter()

    class Meta:
        model = ContractorResume


class ContractorResumeFilesFilterSet(FilterSet):
    id = AllLookupsFilter()
    title = AllLookupsFilter()
    img = AllLookupsFilter()
    resume = RelatedFilter('users.filters.ContractorResumeFilterSet')

    class Meta:
        model = ContractorResumeFiles