update for change to etsicatalog
[modeling/etsicatalog.git] / catalog / samples / views.py
1 # Copyright 2017 ZTE Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import logging
16 import traceback
17
18 from rest_framework import status
19 from rest_framework.response import Response
20 from rest_framework.views import APIView
21
22 logger = logging.getLogger(__name__)
23
24
25 class SampleList(APIView):
26     """
27     List all samples.
28     """
29     def get(self, request, format=None):
30         logger.debug("get")
31         return Response({"status": "active"})
32
33
34 class CallbackSample(APIView):
35     """
36     Callback Sample.
37     """
38     def get(self, request, format=None):
39         logger.debug("Callback Sample")
40         return Response(data={}, status=status.HTTP_204_NO_CONTENT)
41
42
43 class TablesList(APIView):
44     def delete(self, request, modelName):
45         logger.debug("Start delete model %s", modelName)
46         try:
47             modelNames = modelName.split("-")
48             for name in modelNames:
49                 model_obj = eval("models.%s.objects" % name)
50                 model_obj.filter().delete()
51                 logger.debug("End delete model %s", name)
52         except:
53             logger.error(traceback.format_exc())
54             return Response(data={"error": "failed"}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
55         return Response(data={}, status=status.HTTP_204_NO_CONTENT)
56
57     def get(self, request, modelName):
58         logger.debug("Get model %s", modelName)
59         count = 0
60         try:
61             model_obj = eval("models.%s.objects" % modelName)
62             count = len(model_obj.filter())
63         except:
64             logger.error(traceback.format_exc())
65             return Response(data={"error": "failed"}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
66         return Response(data={"count": count}, status=status.HTTP_200_OK)