import os
from enum import Enum


class MXBeamline(Enum):
    X06SA = "X06SA"
    X10SA = "X10SA"
    X06DA = "X06DA"
    SIMULATED = "SIMULATED"


def mx_beamline() -> MXBeamline:
    name = os.getenv("BEAMLINE_XNAME", "SIMULATED").strip().upper()
    return MXBeamline[name] if name in MXBeamline.__members__ else MXBeamline.SIMULATED