feat(admin): first version of export feature.
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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'}
|
||||
)
|
||||
@@ -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' %}
|
||||
|
||||
@@ -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() }}
|
||||
Reference in New Issue
Block a user