98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
"""Shared file-tree access and serialization helpers."""
|
|
import uuid
|
|
from django.core.exceptions import ValidationError
|
|
from django.http import Http404
|
|
from rest_framework.response import Response
|
|
from .models import DriveItem, ItemGrant
|
|
from .spaces import can_read_space, can_write_space
|
|
|
|
|
|
def error(message, code=400):
|
|
return Response({'detail':message}, status=code)
|
|
|
|
|
|
def item_info(item, user=None):
|
|
data = {
|
|
'id':str(item.id), 'name':item.name, 'is_folder':item.is_folder,
|
|
'size':item.size,'mime_type':item.mime_type,'owner_id':item.owner_id,
|
|
'owner':item.owner.username,'parent':str(item.parent_id) if item.parent_id else None,
|
|
'space_id':item.space_id, 'is_shared':item.user_shares.exists() or any(link.revoked_at is None for link in item.public_links.all()),
|
|
'trashed_at':item.trashed_at.isoformat() if item.trashed_at else None,
|
|
'created_at':item.created_at.isoformat(),'updated_at':item.updated_at.isoformat(),
|
|
}
|
|
if user is not None:
|
|
data['can_write'] = writable(item,user)
|
|
return data
|
|
|
|
|
|
def get_item(pk):
|
|
try:
|
|
return DriveItem.objects.select_related('owner','parent','space').get(pk=pk)
|
|
except (DriveItem.DoesNotExist,ValueError,ValidationError):
|
|
raise Http404
|
|
|
|
|
|
def chain(item):
|
|
ancestors=[]
|
|
visited=set()
|
|
while item is not None:
|
|
if item.pk in visited:
|
|
raise Http404
|
|
visited.add(item.pk)
|
|
ancestors.append(item)
|
|
item=item.parent
|
|
return ancestors
|
|
|
|
|
|
def readable(item, user, allow_trashed=False):
|
|
space_allowed = can_read_space(user,item.space)
|
|
if not space_allowed and getattr(user,'is_authenticated',False) and user.is_active:
|
|
space_allowed = ItemGrant.objects.filter(user=user,item_id__in=[node.pk for node in chain(item)]).exists()
|
|
if not space_allowed:
|
|
return False
|
|
if not allow_trashed and any(n.trashed_at is not None for n in chain(item)):
|
|
return False
|
|
return True
|
|
|
|
|
|
def writable(item,user):
|
|
return can_write_space(user,item.space)
|
|
|
|
|
|
def valid_name(name):
|
|
if not isinstance(name,str) or not name.strip() or len(name)>255:
|
|
raise ValueError('Nom invalide (1 à 255 caractères).')
|
|
name=name.strip()
|
|
if name in ('.','..') or any(c in name for c in ('/','\\','\x00','\r','\n')):
|
|
raise ValueError('Ce nom contient des caractères interdits.')
|
|
return name
|
|
|
|
|
|
def target_parent(parent_id, user, space):
|
|
if not parent_id:
|
|
return None
|
|
try:
|
|
parent=DriveItem.objects.select_related('parent','space').get(pk=uuid.UUID(str(parent_id)))
|
|
except (ValueError,DriveItem.DoesNotExist,AttributeError):
|
|
raise ValueError('Dossier de destination introuvable.')
|
|
if parent.space_id != space.id:
|
|
raise PermissionError('Impossible de déplacer un élément entre deux espaces.')
|
|
if not parent.is_folder or not readable(parent,user) or not writable(parent,user):
|
|
raise PermissionError('Dossier de destination inaccessible en écriture.')
|
|
return parent
|
|
|
|
|
|
def descendant_files(node):
|
|
files=[]
|
|
pending=[node.pk]
|
|
while pending:
|
|
batch=pending[:500];pending=pending[500:]
|
|
children=list(DriveItem.objects.filter(parent_id__in=batch).values_list('pk','is_folder','storage_key','storage_location'))
|
|
pending.extend(row[0] for row in children)
|
|
for _,is_folder,key,location in children:
|
|
if not is_folder and key: files.append(location or f'{key[:2]}/{key}')
|
|
if not node.is_folder and node.storage_key:
|
|
key=node.storage_key
|
|
files.append(node.storage_location or f'{key[:2]}/{key}')
|
|
return files
|