2017年5月4日 星期四

Google Apps Script 入門到進階(9) --- 仿 Excel Querytable

Google Sheet中匯入網頁表格資料可用importHTML,在Google Apps Script如果也要有像Excel Querytable功能那就要自己刻了,以下就以Google Apps Script仿Excel Querytable的做法。
筆者以證交所的 三大法人買賣金額統計表 為例,配合以下程式碼。
function Gettable() {
  
  var csvUrl = "http://www.twse.com.tw/ch/trading/fund/BFI82U/BFI82U.php";
    
  var t1 = DateAdd('y', -1911, DateAdd('d', 2, new Date())); 
  var payload = {
    qdate : t1.getFullYear() + "/" + padLeft(parseInt(t1.getMonth())+1, 2) + "/" + padLeft(parseInt(t1.getDay()), 2)
  };
  
  var header ={
    'Content-Type':'application/x-www-form-urlencoded'
  };
  
  var options ={
        'method'  : 'post',
        'payload' : payload,
        'headers' : header
  }; 
  
  var Response = UrlFetchApp.fetch(csvUrl, options);
  if(Response.getResponseCode() == 200)
  {
    Querytable(Response.getContentText(), 0, "你的Google Sheet的sheet id", 0, "A3");
  }
}

function Querytable(contentText, tableNo, sheetID, sheetNo, cellPos)
{ 
  var doc = Xml.parse(contentText, true);
  var bodyHtml = doc.html.body.toXmlString();
  var xml = XmlService.parse(bodyHtml).getRootElement();  
  
  var sheet = SpreadsheetApp.openById(sheetID).getSheets()[sheetNo];
  var baseCell = sheet.getRange(cellPos);
  var row = baseCell.getRow();
  var column = baseCell.getColumn();
  
  var tr = getElementsByTagName(getElementsByTagName(xml, 'table')[tableNo], 'tr');
  for(var i = 0 ; i < tr.length ; i++)
  {
    var th = getElementsByTagName(tr[i], 'td');
    for(var j = 0 ; j < th.length ; j++)
    {
      sheet.getRange(row + i, column + j).setValue(th[j].getValue().trim());    
    }
  }
}

function getElementsByTagName(element, tagName) {
  var data = [], descendants = element.getDescendants();
  for(var i in descendants) {
    var elem = descendants[i].asElement();
    if ( elem != null && elem.getName() == tagName) data.push(elem);
  }
  return data;
}
Querytable參數:
contentText:網頁內容。
tableNo:網頁中表格編號,從0開始。
sheetID:Google Sheet ID。
sheetNo:Google Sheet 中的哪個工作表。
cellPos:指定要從哪個儲存格開始存放表格資料。

執行結果:

參考資料: