"""Shared constants and helpers for PEJ section."""

from database import SessionLocal, ZOPKProject

# Explicit slug list — easy to extend with SMR projects later
NUCLEAR_PROJECT_SLUGS = ['nuclear-plant']

LINK_TYPE_LABELS = {
    'potential_supplier': 'Potencjalny dostawca',
    'partner': 'Partner',
    'investor': 'Inwestor',
    'beneficiary': 'Beneficjent'
}


def get_nuclear_project_ids(db=None):
    """Return IDs of nuclear projects from ZOPK."""
    close = False
    if db is None:
        db = SessionLocal()
        close = True
    try:
        projects = db.query(ZOPKProject.id).filter(
            ZOPKProject.slug.in_(NUCLEAR_PROJECT_SLUGS),
            ZOPKProject.project_type == 'energy'
        ).all()
        return [p.id for p in projects]
    finally:
        if close:
            db.close()