dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2024-04-26 09:19:28 +09:00
parent d5e5316fce
commit 0d18c61fa2
6 changed files with 1392 additions and 256 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
JSDPattern.py
View File

@@ -88,26 +88,9 @@ class JSDPattern:
return data
def getDBData(self, stock_code, day, mins, get_days=14):
def getDBData(self, stock_code, day, get_days=14):
if mins == 3:
table = 'minute3'
elif mins == 5:
table = 'minute5'
elif mins == 10:
table = 'minute10'
elif mins == 20:
table = 'minute20'
elif mins == 30:
table = 'minute30'
elif mins == 60:
table = 'minute60'
elif mins == 200:
table = 'minute200'
elif mins == 1440:
table = 'daily'
else:
table = 'minutely'
table = 'stock'
conn = sqlite3.connect(os.path.join(self.RESOURCE_PATH, 'coins.db'))
cursor = conn.cursor()
@@ -142,35 +125,8 @@ class JSDPattern:
return result
def getCoinData(self, ticker, mins=None, to=None, ymd=None, get_days=14):
result = None
if ymd is not None and datetime.now() < datetime.strptime(ymd, '%Y%m%d'):
ymd = None
if ymd is None:
if to is None:
if mins is None:
df = pyupbit.get_ohlcv(ticker=ticker['ticker_code'])
else:
if mins == 1440:
df = pyupbit.get_ohlcv(ticker=ticker['ticker_code'], interval='minute1', count=1)
else:
df = pyupbit.get_ohlcv(ticker=ticker['ticker_code'], interval='minute' + str(mins))
else:
df = pyupbit.get_ohlcv(ticker=ticker['ticker_code'], interval='minute' + str(mins), to=to)
if df is not None:
df["datetime"] = df.index
df = df[['open', 'high', 'low', 'close', 'volume']].astype(float)
if mins is not None:
result = self.getDBData(ticker['ticker_code'], datetime.today().strftime('%Y%m%d'), mins=mins, get_days=get_days)
data = self.append(df, result)
else:
result = self.getDBData(ticker['ticker_code'], ymd, mins=mins, get_days=get_days)
data = self.append(df=None, result=result)
result = self.getDBData(ticker['ticker_code'], ymd, mins=mins, get_days=get_days)
data = self.append(df=None, result=result)
return data