The creator of the West Wing television series has been ordered to undergo rehabilitation after pleading guilty to possessing cocaine and magic mushrooms.