84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from flask.helpers import send_file
|
|
from flask import render_template, abort, flash, Blueprint
|
|
from Web.main.forms import UploadJavaForm
|
|
from random import randint
|
|
import shutil
|
|
import secrets
|
|
import os
|
|
import logging
|
|
|
|
from gui.utils import nassi
|
|
from interpreter.NassiShneidermann import NassiShneidermanDiagram, OB
|
|
|
|
main = Blueprint('main', __name__)
|
|
|
|
def deleteFile(path):
|
|
file_list = os.listdir(path)
|
|
for f in file_list:
|
|
try:
|
|
os.remove(path + '/' +f)
|
|
print("remove " + f)
|
|
except:
|
|
print("fail to remove " + f)
|
|
|
|
def deleteFolder(path):
|
|
try:
|
|
shutil.rmtree(path)
|
|
print("remove " + path)
|
|
except:
|
|
print("fail to remove " + path)
|
|
|
|
def javaDatei(form_file):
|
|
try:
|
|
random_hex = secrets.token_hex(8)
|
|
_, f_ext = os.path.splitext(form_file.filename)
|
|
file_fname = random_hex + f_ext
|
|
dirctory_path = os.path.abspath(os.path.join('Web', os.pardir))
|
|
file_path = os.path.join(dirctory_path, './tmp/input', file_fname)
|
|
form_file.save(file_path)
|
|
return file_path
|
|
except:
|
|
flash('Hier ist was falsch gelaufen!')
|
|
|
|
|
|
@main.route('/', methods=['POST', 'GET'])
|
|
@main.route('/generator', methods=['POST','GET'])
|
|
def generator():
|
|
form = UploadJavaForm()
|
|
|
|
if form.validate_on_submit():
|
|
if form.java.data:
|
|
input_path = javaDatei(form.java.data)
|
|
output_path = os.path.join(os.path.abspath(os.path.join('Web', os.pardir)), './tmp/input')
|
|
outputname = str(randint(0, 100) )
|
|
remove_tags = None
|
|
comments = None
|
|
behaviour = OB.RANDOM_NAME
|
|
types = None
|
|
|
|
NSD = NassiShneidermanDiagram(True)
|
|
output_directory = output_path + '/' + outputname
|
|
|
|
|
|
try:
|
|
if not os.path.exists(output_directory):
|
|
os.makedirs(output_directory)
|
|
except OSError:
|
|
logging.error('Error: Creating directory. ' + output_directory)
|
|
|
|
|
|
custom_tags = {"comments" : comments, "ignore" : remove_tags, "types" : types}
|
|
|
|
NSD.load_from_file(input_path, custom_tags)
|
|
NSD.convert_to_image(output_directory, on_conflict=behaviour)
|
|
|
|
zip_path = os.path.join(os.path.abspath(os.path.join('Web', os.pardir)), f'../tmp/output/{outputname}')
|
|
shutil.make_archive(zip_path, 'zip', output_directory)
|
|
|
|
deleteFolder(output_path)
|
|
|
|
return send_file(zip_path + '.zip', as_attachment=True)
|
|
|
|
return render_template('upload.html', title='Upload', legend='Upload', form=form )
|
|
|