from server.constants import DSP def is_spotify(item: dict) -> bool: return item["vendor"] == DSP.SPOTIFY.value def is_amazon(item: dict) -> bool: return item["vendor"] == DSP.AMAZON.value def get_data(mapping: dict, item: dict) -> dict or None: record_id = item["id"] if is_spotify(item) or is_amazon(item) else (item["id"], item["country_code"]) return mapping.get(record_id)