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

""" 

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 django.db import transaction 

from storageadmin.models import User, Setup, OauthApp 

from storageadmin.views import UserListView 

from oauth2_provider.models import Application as OauthApplication 

from django.contrib.auth.models import User as DjangoUser 

from rest_framework.response import Response 

 

 

class SetupUserView(UserListView): 

 

    authentication_classes = () 

    permission_classes = () 

 

    @transaction.atomic 

    def post(self, request): 

        setup = Setup.objects.all()[0] 

        setup.setup_user = True 

        setup.save() 

 

        # Create user 

        res = super(SetupUserView, self).post(request) 

        # Create cliapp id and secret for oauth 

        name = 'cliapp' 

        user = User.objects.get(username=request.data['username']) 

        duser = DjangoUser.objects.get(username=request.data['username']) 

        client_type = OauthApplication.CLIENT_CONFIDENTIAL 

        auth_grant_type = OauthApplication.GRANT_CLIENT_CREDENTIALS 

        app = OauthApplication(name=name, client_type=client_type, 

                               authorization_grant_type=auth_grant_type, 

                               user=duser) 

        app.save() 

        oauth_app = OauthApp(name=name, application=app, user=user) 

        oauth_app.save() 

        return res 

 

    def get(self, request): 

        setup = Setup.objects.all()[0] 

        return Response({'new_setup': not setup.setup_system, }) 

 

    @transaction.atomic 

    def put(self, request): 

        setup = Setup.objects.all()[0] 

        setup.setup_system = True 

        setup.save() 

        return Response({'new_setup': not setup.setup_system, })