Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

""" 

Copyright (c) 2012-2013 RockStor, Inc. <http://rockstor.com> 

This file is part of RockStor. 

 

RockStor is free software; you can redistribute it and/or modify 

it under the terms of the GNU General Public License as published 

by the Free Software Foundation; either version 2 of the License, 

or (at your option) any later version. 

 

RockStor is distributed in the hope that it will be useful, but 

WITHOUT ANY WARRANTY; without even the implied warranty of 

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 

General Public License for more details. 

 

You should have received a copy of the GNU General Public License 

along with this program. If not, see <http://www.gnu.org/licenses/>. 

""" 

 

from rest_framework.views import APIView 

from rest_framework.response import Response 

from rest_framework.authentication import (BasicAuthentication, 

                                           SessionAuthentication) 

from storageadmin.auth import DigestAuthentication 

from rest_framework.permissions import IsAuthenticated 

from storageadmin.util import handle_exception 

from storageadmin.models import DashboardConfig 

from storageadmin.serializers import DashboardConfigSerializer 

 

import logging 

logger = logging.getLogger(__name__) 

 

 

class DashboardConfigView(APIView): 

    authentication_classes = (DigestAuthentication, SessionAuthentication, 

                              BasicAuthentication,) 

    permission_classes = (IsAuthenticated,) 

 

    def get(self, request): 

        try: 

            current_user = request.user 

            dcs = DashboardConfig.objects.filter(user__pk=current_user.id) 

            if len(dcs) > 0: 

                dc = dcs[0] 

                return Response(DashboardConfigSerializer(dc).data) 

            else: 

                return Response() 

        except Exception as e: 

            handle_exception(e, request) 

 

    def post(self, request): 

        try: 

            current_user = request.user 

            widgets = request.data['widgets'] 

            dcs = DashboardConfig.objects.filter(user__pk=current_user.id) 

            if len(dcs) > 0: 

                dc = dcs[0] 

                dc.widgets = widgets 

            else: 

                dc = DashboardConfig(widgets=widgets) 

                dc.user_id = current_user.id 

            dc.save() 

            return Response(DashboardConfigSerializer(dc).data) 

        except Exception as e: 

            handle_exception(e, request) 

 

    def put(self, request): 

        try: 

            current_user = request.user 

            widgets = request.data['widgets'] 

            dcs = DashboardConfig.objects.filter(user__pk=current_user.id) 

            if len(dcs) > 0: 

                dc = dcs[0] 

                dc.widgets = widgets 

            else: 

                dc = DashboardConfig(widgets=widgets) 

                dc.user_id = current_user.id 

            dc.save() 

            return Response(DashboardConfigSerializer(dc).data) 

        except Exception as e: 

            handle_exception(e, request)