feat(admin): first version of export feature.

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)

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 = ['<INVENTORY>']
+
+    for (part_num, color_id), quantity in sorted(part_quantities.items()):
+        xml_lines.append(
+            f'<ITEM><ITEMTYPE>P</ITEMTYPE><ITEMID>{part_num}</ITEMID>'
+            f'<COLOR>{color_id}</COLOR><MINQTY>{quantity}</MINQTY></ITEM>'
+        )
+
+    xml_lines.append('</INVENTORY>')
+    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 = ['<INVENTORY>']
+
+    for (part_num, color_id), quantity in sorted(part_quantities.items()):
+        xml_lines.append(
+            f'<ITEM><ITEMTYPE>P</ITEMTYPE><ITEMID>{part_num}</ITEMID>'
+            f'<COLOR>{color_id}</COLOR><MINQTY>{quantity}</MINQTY></ITEM>'
+        )
+
+    xml_lines.append('</INVENTORY>')
+    xml_content = ''.join(xml_lines)
+
+    return Response(
+        xml_content,
+        mimetype='application/xml',
+        headers={'Content-Disposition': 'attachment;filename=bricktracker_problems_bricklink.xml'}
+    )

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' %}

templates/admin/export.html

@@ -0,0 +1,70 @@
+{{ accordion.header('Export', 'export', 'admin', expanded=open_export, icon='download-line') }}
+  <div class="p-3">
+    <p class="text-muted">Export your sets, parts, or missing/damaged parts to various formats for use with Rebrickable, BrickLink, or LEGO Pick-a-Brick.</p>
+
+    <!-- Export Sets Section -->
+    <div class="card mb-3">
+      <div class="card-header">
+        <h6 class="mb-0"><i class="ri-stack-line"></i> Export Sets</h6>
+      </div>
+      <div class="card-body">
+        <p class="small text-muted">Export all your sets to Rebrickable format for tracking your collection.</p>
+        <a href="{{ url_for('admin_export.export_sets_rebrickable') }}" class="btn btn-sm btn-primary">
+          <i class="ri-file-text-line"></i> Download Sets (Rebrickable CSV)
+        </a>
+      </div>
+    </div>
+
+    <!-- Export All Parts Section -->
+    <div class="card mb-3">
+      <div class="card-header">
+        <h6 class="mb-0"><i class="ri-shapes-line"></i> Export All Parts</h6>
+      </div>
+      <div class="card-body">
+        <p class="small text-muted">Export all parts from your collection in different formats.</p>
+        <div class="d-flex gap-2 flex-wrap">
+          <a href="{{ url_for('admin_export.export_parts_rebrickable') }}" class="btn btn-sm btn-success">
+            <i class="ri-file-text-line"></i> Rebrickable CSV
+          </a>
+          <a href="{{ url_for('admin_export.export_parts_lego') }}" class="btn btn-sm btn-warning">
+            <i class="ri-file-text-line"></i> LEGO Pick-a-Brick CSV
+          </a>
+          <a href="{{ url_for('admin_export.export_parts_bricklink') }}" class="btn btn-sm btn-info">
+            <i class="ri-file-code-line"></i> BrickLink XML
+          </a>
+        </div>
+      </div>
+    </div>
+
+    <!-- Export Problems Section -->
+    <div class="card">
+      <div class="card-header">
+        <h6 class="mb-0"><i class="ri-error-warning-line"></i> Export Missing/Damaged Parts</h6>
+      </div>
+      <div class="card-body">
+        <p class="small text-muted">Export only missing or damaged parts to create wanted lists or shopping lists.</p>
+        <div class="d-flex gap-2 flex-wrap">
+          <a href="{{ url_for('admin_export.export_problems_rebrickable') }}" class="btn btn-sm btn-success">
+            <i class="ri-file-text-line"></i> Rebrickable CSV
+          </a>
+          <a href="{{ url_for('admin_export.export_problems_lego') }}" class="btn btn-sm btn-warning">
+            <i class="ri-file-text-line"></i> LEGO Pick-a-Brick CSV
+          </a>
+          <a href="{{ url_for('admin_export.export_problems_bricklink') }}" class="btn btn-sm btn-info">
+            <i class="ri-file-code-line"></i> BrickLink XML
+          </a>
+        </div>
+      </div>
+    </div>
+
+    <!-- Format Info -->
+    <div class="alert alert-info mt-3 mb-0">
+      <h6 class="alert-heading"><i class="ri-information-line"></i> Format Information</h6>
+      <ul class="mb-0 small">
+        <li><strong>Rebrickable CSV:</strong> Part,Color,Quantity format for direct import to Rebrickable</li>
+        <li><strong>LEGO Pick-a-Brick CSV:</strong> Element ID and quantity for LEGO's Pick-a-Brick service</li>
+        <li><strong>BrickLink XML:</strong> Wanted list format for importing to BrickLink</li>
+      </ul>
+    </div>
+  </div>
+{{ accordion.footer() }}