excel write

Author: jaesukhwang95

excel/add_to_sheet.js

@@ -0,0 +1,17 @@
+const xlsx = require('xlsx');
+
+function range_add_cell(range, cell) {
+  var rng = xlsx.utils.decode_range(range);
+  var c = typeof cell === 'string' ? xlsx.utils.decode_cell(cell) : cell;
+  if (rng.s.r > c.r) rng.s.r = c.r;
+  if (rng.s.c > c.c) rng.s.c = c.c;
+
+  if (rng.e.r < c.r) rng.e.r = c.r;
+  if (rng.e.c < c.c) rng.e.c = c.c;
+  return xlsx.utils.encode_range(rng);
+}
+
+module.exports = function add_to_sheet(sheet, cell, type, raw) {
+  sheet['!ref'] = range_add_cell(sheet['!ref'], cell);
+  sheet[cell] = { t: type, v: raw };
+};

excel/excelParser.js

@@ -1,9 +1,11 @@
 const xlsx = require('xlsx');
 const axios = require('axios');
 const cheerio = require('cheerio');
+const add_to_sheet = require('./add_to_sheet');
 
 const workbook = xlsx.readFile('./data.xlsx');
 
+console.log(workbook.SheetNames);
 const ws = workbook.Sheets.영화목록;
 
 const records = xlsx.utils.sheet_to_json(ws);
@@ -13,16 +15,21 @@ for(const [i, r] of records.entries()) {
 }
 
 const crawler = async () => {
+    add_to_sheet(ws, 'C1', 's', '평점');
     for (const [i, r] of records.entries()) {
         const response = await axios.get(r.링크);
         if (response.status === 200) {  
             const html = response.data;
             const $ = cheerio.load(html);
             const text = $('.score.score_left .star_score').text();
             console.log(r.제목, '평점', text.trim());
+            const newCell = 'C' + (i +2 );
+            add_to_sheet(ws, newCell, 'n', parseFloat(text.trim()));
         }
     }
+    xlsx.writeFile(workbook, './result.xlsx');
 };
+//순서 보장
 
 // const crawler = async () => {
 //     await Promise.all(records.map( async (r) => {