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

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