From d468f7605820089b2bb39750814ba3fa657fa4ee Mon Sep 17 00:00:00 2001 From: Bryan Gerlach Date: Tue, 8 Oct 2024 14:52:27 -0500 Subject: [PATCH] image stuff --- rdgenerator/views.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/rdgenerator/views.py b/rdgenerator/views.py index 0e709c4..91d5af3 100644 --- a/rdgenerator/views.py +++ b/rdgenerator/views.py @@ -1,6 +1,6 @@ import io from pathlib import Path -from django.http import HttpResponse +from django.http import HttpResponse, JsonResponse from django.shortcuts import render from django.core.files.base import ContentFile import os @@ -191,8 +191,10 @@ def generator_view(request): create_github_run(myuuid) response = requests.post(url, json=data, headers=headers) print(response) - - return render(request, 'waiting.html', {'filename':filename, 'uuid':myuuid, 'status':"Starting generator...please wait", 'platform':platform}) + if response.status_code == 204: + return render(request, 'waiting.html', {'filename':filename, 'uuid':myuuid, 'status':"Starting generator...please wait", 'platform':platform}) + else: + return JsonResponse({"error": "Something went wrong"}) else: form = GenerateForm() return render(request, 'generator.html', {'form': form}) @@ -225,10 +227,12 @@ def download(request): return response def get_png(request): + print("get png") filename = request.GET['filename'] uuid = request.GET['uuid'] #filename = filename+".exe" file_path = os.path.join('png',uuid,filename) + print(file_path) with open(file_path, 'rb') as file: response = HttpResponse(file, headers={ 'Content-Type': 'application/vnd.microsoft.portable-executable',