Refactor internal data passing to dataclasses and enums, and remove unneeded defensive code

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-08-04 15:13:31 +08:00
parent 8d223177cd
commit e0eb343ce0
11 changed files with 540 additions and 333 deletions
@@ -21,6 +21,7 @@ from pop_fem_audit_tools import config, fetch_lyrics
 from pop_fem_audit_tools.database import Base, DataSource
 from pop_fem_audit_tools.models import (
    Artist,
+    Role,
    Song,
    SongArtist,
 )
@@ -81,7 +82,7 @@ class TestFetchLyrics(unittest.TestCase):
                session.add(song)
                session.add(SongArtist(song=song,
                                       artist=artists[artist],
-                                       role="primary",
+                                       role=Role.PRIMARY,
                                       position=0))
            session.commit()
        finally: