filip/rdgen
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

image stuff

Browse Source
This commit is contained in:
Bryan Gerlach
2024-10-08 14:52:27 -05:00
parent 0e2f682bb5
commit d468f76058
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rdgenerator/views.py
View File

@@ -1,6 +1,6 @@
import io import io
from pathlib import Path from pathlib import Path
from django.http import HttpResponse from django.http import HttpResponse, JsonResponse
from django.shortcuts import render from django.shortcuts import render
from django.core.files.base import ContentFile from django.core.files.base import ContentFile
import os import os
@@ -191,8 +191,10 @@ def generator_view(request):
create_github_run(myuuid) create_github_run(myuuid)
response = requests.post(url, json=data, headers=headers) response = requests.post(url, json=data, headers=headers)
print(response) print(response)
if response.status_code == 204:
return render(request, 'waiting.html', {'filename':filename, 'uuid':myuuid, 'status':"Starting generator...please wait", 'platform':platform}) return render(request, 'waiting.html', {'filename':filename, 'uuid':myuuid, 'status':"Starting generator...please wait", 'platform':platform})
else:
return JsonResponse({"error": "Something went wrong"})
else: else:
form = GenerateForm() form = GenerateForm()
return render(request, 'generator.html', {'form': form}) return render(request, 'generator.html', {'form': form})
@@ -225,10 +227,12 @@ def download(request):
return response return response
def get_png(request): def get_png(request):
print("get png")
filename = request.GET['filename'] filename = request.GET['filename']
uuid = request.GET['uuid'] uuid = request.GET['uuid']
#filename = filename+".exe" #filename = filename+".exe"
file_path = os.path.join('png',uuid,filename) file_path = os.path.join('png',uuid,filename)
print(file_path)
with open(file_path, 'rb') as file: with open(file_path, 'rb') as file:
response = HttpResponse(file, headers={ response = HttpResponse(file, headers={
'Content-Type': 'application/vnd.microsoft.portable-executable', 'Content-Type': 'application/vnd.microsoft.portable-executable',