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
2023-01-30 10:01:07 +09:00
parent 7f052708dd
commit 4fb27a3024
3 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
HTS_etf.py
View File

@@ -28,7 +28,7 @@ class HTS_etf (HTS):
self.RESOURCE_PATH = RESOURCE_PATH self.RESOURCE_PATH = RESOURCE_PATH
self.orderChecker = OrderChecker(self.RESOURCE_PATH) self.orderChecker = OrderChecker(self.RESOURCE_PATH, "ETF")
self.buySellChecker = BuySellChecker() self.buySellChecker = BuySellChecker()
self.labelChecker = LabelChecker(RESOURCE_PATH) self.labelChecker = LabelChecker(RESOURCE_PATH)
self.slackBot = SlackBot() self.slackBot = SlackBot()

2
HTS_stocks.py
View File

@@ -29,7 +29,7 @@ class HTS_Stocks (HTS):
self.RESOURCE_PATH = RESOURCE_PATH self.RESOURCE_PATH = RESOURCE_PATH
self.stockStatus = StockStatus(RESOURCE_PATH) self.stockStatus = StockStatus(RESOURCE_PATH)
self.buySellChecker = BuySellChecker() self.buySellChecker = BuySellChecker()
self.orderChecker = OrderChecker(self.RESOURCE_PATH) self.orderChecker = OrderChecker(self.RESOURCE_PATH, "STOCK")
self.analyzed_day = 120 self.analyzed_day = 120
self.MAX_BUY_PRICE = 200000 self.MAX_BUY_PRICE = 200000

14
hts/OrderChecker.py
View File

@@ -6,13 +6,15 @@ from datetime import datetime, timedelta
class OrderChecker: class OrderChecker:
RESOURCE_PATH = None RESOURCE_PATH = None
order_df = None order_df = None
TYPE = None
def __init__(self, RESOURCE_PATH): def __init__(self, RESOURCE_PATH, type=""):
self.RESOURCE_PATH = RESOURCE_PATH self.RESOURCE_PATH = RESOURCE_PATH
self.TYPE = type
return return
def read(self, ymd): def read(self, ymd):
saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv") saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
if os.path.isfile(saveFileName): if os.path.isfile(saveFileName):
order_df = pd.read_csv(saveFileName) order_df = pd.read_csv(saveFileName)
@@ -23,7 +25,7 @@ class OrderChecker:
if 'Unnamed: 0' in order_df.columns: if 'Unnamed: 0' in order_df.columns:
order_df.drop(['Unnamed: 0'], axis=1, inplace=True) order_df.drop(['Unnamed: 0'], axis=1, inplace=True)
order_df = order_df.fillna(0) order_df = order_df.fillna(0)
order_df = order_df.astype({"stock_code": int, "type": int, "orderNum": int, "canceled": int, "count": int, "price": int}) order_df = order_df.astype({"stock_code": str, "type": int, "orderNum": int, "canceled": int, "count": int, "price": int})
order_df['datetime'] = pd.to_datetime(order_df['datetime']) order_df['datetime'] = pd.to_datetime(order_df['datetime'])
return order_df return order_df
@@ -50,7 +52,7 @@ class OrderChecker:
order_df = order_df.astype({"stock_code": str, "type": int, "orderNum": str, "canceled": int, "count": int, "price": int}) order_df = order_df.astype({"stock_code": str, "type": int, "orderNum": str, "canceled": int, "count": int, "price": int})
# 파일로 기록한다. # 파일로 기록한다.
saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv") saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
order_df.to_csv(saveFileName) order_df.to_csv(saveFileName)
return True return True
@@ -63,7 +65,7 @@ class OrderChecker:
order_df.loc[(order_df["stock_code"] == "A" + stock_code, 'type')] = 1 order_df.loc[(order_df["stock_code"] == "A" + stock_code, 'type')] = 1
# 파일로 기록 한다. # 파일로 기록 한다.
saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv") saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
order_df.to_csv(saveFileName) order_df.to_csv(saveFileName)
return True return True
@@ -83,7 +85,7 @@ class OrderChecker:
for i in range(len(df)): for i in range(len(df)):
order_df.loc[(order_df['stock_code']=="A" + stock_code) & (order_df.index == df.index[i]), 'canceled'] = 1 order_df.loc[(order_df['stock_code']=="A" + stock_code) & (order_df.index == df.index[i]), 'canceled'] = 1
saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv") saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
order_df.to_csv(saveFileName, index=False) order_df.to_csv(saveFileName, index=False)
if ORDER_LIST is not None and len(ORDER_LIST) > 0: if ORDER_LIST is not None and len(ORDER_LIST) > 0: