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',