87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
import win32com.client
|
|
import os
|
|
from datetime import datetime, timedelta
|
|
|
|
from hts.HTS import HTS
|
|
|
|
class DataDownloader (HTS):
|
|
|
|
buySellChecker = None
|
|
stock_code = None
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
return
|
|
|
|
# 주식 현재가 조회
|
|
def writeStockData(self, stock_code, given_day, type, outfp):
|
|
objCpCybos = win32com.client.Dispatch("CpUtil.CpCybos")
|
|
bConnect = objCpCybos.IsConnect
|
|
if (bConnect == 0):
|
|
print("PLUS가 정상적으로 연결되지 않음. ")
|
|
exit()
|
|
|
|
# 차트 객체 구하기
|
|
objStockChart = win32com.client.Dispatch("CpSysDib.StockChart")
|
|
|
|
objStockChart.SetInputValue(0, 'A' + stock_code) # 종목 코드
|
|
objStockChart.SetInputValue(1, ord('2')) # 1: 기간으로 조회, 2: 개수로 조회
|
|
objStockChart.SetInputValue(2, given_day) # 기간 조회 시, 시작일
|
|
objStockChart.SetInputValue(3, given_day) # 기간 조회 시, 종료일
|
|
objStockChart.SetInputValue(4, 400) # 조회 시 가져오는 Line 개수
|
|
objStockChart.SetInputValue(5, [0, 1, 2, 3, 4, 5, 8]) # 날짜,시간,시가,고가,저가,종가,거래량
|
|
objStockChart.SetInputValue(6, ord(type)) # '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
|
|
objStockChart.SetInputValue(7, 1)
|
|
objStockChart.SetInputValue(9, ord('1')) # 수정주가 사용
|
|
objStockChart.BlockRequest()
|
|
|
|
size = objStockChart.GetHeaderValue(3)
|
|
|
|
if size > 0:
|
|
for i in range(size - 1, -1, -1):
|
|
day = objStockChart.GetDataValue(0, i)
|
|
time = objStockChart.GetDataValue(1, i)
|
|
start = objStockChart.GetDataValue(2, i)
|
|
high = objStockChart.GetDataValue(3, i)
|
|
low = objStockChart.GetDataValue(4, i)
|
|
close = objStockChart.GetDataValue(5, i)
|
|
vol = objStockChart.GetDataValue(6, i)
|
|
outfp.write("%d,%s,%d,%d,%d,%d,%d\n" % (day, str(time).zfill(4), start, high, low, close, vol))
|
|
|
|
return size
|
|
|
|
if __name__ == "__main__":
|
|
|
|
today = datetime.today()
|
|
|
|
PROJECT_HOME = os.path.join(os.path.dirname(os.path.join(os.path.dirname(os.path.join(os.path.dirname(__file__))))))
|
|
RESOURCE_DIR = PROJECT_HOME + "/hits/data"
|
|
|
|
stock_codes = ["252670", "122630"]
|
|
dataDownloader = DataDownloader()
|
|
|
|
for stock_code in stock_codes:
|
|
outFp = open(RESOURCE_DIR + "/" + stock_code + "_"+today.strftime('%Y%m%d')+".csv", mode="w", encoding="utf-8")
|
|
outFp.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량"))
|
|
|
|
zero_count_m, zero_count_S = 0, 0
|
|
for i in range(1, 100000):
|
|
given_day = today - timedelta(i)
|
|
|
|
# '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
|
|
if zero_count_m < 10:
|
|
count_m = dataDownloader.writeStockData(stock_code, given_day.strftime('%Y%m%d'), 'm', outFp_m)
|
|
if count_m == 0:
|
|
zero_count_m += 1
|
|
else:
|
|
zero_count_m = 0
|
|
print(given_day, "m", count_m)
|
|
|
|
if zero_count_m > 10:
|
|
break
|
|
|
|
outFp.close()
|
|
|
|
print ("done...")
|