st-cloud/folder/views.py

from django.http import HttpResponse
from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from .models import Folder
from file.models import File
from django.contrib.auth.models import User
import json

# Create your views here.
DEBUG = 1


# 在根目录下列出所有文件夹与文件
@login_required(login_url='/account/login/')
def folder_list(request):
    if request.method == 'POST':
        data = request.POST
        # 获取所有文件夹
        folders = Folder.objects.filter(father_folder__folder_id=data['folder_id']).get()
        files = File.objects.filter(folder__folder_id=data['folder_id']).get()
        response = []
        for i in folders:
            response.append(i)
        for i in files:
            response.append(i)
        return HttpResponse(json.dumps(response), status=200)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'folder_list.html')
    else:
        return HttpResponse(status=400)


# 增文件夹
@login_required(login_url='/account/login/')
def add_folder(request):
    if request.method == 'POST':
        data = request.POST
        folder = Folder.objects.filter(father_folder__folder_id=data['folder_id']).get()
        if not folder:
            # 查无此文件夹
            return HttpResponse(status=421)
        else:
            Folder.objects.create(folder_id=data['folder_id'],
                                  folder_name=data['folder_name'],
                                  father_folder=folder)
            return HttpResponse(status=200)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'add_folder.html')
    else:
        return HttpResponse(status=400)

# 删除文件夹
@login_required(login_url='/account/login/')
def delete_folder(request):
    if request.method == 'POST':
        data = request.POST
        folder = Folder.objects.filter(folder_id=data['folder_id']).get()
        if not folder:
            # 查无此文件夹
            return HttpResponse(status=421)
        else:
            folder.delete()
            return HttpResponse(status=200)
    elif request.method == 'GET':
        if DEBUG:
            return render(request, 'delete_folder.html')
    else:
        return HttpResponse(status=400)