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-10-16 11:54:19 +09:00
parent 9fc8f27235
commit 997b1dd99f
5 changed files with 94 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
hts/OrderChecker.py
View File

@@ -8,13 +8,13 @@ class OrderChecker:
order_df = None
TYPE = None
def __init__(self, RESOURCE_PATH, type=""):
def __init__(self, RESOURCE_PATH, stock_code=""):
self.RESOURCE_PATH = RESOURCE_PATH
self.TYPE = type
self.stock_code = stock_code
return
def read(self, ymd):
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.stock_code+"_"+ymd + ".csv")
if os.path.isfile(saveFileName):
order_df = pd.read_csv(saveFileName)
@@ -53,7 +53,7 @@ class OrderChecker:
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, "order", self.TYPE+"_"+ymd + ".csv")
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.stock_code+"_"+ymd + ".csv")
order_df.to_csv(saveFileName)
return True
@@ -66,7 +66,7 @@ class OrderChecker:
order_df.loc[(order_df["stock_code"] == stock_code, 'type')] = 1
# 파일로 기록 한다.
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.stock_code+"_"+ymd + ".csv")
order_df.to_csv(saveFileName)
return True
@@ -86,7 +86,7 @@ class OrderChecker:
for i in range(len(df)):
order_df.loc[(order_df['stock_code']==stock_code) & (order_df.index == df.index[i]), 'canceled'] = 1
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.TYPE+"_"+ymd + ".csv")
saveFileName = os.path.join(self.RESOURCE_PATH, "order", self.stock_code+"_"+ymd + ".csv")
order_df.to_csv(saveFileName, index=False)
if ORDER_LIST is not None and len(ORDER_LIST) > 0: