[PAY-2527] Return playlists_containing_track from track apis (#7764)

AudiusProject · Mar 1, 2024 · 4f6972f · 4f6972f
1 parent dbc6dc1
commit 4f6972f
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 0 deletions.
diff --git a/packages/discovery-provider/integration_tests/tasks/entity_manager/test_utils.py b/packages/discovery-provider/integration_tests/tasks/entity_manager/test_utils.py
@@ -189,6 +189,7 @@ def test_valid_parse_metadata(app):
                 "audio_upload_id": None,
                 "placement_hosts": None,
                 "ddex_app": None,
+                "playlists_containing_track": None,
             },
             "QmUpdatePlaylist1": {
                 "playlist_id": 1,

diff --git a/packages/discovery-provider/plugins/pedalboard/packages/storage/src/index.ts b/packages/discovery-provider/plugins/pedalboard/packages/storage/src/index.ts
@@ -987,6 +987,7 @@ export type Tracks = {
   is_download_gated: boolean;
   download_conditions: unknown | null;
   is_original_available: boolean;
+  playlists_containing_track: number[] | null;
   orig_file_cid: string | null;
   orig_filename: string | null;
   ddex_app: string | null;

diff --git a/packages/discovery-provider/src/api/v1/models/tracks.py b/packages/discovery-provider/src/api/v1/models/tracks.py
@@ -117,6 +117,7 @@
         "permalink": fields.String,
         "is_streamable": fields.Boolean,
         "ddex_app": fields.String(allow_null=True),
+        "playlists_containing_track": fields.List(fields.Integer),
     },
 )
 

diff --git a/packages/discovery-provider/src/schemas/track_schema.json b/packages/discovery-provider/src/schemas/track_schema.json
@@ -58,6 +58,13 @@
           "type": "integer",
           "default": 0
         },
+        "playlists_containing_track": {
+          "type": "array",
+          "default": [],
+          "items": {
+            "type": "integer"
+          }
+        },
         "preview_start_seconds": {
           "type": [
             "float",

diff --git a/packages/discovery-provider/src/tasks/metadata.py b/packages/discovery-provider/src/tasks/metadata.py
@@ -74,6 +74,7 @@ class TrackMetadata(TypedDict):
     is_download_gated: Optional[bool]
     download_conditions: Optional[Any]
     is_playlist_upload: Optional[bool]
+    playlists_containing_track: Optional[List[int]]
     ai_attribution_user_id: Optional[int]
     placement_hosts: Optional[str]
     ddex_app: Optional[str]
@@ -121,6 +122,7 @@ class TrackMetadata(TypedDict):
     "is_download_gated": False,
     "download_conditions": None,
     "is_playlist_upload": False,
+    "playlists_containing_track": None,
     "ai_attribution_user_id": None,
     "placement_hosts": None,
     "ddex_app": None,