diff --git a/quantcommon.py b/quantcommon.py index 3990fa8..4398ad4 100644 --- a/quantcommon.py +++ b/quantcommon.py @@ -1,6 +1,7 @@ import os from urllib.parse import quote_plus +import pandas as pd import pymysql from dotenv import load_dotenv from sqlalchemy import create_engine @@ -26,3 +27,30 @@ class QuantCommon: port=self.port, db=self.db, charset='utf8') + + def get_ticker_list(self): + engine = self.create_engine() + + try: + ticker_list = pd.read_sql(""" + select * from kor_ticker + where 기준일 = (select max(기준일) from kor_ticker) + and 종목구분 = '보통주'; + """, con=engine) + finally: + engine.dispose() + + return ticker_list + + def get_value_list(self): + engine = self.create_engine() + + try: + value_list = pd.read_sql(""" + select * from kor_value + where 기준일 = (select max(기준일) from kor_value); + """, con=engine) + finally: + engine.dispose() + + return value_list \ No newline at end of file