I prefer a simpler rule
They are told not to do it - they do it - they should suffer the consequences
Anyone who has flown has heard the warning before and in most countries there are other restrictions on where you can smoke - this isn't the 1970's.