496 lines
19 KiB
Python
496 lines
19 KiB
Python
import json
|
|
import os
|
|
import time
|
|
import shutil
|
|
from stockpredictor.analysis.Common import Common
|
|
from stockpredictor.analysis.MACD import MACD
|
|
from stockpredictor.analysis.RSI import RSI
|
|
from stockpredictor.analysis.Stochastic import Stochastic
|
|
from stockpredictor.analysis.IchimokuCloud import IchimokuCloud
|
|
import matplotlib.pyplot as plt
|
|
import datetime
|
|
import sqlite3
|
|
from datetime import datetime
|
|
|
|
from matplotlib import rc
|
|
import math
|
|
|
|
rc('font', family='AppleGothic')
|
|
plt.rcParams['axes.unicode_minus'] = False
|
|
|
|
import pandas as pd
|
|
import plotly.graph_objs as go
|
|
from plotly import tools, subplots
|
|
import plotly.io as po
|
|
|
|
class Analyzer:
|
|
tableName = 'stock'
|
|
PROJECT_HOME = None
|
|
|
|
stocks = None
|
|
candidate = None
|
|
|
|
macd = None
|
|
rsi = None
|
|
stochastic = None
|
|
ichimokuCloud = None
|
|
|
|
common = None
|
|
inFileName = None
|
|
fnguideFileName = None
|
|
|
|
fnguide = {}
|
|
|
|
def __init__(self, PROJECT_HOME, inFileName, fnguideFileName):
|
|
self.PROJECT_HOME = PROJECT_HOME
|
|
self.inFileName = inFileName
|
|
self.fnguideFileName = fnguideFileName
|
|
|
|
self.stocks = []
|
|
self.candidate = []
|
|
|
|
self.common = Common()
|
|
|
|
self.macd = MACD()
|
|
self.rsi = RSI()
|
|
self.stochastic = Stochastic()
|
|
self.ichimokuCloud = IchimokuCloud()
|
|
|
|
self.readFnguide()
|
|
return
|
|
|
|
def readFnguide(self):
|
|
conn = sqlite3.connect(self.fnguideFileName)
|
|
cursor = conn.cursor()
|
|
|
|
today = datetime.today()
|
|
year1 = str(today.year - 1) + ".12.01"
|
|
year2 = str(today.year - 2) + ".12.01"
|
|
year3 = str(today.year - 3) + ".12.01"
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM fnguide WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
data = json.loads(result[2])
|
|
self.fnguide[result[0]] = True
|
|
|
|
if (year1 in data and year2 in data and year3 in data):
|
|
if (data[year1]['영업이익'] < 0 and data[year2]['영업이익'] < 0 and data[year3]['영업이익'] < 0):
|
|
# 3년 연속 영업이익이 적자이면 매수하지 않는다.
|
|
self.fnguide[result[0]] = False
|
|
if (data[year1]['영업이익'] < -100):
|
|
# 전년 영억적자가 100억 이상이면 매수하지 않는다.
|
|
self.fnguide[result[0]] = False
|
|
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM fnguide WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
cursor.close()
|
|
conn.close()
|
|
return
|
|
|
|
def draw(self, stock):
|
|
|
|
last_index = self.get_last_index(stock)
|
|
if last_index > 300:
|
|
index = 300 # 최대 300일치 그래프 확인
|
|
df_stock = pd.DataFrame(stock["PRICE"][len(stock["PRICE"]) - index:])
|
|
df_macd = pd.DataFrame(stock["MACD"][len(stock["MACD"]) - index:last_index+1])
|
|
df_stochastic = pd.DataFrame(stock["STOCHASTIC"][len(stock["STOCHASTIC"]) - index:last_index+1])
|
|
df_rsi = pd.DataFrame(stock["RSI"][len(stock["RSI"]) - index:last_index+1])
|
|
df_ichimoku = pd.DataFrame(stock["ICHIMOKU"][len(stock["ICHIMOKU"]) - index:])
|
|
else:
|
|
index = last_index
|
|
df_stock = pd.DataFrame(stock["PRICE"][:index+1])
|
|
df_macd = pd.DataFrame(stock["MACD"][:index+1])
|
|
df_stochastic = pd.DataFrame(stock["STOCHASTIC"][:index+1])
|
|
df_ichimoku = pd.DataFrame(stock["ICHIMOKU"][:index+1])
|
|
df_rsi = pd.DataFrame(stock["RSI"][:index+1])
|
|
|
|
# general
|
|
volume = go.Bar(x=df_stock.DATE, y=df_stock['volume'], name="volume")
|
|
volume_data = [volume]
|
|
|
|
leadingSpan1 = go.Scatter(x=df_ichimoku.DATE, y=df_ichimoku['leadingSpan1'], name="선행스팬", line_color='#8B4513')
|
|
leadingSpan2 = go.Scatter(x=df_ichimoku.DATE, y=df_ichimoku['leadingSpan2'], name="후행스팬", line_color='#4169E1')
|
|
cnadle = go.Candlestick(x=df_stock.DATE, open=df_stock.open, high=df_stock.high, low=df_stock.low, close=df_stock.close, increasing_line_color= 'red', decreasing_line_color= 'blue')
|
|
ichimokuCloud_data = [leadingSpan1, leadingSpan2, cnadle]
|
|
|
|
# macd
|
|
macd = go.Scatter(x=df_macd.DATE, y=df_macd['macd'], name="MACD", line_color='#8B4513')
|
|
macd_signal = go.Scatter(x=df_macd.DATE, y=df_macd['macds'], name="MACD Signal", line_color='#4169E1')
|
|
oscillator = go.Bar(x=df_macd.DATE, y=df_macd['macdo'], name="oscillator")
|
|
macd_data = [macd, macd_signal, oscillator]
|
|
|
|
# stochastic
|
|
slow_k = go.Scatter(x=df_stochastic.DATE, y=df_stochastic['slow_k'], name="Slow%K", line_color='#8B4513')
|
|
slow_d = go.Scatter(x=df_stochastic.DATE, y=df_stochastic['slow_d'], name="Slow%D", line_color='#4169E1')
|
|
stochastic_data = [slow_k, slow_d]
|
|
|
|
# rsi
|
|
rsi = go.Scatter(x=df_macd.DATE, y=df_rsi['rsi'], name="RSI", line_color='#8B4513')
|
|
rsi_signal = go.Scatter(x=df_macd.DATE, y=df_rsi['rsis'], name="RSI Signal", line_color='#4169E1')
|
|
rsi_data = [rsi, rsi_signal]
|
|
|
|
fig = subplots.make_subplots(rows=5, cols=1, subplot_titles=('MACD', 'Stochastic', 'RSI', '거래량', '일목균형표'))
|
|
|
|
for trace in macd_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)
|
|
for trace in volume_data:
|
|
fig.append_trace(trace, 4, 1)
|
|
for trace in ichimokuCloud_data:
|
|
fig.append_trace(trace, 5, 1)
|
|
|
|
fig.update_layout(height=2000)
|
|
|
|
return fig
|
|
|
|
def get_last_index(self, stock):
|
|
for i in range(0, len(stock['PRICE'])):
|
|
if (stock['PRICE'][i]['close'] == 0 and stock['PRICE'][i]['open'] == 0 and stock['PRICE'][i]['volume'] == 0):
|
|
return i-1
|
|
return len(stock['PRICE']) - 1
|
|
|
|
def analyzeMACD(self):
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
|
|
# 기존 분석 데이터를 모두 지움
|
|
cursor.execute('update ' + self.tableName + ' set MACD = ""')
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": json.loads(result[2])}
|
|
results = self.macd.analyze(stock)
|
|
text = json.dumps(results, ensure_ascii=False)
|
|
cursor.execute("UPDATE " + self.tableName + " SET MACD=? WHERE CODE=?", (text, stock["CODE"]))
|
|
|
|
print("#analyzeMACD", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
cursor.close()
|
|
conn.close()
|
|
|
|
return
|
|
|
|
def analyzeRSI(self):
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
|
|
# 기존 분석 데이터를 모두 지움
|
|
cursor.execute('update ' + self.tableName + ' set RSI = ""')
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
prices = json.loads(result[2])
|
|
for price in prices:
|
|
del price['diff']
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": prices}
|
|
results = self.rsi.analyze(stock)
|
|
text = json.dumps(results, ensure_ascii=False)
|
|
cursor.execute("UPDATE " + self.tableName + " SET RSI=? WHERE CODE=?", (text, stock["CODE"]))
|
|
|
|
print("#analyzeRSI", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
cursor.close()
|
|
conn.close()
|
|
|
|
return
|
|
|
|
def analyzeStochastic(self):
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
|
|
# 기존 분석 데이터를 모두 지움
|
|
cursor.execute('update ' + self.tableName + ' set STOCHASTIC = ""')
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": json.loads(result[2])}
|
|
results = self.stochastic.analyze(stock)
|
|
text = json.dumps(results, ensure_ascii=False)
|
|
cursor.execute("UPDATE " + self.tableName + " SET STOCHASTIC=? WHERE CODE=?", (text, stock["CODE"]))
|
|
|
|
print("#analyzeStochastic", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
cursor.close()
|
|
conn.close()
|
|
return
|
|
|
|
def analyzeIchimokuCloud(self):
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
|
|
# 기존 분석 데이터를 모두 지움
|
|
cursor.execute('update ' + self.tableName + ' set ICHIMOKU = ""')
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": json.loads(result[2])}
|
|
results = self.ichimokuCloud.analyze(stock)
|
|
text = json.dumps(results, ensure_ascii=False)
|
|
cursor.execute("UPDATE " + self.tableName + " SET ICHIMOKU=? WHERE CODE=?", (text, stock["CODE"]))
|
|
|
|
print("#analyzeIchimokuCloud", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
cursor.close()
|
|
conn.close()
|
|
return
|
|
|
|
def analyzeFinalScore(self, last_index, STOCK, MACD, STOCHASTIC, ICHIMOKU, RSI):
|
|
"""
|
|
매수 조건
|
|
#0. 최소 매수 조건은 거래량은 20만건, 종가는 2천원 이상인 종목이어야 한다.
|
|
#1. 골든크로스: 5일선, 20일선, 60일선, 120일선이 순서대로 나열되는 순간
|
|
"""
|
|
i = last_index
|
|
buy_price = 0
|
|
count = 0
|
|
for idx in range(i, i-5, -1):
|
|
if idx-1 < 0:
|
|
break
|
|
buy_price += STOCK[idx-1]['close'] - STOCK[idx]['low']
|
|
count += 1
|
|
if count == 0:
|
|
buy_price = STOCK[i]['close']
|
|
else:
|
|
# 종가 - 최저가의 최근 3일 평균 가격을 산정한다.
|
|
buy_price = round(STOCK[i]['close'] - (buy_price/count))
|
|
|
|
status = ""
|
|
if STOCK[i]['volume'] > 100000 and STOCK[i]['close'] > 2000:
|
|
# 거래량이 100만 이상이고, 종가가 1천원 이상인지 체크 (https://happpy-rich.tistory.com/94)
|
|
|
|
all_upper_cross_status = self.common.checkAllUpperCross(STOCK, i)
|
|
if all_upper_cross_status != "":
|
|
status += all_upper_cross_status
|
|
|
|
# GOLDENCROSS#1은 바로 매수하지 않고, 이 시점 이후로 5일선이 20일선을 하방으로 뚫었다가 다시 20일선을 상방으로 뚫는 순간 매수를 시도한다.
|
|
# GOLDENCROSS#2은 바로 매수 가능
|
|
# GOLDENCROSS#3은 바로 매수 가능
|
|
golden_cross_status = self.common.check_golded_cross(STOCK, i)
|
|
if golden_cross_status != "":
|
|
status += golden_cross_status
|
|
|
|
# BUYINGBEARMARKET#1은 바로 매수 가능
|
|
# BUYINGBEARMARKET#2은 바로 매수 가능
|
|
bearmarket_buying_status = self.common.check_bearmarket_buying(STOCK, STOCHASTIC, i)
|
|
if bearmarket_buying_status != "":
|
|
status += bearmarket_buying_status
|
|
|
|
# BUYINGSTOCHASTIC#1은 바로 매수 가능
|
|
stochastic_buying_status = self.common.check_stochastic_buying(STOCK, STOCHASTIC, ICHIMOKU, i)
|
|
if stochastic_buying_status != "":
|
|
status += stochastic_buying_status
|
|
|
|
# 20
|
|
if self.common.check_20days_line_buying(STOCK, i):
|
|
if STOCHASTIC[i]['slow_k'] < 40:
|
|
status += '20_'
|
|
|
|
# 60
|
|
if self.common.check_60days_line_buying(STOCK, i):
|
|
if STOCHASTIC[i]['slow_k'] < 40:
|
|
status += '60_'
|
|
|
|
# STOCHASTIC
|
|
stochastic_status = self.common.check_stochastic(STOCK, STOCHASTIC, i)
|
|
if stochastic_status != "":
|
|
status += stochastic_status
|
|
|
|
# YANGBONG
|
|
if self.common.checkLongYangBongAfterUmBong(STOCK, i):
|
|
# 어제 음봉 이후 장대양봉이었다면,
|
|
status += 'YANGBONG_'
|
|
|
|
return status, buy_price
|
|
|
|
# 그래프 출력
|
|
def analyzeToHtml(self, outPath):
|
|
tmp_path = outPath + "/tmp"
|
|
if os.path.isdir(tmp_path):
|
|
os.rmdir(tmp_path)
|
|
os.mkdir(tmp_path)
|
|
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
item_code = result[0]
|
|
item_name = result[1]
|
|
|
|
# 부실 기업은 매수하지 않고 그냥 넘긴다.
|
|
# kospi 지수와 kosdak 지수도 그냥 넘긴다.
|
|
if ((item_code in self.fnguide and not self.fnguide[item_code]) or (item_code == "KOSPI" or item_code == "KOSDAK")):
|
|
rowid += 1
|
|
# 다음 종목을 가져옴
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
continue
|
|
|
|
result_3 = result[3]
|
|
result_4 = result[4]
|
|
result_5 = result[5]
|
|
result_6 = result[6]
|
|
if result[3] != result[3]: result_3 = result[3].replace("NaN", "0")
|
|
if result[4] != result[4]: result_4 = result[4].replace("NaN", "0")
|
|
if result[5] != result[5]: result_5 = result[5].replace("NaN", "0")
|
|
if result[6] != result[6]: result_6 = result[6].replace("NaN", "0")
|
|
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": json.loads(result[2]), "MACD": json.loads(result_3), "STOCHASTIC": json.loads(result_4), "ICHIMOKU": json.loads(result_5), "RSI": json.loads(result_6)}
|
|
|
|
last_index = self.get_last_index(stock)
|
|
STOCK = stock['PRICE']
|
|
MACD = stock['MACD']
|
|
STOCHASTIC = stock['STOCHASTIC']
|
|
ICHIMOKU = stock['ICHIMOKU']
|
|
RSI = stock['RSI']
|
|
|
|
# 종목 상태 체크 분석
|
|
state, buy_price = self.analyzeFinalScore(last_index, STOCK, MACD, STOCHASTIC, ICHIMOKU, RSI)
|
|
|
|
stochastic_score = STOCHASTIC[last_index]['slow_k']
|
|
macd_score = MACD[last_index]['macd']
|
|
rsi_score = RSI[last_index]['rsi']
|
|
ichimoku_score = ICHIMOKU[last_index]['ichimoku_buy']
|
|
|
|
if state != "":
|
|
fig = self.draw(stock)
|
|
title = "%s (%s), %s, buy_price (%d), stochastic(%.3f), rsi(%.3f), macd(%.3f), ichimoku(%d)) 차트" % (item_name, item_code, state, buy_price, stochastic_score, rsi_score, macd_score, ichimoku_score)
|
|
fig['layout'].update(title=title)
|
|
fileName = "%s/%s__%.3f__%.3f__%s_%s.html" % (outPath, state, stochastic_score, rsi_score, item_name.replace(" ", ""), item_code)
|
|
po.write_html(fig, file=fileName, auto_open=False)
|
|
else:
|
|
if (RSI[last_index]['rsi_buy'] == 1) and STOCK[last_index]['volume'] > 1000000:
|
|
fig = self.draw(stock)
|
|
title = "%s (%s) buy_price (%d), stochastic(%.3f), rsi(%.3f), macd(%.3f), ichimoku(%d)) 차트"%(item_name, item_code, buy_price, stochastic_score, rsi_score, macd_score, ichimoku_score)
|
|
fig['layout'].update(title=title)
|
|
fileName = "%s/%.3f__%.3f__%s_%s.html"%(tmp_path, stochastic_score, rsi_score, item_name.replace(" ", ""), item_code)
|
|
po.write_html(fig, file=fileName, auto_open=False)
|
|
|
|
print ("#html", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
cursor.close()
|
|
conn.close()
|
|
return
|
|
|
|
def analyze(self):
|
|
conn = sqlite3.connect(self.inFileName)
|
|
cursor = conn.cursor()
|
|
|
|
# 기존 분석 데이터를 모두 지움
|
|
cursor.execute('update ' + self.tableName + ' set ICHIMOKU = "", MACD = "", STOCHASTIC = "", RSI = ""')
|
|
|
|
rowid = 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
while result != None:
|
|
stock = {"CODE": result[0], "NAME": result[1], "PRICE": json.loads(result[2])}
|
|
|
|
try:
|
|
results_ICHIMOKU = self.ichimokuCloud.analyze(stock)
|
|
text_ICHIMOKU = json.dumps(results_ICHIMOKU, ensure_ascii=False)
|
|
|
|
results_MACD = self.macd.analyze(stock)
|
|
text_MACD = json.dumps(results_MACD, ensure_ascii=False)
|
|
|
|
results_STOCHASTIC = self.stochastic.analyze(stock)
|
|
text_STOCHASTIC = json.dumps(results_STOCHASTIC, ensure_ascii=False)
|
|
|
|
results_RSI = self.rsi.analyze(stock)
|
|
text_RSI = json.dumps(results_RSI, ensure_ascii=False)
|
|
except:
|
|
print("#", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
continue
|
|
|
|
cursor.execute("UPDATE " + self.tableName + " SET ICHIMOKU=?, MACD=?, STOCHASTIC=?, RSI=? WHERE CODE=?", (text_ICHIMOKU,text_MACD,text_STOCHASTIC,text_RSI, stock["CODE"]))
|
|
print("#", rowid, stock['NAME'])
|
|
rowid += 1
|
|
cursor.execute('SELECT * FROM ' + self.tableName + ' WHERE rowid=?', (rowid,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
cursor.close()
|
|
conn.close()
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
start = time.time()
|
|
PROJECT_HOME = "../.."
|
|
inFileName = PROJECT_HOME + '/resources/stock.db'
|
|
inFnguideFileName = PROJECT_HOME + '/resources/fnguide.db'
|
|
analyzer = Analyzer(PROJECT_HOME, inFileName, inFnguideFileName)
|
|
|
|
# 분석 & update DB
|
|
"""
|
|
#print ("analyze IchimokuCloud...")
|
|
analyzer.analyzeIchimokuCloud()
|
|
#print ("analyze MACD...")
|
|
analyzer.analyzeMACD()
|
|
#print ("analyze Stochastic...")
|
|
analyzer.analyzeStochastic()
|
|
#print ("analyze RSI...")
|
|
analyzer.analyzeRSI()
|
|
"""
|
|
|
|
###analyzer.analyze()
|
|
|
|
day = datetime.today().strftime("%Y%m%d")
|
|
|
|
# HTML 출력
|
|
outPath = PROJECT_HOME + "/resources/analysis/"+day
|
|
if os.path.isdir(outPath):
|
|
shutil.rmtree(outPath)
|
|
os.mkdir(outPath)
|
|
print("print to Html...")
|
|
analyzer.analyzeToHtml(outPath)
|
|
|
|
|
|
# 파일 출력
|
|
#print("print to File...")
|
|
#outFileName = PROJECT_HOME + '/resources/analysis/'+day+'.json'
|
|
#analyzer.analyzeToFile(outFileName)
|
|
|
|
print("time : %6.2f 초", (time.time() - start))
|
|
|
|
print("done...")
|