st-cloud/file/views.py

import random
import string

from django.shortcuts import render, redirect
from django.contrib.auth.decorators import login_required
from file.models import File
from django.http import FileResponse, JsonResponse, HttpResponse
from django.utils import timezone
from .judgement_function import judge_filepath, format_size
from django.utils.http import urlquote
from django.contrib.auth.models import User
import json
import os

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

DEBUG = 1


# Create your views here.


#


@login_required
def upload_file(request):
    if request.method == "POST":
        data = request.POST
        file_obj = request.FILES.get('file')
        file_type = judge_filepath(file_obj.name.split('.')[-1].lower())
        file_folder = request.POST.get('file_path')
        update_time = timezone.now().strftime("%Y-%m-%d %H:%M:%S")
        file_size = format_size(file_obj.size)
        file_name = file_obj.name
        save_path = BASE_DIR + '/' + file_folder
        file_path = file_folder + '/' + file_name
        file_id_1 = ''.join(random.sample(string.digits, 8))
        File.objects.create(file_id=file_id_1,
                            file_name=file_name,
                            folder=file_folder,
                            file_path=file_path,
                            update_time=update_time,
                            file_size=file_size,
                            file_type=file_type)
        with open(save_path + file_name, 'wb+') as f:
            for chunk in file_obj.chunks():
                f.write(chunk)
        return HttpResponse(status=200)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'upload_file.html')
    else:
        return HttpResponse(status=400)


@login_required
def download_file(request):
    if request.method == "POST":
        file_path = request.GET.get('file_path')
        file_name = file_path.split('/')[-1]
        file_dir = BASE_DIR + '/' + file_path
        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
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'download_file.html')
    else:
        return HttpResponse(status=400)


@login_required
def delete_file(request):
    if request.method == "POST":
        file_path = request.GET.get('file_path')
        File.objects.get(file_path=file_path, file_id=request.File.file_id).delete()
        try:
            os.remove(BASE_DIR + '/' + file_path)
        except Exception as e:
            print(e)
        return HttpResponse(status=200)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'delete_file.html')
    else:
        return HttpResponse(status=400)