from typing import Optional, List
from db import db
from models import MarketingAccount


class DataHealthRepository:
    def get_marketing_accounts_by_label_ids(self, label_ids: Optional[List[int]]) -> List[MarketingAccount]:
        query = db.session.query(MarketingAccount)
        if label_ids:
            query = query.filter(MarketingAccount.label_id.in_(label_ids))
        return query.all()