How about you declare party affiliation when you register and if you want to change you may do so at any time. But you may not vote in a party primary or participate in a caucus for one election cycle.

Truth be told, my preference is Nader's solution - the total destruction of the two-party system.