The US has threatened Australia with punishment if it signs or even praises the global treaty to prohibit nuclear weapons.