Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions src/enablebanking_sdk/service/integration.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import logging
from datetime import datetime, timedelta
from typing import Optional, Tuple
from typing import Tuple
from requests.exceptions import HTTPError

import jwt
Expand All @@ -15,7 +15,7 @@

class EnableBankingIntegration:
token: str
token_expiration: Optional[datetime] = None
token_expiration: datetime | None = None

base_url: str
app_id: str
Expand Down Expand Up @@ -121,7 +121,7 @@ def start_user_session(self, request: dict) -> dict:
json=request,
)

def delete_user_session(self, session_id: str, psu_headers: dict):
def delete_user_session(self, session_id: str, psu_headers: dict | None = None):
return self._request(
method="DELETE",
path=f"/sessions/{session_id}",
Expand All @@ -144,7 +144,7 @@ def get_account_transactions(
psu_headers: dict | None = None,
continuation_key: str | None = None,
) -> dict:
def _fmt_dt(date: Optional[datetime]) -> Optional[str]:
def _fmt_dt(date: datetime | None) -> str | None:
return date.replace(tzinfo=None).strftime("%Y-%m-%d") if date else None

return self._request(
Expand All @@ -162,7 +162,7 @@ def _fmt_dt(date: Optional[datetime]) -> Optional[str]:
def get_account_balances(
self,
account_uid: str,
psu_headers: Optional[dict] = None,
psu_headers: dict | None = None,
) -> dict:
return self._request(
method="GET",
Expand Down