如何使按钮在单击时在div之间切换？

<div id="container">
  <button id="toggleButton">切换</button>
  <div id="div1">内容1</div>
  <div id="div2">内容2</div>
</div>

#div2 {
  display: none;
}

var toggleButton = document.getElementById("toggleButton");
var div1 = document.getElementById("div1");
var div2 = document.getElementById("div2");

toggleButton.addEventListener("click", function() {
  if (div1.style.display === "none") {
    div1.style.display = "block";
    div2.style.display = "none";
  } else {
    div1.style.display = "none";
    div2.style.display = "block";
  }
});