node js modified(메세지 추가 및 케이스 다양화)

 var express = require('express');
 const request = require('request');
 const TARGET_URL = 'https://api.line.me/v2/bot/message/reply'
-const TOKEN = 'OVIxKODBqM8Pn2dpFtFeSLsDbBvApfTu88rh8wFGOzfvgvPjmBH0A4XKii97VxIDO9shYyTix4qGq32vwvL895Rbss5VSVEiU/XG5lOdyTLgECkSQKOdObSetZwcVHbGmzZJ+0Cz5vZrB5KuImOwrwdB04t89/1O/w1cDnyilFU='
+const TOKEN = "sGB5EZq14+Lu4sc6Gt6NJ1Sx/rDtE1BsTyfZeNX42XVVjcgX1kk6/Uj1H40IQ1X3y8MzJLvKGifhioNCJLWQkWH2i95EtNDQyGn7Iqu6MnQRE7pg/z7klGgswS9974s0EKX90FcCfAIndNSYAG+d5gdB04t89/1O/w1cDnyilFU="
 const fs = require('fs');
 const path = require('path');
 const HTTPS = require('https');
-const domain = "2017103989.oss2021.tk"
+const domain = "2016100990.oss2021.tk"
 const sslport = 23023;
 const bodyParser = require('body-parser');
 var app = express();
 
 var stockarr = [];
