Monarch Money cashflow sensor bugfix (#125774)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
This commit is contained in:
Jeef 2024-09-27 01:18:37 -06:00 committed by Franck Nijhof
parent ebfd442b51
commit bb73529770
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3

View File

@ -2,7 +2,7 @@
import asyncio import asyncio
from dataclasses import dataclass from dataclasses import dataclass
from datetime import timedelta from datetime import datetime, timedelta
from aiohttp import ClientResponseError from aiohttp import ClientResponseError
from gql.transport.exceptions import TransportServerError from gql.transport.exceptions import TransportServerError
@ -63,9 +63,13 @@ class MonarchMoneyDataUpdateCoordinator(DataUpdateCoordinator[MonarchData]):
async def _async_update_data(self) -> MonarchData: async def _async_update_data(self) -> MonarchData:
"""Fetch data for all accounts.""" """Fetch data for all accounts."""
now = datetime.now()
account_data, cashflow_summary = await asyncio.gather( account_data, cashflow_summary = await asyncio.gather(
self.client.get_accounts_as_dict_with_id_key(), self.client.get_accounts_as_dict_with_id_key(),
self.client.get_cashflow_summary(), self.client.get_cashflow_summary(
start_date=f"{now.year}-01-01", end_date=f"{now.year}-12-31"
),
) )
return MonarchData(account_data=account_data, cashflow_summary=cashflow_summary) return MonarchData(account_data=account_data, cashflow_summary=cashflow_summary)