Marijuana legality question?

I'm pretty sure there is only one country in Europe where it is legal, but you still have to buy it in specialized shops, sorta like liquor in the US. Even if you smoke there and don't smoke when you get back you'd get in trouble foe failing it.