import json
from django.http import JsonResponse
from .models import WpEsiRadar, WpEsiIndustry, WpEsiFunction, WpEsiMedia, WpEsiTechnology, WpEsiType, WpEsiRegion, \
    WpEsiTag, WpEsiEntity


def auto(request, slug):
    print(slug)
    if slug == 'entity':
        return select(WpEsiEntity)
    if slug == 'radar':
        return select(WpEsiRadar)
    if slug == 'industry':
        return select(WpEsiIndustry)
    if slug == 'function':
        return select(WpEsiFunction)
    if slug == 'media':
        return select(WpEsiMedia)
    if slug == 'technology':
        return select(WpEsiTechnology)
    if slug == 'type':
        return select(WpEsiType)
    if slug == 'region':
        return select(WpEsiRegion)
    if slug == 'tags':
        return select(WpEsiTag)


def select(obj):
    d = [{'text': i.name, 'value': i.id} for i in obj.objects.all()]
    return JsonResponse(d, safe=False)