https://www.pc.gc.ca/en/pn-np/ab/waterton/activ/camping
The townsite has a campground. If you drive closer to the US border, there's one called Belly River. Both are Parks Canada campgrounds.
Waterton Springs and Crooked Creek are privately owned campgrounds nearby.
Pincher Creek has some more options if you are willing to drive a little more.