from django.contrib import admin

from rest_framework.authtoken.admin import TokenAdmin
from .models import Profile, Team



TokenAdmin.raw_id_fields = ['user']


@admin.register(Profile)
class ProfileAdmin(admin.ModelAdmin):
    list_display = ('user', 'likedSongs')  # Add other fields if necessary


@admin.register(Team)
class TeamAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'last_name', 'position', 'img_person')