test-exe-getStockMajorData.R

USER_LIB_PATH = Sys.getenv("USER_LIB_PATH", "C:/Users/chiachi/Documents/R/win-library/3.4")
.libPaths(USER_LIB_PATH)

library(httr)
library(XML)
# library(stringr)


getStockMajorData = function(stockId) {
  # Connector
  # url = "http://tw.stock.yahoo.com/d/s/major_2451.html"
  # stockId = "2451"
  url = sprintf("http://tw.stock.yahoo.com/d/s/major_%s.html",as.character(stockId))
  res <- GET(url)
  content(res, "text", encoding = "big5")
  html <- htmlParse(content(res, "text", encoding = "big5"), encoding = "utf8")
  
  # Parser
  tables <- readHTMLTable(html)
  
  # figure out filtering condition ... 
  filter_condition <- (sapply(tables,NCOL)==8)&(sapply(tables,NROW) <= 15)
  data_table <- tables[filter_condition][[1]]
  
  # extract date info
  DataString_source = content(res, "text", encoding = "big5")
  # DataString_regexp <- "([[:digit:]]{3}) /([[:digit:]]{2}) /([[:digit:]]{2})"
  # DataString_Location = str_locate_all(DataString_source,DataString_regexp)[[1]]
  # DataString = str_sub(DataString_source, DataString_Location[1],DataString_Location[2])
  DateString = regmatches(DataString_source,regexpr("([0-9]+) /([0-9]+) /([0-9]+)",DataString_source))
  DateVector = as.numeric(unlist(strsplit(DateString,split = " /")))
  DateVector[1] = DateVector[1] + 1911
  DataDate = as.Date(paste(DateVector,collapse = "-"))
  
  
  # change the data type of each column
  Data_Table = data_table
  Data_Table[,1] = as.factor(Data_Table[,1])
  Data_Table[,2] = as.integer(as.character(Data_Table[,2]))
  Data_Table[,3] = as.integer(as.character(Data_Table[,3]))
  Data_Table[,4] = as.integer(as.character(Data_Table[,4]))
  Data_Table[,5] = as.factor(Data_Table[,5])
  Data_Table[,6] = as.integer(as.character(Data_Table[,6]))
  Data_Table[,7] = as.integer(as.character(Data_Table[,7]))
  Data_Table[,8] = as.integer(as.character(Data_Table[,8]))
  
  # Convert data to table in db
  
  names(Data_Table)[c(1,5)] <- "Broker"
  Data_Table <- rbind(Data_Table[,1:3],Data_Table[,5:7])
  
  names(Data_Table)
  names(Data_Table)[2:3]<-c("Buy","Sell")
  
  Data_Table = data.frame(StockId=stockId,Date=DataDate,Data_Table)
  
  return(Data_Table)
}


library(readr)
STOCK_ID = Sys.getenv("STOCK_ID", "2330")
OUTPUT_PATH = Sys.getenv("OUTPUT_PATH", sprintf("%s.csv",STOCK_ID))

df = getStockMajorData(STOCK_ID)
write_csv(df,OUTPUT_PATH)