Coverage for src/utils/portfolio_util.py: 0%

1import pandas as pd 

2from dateutil import parser 

3 

4from src.configuration import store 

5from src.utils import store_util 

6 

7 

8def update_investment(account, amount, my_date): 

9 df = store.portfolio_investments.copy() 

10 my_date = parser.parse(my_date) 

11 if df.empty or df.loc[(df.date == my_date) & (df.account_name == account)].empty: 

12 new_df = pd.DataFrame({'date': pd.to_datetime(my_date), 

13 'account_name': account, 

14 'amount': amount}, 

15 index=[0]) 

16 # new_df.date = new_df.date.dt.date 

17 df = pd.concat([df, new_df], ignore_index=True) 

18 

19 else: 

20 df.loc[ 

21 (df.date == my_date) & 

22 (df.account_name == account), 'amount'] += amount 

23 df.date = pd.to_datetime(df.date) 

24 store.portfolio_investments = df 

25 store_util.save_stores()