They already get generous pensions.

But yes, cap their wealth, limit their stock investments to broad market ETF for stocks and bonds. Extend this to spouses and probably children (for a term). Perhaps provide them with housing in Washington DC.

And attempts to bypass the limits should have very harsh punishments.

Politics should be like teaching - no one does it for the money.