Spaces:
Sleeping
Sleeping
from flask import Flask, request, send_file, render_template_string | |
import os | |
import subprocess | |
app = Flask(__name__) | |
UPLOAD_FOLDER = 'uploads' | |
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER | |
# HTML template for the index page | |
HTML_TEMPLATE = """ | |
<!DOCTYPE html> | |
<html lang="en"> | |
<head> | |
<meta charset="UTF-8"> | |
<meta name="viewport" content="width=device-width, initial-scale=1.0"> | |
<title>NSIS Installer Creator</title> | |
</head> | |
<body> | |
<h1>Upload Files to Create NSIS Installer</h1> | |
<form action="/upload" method="post" enctype="multipart/form-data"> | |
<label for="bat_file">Upload BAT file:</label> | |
<input type="file" name="bat_file" required><br><br> | |
<label for="icon_file">Upload ICO file:</label> | |
<input type="file" name="icon_file" required><br><br> | |
<input type="submit" value="Upload and Compile"> | |
</form> | |
</body> | |
</html> | |
""" | |
def index(): | |
return render_template_string(HTML_TEMPLATE) | |
def upload_files(): | |
if 'bat_file' not in request.files or 'icon_file' not in request.files: | |
return "No file part", 400 | |
bat_file = request.files['bat_file'] | |
icon_file = request.files['icon_file'] | |
if bat_file.filename == '' or icon_file.filename == '': | |
return "No selected file", 400 | |
bat_path = os.path.join(app.config['UPLOAD_FOLDER'], bat_file.filename) | |
icon_path = os.path.join(app.config['UPLOAD_FOLDER'], icon_file.filename) | |
# Save the uploaded files | |
bat_file.save(bat_path) | |
icon_file.save(icon_path) | |
# Construct the NSIS script | |
nsi_script = f""" | |
!include nsDialogs.nsh | |
!include LogicLib.nsh | |
Outfile "output.exe" | |
InstallDir "$PROGRAMFILES\\YourApp" | |
Name "Your App" | |
Caption "Your App Installer" | |
VIProductVersion "1.0.0" | |
RequestExecutionLevel admin | |
Page instfiles | |
Section "Install" | |
SetOutPath "$INSTDIR" | |
File "{bat_path}" | |
File "{icon_path}" | |
SectionEnd | |
""" | |
nsi_path = os.path.join(app.config['UPLOAD_FOLDER'], 'installer.nsi') | |
with open(nsi_path, 'w') as nsi_file: | |
nsi_file.write(nsi_script) | |
# Compile the NSIS script | |
try: | |
result = subprocess.run(['makensis', nsi_path], check=True, capture_output=True, text=True) | |
return f"Compilation successful! Download here: <a href='/download/output.exe'>output.exe</a>", 200 | |
except subprocess.CalledProcessError as e: | |
return f"Compilation failed: {e.stderr}", 400 | |
def download_file(filename): | |
return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename), as_attachment=True) | |
if __name__ == '__main__': | |
if not os.path.exists(UPLOAD_FOLDER): | |
os.makedirs(UPLOAD_FOLDER) | |
app.run(host='0.0.0.0', port=7860) | |