backtest update

박하늘
Commit 5dc20a06c681c36e6957f6aa6315cdd4e4538655 5dc20a06 1 parent 37ee1c4f
Showing 2 changed files with 59 additions and 54 deletions
server/app.js
server/optimizer.py
--- a/server/app.js
View file @5dc20a0
+++ b/server/app.js
View file @5dc20a0
 var express = require('express');
 const request = require('request');
 const TARGET_URL = 'https://api.line.me/v2/bot/message/reply'
-const TOKEN = "sGB5EZq14+Lu4sc6Gt6NJ1Sx/rDtE1BsTyfZeNX42XVVjcgX1kk6/Uj1H40IQ1X3y8MzJLvKGifhioNCJLWQkWH2i95EtNDQyGn7Iqu6MnQRE7pg/z7klGgswS9974s0EKX90FcCfAIndNSYAG+d5gdB04t89/1O/w1cDnyilFU="
+const PUSH_TARGET_URL = 'https://api.line.me/v2/bot/message/push'
+const TOKEN = "OVIxKODBqM8Pn2dpFtFeSLsDbBvApfTu88rh8wFGOzfvgvPjmBH0A4XKii97VxIDO9shYyTix4qGq32vwvL895Rbss5VSVEiU/XG5lOdyTLgECkSQKOdObSetZwcVHbGmzZJ+0Cz5vZrB5KuImOwrwdB04t89/1O/w1cDnyilFU="
 const fs = require('fs');
 const path = require('path');
 const HTTPS = require('https');
-const domain = "2016100990.oss2021.tk"
+const domain = "2017103989.oss2021.tk"
 const sslport = 23023;
 const bodyParser = require('body-parser');
 var app = express();
@@ -13,6 +14,7 @@ var app = express();
 var stockarr = [];
 var pastreply = "";
 var tatic ="";
