Hmm... I actually got it to work I think.
However it may be a bit "hacky" way of doing it, I got it working by looking at which button has the selected light on it. Code attached below.
Though I've only been looking at this four about 15-20min, and with no experience with the simple radio or dcs lua files, so there may well be a better way of doing it.