#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, '/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 += "