Coverage for src/data_manager.py: 89%

1import json 

2from .const import (PREFECTURES_JSON_PATH, PREDICTION_TOTAL_JSON_PATH, 

3 HISTORY_TOTAL_JSON_PATH, TODAY_TOTAL_JSON_PATH, 

4 STATISTICS_JSON_PATH, POSITIVE_DETAIL_JSON_PATH) 

5 

6 

7class DataManager: 

8 def __init__(self): 

9 with open(PREFECTURES_JSON_PATH) as f: 

10 self.prefectures_json = f.read() 

11 with open(HISTORY_TOTAL_JSON_PATH) as f: 

12 self.history_total_json = f.read() 

13 with open(TODAY_TOTAL_JSON_PATH) as f: 

14 self.today_total_json = f.read() 

15 with open(PREDICTION_TOTAL_JSON_PATH) as f: 

16 self.prediction_total_json = f.read() 

17 with open(STATISTICS_JSON_PATH) as f: 

18 self.statistics_json = f.read() 

19 with open(POSITIVE_DETAIL_JSON_PATH) as f: 

20 self.positive_detail_json = f.read() 

21 

22 def get_positive_detail_json(self, prefecture): 

23 json_dict = [x for x in json.loads(self.positive_detail_json) if x['prefecture'] == prefecture] 

24 return json.dumps(json_dict, indent=2, ensure_ascii=False)