mirror of
https://github.com/bszczuka/steel-training.git
synced 2025-12-21 13:23:02 +01:00
add code
This commit is contained in:
bogo
161
app.py
Normal file
161
app.py
Normal file
@@ -0,0 +1,161 @@
|
||||
import math
|
||||
from shutil import posix
|
||||
|
||||
from flask import Flask, render_template, request, send_file
|
||||
from weasyprint import HTML
|
||||
import io
|
||||
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
# Original values dictionary
|
||||
VALUES = {
|
||||
"five_to_go": {
|
||||
"t1": {"len_a": 3100, "len_h": 9100, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 1000, "len_h": 11000, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 1000, "len_h": 13700, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t4": {"len_a": 3100, "len_h": 16500, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 5200, "len_h": 6400, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
},
|
||||
"showdown_left": {
|
||||
"t1": {"len_a": 1800, "len_h": 22900, 'width': 400, 'height': 600, 'elevation': 200, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 200, "len_h": 9000, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t3": {"len_a": 1000, "len_h": 11000, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
"t4": {"len_a": 3800, "len_h": 22900, 'width': 400, 'height': 600, 'elevation': 200, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 2200, "len_h": 9000, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
},
|
||||
"showdown_right": {
|
||||
"t1": {"len_a": 2200, "len_h": 9000, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 3800, "len_h": 22900, 'width': 400, 'height': 600, 'elevation': 200, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 1000, "len_h": 11000, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'left'},
|
||||
"t4": {"len_a": 200, "len_h": 9000, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t5": {"len_a": 1800, "len_h": 22900, 'width': 400, 'height': 600, 'elevation': 200, 'stop_plate': False, 'side': 'right'},
|
||||
|
||||
},
|
||||
"smoke_and_hope": {
|
||||
"t1": {"len_a": 4300, "len_h": 6400, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 2700, "len_h": 8200, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 0, "len_h": 12800, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
"t4": {"len_a": 2700, "len_h": 8200, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 4300, "len_h": 6400, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'right'},
|
||||
|
||||
},
|
||||
"accelerator": {
|
||||
"t1": {"len_a": 3600, "len_h": 9100, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 1200, "len_h": 9100, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 0, "len_h": 13700, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
"t4": {"len_a": 1900, "len_h": 18300, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 6200, "len_h": 18300, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': False, 'side': 'right'},
|
||||
|
||||
},
|
||||
"pendulum": {
|
||||
"t1": {"len_a": 3800, "len_h": 16500, 'width': 300, 'height': 300, 'elevation': 400, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 1900, "len_h": 16500, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 0, "len_h": 9100, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
"t4": {"len_a": 1900, "len_h": 16500, 'width': 250, 'height': 250, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 3800, "len_h": 16500, 'width': 300, 'height': 300, 'elevation': 400, 'stop_plate': False, 'side': 'right'},
|
||||
|
||||
},
|
||||
"speed_option": {
|
||||
"t1": {"len_a": 3700, "len_h": 9100, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 6600, "len_h": 32000, 'width': 400, 'height': 600, 'elevation': 150, 'stop_plate': True, 'side': 'left'},
|
||||
"t3": {"len_a": 1800, "len_h": 18300, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t4": {"len_a": 2000, "len_h": 7300, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 6400, "len_h": 13700, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
},
|
||||
"roundabout": {
|
||||
"t1": {"len_a": 2700, "len_h": 13700, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t2": {"len_a": 600, "len_h": 6400, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'left'},
|
||||
"t3": {"len_a": 600, "len_h": 9100, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': True, 'side': 'right'},
|
||||
"t4": {"len_a": 2500, "len_h": 13700, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
"t5": {"len_a": 2500, "len_h": 6400, 'width': 300, 'height': 300, 'elevation': 0, 'stop_plate': False, 'side': 'right'},
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
def _target_info(distance, stage, size, target):
|
||||
values = VALUES[stage][target]
|
||||
|
||||
a_len, h_len = values["len_a"], values["len_h"]
|
||||
org_width, org_height, org_elevation = values['width'], values['height'], values['elevation']
|
||||
x_len = math.sqrt(a_len ** 2 + h_len ** 2)
|
||||
|
||||
scale = distance / h_len
|
||||
target_scale = distance / x_len
|
||||
|
||||
sim_a_len = a_len * scale
|
||||
|
||||
width, height, elevation = org_width * target_scale, org_height * target_scale, org_elevation * target_scale
|
||||
post_width = 40 * target_scale
|
||||
post_height = 150 if size == 'a3' else 100
|
||||
if distance < 1000:
|
||||
post_height = 15
|
||||
if target == 't1':
|
||||
position = 0
|
||||
else:
|
||||
first_target = _target_info(distance, stage, size, 't1')
|
||||
if values['side'] == 'left':
|
||||
position = first_target['sim_a_len'] - sim_a_len
|
||||
else:
|
||||
position = first_target['sim_a_len'] + sim_a_len
|
||||
|
||||
return {
|
||||
'distance': distance,
|
||||
'stage': stage,
|
||||
'target': target,
|
||||
'width': width,
|
||||
'height': height,
|
||||
'elevation': elevation,
|
||||
'sim_a_len': sim_a_len,
|
||||
'stop_plate': values['stop_plate'],
|
||||
'post_height': post_height + elevation,
|
||||
'post_width': post_width,
|
||||
'side': values['side'],
|
||||
'position': int(position),
|
||||
}
|
||||
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
return render_template('index.html')
|
||||
|
||||
|
||||
@app.route('/generate-pdf', methods=['POST'])
|
||||
def generate_pdf():
|
||||
distance = int(request.form.get('distance', 0))
|
||||
distance_in_mm = distance * 10
|
||||
stage = request.form.get('stage')
|
||||
size = request.form.get('size')
|
||||
|
||||
targets = [f't{i}' for i in range(1, 6)]
|
||||
target_info = [_target_info(distance_in_mm, stage, size, target) for target in targets]
|
||||
wall_extra_space_for_paper = 297 if size == 'a3' else 210
|
||||
wall_length = target_info[-1]['position'] + wall_extra_space_for_paper
|
||||
preview_size = 267 if size == 'a3' else 190
|
||||
preview_scale = preview_size / target_info[-1]['position']
|
||||
preview_distance = distance_in_mm * preview_scale
|
||||
preview_target_info = [_target_info(preview_distance, stage, size, target) for target in targets]
|
||||
box_position = int(target_info[0]['sim_a_len']) / 10
|
||||
|
||||
rendered_html = render_template(
|
||||
'pdf_template.html',
|
||||
distance=distance,
|
||||
size=size,
|
||||
stage=stage,
|
||||
target_info=target_info,
|
||||
preview_target_info=preview_target_info,
|
||||
wall_length=wall_length,
|
||||
box_position=box_position,
|
||||
)
|
||||
# return rendered_html
|
||||
|
||||
pdf_file = io.BytesIO()
|
||||
HTML(string=rendered_html).write_pdf(pdf_file)
|
||||
pdf_file.seek(0)
|
||||
filename = 'Paper Challange - '+stage.replace('_', ' ')+' - '+str(distance)+'cm-'+size+'.pdf'
|
||||
|
||||
return send_file(pdf_file, download_name=filename, as_attachment=False)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True)
|
||||
Reference in New Issue
Block a user