Support access_authorities in entity manager (#13810)

Author: raymondjacobson

packages/discovery-provider/src/models/tracks/track.py

@@ -83,6 +83,7 @@ class Track(Base, RepresentableMixin):
     slot = Column(Integer)
     is_available = Column(Boolean, nullable=False, server_default=text("true"))
     allowed_api_keys = Column(ARRAY(String))
+    access_authorities = Column(ARRAY(String))
     is_stream_gated = Column(Boolean, nullable=False, server_default=text("false"))
     stream_conditions = Column(JSONB(True))
     is_download_gated = Column(Boolean, nullable=False, server_default=text("false"))

packages/discovery-provider/src/tasks/entity_manager/entities/track.py

@@ -274,6 +274,16 @@ def populate_track_record_metadata(track_record: Track, track_metadata, handle,
                         api_key.lower()
                         for api_key in track_metadata["allowed_api_keys"]
                     ]
+        elif key == "access_authorities":
+            if key in track_metadata:
+                if track_metadata[key] is None:
+                    track_record.access_authorities = None
+                elif isinstance(track_metadata[key], list):
+                    track_record.access_authorities = [
+                        str(addr).strip()
+                        for addr in track_metadata["access_authorities"]
+                        if isinstance(addr, str)
+                    ]
         elif key == "stem_of":
             if "stem_of" in track_metadata and is_valid_json_field(
                 track_metadata, "stem_of"

packages/discovery-provider/src/tasks/entity_manager/utils.py

@@ -432,6 +432,14 @@ def parse_metadata(metadata: str, action: str, entity_type: str):
         cid = data["cid"]
         metadata_json = data["data"]
 
+        if entity_type == EntityType.TRACK and "access_authorities" in data:
+            if (
+                "access_authorities" not in metadata_json
+                or metadata_json["access_authorities"] is None
+            ):
+                metadata_json = dict(metadata_json)
+                metadata_json["access_authorities"] = data["access_authorities"]
+
         # Don't format metadata for UPDATEs
         # This is to support partial updates
         # Individual entities are responsible for updating existing records with metadata

packages/discovery-provider/src/tasks/metadata.py

@@ -142,6 +142,7 @@ class TrackMetadata(TypedDict):
     producer_copyright_line: Optional[Copyright]
     parental_warning_type: Optional[str]
     allowed_api_keys: Optional[str]
+    access_authorities: Optional[List[str]]
     bpm: Optional[float]
     is_custom_bpm: Optional[bool]
     musical_key: Optional[str]
@@ -203,6 +204,7 @@ class TrackMetadata(TypedDict):
     "producer_copyright_line": None,
     "parental_warning_type": None,
     "allowed_api_keys": None,
+    "access_authorities": None,
     "bpm": None,
     "is_custom_bpm": False,
     "musical_key": None,