from sqlalchemy.orm.query import Query
from typing import List

from apollo_main_db.apollo.models import UserMarket
from src.db.base import session


def get_emails_query(user_id_list: List[str]) -> Query:
    """Get emails query.

    Args:
        user_id_list: User ID list.

    Returns:
        Emails query.

    """
    return session.query(UserMarket.user_id, UserMarket.email).filter(UserMarket.user_id.in_(user_id_list))