+var backtest = 0;
 app.use('/simages', express.static(__dirname + '/src'));
 app.use(bodyParser.json());
 app.post('/hook', function (req, res) {
@@ -26,34 +28,32 @@ app.post('/hook', function (req, res) {
     console.log('[request message]', eventObj.message);
     if(eventObj.type == 'postback')
     {   
-        if(eventObj.postback.data == 'action=datetemp&selectId=1' && pastreply == "비중 추천"){
+        if(eventObj.postback.data == 'action=datetemp&selectId=1' && backtest == 0)
-            console.log(stockarr);
+        {
             weight_recommend(eventObj.replyToken, eventObj.postback.params.date)
+            stockarr.splice(0, stockarr.length);
             tatic = ""
             pastreply = ""
         }
-        else
+        else if(eventObj.postback.data == 'action=datetemp&selectId=1' && backtest == 1)
         {
-            optimizer(eventObj.replyToken, eventObj.postback.params.date)
+            optimizer(eventObj.source.userId, eventObj.postback.params.date)
             stockarr.splice(0, stockarr.length);
+            tatic = ""
+            pastreply = ""
+            backtest = 0;
         }
     }
     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 == '백테스트')
-        {   
-            date(eventObj.replyToken, eventObj.message.text)
-        }
         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)
@@ -62,9 +62,13 @@ app.post('/hook', function (req, res) {
             weight_1(eventObj.replyToken, eventObj.message.text)
             pastreply = '비중 추천'
         }
-        else if (pastreply = '비중 추천' && eventObj.message.text.indexOf(' ') != -1){
+        else if (eventObj.message.text == '백테스트'){
+            weight_1(eventObj.replyToken, eventObj.message.text)
+            pastreply = '비중 추천'
+            backtest = 1
+        }
+        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++)
             {
@@ -74,7 +78,7 @@ app.post('/hook', function (req, res) {
             pastreply = "전략"
         }
-        else if (pastreply = "전략" && (eventObj.message.text == 'gmv' || eventObj.message.text == 'ms') || eventObj.message.text == 'rp'){
+        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)
@@ -257,7 +261,7 @@ function printhelp(replyToken, message){
                     "messages":[
                         {
                             "type":"text",
-                            "text":"사용자 : 비중 추천 도움말\n챗봇 : 포트폴리오에 넣을 종목을 선택해주세요(2 종목 이상)\n사용자 : 종목명 입력 (ex 삼성전자, LG전자 ...)\n챗봇: 전략을 선택해주세요. (gmv, ms , rp)\n사용자: gmv, ms, rp 중 입력\n챗봇 : 케이스에 맞게 함수 실행 후 비중 출력"
+                            "text":"사용자 : 비중 추천\n챗봇 : 포트폴리오에 넣을 종목을 선택해주세요(2 종목 이상)\n사용자: 종목명 입력\n챗봇: 시작할 날짜를 입력해주세요\n사용자 : 종목명 입력 (ex 삼성전자, LG전자 ...)\n챗봇: 전략을 선택해주세요. (gmv, ms , rp)\n사용자: gmv, ms, rp 중 입력\n챗봇 : 케이스에 맞게 함수 실행 후 비중 출력"
                         }
                     ]
                 }
@@ -278,7 +282,7 @@ function printhelp(replyToken, message){
                     "messages":[
                         {
                             "type":"text",
-                            "text":"사용자 : 백테스트 도움말\n챗봇: 포트폴리오에 넣을 종목을 선택해주세요(2 종목 이상)\n사용자: 종목명 입력\n챗봇: 시작할 날짜를 입력해주세요\n사용자: 입력\n챗봇: 전략을 선택해주세요 (gmv, ms , rp)\n사용자: gmv,ms,rp 중 입력\n챗봇: 함수 실행 후 그래프 출력"
+                            "text":"사용자 : 백테스트\n챗봇: 포트폴리오에 넣을 종목을 선택해주세요(2 종목 이상)\n사용자: 종목명 입력\n챗봇: 시작할 날짜를 입력해주세요\n사용자: 입력\n챗봇: 전략을 선택해주세요 (gmv, ms , rp)\n사용자: gmv,ms,rp 중 입력\n챗봇: 함수 실행 후 그래프 출력"
                         }
                     ]
                 }
@@ -299,7 +303,7 @@ function printhelp(replyToken, message){
                     "messages":[
                         {
                             "type":"text",
-                            "text":"사용자 : 주가 도움말\n챗봇: 종목명을 알려주세요.\n사용자: 종목명 입력 (ex 삼성전자)\n챗봇 : 현재가 거래량 전일대비 수익률"
+                            "text":"사용자 : 주가\n챗봇: 종목명을 알려주세요.\n사용자: 종목명 입력 (ex 삼성전자)\n챗봇 : 현재가 거래량 전일대비 수익률"
                         }
                     ]
                 }
@@ -394,7 +398,7 @@ function basicinform(replyToken, message) {
 }
-function optimizer(replyToken, sdate) {
+function optimizer(userid, sdate) {
     sdate = sdate.toString();
     var pystring = ''
     var i;
@@ -406,16 +410,16 @@ function optimizer(replyToken, sdate) {
     pystring += sdate;
     console.log(pystring);
     const spawn = require("child_process").spawn;
-    const process = spawn("python", ["optimizer.py", pystring]);
+    const process = spawn("python", ["optimizer.py", pystring, 'backtest', tatic]);
     const Callback = (data) => {
         request.post(
             {
-                url: TARGET_URL,
+                url: PUSH_TARGET_URL,
                 headers: {
                     'Authorization': `Bearer ${TOKEN}`
                 },
                 json: {
-                    "replyToken":replyToken,
+                    "to": userid,
                     "messages":[
                         {
                             "type":"text",
--- a/server/optimizer.py
View file @5dc20a0
+++ b/server/optimizer.py
View file @5dc20a0
@@ -606,36 +606,37 @@ else:
     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', sys.argv[3])
-    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')
+#data = back_test().backtest_data(['삼성전자','LG전자','카카오','호텔신라'],[0,0,0,0],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'ms')
-    x = data['pfo_return'][0]['Date']
+# data = back_test().backtest_data(['삼성전자','LG전자'],[0.1,0.1],'2020-01-01', '2021-01-02',10000000,6, 'monthly', 'gmv')
-    y = data['pfo_return'][0]['acc_return ratio']
+x = data['pfo_return'][0]['Date']
-    y2 = data['bench'][0]['KOSPI_acc_return']
+y = data['pfo_return'][0]['acc_return ratio']
-    y3 = data['bench'][0]['S&P500_acc_return']
+y2 = data['bench'][0]['KOSPI_acc_return']
-    x_ticks = []
+y3 = data['bench'][0]['S&P500_acc_return']
-    for i,j in enumerate(x):
+x_ticks = []
-        if (i % 6) == 0:
+for i,j in enumerate(x):
-            x_ticks.append(j)
+    if (i % 6) == 0:
-        else:
+        x_ticks.append(j)
-            x_ticks.append('')
+    else:
-    x_ticks[-1]= x[-1]
+        x_ticks.append('')
-    plt.figure(figsize=(10,5))
+x_ticks[-1]= x[-1]
-    ax=plt.gca()
+plt.figure(figsize=(10,5))
-    ax.xaxis.set_major_locator(ticker.MultipleLocator(12))
+ax=plt.gca()
-    plt.plot(x,y,label = 'gmv result')
+ax.xaxis.set_major_locator(ticker.MultipleLocator(12))
-    plt.plot(x,y2 ,label = 'kospi result')
+plt.plot(x,y,label = 'gmv result')
-    plt.plot(x,y3, label = 's&p500 result')
+plt.plot(x,y2 ,label = 'kospi result')
-    plt.xticks(x_ticks,rotation=60)
+plt.plot(x,y3, label = 's&p500 result')
-    plt.xlabel('Date')
+plt.xticks(x_ticks,rotation=60)
-    plt.ylabel('Return')
+plt.xlabel('Date')
-    plt.title('result')
+plt.ylabel('Return')
-    plt.legend()
+plt.title('result')
-    plt.show()
+plt.legend()
-    plt.savefig("./src/test.png", dpi = 100)
+plt.show()
-    print("end")
+plt.savefig("./src/test.png", dpi = 100)
+print("end")