f8652c106b
An "Export" button on the watchlist filter bar opens a modal (same aesthetic as the artist DB-record inspector) to export your whole watchlist roster — each artist's name + source IDs (spotify / musicbrainz / deezer / discogs / itunes / amazon), with an optional "external links" toggle that adds the discography URLs built from those IDs. Live preview, copy, and download in the chosen format. - core/exports/watchlist_export.py: pure builder (json/csv/txt + links, present-IDs only, deterministic columns) — the single source of truth, fully unit-tested. - GET /api/watchlist/export?format=&links= shapes the roster + returns it (with X-Export-Count / X-Export-Ext headers for the modal). - Frontend reuses the DB-record helpers (_jsonSyntaxHighlight / _arecCopy). Tests (8): builder across json/csv/txt, links on/off, present-ids-only, empty + bad-format fallback, mime/ext, and endpoint wiring. ruff clean; 64 integrity green. Scoped to the watchlist for v1; library-wide export + a "library contents" (owned albums/tracks) option are natural follow-ups.
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
"""Watchlist roster export builder (corruption's request) — JSON / CSV / txt,
|
|
optional external links, deterministic columns."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import csv
|
|
import io
|
|
import json
|
|
|
|
from core.exports.watchlist_export import build_watchlist_export, export_mime_and_ext
|
|
|
|
|
|
_ARTISTS = [
|
|
{'artist_name': 'Rob Zombie', 'spotify_artist_id': 'sp1',
|
|
'musicbrainz_artist_id': 'mb1', 'deezer_artist_id': 'dz1'},
|
|
{'artist_name': 'Nobody IDs', 'spotify_artist_id': None},
|
|
]
|
|
|
|
|
|
def test_txt_is_names_one_per_line():
|
|
out = build_watchlist_export(_ARTISTS, fmt='txt')
|
|
assert out == 'Rob Zombie\nNobody IDs'
|
|
|
|
|
|
def test_json_includes_present_ids_only():
|
|
out = json.loads(build_watchlist_export(_ARTISTS, fmt='json'))
|
|
assert out[0]['name'] == 'Rob Zombie'
|
|
assert out[0]['spotify_artist_id'] == 'sp1' and out[0]['musicbrainz_artist_id'] == 'mb1'
|
|
assert 'deezer_artist_id' in out[0]
|
|
assert out[1] == {'name': 'Nobody IDs'} # null id dropped, no links key
|
|
|
|
|
|
def test_json_links_when_requested():
|
|
out = json.loads(build_watchlist_export(_ARTISTS, fmt='json', include_links=True))
|
|
assert out[0]['links']['spotify'] == 'https://open.spotify.com/artist/sp1'
|
|
assert out[0]['links']['musicbrainz'] == 'https://musicbrainz.org/artist/mb1'
|
|
assert 'links' not in out[1] # no ids → no links
|
|
|
|
|
|
def test_csv_header_and_rows():
|
|
out = build_watchlist_export(_ARTISTS, fmt='csv')
|
|
rows = list(csv.reader(io.StringIO(out)))
|
|
assert rows[0][0] == 'name' and 'spotify_artist_id' in rows[0]
|
|
assert rows[1][0] == 'Rob Zombie'
|
|
assert rows[2][0] == 'Nobody IDs'
|
|
|
|
|
|
def test_csv_adds_url_columns_with_links():
|
|
out = build_watchlist_export(_ARTISTS, fmt='csv', include_links=True)
|
|
header = next(csv.reader(io.StringIO(out)))
|
|
assert 'spotify_url' in header and 'discogs_url' in header
|
|
|
|
|
|
def test_empty_and_bad_format():
|
|
assert build_watchlist_export([], fmt='txt') == ''
|
|
assert build_watchlist_export(None, fmt='json') == '[]'
|
|
assert build_watchlist_export(_ARTISTS, fmt='nonsense').startswith('[') # falls back to json
|
|
|
|
|
|
def test_mime_and_ext():
|
|
assert export_mime_and_ext('csv') == ('text/csv', 'csv')
|
|
assert export_mime_and_ext('txt') == ('text/plain', 'txt')
|
|
assert export_mime_and_ext('weird') == ('application/json', 'json')
|
|
|
|
|
|
# ── endpoint wiring (empty watchlist → valid shapes + headers) ──────────────
|
|
import os, tempfile # noqa: E402
|
|
os.environ['DATABASE_PATH'] = os.path.join(tempfile.mkdtemp(prefix='soulsync-testdb-wlexp-'), 'w.db')
|
|
os.environ['SOULSYNC_TEST_DB_READY'] = '1'
|
|
import pytest # noqa: E402
|
|
web_server = pytest.importorskip('web_server')
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
return web_server.app.test_client()
|
|
|
|
|
|
def test_export_endpoint_wiring(client):
|
|
r = client.get('/api/watchlist/export?format=json')
|
|
assert r.status_code == 200
|
|
assert r.data.decode().strip() == '[]'
|
|
assert r.headers.get('X-Export-Count') == '0'
|
|
assert r.headers.get('X-Export-Ext') == 'json'
|
|
|
|
r2 = client.get('/api/watchlist/export?format=csv&links=1')
|
|
assert r2.status_code == 200 and r2.headers.get('X-Export-Ext') == 'csv'
|
|
assert 'spotify_url' in r2.data.decode().splitlines()[0] # header row with links
|