Coverage for utils / kr_stock_data.py: 100%

1import pandas as pd 

2 

3class StockCodeNameResolver: 

4 def __init__(self, csv_path="data/stock_code_list.csv"): 

5 self.df = pd.read_csv(csv_path, dtype=str) 

6 self.code_to_name = dict(zip(self.df['code'], self.df['name'])) 

7 self.name_to_code = dict(zip(self.df['name'], self.df['code'])) 

8 

9 def get_name(self, code): 

10 return self.code_to_name.get(code) 

11 

12 def get_code(self, name): 

13 return self.name_to_code.get(name)