def save_portfolio_info(strategy_name):
date = datetime.datetime.now(
tz=pytz.timezone('US/Eastern')).strftime('%Y-%m-%d')
dic = PortfolioDAO.read_portfolio_info(strategy_name)
dic[date] = API().get_portfolio_info().to_dict()
file_path = PathMgr.get_strategies_portfolio_file(strategy_name)
ensure_parent_dir_exists(file_path)
write_to_file(file_path, json.dumps(dic, indent=4, sort_keys=True))
def read_portfolio_info(strategy_name):
file_path = PathMgr.get_strategies_portfolio_file(strategy_name)
if os.path.exists(file_path):
content = read_file_to_string(file_path)
dic = json.loads(content)
# for key in dic.keys():
# dic[key] = Portfolio.from_dict(dic[key])
return dic
else:
return {}
