Пример #1
0
def get_image_path(generator_id, fname, size=None):
    gen = query.get_generator(generator_id)
    pack_path = get_package_path(gen)

    pack_id = gen.package.pack_id
    gen_id = gen.gen_id.split(pack_id, 1)[-1].lstrip('.')

    image_path = os.path.sep.join([pack_path, 'screenshots', gen_id, fname])

    if size:
        id = '+'.join([generator_id, fname, size])
        fsize = [int(s) for s in size.split('x', 1)]
        thumb_path = compute_cache_path(config.cache_dir, id)
        if not os.path.exists(thumb_path):
            im = Image.open(image_path)
            im.thumbnail(fsize, Image.ANTIALIAS)
            im.save(thumb_path, "JPEG")
        image_path = thumb_path

    return image_path
Пример #2
0
def download(request, _id):
    gen = query.get_generator(_id)
    if not gen:
        return apiresponse(code='err_record')

    pack_path = get_package_path(gen)
    zip_file = '.'.join([pack_path, 'zip'])

    if not os.path.exists(zip_file):
        return apiresponse(code='err_record')

    data = open(zip_file, 'rb').read()

    resp = HttpResponse(mimetype='application/zip')
    resp['Content-Disposition'] = 'attachment; filename=%s.zip' % _id
    resp.write(data)

    pack = gen.package
    num_downloaded = pack.num_downloaded+1 if pack.num_downloaded else 1
    pack.num_downloaded = num_downloaded
    pack.save()

    return resp