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 os
import json
import re
import base64
path = "/home/bscheibel/app/app"
db_params = "localhost"
path_extraction = '/home/bscheibel/technical_drawings_extraction/main.py'
path_image = path + "/temporary/out"
#path = "/home/centurio/Projects/engineering_drawings_ui/app"
#db_params = "unix_socket_path='/tmp/redis.sock',db=7"
#path_image = "/home/centurio/web/edi2/out"
#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, path_image])
def extract_all(uuid, filename, db):
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()
orientation_degrees = p.get('/Rotate')
if orientation_degrees != 0 :
orientation = "landscape"
else:
orientation = "portrait"
print(w, h, orientation_degrees)
return w, h, orientation
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
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(" ", "")
url1 = name + ".PDF"
file = send_from_directory("static/isos",url1)
url = "isos/" + url1
link_names[name] = url
except:
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, db_params)
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"
if filename.endswith(".pdf") or filename.endswith(".PDF"):
w,h, orientation = get_file_size(UPLOAD_FOLDER +"/" + filename)
convert_pdf_img(filename)
db = redis.Redis(db_params)
print(db_params)
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*)"
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):
for det in details:
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 = ''
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]
if len(search_terms) < 1:
search_terms["Beginn"] = 1
terms = json.dumps(search_terms)
terms = base64.b64encode(terms.encode()).decode('utf-8')
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 += "
"
html_links = ""
for link in links:
html_links += " Open " + link + "
"
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)
@app.route('/uploads/')
def send_file(filename):
return send_from_directory(UPLOAD_FOLDER, filename)
@app.after_request
def add_header(response):
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(db_params)
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(db_params)
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 = {value_name: value_v}
json_dict = json.dumps(dict_res)
db.set(key, json_dict)
return "OK"