import os from datetime import datetime, timedelta import pandas as pd import plotly.graph_objects as go from plotly import subplots from BS import BS class Simulation: bs = None def __init__(self): self.bs = BS() #self.connect() return def getCSV(self, fileName, given_day, result): data = pd.read_csv(fileName) days = data.날짜 time = data.시간 open = data.시가 close = data.종가 high = data.고가 low = data.저가 vol = data.거래량 start_time = datetime.strptime(given_day + " 090000", '%Y%m%d %H%M%S') for i in range(len(data)): temp = datetime.strptime(str(days[i]) + " " + str(time[i]).zfill(4)+"00", '%Y%m%d %H%M%S') if temp < start_time: continue if temp not in result["check"]: result["check"].add(temp) result["time"].append(temp) result["open"].append(open[i]) result["close"].append(close[i]) result["high"].append(high[i]) result["low"].append(low[i]) result["vol"].append(vol[i]) return def checkTransaction(self, data): size = len(data["Close"]) bsLine = {} bsLine['buy'] = [-1 for i in range(size)] bsLine['weight'] = [-1 for i in range(size)] bsLine['sell'] = [-1 for i in range(size)] for i in range(21, size-5): if data["High"][i] > data["upper"][i]: bsLine['sell'][i] = data["High"][i] if data["slow_k"][i] <= 36: if data["Low"][i] < data["lower"][i]: bsLine['buy'][i] = data["Close"][i] - 5 if data["slow_k"][i] <= 25: if data["slow_k"][i - 1] < data["slow_d"][i - 1] and data["slow_d"][i] < data["slow_k"][i]: bsLine['buy'][i] = data["Close"][i] - 5 # rsi가 rsis 위로 올라오며 15 이하일 경우 10배로 주문함 (14:30 이전) if data["rsi"][i] < 15 and data["rsis"][i] < 15 and data["rsi"][i - 1] < data["rsis"][i - 1] and data["rsis"][i] < data["rsi"][i]: bsLine['buy'][i] = data["Close"][i] - 5 bsLine['weight'][i] = 10 if data["slow_k"][i] == 1: bsLine['weight'][i] = 8 elif data["slow_k"][i] in(2,3): bsLine['weight'][i] = 7 elif data["slow_k"][i] in(4,5,6): bsLine['weight'][i] = 6 elif data["slow_k"][i] in(7,8,9,10): bsLine['weight'][i] = 5 elif data["slow_k"][i] in(11,12,13,14,15): bsLine['weight'][i] = 4 elif data["slow_k"][i] in(16,17,18,19,20,21): bsLine['weight'][i] = 3 elif data["slow_k"][i] in(22,23,24,25,26,27,28): bsLine['weight'][i] = 2 elif data["slow_k"][i] in(29,30,31,32,33,34,35,36): bsLine['weight'][i] = 1 if data["rsi"][i] < 10: bsLine['weight'][i] = 8 if i<=20: bsLine['weight'][i] = 1 return bsLine def draw(self, stock_code, given_day, data, bsLine): buy_line = bsLine['buy'] sell_line = bsLine['sell'] # 그래프 설정을 위한 변수를 생성한다. data['Open'] = pd.to_numeric(data['Open']) data['High'] = pd.to_numeric(data['High']) data['Low'] = pd.to_numeric(data['Low']) data['Close'] = pd.to_numeric(data['Close']) data['Volume'] = pd.to_numeric(data['Volume']) data["fast_k"] = pd.to_numeric(data['fast_k']) data["slow_k"] = pd.to_numeric(data['slow_k']) data["slow_d"] = pd.to_numeric(data['slow_d']) data["rsi"] = pd.to_numeric(data['rsi']) data["rsis"] = pd.to_numeric(data['rsis']) buy_colors = [] for i in range(len(buy_line)): if buy_line[i] < 0: buy_colors.append("#ffffff") buy_line[i] = data["lower"][0] else: buy_colors.append("#ff00ff") sell_colors = [] for i in range(len(sell_line)): if sell_line[i] < 0: sell_colors.append("#ffffff") sell_line[i] = data["lower"][0] else: sell_colors.append("#00ced1") # 그래프를 설정한다. buy_check = go.Scatter(x=data['Date'], y=buy_line, mode='markers', name="buy", marker=dict(size=14, color=buy_colors, line_width=0)) sell_check = go.Scatter(x=data['Date'], y=sell_line, mode='markers', name="sell", marker=dict(size=14, color=sell_colors, line_width=0)) bolinger_upper = go.Scatter(x=data['Date'], y=data["upper"], name="upper", line_color='#8B4513') bolinger_lower = go.Scatter(x=data['Date'], y=data["lower"], name="lower", line_color='#8B4513') candle_stick = go.Candlestick(x=data['Date'], open=data['Open'], high=data['High'], low=data['Low'], close=data['Close'], increasing_line_color='red', decreasing_line_color='blue') fast_k_line = go.Scatter(x=data['Date'], y=data["fast_k"], mode='lines', name='fast_k') slow_k_line = go.Scatter(x=data['Date'], y=data["slow_k"], mode='lines', name='slow_k') slow_d_line = go.Scatter(x=data['Date'], y=data["slow_d"], mode='lines', name='slow_d') rsi_line = go.Scatter(x=data['Date'], y=data["rsi"], mode='lines', name='rsi') rsis_line = go.Scatter(x=data['Date'], y=data["rsis"], mode='lines', name='rsis') #candle_data = [candle_stick, bolinger_upper, bolinger_lower, buy_check, sell_check, avg1, avg2, avg5, avg10, avg20, avg30, avg40, avg50, avg60] candle_data = [candle_stick, bolinger_upper, bolinger_lower, buy_check, sell_check] stochastic_data = [slow_k_line, slow_d_line] rsi_data = [rsi_line, rsis_line] # 그래프를 그린다. """ fig = go.Figure(data=candle_data) fig.update_layout(title=stock_code + "_" + given_day) fig.show() """ fig = subplots.make_subplots(rows=3, cols=1, subplot_titles=('캔들', "스토캐스틱", "RSI")) for trace in candle_data: fig.append_trace(trace, 1, 1) for trace in stochastic_data: fig.append_trace(trace, 2, 1) for trace in rsi_data: fig.append_trace(trace, 3, 1) #fig.update_xaxes(nticks=5) #fig.update_layout(height=1800, title=stock_code + "_" + given_day, xaxis_rangeslider_visible=False) fig.update_layout(height=1800, title=stock_code + "_" + given_day) fig.show() return def simulate(self, stock_code, given_day): result = {"check": set(), "time": [], "open": [], "close": [], "high": [], "low": [], "vol": []} # 데이터를 가지고 온다. self.getCSV("./data/"+stock_code+"_"+given_day+".csv", given_day, result) # 분석을 통해서 볼린저밴드 상/하단을 계산한다. data = self.bs.analyze(result) # 사야 할 시점과 팔아야 할 시점을 체크한다. bsLine = self.checkTransaction(data) # 그래프를 그린다. self.draw(stock_code, given_day, data, bsLine) return if __name__ == "__main__": today = datetime.today() PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__)))) RESOURCE_DIR = PROJECT_HOME + "/resources/analysis/"+today.strftime("%Y%m%d") stock_codes = ["252670", "122630"] given_days = ['20210901','20210902','20210903','20210906','20210907','20210908','20210909','20210910','20210913','20210914','20210915','20210916','20210917','20210923','20210924','20210927','20210928','20210929','20210930','20211001', '20211005'] given_days = ['20211008'] simulation = Simulation() simulation.simulate(stock_codes[0], given_days[0]) print ("done...")