from typing import List

from apollo_main_db.apollo.models import StarredContent
from src.db.base import session as db_session


def get_starred_content(user_id_list: List[str] or None, isrc_list: List[str] or None):
    """Get users starred content.

    Args:
        user_id_list: List[str] Users IDs.
        isrc_list: List[str] Users IDs.
    """
    query = db_session.query(StarredContent).with_entities(
        StarredContent.id,
        StarredContent.user_id,
        StarredContent.uri,
        StarredContent.isrc,
    )
    if user_id_list:
        query = query.filter(StarredContent.user_id.in_(user_id_list))
    if isrc_list:
        query = query.filter(StarredContent.isrc.in_(isrc_list))
    return query