Django+Bootstrap上傳壓縮文件包並解析

url.py中

url(r'^blind/(\w+)/project_type/$', 'encode.views.blind_project_type', name="blind_project_type"),

html中

<td>
   <form class="margin-10" action="" method="post" enctype="multipart/form-data">
       {% bootstrap_form form %}
       <button type="submit" class="btn btn-info" style="height: 34px;">上傳壓縮包</button>
   </form>
   <br>
   <h4 style="color: #DD1144">{{ message }}</h4>
</td>

form.py中

class UploadComPackageForm(forms.Form):
    files_field = forms.FileField(label=u"上傳壓縮包", required=True, widget=forms.ClearableFileInput(attrs={'multiple': True}))

view.py中

from utils.img_util import *
from utils.definition import RESOLUTION_DEFINITION, RESOLUTION_DEFINITION_CONVERT, VIDEO_EXTENSION_LIST, RESOLUTION_LIST
# COMPRESS_PACKAGE_PREFIX = "/mnt/competitor/new-comparison-platform/"


@login_required
def blind_project_type(request, blind_task_id=None):
    blind_task = BlindTask.objects.get(id=ObjectId(blind_task_id))
    if request.method == "POST":
        form = UploadComPackageForm(request.POST, request.FILES)
        if form.is_valid():
            try:
                files = request.FILES.getlist('files_field')
                for f in files:
                    file_name = os.path.splitext(f.name)[0]
                    project = Project(name=file_name, user=request.user.username)
                    try:
                        project.save()
                        blind_task.projects.append(project)
                        blind_task.save()
                    except NotUniqueError:
                        message = "盲評任務已存在,請重命名文件夾後再壓縮上傳！"
                        return render(request, 'blind_task_type.html', locals())

                    is_or_not, file_extension = is_or_not_compress(f.name)
                    if is_or_not == "compress":
                        compress_data = f.read()
                        path = config.COMPRESS_PACKAGE_PREFIX
                        filepath = os.path.join(path, f.name)
                        with open(filepath, 'wb+')as destination:
                            for chunk in f.chunks():
                                destination.write(chunk)
                        if file_extension == "zip":
                            uncompress_dir = un_zip(filepath)
                        if file_extension == "tar":
                            uncompress_dir = un_tar(filepath)
                        # if file_extension == "gz":
                        #     un_gz(filepath)
                        project_list = os.listdir(uncompress_dir)
                        for project_index in range(0, len(project_list)):
                            project_dir = os.path.join(uncompress_dir, project_list[project_index])
                            if os.path.isdir(project_dir):
                                entry_list = os.listdir(project_dir)
                                for entry_index in range(0, len(entry_list)):
                                    entry_dir = os.path.join(project_dir, entry_list[entry_index])
                                    entry = Entry(desc=entry_list[entry_index])
                                    entry.save()
                                    project.entries.append(entry)
                                    project.save()
                                    if os.path.isdir(entry_dir):
                                        image_list = os.listdir(entry_dir)
                                        display_name = 0
                                        master_info = []
                                        other_info = {}
                                        for image_index in range(0, len(image_list)):
                                            image_dir = os.path.join(entry_dir, image_list[image_index])
                                            if os.path.isdir(image_dir):
                                                message = "不符合規定的壓縮包！請採用如下規定三級存儲結構！"
                                            else:
                                                image_ture_name = image_list[image_index]
                                                real_name = image_ture_name
                                                real_com = real_name.split(".")[0]
                                                display_name = display_name + 1
                                                myimage = open(os.path.join(entry_dir, image_ture_name), 'rb')
                                                image_data = myimage.read()
                                                tool = ImageMetaTool(image_data)
                                                image_doc = ImageDoc.create(
                                                    real_name,
                                                    real_com,
                                                    "圖片%d" % (display_name),
                                                    tool.get_resolution(),
                                                    image_data,
                                                    real_name,
                                                    tool.get_content_type(),
                                                    )
                                                entry.images.append(image_doc)
                                                entry.save()
                redirect_url = reverse('blind_add_task_self', args=[blind_task.id])
                return HttpResponseRedirect(redirect_url)
            except ResolutionException as e:
                error = str(e)
            except Exception as e:
                print traceback.format_exc(e)
                error = str(e)
    elif request.method == "GET":
        form = UploadComPackageForm()
    return render(request, 'blind_task_type.html', locals())

img_util.py中

def un_zip(filename):
    zip_file = zipfile.ZipFile(filename)
    compress_dir = os.path.splitext(filename)[0]
    if os.path.isdir(compress_dir):
        pass
    else:
        os.mkdir(compress_dir)
    for names in zip_file.namelist():
        zip_file.extract(names, compress_dir)
    zip_file.close()
    return compress_dir

def un_gz(filename):
    print "this is tar.gz解壓"
    g_file = gzip.GzipFile(filename)
    print g_file
    if os.path.isdir(os.path.splitext(filename)[0]):
        pass
    else:
        os.mkdir(os.path.splitext(filename)[0])
    print os.path.splitext(filename)[0]
    un_tar(os.path.splitext(filename)[0])
    g_file.close()


def is_or_not_compress(filename):
    file_extension = os.path.splitext(filename)[1]
    file_extension = file_extension.replace('.', '')
    if file_extension in COMPRESS_EXTENSION_LIST:
        return "compress", file_extension