Need to restrict Context DFF Segments based on PO header Business unit

Summary:
Customer is having multiple business units under single POD, due to security/audit compliances business wants to restrict the DFF context segment list to the respective business units.

If there are two BU A and B and two Context segments "DOA of BU A" and "DOA of BU B," if PO is created for BU A then DFF context "DOA of BU A" automatically should get selected, user should not be able to see or select the context segment "DOA of BU B", vise a versa.

Please let us know is there any way or workaround to satisfy the above business scenario.