I'd say, yes: you count the shake. I guess you're asking if the training "consumes" the protein so that you don't need to count it, but I don't think that's the case. At least, not in a way that you'll notice more right after training than at any other time when consuming protein if still training regularly.
You'll probably want to use the TAG model: count 100% of carbs, 50% of protein as if it were carbs, and 10% of fat as if it were carbs. Other models seem to ignore the first 30g of protein, then half the rest, depending on whether it's taken with carbs, etc., but that seems overcomplicated, and that model seems to ignore fat too.