FrederikBaerentsen/BrickTracker
4
6
Fork 6
Code Issues 13 Pull Requests 2 Actions Packages Projects 1 Releases 3 Wiki Activity

Moved code from views to instructions.py

Browse Source
This commit is contained in:
FrederikBaerentsen 2025-01-24 17:20:53 +01:00
parent 4a785df532
commit 6320629b07
2 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
bricktracker/instructions.py
View File

@ -139,6 +139,18 @@ class BrickInstructions(object):
found_tags.append((img_tag['alt'].replace('LEGO Building Instructions for ', ''), a_tag['href'])) # Save alt and href
return found_tags
def get_list(self, request_form, /) -> list:
selected_instructions = []
# Get the list of instructions
for key in request_form:
if key.startswith('instruction-') and request_form.get(key) == 'on': # Checkbox is checked
index = key.split('-')[-1]
alt_text = request_form.get(f'instruction-alt-text-{index}')
href_text = request_form.get(f'instruction-href-text-{index}')
selected_instructions.append((href_text,alt_text))
return selected_instructions
def download(self, href: str, /) -> None:
target = self.path(secure_filename(self.filename))

14
bricktracker/views/instructions.py
View File

@ -147,6 +147,7 @@ def download() -> str:
def do_download() -> Response:
set_id: str = request.form.get('add-set', '')
# BrickInstructions require an argument. Not sure which makes sense here.
found_tags = BrickInstructions(set_id).find_instructions(set_id)
return render_template('instructions.html', download=True, found_tags=found_tags)
@ -155,20 +156,15 @@ def do_download() -> Response:
@login_required
@exception_handler(__file__, post_redirect='instructions.download')
def confirm_download() -> Response:
selected_instructions = []
for key in request.form:
if key.startswith('instruction-') and request.form.get(key) == 'on': # Checkbox is checked
index = key.split('-')[-1]
alt_text = request.form.get(f'instruction-alt-text-{index}')
href_text = request.form.get(f'instruction-href-text-{index}')
selected_instructions.append((href_text,alt_text))
# BrickInstructions require an argument. Not sure which makes sense here.
selected_instructions = BrickInstructions("").get_list(request.form)
if not selected_instructions:
flash("No instructions selected", 'danger')
# No instructions selected
return redirect(url_for('instructions.download'))
for href, filename in selected_instructions:
print(f"Downloading {filename} from {href}")
BrickInstructions(f"{filename}.pdf").download(href)