diff --git a/bricktracker/app.py b/bricktracker/app.py index 1f28f64..e791597 100644 --- a/bricktracker/app.py +++ b/bricktracker/app.py @@ -17,6 +17,7 @@ from bricktracker.version import __version__ from bricktracker.views.add import add_page from bricktracker.views.admin.admin import admin_page from bricktracker.views.admin.database import admin_database_page +from bricktracker.views.admin.export import admin_export_page from bricktracker.views.admin.image import admin_image_page from bricktracker.views.admin.instructions import admin_instructions_page from bricktracker.views.admin.owner import admin_owner_page @@ -149,6 +150,7 @@ def setup_app(app: Flask) -> None: # Register admin routes app.register_blueprint(admin_page) app.register_blueprint(admin_database_page) + app.register_blueprint(admin_export_page) app.register_blueprint(admin_image_page) app.register_blueprint(admin_instructions_page) app.register_blueprint(admin_retired_page) diff --git a/bricktracker/views/admin/export.py b/bricktracker/views/admin/export.py new file mode 100644 index 0000000..dabb6d4 --- /dev/null +++ b/bricktracker/views/admin/export.py @@ -0,0 +1,295 @@ +import csv +import io + +from flask import ( + Blueprint, + request, +) +from flask_login import login_required +from werkzeug.wrappers.response import Response + +from ..exceptions import exception_handler +from ...part_list import BrickPartList +from ...set_list import BrickSetList + +admin_export_page = Blueprint( + 'admin_export', + __name__, + url_prefix='/admin/export' +) + + +# Export all sets to Rebrickable CSV format +@admin_export_page.route('/sets/rebrickable-csv', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_sets_rebrickable() -> Response: + + set_list = BrickSetList() + all_sets = set_list.all() + + output = io.StringIO() + writer = csv.writer(output) + + writer.writerow(['Set Number', 'Quantity', 'Includes Spares', 'Inventory Ver']) + + for set_item in all_sets.records: + writer.writerow([ + set_item.fields.set, + 1, # Each set instance counts as 1 + 'True', # BrickTracker tracks spares separately + 1 # Inventory version (always 1 for now) + ]) + + # Prepare response + output.seek(0) + return Response( + output.getvalue(), + mimetype='text/csv', + headers={'Content-Disposition': 'attachment;filename=bricktracker_sets_rebrickable.csv'} + ) + + +# Export all parts to Rebrickable CSV format +@admin_export_page.route('/parts/rebrickable-csv', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_parts_rebrickable() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.all_filtered(owner_id, color_id, theme_id, year) + + part_quantities = {} + for part in part_list.records: + key = (part.fields.part, part.fields.color) + if key in part_quantities: + part_quantities[key] += part.fields.quantity + else: + part_quantities[key] = part.fields.quantity + + output = io.StringIO() + writer = csv.writer(output) + + writer.writerow(['Part', 'Color', 'Quantity']) + + for (part_num, color_id), quantity in sorted(part_quantities.items()): + writer.writerow([part_num, color_id, quantity]) + + output.seek(0) + return Response( + output.getvalue(), + mimetype='text/csv', + headers={'Content-Disposition': 'attachment;filename=bricktracker_parts_rebrickable.csv'} + ) + + +# Export all parts to LEGO Pick-a-Brick CSV format +@admin_export_page.route('/parts/lego-csv', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_parts_lego() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.all_filtered(owner_id, color_id, theme_id, year) + + element_quantities = {} + for part in part_list.records: + if part.fields.element: + element_id = part.fields.element + if element_id in element_quantities: + element_quantities[element_id] += part.fields.quantity + else: + element_quantities[element_id] = part.fields.quantity + + output = io.StringIO() + writer = csv.writer(output) + + writer.writerow(['elementId', 'quantity']) + + for element_id, quantity in sorted(element_quantities.items()): + writer.writerow([element_id, quantity]) + + output.seek(0) + return Response( + output.getvalue(), + mimetype='text/csv', + headers={'Content-Disposition': 'attachment;filename=bricktracker_parts_lego.csv'} + ) + + +# Export all parts to BrickLink XML format +@admin_export_page.route('/parts/bricklink-xml', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_parts_bricklink() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.all_filtered(owner_id, color_id, theme_id, year) + + part_quantities = {} + for part in part_list.records: + part_num = part.fields.bricklink_part_num or part.fields.part + color_id = part.fields.bricklink_color_id or part.fields.color + + key = (part_num, color_id) + if key in part_quantities: + part_quantities[key] += part.fields.quantity + else: + part_quantities[key] = part.fields.quantity + + xml_lines = [''] + + for (part_num, color_id), quantity in sorted(part_quantities.items()): + xml_lines.append( + f'P{part_num}' + f'{color_id}{quantity}' + ) + + xml_lines.append('') + xml_content = ''.join(xml_lines) + + return Response( + xml_content, + mimetype='application/xml', + headers={'Content-Disposition': 'attachment;filename=bricktracker_parts_bricklink.xml'} + ) + + +# Export missing/damaged parts to Rebrickable CSV format +@admin_export_page.route('/problems/rebrickable-csv', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_problems_rebrickable() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.problem_filtered(owner_id, color_id, theme_id, year) + + part_quantities = {} + for part in part_list.records: + qty = (part.fields.missing or 0) + (part.fields.damaged or 0) + if qty > 0: + key = (part.fields.part, part.fields.color) + if key in part_quantities: + part_quantities[key] += qty + else: + part_quantities[key] = qty + + output = io.StringIO() + writer = csv.writer(output) + + writer.writerow(['Part', 'Color', 'Quantity']) + + for (part_num, color_id), quantity in sorted(part_quantities.items()): + writer.writerow([part_num, color_id, quantity]) + + output.seek(0) + return Response( + output.getvalue(), + mimetype='text/csv', + headers={'Content-Disposition': 'attachment;filename=bricktracker_problems_rebrickable.csv'} + ) + + +# Export missing/damaged parts to LEGO Pick-a-Brick CSV format +@admin_export_page.route('/problems/lego-csv', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_problems_lego() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.problem_filtered(owner_id, color_id, theme_id, year) + + element_quantities = {} + for part in part_list.records: + qty = (part.fields.missing or 0) + (part.fields.damaged or 0) + if qty > 0 and part.fields.element: + element_id = part.fields.element + if element_id in element_quantities: + element_quantities[element_id] += qty + else: + element_quantities[element_id] = qty + + output = io.StringIO() + writer = csv.writer(output) + + writer.writerow(['elementId', 'quantity']) + + for element_id, quantity in sorted(element_quantities.items()): + writer.writerow([element_id, quantity]) + + output.seek(0) + return Response( + output.getvalue(), + mimetype='text/csv', + headers={'Content-Disposition': 'attachment;filename=bricktracker_problems_lego.csv'} + ) + + +# Export missing/damaged parts to BrickLink XML format +@admin_export_page.route('/problems/bricklink-xml', methods=['GET']) +@login_required +@exception_handler(__file__) +def export_problems_bricklink() -> Response: + + owner_id = request.args.get('owner') + color_id = request.args.get('color') + theme_id = request.args.get('theme') + year = request.args.get('year') + + part_list = BrickPartList() + part_list.problem_filtered(owner_id, color_id, theme_id, year) + + part_quantities = {} + for part in part_list.records: + qty = (part.fields.missing or 0) + (part.fields.damaged or 0) + if qty > 0: + part_num = part.fields.bricklink_part_num or part.fields.part + color_id = part.fields.bricklink_color_id or part.fields.color + + key = (part_num, color_id) + if key in part_quantities: + part_quantities[key] += qty + else: + part_quantities[key] = qty + + xml_lines = [''] + + for (part_num, color_id), quantity in sorted(part_quantities.items()): + xml_lines.append( + f'P{part_num}' + f'{color_id}{quantity}' + ) + + xml_lines.append('') + xml_content = ''.join(xml_lines) + + return Response( + xml_content, + mimetype='application/xml', + headers={'Content-Disposition': 'attachment;filename=bricktracker_problems_bricklink.xml'} + ) diff --git a/templates/admin.html b/templates/admin.html index 2da5000..afd4b04 100644 --- a/templates/admin.html +++ b/templates/admin.html @@ -36,6 +36,7 @@ {% include 'admin/set/refresh.html' %} {% else %} {% include 'admin/logout.html' %} + {% include 'admin/export.html' %} {% include 'admin/instructions.html' %} {% if not config['USE_REMOTE_IMAGES'] %} {% include 'admin/image.html' %} diff --git a/templates/admin/export.html b/templates/admin/export.html new file mode 100644 index 0000000..6631f89 --- /dev/null +++ b/templates/admin/export.html @@ -0,0 +1,70 @@ +{{ accordion.header('Export', 'export', 'admin', expanded=open_export, icon='download-line') }} +
+

Export your sets, parts, or missing/damaged parts to various formats for use with Rebrickable, BrickLink, or LEGO Pick-a-Brick.

+ + +
+
+
Export Sets
+
+
+

Export all your sets to Rebrickable format for tracking your collection.

+ + Download Sets (Rebrickable CSV) + +
+
+ + +
+
+
Export All Parts
+
+
+

Export all parts from your collection in different formats.

+
+ + Rebrickable CSV + + + LEGO Pick-a-Brick CSV + + + BrickLink XML + +
+
+
+ + +
+
+
Export Missing/Damaged Parts
+
+
+

Export only missing or damaged parts to create wanted lists or shopping lists.

+
+ + Rebrickable CSV + + + LEGO Pick-a-Brick CSV + + + BrickLink XML + +
+
+
+ + +
+
Format Information
+ +
+
+{{ accordion.footer() }}