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

81

""" 

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.response import Response 

from django.db import transaction 

from storageadmin.models import (Share, Snapshot, NFSExport, SambaShare) 

from fs.btrfs import (update_quota, rollback_snap) 

from storageadmin.serializers import ShareSerializer 

from storageadmin.util import handle_exception 

import rest_framework_custom as rfc 

from clone_helpers import create_clone 

from share import ShareMixin 

from django.core.exceptions import ObjectDoesNotExist 

import logging 

logger = logging.getLogger(__name__) 

 

 

class ShareCommandView(ShareMixin, rfc.GenericView): 

    serializer_class = ShareSerializer 

 

    def _validate_share(self, request, sid): 

        try: 

            return Share.objects.get(id=sid) 

        except ObjectDoesNotExist: 

            e_msg = ('Share with id: {} does not exist'.format(sid)) 

            handle_exception(Exception(e_msg), request) 

 

    def _validate_snapshot(self, request, share): 

        try: 

            snap_name = request.data.get('name', '') 

            return Snapshot.objects.get(share=share, name=snap_name) 

        except ObjectDoesNotExist: 

            e_msg = ('Snapshot(%s) does not exist for this Share(%s)' % 

                     (snap_name, share.name)) 

            handle_exception(Exception(e_msg), request) 

 

    @transaction.atomic 

    def post(self, request, sid, command): 

        with self._handle_exception(request): 

            share = self._validate_share(request, sid) 

 

            if (command == 'clone'): 

                new_name = request.data.get('name', '') 

                return create_clone(share, new_name, request, logger) 

 

            if (command == 'rollback'): 

                snap = self._validate_snapshot(request, share) 

 

                if (NFSExport.objects.filter(share=share).exists()): 

                    e_msg = ('Share(%s) cannot be rolled back as it is ' 

                             'exported via nfs. Delete nfs exports and ' 

                             'try again' % share.name) 

                    handle_exception(Exception(e_msg), request) 

 

                if (SambaShare.objects.filter(share=share).exists()): 

                    e_msg = ('Share(%s) cannot be rolled back as it is shared' 

                             ' via Samba. Unshare and try again' % share.name) 

                    handle_exception(Exception(e_msg), request) 

 

                rollback_snap(snap.real_name, share.name, share.subvol_name, 

                              share.pool) 

                update_quota(share.pool, snap.qgroup, share.size * 1024) 

                share.qgroup = snap.qgroup 

                share.save() 

                snap.delete() 

                return Response()