bscheibel
/
ui


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287
							#encoding
from app import app
from flask import request, redirect, url_for, send_from_directory, render_template
import subprocess
import redis
import random
import PyPDF2
import json
import base64
import os
import json
import re
import base64
#https://medium.com/@emerico/convert-pdf-to-image-using-python-flask-2864fb655e01


UPLOAD_FOLDER = "/home/centurio/Projects/engineering_drawings_extraction/temporary"
app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER
ALLOWED_EXTENSIONS = set(['pdf', 'png', 'jpg', 'jpeg', 'PDF'])


def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS

def convert_pdf_img(filename):
    PDFFILE = UPLOAD_FOLDER +"/" + filename
    subprocess.call(['pdftoppm', '-jpeg', '-singlefile',
                     PDFFILE, '/home/bscheibel//app/app/temporary/out'])

def extract_all(uuid, filename, db):
    #order_bounding_boxes_in_each_block.main(uuid, UPLOAD_FOLDER + "/" + filename)
    subprocess.call(['pipenv run python main.py','/home/centurio/Projects/engineering_drawings_extraction/main.py', str(uuid),UPLOAD_FOLDER + "/" + filename, db, str(0)])

def get_file_size(file):
    pdf = PyPDF2.PdfFileReader(file)
    p = pdf.getPage(0)

    w = p.mediaBox.getWidth()
    h= p.mediaBox.getHeight()
    OrientationDegrees = p.get('/Rotate')
    if OrientationDegrees != 0 :
        orientation = "landscape"
    else:
        orientation = "portrait"

    print(w,h,OrientationDegrees)
    return w,h, orientation


def check_config_file(d):
    reg_search = d
    #print(reg_search)

    for conf in d:
        print(conf)

    return "blub"


def check_links(isos):
    link_names = {}
    isos_names = []
    isos = list(set(isos))
    reg_isos = r"(ISO\s\d*)\s1\-(\d?)"
    print(isos)
    isos_new = []
    for name in isos:
        if re.search(reg_isos, name):
            n = 1
            #print(name)
            new_isos = re.search(reg_isos,name).group(1)
            number = re.search(reg_isos,name).group(2)
            while n <= int(number):
                isos_new.append(new_isos+"-"+str(n))
                n += 1
        else:
            isos_new.append(name)
    for name in isos_new:
        try:
            name = name.replace(" ", "")
            #name = name.replace("-"," ")
            url1 = name + ".PDF"
            #print(url)
            file = send_from_directory("static/isos",url1)
            url = "isos/" + url1
            #link_names.append(url)
            link_names[name] = url
            #print(link_names)
        except:
            #print(name)
            isos_names.append(name)
    return link_names, isos_names


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        file = request.files['file']
        if file and allowed_file(file.filename):
            filename = file.filename
            basedir = os.path.abspath(os.path.dirname(__file__))
            file.save(os.path.join(basedir,app.config["UPLOAD_FOLDER"], filename))
            uuid = random.randint(100,10000000)
            extract_all(uuid, filename, 'localhost')
            return redirect(url_for('uploaded_file', filename=filename, uuid=uuid))
    return '''
    <!doctype html>
    <title>Upload new File</title>
    <h1>Upload new File</h1>
    <form action="" method=post enctype=multipart/form-data>
      <p><input type=file name=file>
         <input type=submit value=Upload>
    </form>
    '''

