|
1 | | -from django.urls import NoReverseMatch |
2 | 1 | from django.contrib import messages |
| 2 | +from django.db import transaction |
| 3 | +from django.db.models import F |
3 | 4 | from django.shortcuts import redirect |
4 | | -from django.views.generic import FormView |
5 | | -from django.urls import reverse_lazy |
6 | | -from admin.base.forms import GuidForm |
| 5 | +from django.urls import NoReverseMatch, reverse, reverse_lazy |
| 6 | +from django.utils import timezone |
| 7 | +from django.views.generic import FormView, View |
7 | 8 | from django.contrib.auth.mixins import PermissionRequiredMixin |
| 9 | + |
| 10 | +from admin.base.forms import GuidForm |
8 | 11 | from admin.base.views import GuidView |
9 | | -from osf.models import Guid, GuidMetadataRecord, BaseFileNode |
| 12 | +from admin.files.tasks import purge_file_version_task |
| 13 | +from framework.postcommit_tasks.handlers import enqueue_postcommit_task |
| 14 | +from osf.models import Guid, GuidMetadataRecord, BaseFileNode, NodeLog |
| 15 | +from osf.models.admin_log_entry import ( |
| 16 | + update_admin_log, |
| 17 | + FILE_REMOVED, |
| 18 | + FILE_VERSION_REMOVED, |
| 19 | +) |
| 20 | +from osf.models.files import File, FileVersion, TrashedFile, TrashedFolder |
| 21 | +from website.files.exceptions import FileNodeCheckedOutError, FileNodeIsPrimaryFile |
10 | 22 |
|
11 | 23 |
|
12 | 24 | class FileSearchView(PermissionRequiredMixin, FormView): |
@@ -52,12 +64,120 @@ def get_context_data(self, **kwargs): |
52 | 64 | ) |
53 | 65 | file = context['object'] |
54 | 66 | node = file.target |
55 | | - latest_modified = file.versions.latest('created') |
| 67 | + is_trashed = isinstance(file, (TrashedFile, TrashedFolder)) |
| 68 | + # Annotate version_id because django templates prohibit accessing attributes that start with underscores |
| 69 | + versions = file.versions.all().order_by('-created').annotate(version_id=F('_id')) if isinstance(file, File) else FileVersion.objects.none() |
| 70 | + |
| 71 | + selected_version_id = self.request.GET.get('version') |
| 72 | + selected_version = versions.filter(version_id=selected_version_id).first() if selected_version_id else None |
| 73 | + if selected_version is None: |
| 74 | + selected_version = versions.first() |
| 75 | + |
56 | 76 | context.update({ |
57 | 77 | 'guid': guid, |
58 | 78 | 'node_id': node._id if node else None, |
59 | 79 | 'node': node, |
60 | 80 | 'file_metadata': metadata_record, |
61 | | - 'version': latest_modified.location.get('version', '') |
| 81 | + 'version': selected_version.location.get('version', '') if selected_version else '', |
| 82 | + 'versions': versions, |
| 83 | + 'selected_version': selected_version, |
| 84 | + 'is_trashed': is_trashed, |
62 | 85 | }) |
63 | 86 | return context |
| 87 | + |
| 88 | + |
| 89 | +class FileDeleteView(FileMixin, View): |
| 90 | + """ Allows authorized users to delete a file or folder (soft delete / trash). |
| 91 | + """ |
| 92 | + permission_required = 'osf.delete_basefilenode' |
| 93 | + raise_exception = True |
| 94 | + |
| 95 | + def post(self, request, *args, **kwargs): |
| 96 | + file = self.get_object() |
| 97 | + if isinstance(file, (TrashedFile, TrashedFolder)): |
| 98 | + messages.error(request, 'This file has already been deleted.') |
| 99 | + return redirect(self.get_success_url()) |
| 100 | + |
| 101 | + node = file.target |
| 102 | + file_path = getattr(file, 'materialized_path', None) or getattr(file, 'path', None) or '' |
| 103 | + guid = self.kwargs['guid'] |
| 104 | + |
| 105 | + try: |
| 106 | + with transaction.atomic(): |
| 107 | + file.delete(user=request.user) |
| 108 | + if node is not None and hasattr(node, 'add_log'): |
| 109 | + params = dict(getattr(node, 'log_params', {})) |
| 110 | + params.update({ |
| 111 | + 'pathType': 'file', |
| 112 | + 'path': file_path, |
| 113 | + }) |
| 114 | + node.add_log( |
| 115 | + action=NodeLog.FILE_REMOVED, |
| 116 | + auth=None, |
| 117 | + foreign_user=NodeLog.SUPPORT_USER_LABEL, |
| 118 | + params=params, |
| 119 | + log_date=timezone.now(), |
| 120 | + should_hide=False, |
| 121 | + ) |
| 122 | + except FileNodeCheckedOutError: |
| 123 | + messages.error(request, 'This file is checked out and cannot be deleted until it is checked in.') |
| 124 | + return redirect(self.get_success_url()) |
| 125 | + except FileNodeIsPrimaryFile: |
| 126 | + messages.error(request, 'This file is the primary file of a preprint and cannot be deleted.') |
| 127 | + return redirect(self.get_success_url()) |
| 128 | + |
| 129 | + update_admin_log( |
| 130 | + user_id=request.user.id, |
| 131 | + object_id=guid, |
| 132 | + object_repr='BaseFileNode', |
| 133 | + message=f'File {guid} deleted by admin.', |
| 134 | + action_flag=FILE_REMOVED, |
| 135 | + ) |
| 136 | + messages.success(request, 'File deleted.') |
| 137 | + return redirect(reverse('home')) |
| 138 | + |
| 139 | + |
| 140 | +class FileVersionDeleteView(FileMixin, View): |
| 141 | + """ Allows authorized users to delete a single version of a file, unlinking |
| 142 | + it from the file and enqueueing a task to purge the underlying storage blob. |
| 143 | + """ |
| 144 | + permission_required = 'osf.delete_fileversion' |
| 145 | + raise_exception = True |
| 146 | + |
| 147 | + def post(self, request, *args, **kwargs): |
| 148 | + file = self.get_object() |
| 149 | + guid = self.kwargs['guid'] |
| 150 | + |
| 151 | + if not isinstance(file, File): |
| 152 | + messages.error(request, 'Only individual files have versions.') |
| 153 | + return redirect(self.get_success_url()) |
| 154 | + |
| 155 | + version_id = self.kwargs.get('version_id') |
| 156 | + version = FileVersion.load(version_id) |
| 157 | + if version is None: |
| 158 | + messages.error(request, 'Version not found.') |
| 159 | + return redirect(self.get_success_url()) |
| 160 | + |
| 161 | + through = version.get_basefilenode_version(file) |
| 162 | + if through is None: |
| 163 | + messages.error(request, 'This version does not belong to this file.') |
| 164 | + return redirect(self.get_success_url()) |
| 165 | + |
| 166 | + if file.versions.count() <= 1: |
| 167 | + messages.error(request, 'Cannot delete the only version of a file. Delete the whole file instead.') |
| 168 | + return redirect(self.get_success_url()) |
| 169 | + |
| 170 | + with transaction.atomic(): |
| 171 | + through.delete() |
| 172 | + |
| 173 | + enqueue_postcommit_task(purge_file_version_task, (version.pk,), {}, celery=True) |
| 174 | + |
| 175 | + update_admin_log( |
| 176 | + user_id=request.user.id, |
| 177 | + object_id=guid, |
| 178 | + object_repr='FileVersion', |
| 179 | + message=f'Version {version_id} of file {guid} unlinked by admin; GCS purge enqueued.', |
| 180 | + action_flag=FILE_VERSION_REMOVED, |
| 181 | + ) |
| 182 | + messages.success(request, 'File version deleted.') |
| 183 | + return redirect(self.get_success_url()) |
0 commit comments