st-cloud/file/views.py

from account.decorators import login_required
from file.models import File
from django.http import FileResponse, JsonResponse, HttpResponse
from django.utils import timezone

from folder.models import Folder
from .judgement_function import judge_filepath, format_size
from django.utils.http import urlquote
import os
from account.models import get_user
from utils.decorators import debug_view
from utils.http import make_json_response
from utils.permission import can_delete

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

# Create your views here.


@debug_view(template_name='upload_file.html')
@login_required
def upload_file(request):
    user = get_user(request)
    try:
        file_obj = request.FILES.get('file')
    except:
        return make_json_response(code=401, error='文件不存在')
    update_time = timezone.now().strftime("%Y-%m-%d %H:%M:%S")
    file_size = file_obj.size
    file_name = file_obj.name
    file_type = judge_filepath(file_name.split('.')[-1].lower()) if '.' in file_name else ''
    father_folder_id = request.POST.get('father_folder_id')
    try:
        folder = Folder.objects.get(folder_id=father_folder_id)
    except:
        return make_json_response(code=402, error='文件夹不存在')
    if not folder.check_permission(user=user):
        return make_json_response(code=404, error='没有上传文件的权限')
    file = File.objects.create(file_name=file_name,
                               father_folder=folder,
                               update_time=update_time,
                               file_size=file_size,
                               file_type=file_type,
                               owner=user,
                               group=folder.group)
    # TODO: 文件hash
    try:
        file_dir = BASE_DIR + '/' + str(file.file_id)
        with open(file_dir, 'wb+') as f:
            for chunk in file_obj.chunks():
                f.write(chunk)
    except:
        file.delete()
        return make_json_response(code=500, error='文件保存失败')
    return make_json_response()
    if request.method == "POST":
        user = get_user(request)
        try:
            file_obj = request.FILES.get('file')
            print(file_obj)
        except Exception as e:
            print(e)
            print('文件不存在')
            return JsonResponse({'code': 401, 'error': '文件不存在'})
        update_time = timezone.now().strftime("%Y-%m-%d %H:%M:%S")
        file_size = file_obj.size
        file_name = file_obj.name
        file_type = judge_filepath(file_name.split('.')[-1].lower()) if '.' in file_name else ''
        father_folder_id = request.POST.get('father_folder_id')
        print(f'file_name={file_name}, father_folder_id={father_folder_id}')
        try:
            father_folder = Folder.objects.get(folder_id=father_folder_id)
        except Exception as e:
            print(e)
            print('文件夹不存在')
            return JsonResponse({'code': 402, 'error': '文件夹不存在'})
        if not father_folder.check_permission(user=user):
            print('没有上传文件的权限')
            return JsonResponse({'code': 404, 'error': '没有上传文件的权限'})
        file = File.objects.create(file_name=file_name,
                                   father_folder=father_folder,
                                   update_time=update_time,
                                   file_size=file_size,
                                   file_type=file_type,
                                   owner=user,
                                   group=father_folder.group)
        # TODO: 文件hash
        try:
            file_dir = BASE_DIR + '/' + str(file.file_id)
            with open(file_dir, 'wb+') as f:
                for chunk in file_obj.chunks():
                    f.write(chunk)
        except:
            print('文件保存失败')
            file.delete()
            return JsonResponse({'code': 500, 'error': '文件保存失败'})
        return JsonResponse({'code': 200})
    elif request.method == 'GET' and DEBUG:
        return render(request, 'upload_file.html')
    else:
        return HttpResponse(status=400)


@debug_view('username', 'token', 'file_id')
@login_required
def download_file(request):
    user = get_user(request)
    file_id = request.POST.get('file_id')
    try:
        file = File.objects.get(file_id=file_id)
    except:
        return make_json_response(code=401, error='文件不存在')
    if not file.father_folder.check_permission(user=user):
        return make_json_response(code=404, error='没有下载文件的权限')
    file_name = file.file_name
    file_dir = BASE_DIR + '/' + str(file.file_id)
    file = open(file_dir, 'rb')
    response = FileResponse(file)
    response['Content-Type'] = 'application/octet-stream'
    response['Content-Disposition'] = 'attachment;filename={}'.format(urlquote(file_name))
    return response


@debug_view('username', 'token', 'file_id')
@login_required
def delete_file(request):
    data = request.POST
    user = get_user(request)
    file_id = data.get('file_id')
    try:
        file = File.objects.get(file_id=file_id)
    except:
        return make_json_response(code=401, error='文件不存在')
    if not can_delete(user=user, f=file):
        return make_json_response(code=404, error='没有删除文件的权限')
    file.delete()
    return make_json_response()