@app.route('/show/<filename>&<uuid>')
def uploaded_file(filename, uuid):
    file_out = "out.jpg"
    #file_out = filename
    #if request.method == 'POST':
    #    uuid = 433
    if filename.endswith(".pdf") or filename.endswith(".PDF"):
        w,h, orientation = get_file_size(UPLOAD_FOLDER +"/" + filename)
        convert_pdf_img(filename)
        db = redis.Redis("localhost")
        #isos = db.get(uuid+"dims")
        #print(iso)
        gen_tol = db.get(str(uuid)+"tol")
        print(gen_tol)
        isos = json.loads(db.get(str(uuid)+"isos"))
        links, isos_names = check_links(isos)
        dims = json.loads(db.get(str(uuid)+"dims"))
        details = json.loads(db.get(str(uuid) + "details"))
        number_blocks = db.get(str(uuid)+"eps")
        html_code = "General tolerances according to: " + gen_tol + "<br>"
        html_general = ""
        reg = r"(-?\d{1,}\.?\d*)"
        #re_gewinde = r"M\d{1,2}"
        #re_passungen = r"h\d{1,2}|H\d{1,2}"
        det_coords= "0,0,0,0"
        with open('/home/bscheibel/app/app/config.json') as f:
            config_file = json.load(f)

        for dim in sorted(dims):
            #print(dim)
            for det in details:
                #print(det)
                try:
                    if dim == det:
                        det_coords = details[det]
                        det_coords = ",".join(str(det) for det in det_coords)
                except:
                    det_coords = "0,0,0,0"
            if "ZZZZ" in dim:
                for d in dims[dim]:
                    html_general += d + "<br>"
                continue
            else:
                html_code += "<td><h4>" + dim + "</h4></td>"
            for d in dims[dim]:
                relevant_isos = []
                search_terms = {}
                terms = ''
                #if "Ra" in d or "Rz" in d or "Rpk" in d:
                #    relevant_isos.append("ISO4287.PDF")
                #if u"\u27C2" in d or u"\u25CE" in d or u"\u232D" in d or u"\u2225" in d or u"\u232F" in d or u"\u2316" in d or u"\u2313" in d or u"\u23E5" in d:
                #    relevant_isos.append("ISO1101.PDF")
                #if re.search(re_gewinde,d):
                #    relevant_isos.append("ISO6410.PDF")
                #if "GG" in d or "CT" in d or "GX" in d:
                #    relevant_isos.append("ISO14405-1.PDF")
                #if re.search(re_passungen,d):
                #    relevant_isos.append("ISO286-1.PDF")
                for conf in config_file:
                    if re.search(conf,d):
                        iso = config_file[conf]
                        for key in iso:
                            relevant_isos.append(key)
                            for blub in iso[key]:
                                search_terms[blub] = iso[key][blub]

                        #terms = '{"Symbole":1,"Tabelle":2,"Definition":3}'
                        if len(search_terms) < 1:
                            search_terms["Beginn"] = 1
                        terms = json.dumps(search_terms)
                        #print(terms)
                        terms = base64.b64encode(terms.encode())
                        #terms = "blub"


                try:
                    number = re.search(reg, d)
                    number = number.group(1)
                    try:
                        floats = len(number.split(".")[1])
                        if floats <= 1:
                            steps = 0.1
                        elif floats == 2:
                            steps = 0.01
                        elif floats == 3:
                            steps = 0.001
                        else:
                            steps = 0.001
                    except:
                        steps = 0.1
                except:
                    number = d
                    steps = 0.1
                coords = ",".join(str(e) for e in dims[dim][d])
                html_code += "<tr><td style='text-align:center'> <input type='checkbox' name='relevant." + d + "' value='checked'> </td>" + \
                             "<td style='text-align:center'>" + d + "</td>" + \
                             "<td max='3' style='text-align:center'> <input type='number' step='" + str(steps) + "' data-coords='" + coords + " 'data-details='" + det_coords  +"'' name='" + d + "' value='" + number + "'> </td>"

                relevant_isos = list(set(relevant_isos))
                for x in relevant_isos:
                    #html_code += "<td style='text-align:left'> <a href=" + url_for('static', filename="isos/"+x) + " >"+ x.partition(".")[0]  +"</a>  </td>"
                    html_code += "<td style='text-align:left' data-terms='" + terms + "'> <a onclick=ui_add_tab_active('#main','" + x.partition(".")[0] + "','" + x.partition(".")[0] +"',true,'isotab','"+terms+"')>" + x.partition(".")[0] + "</a>  </td>"
                #print(html_code)
                html_code += "</tr>"
                html_links = ""
                for link in links:
                    html_links += "<a onclick =ui_add_tab_active('#main','" + link + "','" + link +"',true,'isotab','empty')> Open " + link + "</a> <br>"
                    #html_links += "<tr> <td> <a onclick =ui_add_tab_active('#main','iso1','iso1',true,'isotab')> Open " + link + " in Tab </a> </td> </tr>"""
        #print("teeest")
        return render_template('index.html', filename=file_out, isos=isos, dims=dims, text=html_code,html_general=html_general, number=number_blocks, og_filename=filename, w=w, h=h, html_links=html_links, isos_names=isos_names, orientation=orientation)
    #return render_template('test_pdfjs_textlayer.html', og_filename=filename)

    #else:
    #    filename = filename
    #    return render_template('show_image.html', filename=filename)


@app.route('/uploads/<filename>')
def send_file(filename):
    return send_from_directory(UPLOAD_FOLDER, filename)

# No caching at all for API endpoints.
@app.after_request
def add_header(response):
    # response.cache_control.no_store = True
    response.headers['Cache-Control'] = 'no-store, no-cache, must-revalidate, post-check=0, pre-check=0, max-age=0'
    response.headers['Pragma'] = 'no-cache'
    response.headers['Expires'] = '-1'
    response.headers['Access-Control-Allow-Origin'] = '*'
    response.headers.add("Access-Control-Allow-Headers", "*")
    response.headers.add("Access-Control-Allow-Methods", "*")
    return response


@app.route('/redis/get/<key>',methods=['GET'])
def redis_get(key):
    db = redis.Redis("localhost")
    result = json.loads(db.get(key))
    return result

@app.route('/redis/set/<key>',methods=['POST'])
def redis_set(key):
    value = request.get_json(force=True)
    value = value["value"]
    db = redis.Redis("localhost")

    value_name = value[0]
    value_v = value[1]
    try:
        result = json.loads(db.get(key))
        result[value_name] = value_v
        json_res = json.dumps(result)
        db.set(key,json_res)
    except:
        dict_res = {}
        dict_res[value_name] = value_v
        json_dict = json.dumps(dict_res)
        db.set(key, json_dict)
    return "OK"


@app.route('/index')
def test():
    return render_template('index.html')