diff --git a/Simulation.py b/Simulation.py index 5c6eeec..37c3b03 100644 --- a/Simulation.py +++ b/Simulation.py @@ -3,6 +3,7 @@ import pandas as pd import plotly.graph_objects as go from plotly import subplots import os +import sqlite3 from hts.HTS import HTS from stock.util.Stock2Vector import Stock2Vector @@ -126,19 +127,27 @@ class Simulation (HTS): return - def simulate(self, stock_code, today): + def getData(self): + return + + def simulate(self, stock_code, today, type="rule"): LAST_DATA = self.stock2Vector.getLastData(stock_code, today) result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA) - # 규칙 기반의 분석을 통해서 볼린저밴드 상/하단을 계산한다. - data = self.buySellChecker.analyzeByRule(result) + if type == "rule": + # 규칙 기반의 분석을 통해서 볼린저밴드 상/하단을 계산한다. + data = self.buySellChecker.analyzeByRule(result) + # 사야 할 시점과 팔아야 할 시점을 체크한다. + bsLine, data = self.buySellChecker.checkTransaction(data, stock_code, False) + elif type == "labeltest": + data = self.getData(result) + else: + data = None - # 사야 할 시점과 팔아야 할 시점을 체크한다. - bsLine, data = self.buySellChecker.checkTransaction(data, stock_code, False) - - # 그래프를 그린다. - self.draw(stock_code, today, data, bsLine) + if data is not None: + # 그래프를 그린다. + self.draw(stock_code, today, data, bsLine) return diff --git a/stock/util/LabelMaker.py b/stock/util/LabelMaker.py new file mode 100644 index 0000000..50a5e09 --- /dev/null +++ b/stock/util/LabelMaker.py @@ -0,0 +1,8 @@ +import sqlite3 + +class LabelMaker: + def __init__(self): + return + +if __name__ == "__main__": + labelMaker = LabelMaker() \ No newline at end of file