整理股票每日交易資訊後,把三大法人資料(區分期貨與選擇權二類-依日期)也模組化,也可以用get_and_save
取資料並且匯入資料庫中。
本日程式碼使用:legal_person.py
這邊製作一個模組檔legalPerson
,用來把三大法人相關的class放進去,目前是放三大法人期貨選擇權的最近一期交易資料,以及匯入歷史資料兩個功能。未來如果有屬於三大法人的部分,可以直接擴充進來,新增新的class,再隨著情境引入到程式中。
主要的Function都做好了,需要調整的是一些瑣碎的部分,像是註解,或是程式碼整理。最主要的是新增funciton:get_and_save()
,讓我們可以一個function中取得資料並且存起來。
我們已經有了_get_csv_data()
(改名字)可以取得資料,_insert_mysql()
(改名字)可以寫入資料庫,接著就把這兩個動作包起來,放在get_and_save()
,並且有路徑參數可以帶入:
def get_and_save(self, url=None) -> bool:
if url:
r = self._get_csv_data(os.path.join(os.path.dirname(__file__), url))
else:
r = self._get_csv_data(url=None)
if r:
r = self._insert_mysql()
return True
else:
return False
並且有回傳值告訴我們說這次作業有沒有成功。
同樣的製作出get_and_save()
,不一樣的是他是網址路徑,非檔案路徑。
def get_and_save(self, url=None) -> bool:
r = self._get_csv_data(self.url if url else None)
if r:
r = self._insert_mysql()
return True
else:
return False
這次整理了兩個class,並且整理註解的部分。變動不多,但是花不少時間在整理與調整,分佈在兩個檔案的class,全部放在一起,以後調用三大法人資料會更方便。