Add -z commend for Chinese columns csv output

�:wq�kb�kbq

:q

Author: cameron93lin

README.md

@@ -19,19 +19,22 @@ Great thanks to leetcode.com, a really awesome website!
 [Advanced](https://skygragon.github.io/leetcode-cli/advanced) ⦙
 [Plugins](https://github.com/skygragon/leetcode-cli-plugins) ⦙
 
-* A very <kbd>[**EFFICIENT**](#quick-start)</kbd> way to fight questions.
-* <kbd>[**CACHING**](https://skygragon.github.io/leetcode-cli/advanced#cache)</kbd> questions to ease offline thinking.
-* <kbd>[**GENERATING**](https://skygragon.github.io/leetcode-cli/commands#show)</kbd> source code before coding.
-* Live <kbd>[**TEST**](https://skygragon.github.io/leetcode-cli/commands#test)</kbd> and <kbd>[**SUBMIT**](https://skygragon.github.io/leetcode-cli/commands#submit)</kbd> with leetcode.com.
-* Download your previous <kbd>[**SUBMISSION**](https://skygragon.github.io/leetcode-cli/commands#submission)</kbd>.
-* Trace your coding <kbd>[**STATUS**](https://skygragon.github.io/leetcode-cli/commands#stat)</kbd>.
-* <kbd>[**AUTO LOGIN**](https://skygragon.github.io/leetcode-cli/advanced#auto-login)</kbd> among multiple agents with single account.
-* Multiple <kbd>[**THEMES**](https://skygragon.github.io/leetcode-cli/advanced#color-themes)</kbd> support.
-* More <kbd>[**PLUGINS**](https://skygragon.github.io/leetcode-cli/advanced#plugins)</kbd> to enjoy extra features!
+- A very <kbd>[**EFFICIENT**](#quick-start)</kbd> way to fight questions.
+- <kbd>[**CACHING**](https://skygragon.github.io/leetcode-cli/advanced#cache)</kbd> questions to ease offline thinking.
+- <kbd>[**GENERATING**](https://skygragon.github.io/leetcode-cli/commands#show)</kbd> source code before coding.
+- Live <kbd>[**TEST**](https://skygragon.github.io/leetcode-cli/commands#test)</kbd> and <kbd>[**SUBMIT**](https://skygragon.github.io/leetcode-cli/commands#submit)</kbd> with leetcode.com.
+- Download your previous <kbd>[**SUBMISSION**](https://skygragon.github.io/leetcode-cli/commands#submission)</kbd>.
+- Trace your coding <kbd>[**STATUS**](https://skygragon.github.io/leetcode-cli/commands#stat)</kbd>.
+- <kbd>[**AUTO LOGIN**](https://skygragon.github.io/leetcode-cli/advanced#auto-login)</kbd> among multiple agents with single account.
+- Multiple <kbd>[**THEMES**](https://skygragon.github.io/leetcode-cli/advanced#color-themes)</kbd> support.
+- More <kbd>[**PLUGINS**](https://skygragon.github.io/leetcode-cli/advanced#plugins)</kbd> to enjoy extra features!
 
 ## Personal Change
+
 Add three more features for List command.
+
 1. `-c` to generate .csv format file for all questions. will include companies tag and companies with 3 differnt time period column. Example useage: `leetcode list -c`
+   - combine '-z' for Chinese column output Example useage: `leetcode list -c -z` (design for better update experience for [Leetcode Spreadsheet](https://docs.google.com/spreadsheets/d/1XFkQJqZlCMitKRnJXEZpWkAx_1dfjbmLG_RkFoz_TPE/edit?usp=sharing))
 2. `-o` to set .csv format file output directory. Default directory: `~/.lc`
 3. `-p` to filter questions by 3 different time period:
    - `1` for questions in recent 0~6 months
@@ -41,8 +44,9 @@ Add three more features for List command.
    - Example useage: `leetcode list -t netfilx -p 1`
 
 Add updateData function for Company plugin. function will include following feature.
+
 1. Update company tag if current user is vip user.
-2. Will use default value to set company tag if current user is not vip user.(data updated at 2020.06.21)
+2. Will use default value to set company tag if current user is not vip user.(data updated at 2021.03.19)
 3. Will store current company tags setting in `~/.lc/leetcode/cache/COMPONIES.json` and `~/.lc/leetcode/cache/TAGS.json` and try to update it if the file is missing.
 
 ## Screenshot
@@ -51,11 +55,10 @@ Add updateData function for Company plugin. function will include following feat
 
 ## Quick Start
 
-	Read help first                         $ leetcode help
-	Login with your leetcode account        $ leetcode user -l
-	Browse all questions                    $ leetcode list
-	Choose one question                     $ leetcode show 1 -g -l cpp
-	Coding it!
-	Run test(s) and pray...                 $ leetcode test ./two-sum.cpp -t '[3,2,4]\n7'
-	Submit final solution!                  $ leetcode submit ./two-sum.cpp
-
+    Read help first                         $ leetcode help
+    Login with your leetcode account        $ leetcode user -l
+    Browse all questions                    $ leetcode list
+    Choose one question                     $ leetcode show 1 -g -l cpp
+    Coding it!
+    Run test(s) and pray...                 $ leetcode test ./two-sum.cpp -t '[3,2,4]\n7'
+    Submit final solution!                  $ leetcode submit ./two-sum.cpp

lib/commands/list.js

@@ -9,6 +9,7 @@ var core = require('../core');
 var file = require('../file');
 var session = require('../session');
 var path = require('path');
+var fs = require('fs');
 const createCsvWriter = require('csv-writer').createObjectCsvWriter;
 
 const cmd = {
@@ -36,6 +37,12 @@ const cmd = {
         default:  path.join(file.homeDir(), 'out.csv'),
         describe: 'Target directory for output file'
       })
+      .option('z', {
+        alias:    'zh',
+        type:     'boolean',
+        default:  false,
+        describe: 'Set output csv as chinese column'
+      })
       .option('t', core.filters.tag)
       .option('x', {
         alias:    'extra',
@@ -79,16 +86,16 @@ cmd.handler = function(argv) {
     const csvWriter = createCsvWriter({
       path: argv.output,
       header: [
-        {id: 'id', title: 'id'},
-        {id: 'name', title: 'name'},
-        {id: 'level', title: 'level'},
-        {id: 'tags', title: 'tags'},
-        {id: 'companies', title: 'companies'},
-        {id: 'companies06', title: 'companies 6 months'},
-        {id: 'companies612', title: 'companies 6-12 months'},
-        {id: 'companies1224', title: 'companies 12-24 months'},
-        {id: 'percent', title: 'percent'},
-        {id: 'category', title: 'category'},
+        {id: 'id', title: argv.z ? "题号" : 'id'},
+        {id: 'name', title: argv.z ? "标题" : 'name'},
+        {id: 'level', title: argv.z ? "难度" : 'level'},
+        {id: 'tags', title: argv.z ? "分类" : 'tags'},
+        {id: 'companies', title: argv.z ? "公司" :'companies'},
+        {id: 'companies06', title: argv.z ? "6个月内出现" :'companies 6 months'},
+        {id: 'companies612', title: argv.z ? "一年内出现" :'companies 6-12 months'},
+        {id: 'companies1224', title: argv.z ? "两年内出现" :'companies 12-24 months'},
+        {id: 'percent', title: argv.z ? "接受率" : 'percent'},
+        {id: 'category', title: argv.z ? "类别" : 'category'},
       ]
     });
     const stat = {};
@@ -102,6 +109,12 @@ cmd.handler = function(argv) {
       if (problem.locked) ++stat.locked;
       if (problem.starred) ++stat.starred;
 
+      var categoryTrans = {
+        "algorithms":"算法",
+        "database":"数据库",
+        "shell":"命令行"
+      }
+
       //build data for csvWriter
       if(argv.csv){
         if(problem.companyStats!=null){
@@ -114,7 +127,7 @@ cmd.handler = function(argv) {
             companies612: problem.companyStats["2"].join(','),
             companies1224: problem.companyStats["3"].join(','),
             percent: problem.percent.toFixed(2),
-            category: problem.category
+            category: argv.z ? categoryTrans[problem.category] : problem.category
           });
         }
         else{
@@ -124,7 +137,7 @@ cmd.handler = function(argv) {
             tags: problem.tags,
             companies: problem.companies,
             percent: problem.percent.toFixed(2),
-            category: problem.category
+            category: argv.z ? categoryTrans[problem.category] : problem.category
           });
         }
       }
@@ -161,7 +174,11 @@ cmd.handler = function(argv) {
     if(argv.csv){
       csvWriter
       .writeRecords(data)
-      .then(()=> console.log('The CSV file was written successfully, output path: '+argv.output));
+      .then(()=> {
+        fs.writeFileSync(argv.output, '\ufeff' + fs.readFileSync(argv.output));
+        console.log('The CSV file was written successfully, output path: '+argv.output);
+      });
+
     }
     if (argv.stat) {
       log.info();