That's not a perk of legalization, since you can do that when the drug is illegal, as well. In fact, many leases do do that for all illegal drugs (which marijuana remains, due to federal prohibition, no matter what happens under state law.)
Same thing with public smoking laws. People smoking up in a park upwind of people deserve a small ticket, an age check, and a harshed mellow. They do not deserve arrest and prosecution on drug laws.
Enforcement of disproportionate punishment is less attractive to sane humans.
Again, legalization doesn't enable anything here.