#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 #path = "/home/bscheibel/app/app" path = "/home/centurio/Projects/engineering_drawings_ui/app" #path_extraction = '/home/bscheibel/PycharmProjects/dxf_reader/main.py' path_extraction = "/home/centurio/Projects/engineering_drawings_extraction/main.py" UPLOAD_FOLDER = path + "/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/centurio/web/edi2/out']) def extract_all(uuid, filename, db): #order_bounding_boxes_in_each_block.main(uuid, UPLOAD_FOLDER + "/" + filename) subprocess.call(['python3', path_extraction, 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 ''' Upload new File

Upload new File

''' @app.route('/show/&') 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(unix_socket_path='/tmp/redis.sock',db=7) #db = redis.Redis("unix_socket_path='127.0.0.1',6379,db=7") #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: " + str(gen_tol) + "
" 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(path +'/static/config.json') as f: config_file = json.load(f) print(config_file) 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 + "
" continue else: html_code += "

" + dim + "

" 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()).decode('utf-8') #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 += " " + \ "" + d + "" + \ " " relevant_isos = list(set(relevant_isos)) for x in relevant_isos: #html_code += " "+ x.partition(".")[0] +" " html_code += " " + x.partition(".")[0] + " " #print(html_code) html_code += "" html_links = "" for link in links: html_links += " Open " + link + "
" #html_links += " Open " + link + " in Tab """ #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/') 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/',methods=['GET']) def redis_get(key): db = redis.Redis(unix_socket_path='/tmp/redis.sock',db=7) result = json.loads(db.get(key)) return result @app.route('/redis/set/',methods=['POST']) def redis_set(key): value = request.get_json(force=True) value = value["value"] db = redis.Redis(unix_socket_path='/tmp/redis.sock',db=7) 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')