+ web_app

web_app/main/routes.py

@@ -0,0 +1,83 @@
+from flask.helpers import send_file
+from flask import render_template, abort, flash, Blueprint
+from web_app.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 deleteFilesInFolder(path):
+    file_list = os.listdir(path)
+    for f in file_list:
+        try:
+            os.remove(path + '/' +f)
+            # print("remove " + f)
+        except:
+            try:
+                shutil.rmtree(path + '/' + f)
+                # print("remove " + f)
+            except:
+                logging.error("fail to remove " + f)
+
+
+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:
+            deleteFilesInFolder(os.path.join(os.path.abspath(os.path.join('Web', os.pardir)), f'../tmp/output/'))
+            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) 
+            
+            deleteFilesInFolder(output_path)
+            return send_file(zip_path + '.zip', as_attachment=True)
+            
+            
+
+    return render_template('upload.html', title='Upload', legend='Upload', form=form )
+