mirror of
https://github.com/Horizontal42/Sort_YM.git
synced 2026-11-10 15:42:02 +03:00
ff452b0162
report/apply/digest can now point at any Yandex Music playlist via a share link instead of only the account's likes. Fetches the playlist's tracks and artist genres straight into memory (no disk cache, nothing persisted) and feeds them through the existing build_rows/render_digest/ apply_classification pipeline. apply supports --label to keep a foreign playlist's tracks in separate genre playlists instead of merging them into your own. Closes a gap where UnauthorizedError went uncaught anywhere.
82 lines
3.2 KiB
Python
82 lines
3.2 KiB
Python
from sort_ym import report
|
||
|
||
CATALOG: dict = {} # classify_track с пустыми artist_genre_lists использует только genre_raw
|
||
|
||
|
||
def _track(id_, title, artists, genre_raw, artist_ids=None):
|
||
return {
|
||
"id": id_,
|
||
"album_id": id_ * 10,
|
||
"title": title,
|
||
"artists": artists,
|
||
"artist_ids": artist_ids or [],
|
||
"genre_raw": genre_raw,
|
||
"lyrics_available": False,
|
||
}
|
||
|
||
|
||
def test_small_fine_group_collapses_to_coarse_bucket():
|
||
# 15 "punk"-треков (>= small_group_min=12) остаются отдельным под-жанром,
|
||
# 2 "indie"-трека (< 12) схлопываются в родительскую крупную корзину "indie-alt".
|
||
tracks_cache = {}
|
||
for i in range(15):
|
||
tid = f"{i}:{i}"
|
||
tracks_cache[tid] = _track(i, f"Punk track {i}", ["Punk Artist"], "punk")
|
||
|
||
for i in range(15, 17):
|
||
tid = f"{i}:{i}"
|
||
tracks_cache[tid] = _track(i, f"Indie track {i}", ["Indie Artist"], "indie")
|
||
|
||
rows = report.build_rows(
|
||
tracks_cache=tracks_cache,
|
||
lang_cache={},
|
||
genre_catalog=CATALOG,
|
||
artist_genres={},
|
||
small_group_min=12,
|
||
)
|
||
|
||
punk_rows = [r for r in rows if r["genre_raw"] == "punk"]
|
||
indie_rows = [r for r in rows if r["genre_raw"] == "indie"]
|
||
|
||
assert len(punk_rows) == 15
|
||
assert all(r["bucket"] == "punk" for r in punk_rows), "крупная под-группа - отдельный плейлист"
|
||
assert all(r["fine_bucket"] == "punk" for r in punk_rows)
|
||
|
||
assert len(indie_rows) == 2
|
||
assert all(r["bucket"] == "indie-alt" for r in indie_rows), "мелкая под-группа схлопнута в крупную корзину"
|
||
assert all(r["fine_bucket"] == "indie" for r in indie_rows), "fine_bucket сохраняет исходную классификацию"
|
||
|
||
|
||
def test_write_report_uses_default_filename(tmp_path):
|
||
rows = [{"title": "T", "artists": "A", "genre_raw": "punk", "fine_bucket": "punk", "bucket": "punk", "lang": "RU", "target_playlist": "Панк — RU", "id": 1, "album_id": 10}]
|
||
|
||
out_file = report.write_report(rows, tmp_path)
|
||
|
||
assert out_file.name == "report.csv" == report.REPORT_FILE
|
||
assert out_file.exists()
|
||
|
||
|
||
def test_write_report_accepts_custom_filename_for_source_runs(tmp_path):
|
||
rows = [{"title": "T", "artists": "A", "genre_raw": "punk", "fine_bucket": "punk", "bucket": "punk", "lang": "RU", "target_playlist": "Панк — RU", "id": 1, "album_id": 10}]
|
||
|
||
out_file = report.write_report(rows, tmp_path, report.REPORT_SOURCE_FILE)
|
||
|
||
assert out_file.name == "report_source.csv" == report.REPORT_SOURCE_FILE
|
||
assert out_file.exists()
|
||
|
||
|
||
def test_collapse_is_single_pass_and_terminates():
|
||
# Группа ровно на границе порога (== small_group_min) НЕ схлопывается (строгое <).
|
||
tracks_cache = {
|
||
f"{i}:{i}": _track(i, f"T{i}", ["Artist"], "punk") for i in range(12)
|
||
}
|
||
|
||
rows = report.build_rows(
|
||
tracks_cache=tracks_cache,
|
||
lang_cache={},
|
||
genre_catalog=CATALOG,
|
||
artist_genres={},
|
||
small_group_min=12,
|
||
)
|
||
|
||
assert all(r["bucket"] == "punk" for r in rows)
|