JavaScript封裝——封裝獲取元素的函數

封裝獲取元素的函數

函數的實參形式
1、#id 通過id名獲取元素
2、.class名 通過class名獲取元素
3、ele 通過標籤名獲取元素

<script>

      function getEle(ele) {
        //substr（參數1，參數2）
        //參數1表示開始的索引
        //參數2表示截取個數，如果沒有參數2表示從開始位置截取到最後
        var str1 = ele.substr(0, 1);
        var str2 = ele.substr(1);
        console.log(str2);
        if (str1 === "#") {
          return document.getElementById(str2);
        } else if (str1 === ".") {
          return document.getElementsByClassName(str2);
        } else {
          return document.getElementsByTagName(ele);
        }
      }
    </script>