Merge branch 'master' of code.simcu.com:jumpserver/jumpserver

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

apps/assets/urls.py

@@ -1,7 +1,14 @@
 # coding:utf-8
-from django.conf.urls import url
+from django.conf.urls import url,include
 from .views import *
-
+from .api import (
+    AssetGroupViewSet,AssetViewSet,IDCViewSet
+)
+from rest_framework import routers
+router = routers.DefaultRouter()
+router.register(r'assetgroup', AssetGroupViewSet)
+router.register(r'asset', AssetViewSet)
+router.register(r'idc', IDCViewSet)
 app_name = 'assets'
 
 urlpatterns = [
@@ -9,4 +16,5 @@ urlpatterns = [
     url(r'^list/$', AssetListView.as_view(), name='asset-list'),
     url(r'^(?P<pk>[0-9]+)/delete/$', AssetDeleteView.as_view(), name='asset-list'),
     url(r'^(?P<pk>[0-9]+)/detail/$', AssetDetailView.as_view(), name='asset-detail'),
+    url(r'^api/v1.0/', include(router.urls)),
 ]