-
+var pastreply = "";
+var tatic ="";
 app.use('/simages', express.static(__dirname + '/src'));
 app.use(bodyParser.json());
 app.post('/hook', function (req, res) {
@@ -25,33 +26,154 @@ app.post('/hook', function (req, res) {
     console.log('[request message]', eventObj.message);
     if(eventObj.type == 'postback')
     {   
-        if(eventObj.postback.data == 'action=datetemp&selectId=1')
+        if(eventObj.postback.data == 'action=datetemp&selectId=1' && pastreply == "비중 추천"){
+            console.log(stockarr);
+            weight_recommend(eventObj.replyToken, eventObj.postback.params.date)
+            tatic = ""
+            pastreply = ""
+        }
+        else
         {
-
             optimizer(eventObj.replyToken, eventObj.postback.params.date)
             stockarr.splice(0, stockarr.length);
         }
     }
     else
-    {   
+    {   console.log(eventObj.message.text)
+        console.log(pastreply)
         if(eventObj.message.text == '도움말' || eventObj.message.text == '주가 도움말' || eventObj.message.text == '비중 추천 도움말' || eventObj.message.text == '백테스트 도움말')
         {
             printhelp(eventObj.replyToken, eventObj.message.text)
         }
-        else if(eventObj.message.text.indexOf(' ') != -1)
+        else if(eventObj.message.text == '백테스트')
         {   
             date(eventObj.replyToken, eventObj.message.text)
         }
-        else
+        else if (eventObj.message.text == '주가')
         {
+            basicinform_pre(eventObj.replyToken, eventObj.message.text)
+            pastreply = '주가'
+            console.log(pastreply);
+        }
+        else if ( pastreply == '주가' && eventObj.message.text.indexOf(' ') == -1){
             basicinform(eventObj.replyToken, eventObj.message.text)
         }
+        else if (eventObj.message.text == '비중 추천'){
+            weight_1(eventObj.replyToken, eventObj.message.text)
+            pastreply = '비중 추천'
+        }
+        else if (pastreply = '비중 추천' && eventObj.message.text.indexOf(' ') != -1){
+            var holder = eventObj.message.text.split(' ')
+            console.log("holder",holder)
+            var i;
+            for(i = 0; i < holder.length; i++)
+            {
+                stockarr[i] = holder[i];
+            }
+            weight_2(eventObj.replyToken, eventObj.message.text);
+            pastreply = "전략"
+        }
+
+        else if (pastreply = "전략" && (eventObj.message.text == 'gmv' || eventObj.message.text == 'ms') || eventObj.message.text == 'rp'){
+            tatic = eventObj.message.text
+            pastreply = "비중 추천"
+            date(eventObj.replyToken, eventObj.message.text)
+        }
+        else {
+            errormessage(eventObj.replyToken, eventObj.message.text);
+            pastreply = ''
+        }
     }
     
     res.sendStatus(200);
     
 });
 
+function errormessage(replyToken, message){
+    request.post(
+        {
+            url: TARGET_URL,
+            headers: {
+                'Authorization': `Bearer ${TOKEN}`
+            },
+            json: {
+                "replyToken":replyToken,
+                "messages":[
+                    {
+                        "type":"text",
+                        "text":"정해진 양식대로 입력하지 않으셨어요. \n 처음부터 다시 진행해주세요 :)"
+                    }
+                ]
+            }
+        },(error, response, body) => {
+            console.log(body)
+        });
+}
+
+
+function weight_1(replyToken, message){
+    request.post(
+        {
+            url: TARGET_URL,
+            headers: {
+                'Authorization': `Bearer ${TOKEN}`
+            },
+            json: {
+                "replyToken":replyToken,
+                "messages":[
+                    {
+                        "type":"text",
+                        "text":"포트폴리오에 넣을 종목을 입력해주세요. \n 두 종목 이상을 띄어쓰기로 구분해서 입력해주세요!"
+                    }
+                ]
+            }
+        },(error, response, body) => {
+            console.log(body)
+        });
+}
+
+function weight_2(replyToken, message){
+    request.post(
+        {
+            url: TARGET_URL,
+            headers: {
+                'Authorization': `Bearer ${TOKEN}`
+            },
+            json: {
+                "replyToken":replyToken,
+                "messages":[
+                    {
+                        "type":"text",
+                        "text":"전략을 입력해주세요 \n gmv, ms, rp 중 하나를 정확히 입력해주세요!"
+                    }
+                ]
+            }
+        },(error, response, body) => {
+            console.log(body)
+        });
+}
+
+function basicinform_pre(replyToken, message){
+    request.post(
+        {
+            url: TARGET_URL,
+            headers: {
+                'Authorization': `Bearer ${TOKEN}`
+            },
+            json: {
+                "replyToken":replyToken,
+                "messages":[
+                    {
+                        "type":"text",
+                        "text":"종목명을 입력해주세요!"
+                    }
+                ]
+            }
+        },(error, response, body) => {
+            console.log(body)
+        });
+}
+
 function printhelp(replyToken, message){
     if(message == '도움말'){
         request.post(
@@ -195,7 +317,7 @@ function basicinform(replyToken, message) {
     const Callback = (data) => {
         pystring = data.toString();
         if(pystring[0] == '1')
-        {
+        {   pastreply =""
             pystring = pystring.replace('1현', '현');
             request.post(
                 {
@@ -209,11 +331,6 @@ function basicinform(replyToken, message) {
                             {
                                 "type":"text",
                                 "text":pystring
-                            },
-                            {
-                                "type":"image",
-                                "originalContentUrl": "https://2017103989.oss2021.tk:23023/simages/test.png",
-                                "previewImageUrl": "https://2017103989.oss2021.tk:23023/simages/test.png"                            
                             }
                         ]
                     }
@@ -222,7 +339,7 @@ function basicinform(replyToken, message) {
                 });
         }
         else
-        {
+        {   
             var candarr = pystring.split('\n')
             request.post(
                 {
@@ -274,6 +391,7 @@ function basicinform(replyToken, message) {
         }
     };
     process.stdout.on("data", Callback);
+    
 }
 
 function optimizer(replyToken, sdate) {
@@ -317,13 +435,59 @@ function optimizer(replyToken, sdate) {
     process.stdout.on("data", Callback);
 }
 
+function weight_recommend(replyToken, sdate) {
+    sdate = sdate.toString();
+    var pystring = ''
+    var i;
+    for(i = 0; i < stockarr.length; i++)
+    {
+        pystring += stockarr[i];
+        pystring += ','
+    }
+    pystring += sdate;
+    console.log(pystring);
+    const spawn = require("child_process").spawn;
+    const process = spawn("python", ["optimizer.py", pystring ,"weight",tatic]);
+    const Callback = (data) => {
+        pystring = data.toString();
+        pystring = pystring.slice(1,-2)
+        pastreply == ""
+        request.post(
+            {
+                url: TARGET_URL,
+                headers: {
+                    'Authorization': `Bearer ${TOKEN}`
+                },
+                json: {
+                    "replyToken":replyToken,
+                    "messages":[
+                        {
+                            "type":"text",
+                            "text": "비중 추천 결과입니다!"
+                        },
+                        {
+                            "type":"text",
+                            "text":pystring
+                        }
+                    ]
+                }
+            },(error, response, body) => {
+            console.log(body)
+            });
+    }
+    process.stdout.on("data", Callback);
+}
+
 function date(replyToken, message) {
+
+    /*
     var holder = message.split(' ')
     var i;
     for(i = 0; i < holder.length; i++)
     {
         stockarr[i] = holder[i];
     }
+    */
     var today = new Date();   
     var year = today.getFullYear();
     var month = today.getMonth() + 1;

@@ -10,12 +10,8 @@ def get_matches(query, choices, limit=3):
     return result
 
 def basicinform(input):
-<<<<<<< HEAD
-    stocks = pd.read_csv('stockcodename.csv', index_col=0)
-=======
     stocks = pd.read_csv('stockcodename.csv', names=['Symbol', 'Market', 'Name'
     , 'Sector', 'Industry', 'ListingDate', 'SettleMonth', 'Represetitive', 'HomePage', 'Region'], index_col=0)
->>>>>>> f1c2d34f3272bc7cfebf6e887b72493185c4c57c
     symbol = ''
 
     for i in enumerate(stocks.Name):
@@ -50,10 +46,7 @@ def basicinform(input):
     return value
 
 
-<<<<<<< HEAD
-=======
 # print(basicinform('호텔신라'))
 
->>>>>>> f1c2d34f3272bc7cfebf6e887b72493185c4c57c
 args = sys.argv
-print(basicinform(args[0]))
+print(basicinform(args[1]))

@@ -583,42 +583,59 @@ class back_test:
 # print(back_test().backtest_data(['삼성전자','LG전자'],[0.9,0.1],'2010-01-01', '2021-01-01',10000000,3, 'monthly', 'gmv')['pfo_return'][0]['acc_return_ratio'])
 # print(back_test().backtest_data(['삼성전자','LG전자'],[0.9,0.1],'2018-01-01', '2021-01-01',10000000,6, 'monthly', 'gmv'))
 
-args = sys.argv[1].split(",")
-
-assets = []
-gravity = []
-for i in range(len(args)-1):
-    assets.append(args[i])
-    gravity.append(0)
-
-data = back_test().backtest_data(assets,gravity,args[-1], '2021-01-02',10000000,6, 'monthly', 'gmv')
-
-
-#data = back_test().backtest_data(['삼성전자','LG전자','호텔신라'],[0.1,0.1,0.8],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'gmv')
-# data = back_test().backtest_data(['삼성전자','LG전자'],[0.1,0.1],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'gmv')
-x = data['pfo_return'][0]['Date']
-y = data['pfo_return'][0]['acc_return ratio']
-y2 = data['bench'][0]['KOSPI_acc_return']
-y3 = data['bench'][0]['S&P500_acc_return']
-x_ticks = []
-for i,j in enumerate(x):
-    if (i % 6) == 0:
-        x_ticks.append(j)
-    else:
-        x_ticks.append('')
-x_ticks[-1]= x[-1]
-plt.figure(figsize=(10,5))
-ax=plt.gca()
-ax.xaxis.set_major_locator(ticker.MultipleLocator(12))
-plt.plot(x,y,label = 'gmv result')
-plt.plot(x,y2 ,label = 'kospi result')
-plt.plot(x,y3, label = 's&p500 result')
-plt.xticks(x_ticks,rotation=60)
-plt.xlabel('Date')
-plt.ylabel('Return')
-plt.title('result')
-plt.legend()
-plt.show()
-plt.savefig("./src/test.png", dpi = 100)
-print("end")
+
+if (sys.argv[2] == "weight"): 
+    args = sys.argv[1].split(",")
+    assets = []
+    gravity = []
+    for i in range(len(args)-1):
+        assets.append(args[i])
+        gravity.append(0)
+
+    if (sys.argv[3] == 'gmv'):     
+        print(c_Models(assets, gravity, args[-1],'2021-06-01').gmv_opt())
+    elif (sys.argv[3] == 'ms'):
+        print(c_Models(assets, gravity, args[-1],'2021-06-01').ms_opt())
+    elif (sys.argv[3] == 'rp'):
+        print(c_Models(assets, gravity, args[-1],'2021-06-01').rp_opt())
+
+else:
+    args = sys.argv[1].split(",")
+    assets = []
+    gravity = []
+    for i in range(len(args)-1):
+        assets.append(args[i])
+        gravity.append(0)
+
+    data = back_test().backtest_data(assets,gravity,args[-1], '2021-06-01',10000000,6, 'monthly', 'gmv')
+
+
+    #data = back_test().backtest_data(['삼성전자','LG전자','호텔신라'],[0.1,0.1,0.8],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'gmv')
+    # data = back_test().backtest_data(['삼성전자','LG전자'],[0.1,0.1],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'gmv')
+    x = data['pfo_return'][0]['Date']
+    y = data['pfo_return'][0]['acc_return ratio']
+    y2 = data['bench'][0]['KOSPI_acc_return']
+    y3 = data['bench'][0]['S&P500_acc_return']
+    x_ticks = []
+    for i,j in enumerate(x):
+        if (i % 6) == 0:
+            x_ticks.append(j)
+        else:
+            x_ticks.append('')
+    x_ticks[-1]= x[-1]
+    plt.figure(figsize=(10,5))
+    ax=plt.gca()
+    ax.xaxis.set_major_locator(ticker.MultipleLocator(12))
+    plt.plot(x,y,label = 'gmv result')
+    plt.plot(x,y2 ,label = 'kospi result')
+    plt.plot(x,y3, label = 's&p500 result')
+    plt.xticks(x_ticks,rotation=60)
+    plt.xlabel('Date')
+    plt.ylabel('Return')
+    plt.title('result')
+    plt.legend()
+    plt.show()
+    plt.savefig("./src/test.png", dpi = 100)
+    print("end")
+