diff --git a/apps/assets/api.py b/apps/assets/api.py index ecbf4289f..e1e1900e8 100644 --- a/apps/assets/api.py +++ b/apps/assets/api.py @@ -1,3 +1,45 @@ # ~*~ coding: utf-8 ~*~ -# +from rest_framework import serializers +from .models import ( + AssetGroup,Asset,IDC,AssetExtend +) +from rest_framework import viewsets,serializers +class AssetGroupSerializer(serializers.ModelSerializer): + class Meta: + model = AssetGroup + #exclude = [ + #'password', 'first_name', 'last_name', 'secret_key_otp', + #'private_key', 'public_key', 'avatar', + #] + +class AssetSerializer(serializers.ModelSerializer): + class Meta: + model = Asset + #fields = ('id', 'title', 'code', 'linenos', 'language', 'style') + +class IDCSerializer(serializers.ModelSerializer): + class Meta: + model = IDC + #fields = ('id', 'title', 'code', 'linenos', 'language', 'style') + +class AssetGroupViewSet(viewsets.ModelViewSet): + """ + API endpoint that allows AssetGroup to be viewed or edited. + """ + queryset = AssetGroup.objects.all() + serializer_class = AssetGroupSerializer + +class AssetViewSet(viewsets.ModelViewSet): + """ + API endpoint that allows Asset to be viewed or edited. + """ + queryset = Asset.objects.all() + serializer_class = AssetSerializer + +class IDCViewSet(viewsets.ModelViewSet): + """ + API endpoint that allows IDC to be viewed or edited. + """ + queryset = IDC.objects.all() + serializer_class = IDCSerializer \ No newline at end of file